Browse Source

Allow specifying TRACKER_USERNAME in config.json

microtasks
tech234a 3 years ago
parent
commit
d125887b58
2 changed files with 10 additions and 1 deletions
  1. +2
    -1
      config.json
  2. +8
    -0
      tracker.py

+ 2
- 1
config.json View File

@@ -1,5 +1,6 @@
{
"HSID": "",
"SSID": "",
"SID" : ""
"SID" : "",
"TRACKER_USERNAME": ""
}

+ 8
- 0
tracker.py View File

@@ -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"



Loading…
Cancel
Save