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

完善命令行错误提示

Browse Source
This commit is contained in:
xaoyaoo 2024-01-09 20:38:42 +08:00
parent 126f03f054
commit eb62def663
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pywxdump/cli.py
View File

@ -289,7 +289,7 @@ class MainExportChatRecords():
sb_decrypt.add_argument("-fs", "--filestorage_path", type=str, sb_decrypt.add_argument("-fs", "--filestorage_path", type=str,
help="(可选)文件夹FileStorage的路径（用于显示图片）", required=False, help="(可选)文件夹FileStorage的路径（用于显示图片）", required=False,
metavar="") metavar="")
sb_decrypt.add_argument("-t", "--type", type=str, help="导出类型(可选:html,txt)", required=False, sb_decrypt.add_argument("-t", "--type", type=str, help="导出类型(可选:html,csv)", required=False,
default="html", default="html",
metavar="") metavar="")
return sb_decrypt return sb_decrypt
@ -297,16 +297,22 @@ class MainExportChatRecords():
def run(self, args): def run(self, args):
# 从命令行参数获取值 # 从命令行参数获取值
t = args.type t = args.type
if t not in ["html", "txt"]: if t not in ["html", "csv"]:
print("[-] 未知的导出类型") print("[-] 未知的导出类型")
return return
if t == "txt": if t == "txt":
try: try:
export_csv(args.username, args.outpath, args.msg_path, page_size=1000000) code, ret = export_csv(args.username, args.outpath, args.msg_path, page_size=1000000)
if not code:
print(ret)
return
print(f"[+] {ret}")
return
except Exception as e: except Exception as e:
print(e) print(e)
print("[-] 导出失败") print("[-] 导出失败")
return return
try: try:
from flask import Flask, request, jsonify, render_template, g from flask import Flask, request, jsonify, render_template, g
import logging import logging