diff --git a/oneget.py b/oneget.py index 355c825..02931ff 100644 --- a/oneget.py +++ b/oneget.py @@ -1,3 +1,4 @@ +import argparse import base64 from datetime import datetime import concurrent.futures @@ -226,6 +227,25 @@ class DMVideoInfo: print(f"[ERROR] 请求失败 vxid={v_xid} : {e}") return None +def parse_args() -> argparse.Namespace: + global MACHINE_ID + parser = argparse.ArgumentParser( + description="Configure worker settings." + ) + parser.add_argument( + "-m", "--machine-id", + type=int, + help=f"Machine identifier (default: {MACHINE_ID})" + ) + + args = parser.parse_args() + + if args.machine_id is not None: + MACHINE_ID = args.machine_id + + if MACHINE_ID is None: + raise ValueError("请指定机器编号") + return args def main(): kwdata = db.get_web_items() @@ -641,7 +661,11 @@ def main(): executor.map(dmvideo_info.get_video_info, tancks) if edges_len < 20: break - +def run(): + parse_args() + while True: + main() + time.sleep(60) if __name__ == '__main__': - main() + main() \ No newline at end of file