Add backup tool

setup.py

@@ -34,8 +34,9 @@
     install_requires=INSTALL_REQUIRES,
     entry_points={
         'console_scripts': [
-            'pull_journeys = via.bin.pull_journeys:main',
-            'generate_geojson = via.bin.generate_geojson:main'
+            'via_pull_journeys = via.bin.pull_external_journeys:main',
+            'via_generate_geojson = via.bin.generate_geojson:main',
+            'via_backup_raw_journeys = via.bin.backup_raw_journeys:main'
         ]
     }
 )

src/via/bin/backup_raw_journeys.py

@@ -0,0 +1,39 @@
+import argparse
+import json
+
+
+def rm_obj_id(journey_data):
+    del journey_data["_id"]
+    return journey_data
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "backup_file",
+        help="Path to the backup json file",
+    )
+    args = parser.parse_args()
+
+    if not args.backup_file.endswith(".json"):
+        raise ValueError("backup file must be a .json file")
+
+    from via.utils import get_mongo_interface
+    from via.settings import MONGO_RAW_JOURNEYS_COLLECTION
+
+    mongo_interface = get_mongo_interface()
+    with open(args.backup_file, "w") as fh:
+        fh.write(
+            json.dumps(
+                [
+                    rm_obj_id(journey_data)
+                    for journey_data in getattr(
+                        mongo_interface, MONGO_RAW_JOURNEYS_COLLECTION
+                    ).find()
+                ]
+            )
+        )
+
+
+if __name__ == "__main__":
+    main()