完善交flag的脚本

Prepare/submit.py

@@ -1,8 +1,36 @@
 import requests
+import sys
 
-token = "47e2ef7aa5bf2bafc6cf95b31070c96f"
-flag = input("input flag: ")
-url = "https://ctf.bugku.com/pvp/submit.html?token=" + token + "&flag=" + flag
 
-response = requests.get(url)
-print(response.text)
+
+
+
+def post_submit(team_token, flag, url):
+    if team_token is None or flag is None:
+        raise Exception('team token or flag wrong')
+    form_data = {
+        "token":team_token,
+        "flag":flag
+    }
+    response = requests.post(url, data=form_data)
+    return response.text
+
+
+def get_submit(team_token, flag, url):
+    if not team_token or not flag:
+        raise Exception('team token or flag wrong')
+    url = url + token + "&flag=" + flag
+    response = requests.get(url)
+    return response.text
+
+
+if __name__ == '__main__':
+    token = ""
+    flag = ""
+    base_url = ' '	#提交flag的服务器地址
+    # url = "https://ctf.bugku.com/pvp/submit.html?token=" + token + "&flag=" + flag
+    method = "" #get或post
+    if method == "get":
+        response = get_submit(token, flag, base_url)
+    elif method == "post":
+        response = post_submit(token, flag, base_url)