Choose segments with import.py

2017-12-21 23:34:27 -05:00 · 2017-12-21 23:34:27 -05:00 · f11b5985d3
parent 6381a86d64
commit f11b5985d3
1 changed files with 9 additions and 2 deletions
--- a/import.py
+++ b/import.py
@ -1,9 +1,16 @@
+"""
+python import.py export.zip challenges,teams,both,metadata
+"""
 from CTFd import create_app
 from CTFd.utils import import_ctf

-import zipfile
 import sys

 app = create_app()
 with app.app_context():
-    import_ctf(sys.argv[1])
+    if sys.argv[2]:
+        segments = sys.argv[2].split(',')
+    else:
+        segments = None
+
+    import_ctf(sys.argv[1], segments=segments)