From fdba82530dc7365b5f042123b665e73161d4d23e Mon Sep 17 00:00:00 2001
From: Smart-SangGe <2251250136@qq.com>
Date: Fri, 28 Jul 2023 17:55:38 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BA=A4flag=E7=9A=84?=
 =?UTF-8?q?=E8=84=9A=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Prepare/submit.py | 38 +++++++++++++++++++++++++++++++++-----
 1 file changed, 33 insertions(+), 5 deletions(-)

diff --git a/Prepare/submit.py b/Prepare/submit.py
index 9cf34e8..b4ef244 100644
--- a/Prepare/submit.py
+++ b/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)
\ No newline at end of file
+
+
+
+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)