Choose segments with import.py

selenium-screenshot-testing
Kevin Chung 2017-12-21 23:34:27 -05:00
parent 6381a86d64
commit f11b5985d3
1 changed files with 9 additions and 2 deletions

View File

@ -1,9 +1,16 @@
"""
python import.py export.zip challenges,teams,both,metadata
"""
from CTFd import create_app from CTFd import create_app
from CTFd.utils import import_ctf from CTFd.utils import import_ctf
import zipfile
import sys import sys
app = create_app() app = create_app()
with app.app_context(): 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)