cyonjan/PyWxDump
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加是否支持局域网访问选项，默认为false

Browse Source
This commit is contained in:
xaoyaoo 2024-01-09 22:36:53 +08:00
parent 975baf8796
commit d4d08fc8f8
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pywxdump/cli.py
View File

@ -185,6 +185,8 @@ class MainShowChatRecords():
metavar="") metavar="")
sb_decrypt.add_argument("-myid", "--my_wxid", type=str, help="(可选)微信账号(本人微信id)", required=False, sb_decrypt.add_argument("-myid", "--my_wxid", type=str, help="(可选)微信账号(本人微信id)", required=False,
default="wxid_vzzcn5fevion22", metavar="") default="wxid_vzzcn5fevion22", metavar="")
sb_decrypt.add_argument("--online", type=bool, help="(可选)是否在线查看(局域网查看)", required=False,
default=False, metavar="")
return sb_decrypt return sb_decrypt
def run(self, args): def run(self, args):
@ -206,22 +208,8 @@ class MainShowChatRecords():
from flask_cors import CORS from flask_cors import CORS
from pywxdump.api import api from pywxdump.api import api
# if getattr(sys, 'frozen', False):
# # The application is run as a bundled executable (PyInstaller)
# base_dir = sys._MEIPASS
# else:
# # The application is run as a script
# base_dir = os.path.abspath(os.path.dirname(__file__))
# template_folder = os.path.join(base_dir, 'ui/web')
# static_folder = os.path.join(base_dir, 'ui/web/assets/')
app = Flask(__name__, template_folder='./ui/web', static_folder='./ui/web/assets/', static_url_path='/assets/') app = Flask(__name__, template_folder='./ui/web', static_folder='./ui/web/assets/', static_url_path='/assets/')
# app.template_folder = template_folder
# app.static_folder = static_folder
# app.static_url_path = '/assets/'
app.logger.setLevel(logging.ERROR) app.logger.setLevel(logging.ERROR)
CORS(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True) # 允许所有域名跨域 CORS(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True) # 允许所有域名跨域
@ -252,8 +240,9 @@ class MainShowChatRecords():
print("Unsupported platform, can't open browser automatically.") print("Unsupported platform, can't open browser automatically.")
except Exception as e: except Exception as e:
pass pass
import socket
def is_port_in_use(host, port): def is_port_in_use(host, port):
import socket
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
try: try:
s.bind((host, port)) s.bind((host, port))
@ -261,8 +250,13 @@ class MainShowChatRecords():
return True return True
return False return False
online = args.online
# 检查端口是否被占用 # 检查端口是否被占用
host = '0.0.0.0' if online:
host = '0.0.0.0'
else:
host = "127.0.0.1"
port = 5000 port = 5000
if is_port_in_use(host, port): if is_port_in_use(host, port):
print(f"Port {port} is already in use. Choose a different port.") print(f"Port {port} is already in use. Choose a different port.")
@ -432,6 +426,7 @@ class MainAll():
args.media_path = merge_save_path args.media_path = merge_save_path
args.wxid_path = filePath args.wxid_path = filePath
args.my_wxid = wxid args.my_wxid = wxid
args.online = False
MainShowChatRecords().run(args) MainShowChatRecords().run(args)