1 files changed, 12 insertions, 8 deletions
diff --git a/to_csv.py b/to_csv.py
index e08bc24..85674fd 100755
--- a/to_csv.py
+++ b/to_csv.py
@@ -1,22 +1,26 @@
 #!/usr/bin/env python3
 import csv
-from mgs import MGSPiracy
+import mosgorsud
+
 from argparse import ArgumentParser
 
+
 if __name__ == '__main__':
     # parse arguments
     argp = ArgumentParser()
-    argp.add_argument('--output', type=str, default='output.csv', help='CSV output file name')
-    argp.add_argument('--from', type=int, default=0, help='First page', dest='_from')
-    argp.add_argument('--to', type=int, default=10, help='Last page')
-    args = argp.parse_args()
+    argp.add_argument('--output', type=str, default='output.csv',
+                      help='CSV output file name')
+    argp.add_argument('--from', type=int, default=1, dest='_from',
+                      help='First page')
+    argp.add_argument('--to', type=int, default=10,
+                      help='Last page')
+    arg = argp.parse_args()
 
     # get cases
-    mgs = MGSPiracy(from_page=args._from, to_page=args.to)
-    cases = mgs.get_cases()
+    cases = mosgorsud.get_cases(from_page=arg._from, to_page=arg.to)
 
     # write to csv
-    f = open(args.output, 'w', newline='')
+    f = open(arg.output, 'w', newline='')
     csv_writer = csv.writer(f)
 
     for case in cases: