Refactor python files

wgmaster/migrate.py

@@ -14,6 +14,7 @@ import csv
 
 
 DB_CNX: Optional[sqlite3.Connection] = None
+QUIET: bool = False
 
 HKID: Optional[str] = None
 WG: Optional[str] = None
@@ -78,7 +79,7 @@ STREET_NAMES: Dict[str, str] = {
 
 
 def success(mgnr: int, key: str, value) -> None:
-    if not args.quiet:
+    if not QUIET:
         print(f'\x1B[1;32m{mgnr:>6}: {key:<12} {value}\x1B[0m', file=sys.stderr)
 
 
@@ -91,7 +92,7 @@ def invalid(mgnr: int, key: str, value) -> None:
 
 
 def renumber_delivery(lsnr_1: str, lsnr_2: str) -> None:
-    if not args.quiet:
+    if not QUIET:
         print(f'\x1B[1m{lsnr_1:<14} -> {lsnr_2:<14}\x1B[0m')
 
 
@@ -104,13 +105,13 @@ def invalid_delivery(lsnr: str, mgnr: int, key: str, value) -> None:
 
 
 def convert(mgnr: int, key: str, old_value: str, new_value) -> None:
-    if not args.quiet:
+    if not QUIET:
         print(f'\x1B[1m{mgnr:>6}: {key:<12} "{old_value}" -> "{new_value}"\x1B[0m', file=sys.stderr)
 
 
 def convert_name(mgnr: int, old_name: Tuple[str, str], new_name: Tuple[str, str, str, str, str],
                  billing: Optional[str] = None) -> None:
-    if not args.quiet:
+    if not QUIET:
         print(f'\x1B[1m{mgnr:>6}: '
               f'{" / ".join([e or "" for e in old_name])} -> '
               f'{" / ".join([e or "" for e in new_name])}'
@@ -883,7 +884,9 @@ def migrate_payments(in_dir: str, out_dir: str) -> None:
     pass  # TODO migrate payments
 
 
-if __name__ == '__main__':
+def main() -> None:
+    global DB_CNX, QUIET, HKID, WG
+
     parser = argparse.ArgumentParser()
     parser.add_argument('in_dir', type=str,
                         help='The input directory where the exported csv files are stored')
@@ -902,6 +905,7 @@ if __name__ == '__main__':
 
     os.makedirs(args.out_dir, exist_ok=True)
 
+    QUIET = args.quiet
     HKID = args.origin
     WG = args.genossenschaft
 
@@ -920,3 +924,7 @@ if __name__ == '__main__':
     migrate_payments(args.in_dir, args.out_dir)
 
     DB_CNX.close()
+
+
+if __name__ == '__main__':
+    main()