Added --add to edscupdate.py to add systems to local db automatically

eyeonus · Mar 2, 2015 · 84d823a · 84d823a
1 parent 514d5bf
commit 84d823a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/edscupdate.py b/edscupdate.py
@@ -95,6 +95,13 @@ def parse_arguments():
             required=False,
             help='Show systems in random order, maximum of 10.',
     )
+    parser.add_argument(
+            '--add-to-local-db', '-A',
+            action='store_true',
+            required=False,
+            help='Add accepted systems to the local database.',
+            dest='add',
+    )
     parser.add_argument(
             '--test',
             required=False,
@@ -375,6 +382,15 @@ def main():
             if ok.lower() != 'y':
                 continue
 
+            if argv.add:
+                tdb.addLocalSystem(
+                    name,
+                    x, y, z,
+                    added='Release 1.00-EDStar',
+                    modified=created,
+                    commit=True
+                )
+
             print("'{}',{},{},{},'Release 1.00-EDStar','{}'".format(
                 name, x, y, z, created,
             ), file=output)