feat(web): 增加命令行指定web端口

This commit is contained in:
imgyh 2023-03-05 20:48:38 +08:00
parent 80825b2fc2
commit 98a9f74d4b

View File

@ -15,7 +15,7 @@ Change Log :
from flask import * from flask import *
from TikTok import TikTok from TikTok import TikTok
import argparse
def work(share_link): def work(share_link):
tk = TikTok() tk = TikTok()
@ -33,6 +33,13 @@ app = Flask(__name__)
# 设置编码 # 设置编码
app.config['JSON_AS_ASCII'] = False 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"]) @app.route("/douyin", methods=["POST"])
def douyin(): def douyin():
@ -57,4 +64,5 @@ def index():
if __name__ == "__main__": 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)