|
@@ -10,6 +10,7 @@ import traceback
|
|
|
import urllib.parse
|
|
|
from tqdm import tqdm
|
|
|
from datetime import datetime
|
|
|
+from argparse import ArgumentParser
|
|
|
|
|
|
|
|
|
from applications import PQMySQL, WeixinSpider, Functions, log, bot, aiditApi
|
|
@@ -593,4 +594,22 @@ def main():
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
- main()
|
|
|
+ parser = ArgumentParser()
|
|
|
+ parser.add_argument("--run_task",
|
|
|
+ help="update: update_job, check: check_job, detail: get_article_detail_job")
|
|
|
+ args = parser.parse_args()
|
|
|
+
|
|
|
+ if args.run_task:
|
|
|
+ run_task = args.run_task
|
|
|
+ match run_task:
|
|
|
+ case "update":
|
|
|
+ update_job()
|
|
|
+ case "check":
|
|
|
+ check_job()
|
|
|
+ case "detail":
|
|
|
+ get_article_detail_job()
|
|
|
+ case _:
|
|
|
+ print("No such task, input update: update_job, check: check_job, detail: get_article_detail_job")
|
|
|
+ else:
|
|
|
+ main()
|
|
|
+
|