From 98a9f74d4b0b319ea1aeb3883c105c0b805105f1 Mon Sep 17 00:00:00 2001
From: imgyh <1974355683@qq.com>
Date: Sun, 5 Mar 2023 20:48:38 +0800
Subject: [PATCH] =?UTF-8?q?feat(web):=20=E5=A2=9E=E5=8A=A0=E5=91=BD?=
 =?UTF-8?q?=E4=BB=A4=E8=A1=8C=E6=8C=87=E5=AE=9Aweb=E7=AB=AF=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TikTokWeb.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/TikTokWeb.py b/TikTokWeb.py
index 5cc86c1..9b56a29 100644
--- a/TikTokWeb.py
+++ b/TikTokWeb.py
@@ -15,7 +15,7 @@ Change Log  :
 
 from flask import *
 from TikTok import TikTok
-
+import argparse
 
 def work(share_link):
     tk = TikTok()
@@ -33,6 +33,13 @@ app = Flask(__name__)
 # 设置编码
 app.config['JSON_AS_ASCII'] = False
 
+def argument():
+    parser = argparse.ArgumentParser(description='抖音去水印工具 使用帮助')
+    parser.add_argument("--port", "-p", help="Web端口",
+                        type=int, required=True, default=5000)
+    args = parser.parse_args()
+
+    return args
 
 @app.route("/douyin", methods=["POST"])
 def douyin():
@@ -57,4 +64,5 @@ def index():
 
 
 if __name__ == "__main__":
-    app.run(debug=False, host="0.0.0.0", port=5000)
+    args = argument()
+    app.run(debug=False, host="0.0.0.0", port=args.port)