diff --git a/config.json b/config.json index 0e3741c..e010b85 100644 --- a/config.json +++ b/config.json @@ -1,5 +1,6 @@ { "HSID": "", "SSID": "", - "SID" : "" + "SID" : "", + "TRACKER_USERNAME": "" } \ No newline at end of file diff --git a/tracker.py b/tracker.py index 0523c3e..5441cb7 100644 --- a/tracker.py +++ b/tracker.py @@ -5,6 +5,9 @@ import requests from requests.adapters import HTTPAdapter from requests.packages.urllib3.util.retry import Retry +from os.path import isfile +from json import loads + # https://github.com/ArchiveTeam/tencent-weibo-grab/blob/9bae5f9747e014db9227821a9c11557267967023/pipeline.py VERSION = "20200921.01" @@ -19,6 +22,11 @@ TRACKER_ENDPOINT = f"http://{TRACKER_HOST}/{TRACKER_ID}" from os import environ if "TRACKER_USERNAME" in environ.keys(): TRACKER_USERNAME = environ["TRACKER_USERNAME"] +elif isfile("config.json"): + try: + TRACKER_USERNAME = loads(open("config.json").read())["TRACKER_USERNAME"] + except: + TRACKER_USERNAME = "Unnamed" else: TRACKER_USERNAME = "Unnamed"