Skip to content

Added optional flag for filtering categories #12

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
abdulrehman-03 wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Added optional flag for filtering categories #12

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 11 additions & 3 deletions download.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,10 +26,11 @@ def main(argv):
'-u (CTF URL)\n'
'-n (CTF name)\n'
'-o (CTF output directory)\n'
'-t (API token) OR -c (Cookie: session=?)\n\n'
'e.g python download.py -u http://ctf.url -n ctf_name -o /home/user/Desktop/ -t api_token')
'-t (API token) OR -c (Cookie: session=?)\n'
'-f (Optional: filter by categories, comma-separated)\n\n'
'e.g python download.py -u http://ctf.url -n ctf_name -o /home/user/Desktop/ -t api_token -f "crypto,web"')
try:
opts, _ = getopt.getopt(argv, 'hu:n:o:t:c:', ['help', 'url=', 'name=', 'output=', 'token=', 'cookie='])
opts, _ = getopt.getopt(argv, 'hu:n:o:t:c:f:', ['help', 'url=', 'name=', 'output=', 'token=', 'cookie=', 'filter='])
except getopt.GetoptError:
print('python download.py -h')
sys.exit(2)
Expand All @@ -42,6 +43,7 @@ def main(argv):
else:
baseUrl, ctfName, outputDir, = "", "", "" # defaults?
headers = {"Content-Type": "application/json"}
categories_filter = [] # Add a variable to store filtered categories
for opt, arg in opts:
if opt in ('-u', '--url'):
baseUrl = arg # URL of the CTFd
Expand All @@ -53,6 +55,8 @@ def main(argv):
headers["Authorization"] = f"Token {arg}" # CTFd API Token
elif opt in ('-c', '--cookie'):
headers["Cookie"] = f"session={arg}" # CTFd API Token
if opt in ('-f', '--filter'):
categories_filter = [cat.strip().lower() for cat in arg.split(",")]

os.makedirs(outputDir, exist_ok=True)

Expand Down Expand Up @@ -82,6 +86,10 @@ def main(argv):

Y = json.loads(S.get(f"{apiUrl}/challenges/{chall['id']}", headers=headers).text)["data"]

# Minimal change: skip challenge if category not in filter
if categories_filter and Y["category"].lower() not in categories_filter:
continue

if Y["category"] not in categories:
categories[Y["category"]] = [Y]
else:
Expand Down