muskit
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from util import project_root
|
|
|
|
holo_en: dict[int, str] = dict()
|
|
holo_id: dict[int, str] = dict()
|
|
niji_en: dict[int, str] = dict()
|
|
niji_exid: dict[int, str] = dict()
|
|
talents: dict[int, str] = dict()
|
|
talents_company: dict[int, str] = dict()
|
|
privated_accounts: dict[int, str] = dict()
|
|
|
|
test_talents = dict()
|
|
|
|
|
|
# TODO: talents(id) -> (name, company)
|
|
def __create_dict(file, _dict, company):
|
|
print(f"Initializing talents' account list from {file}...")
|
|
global talents
|
|
with open(file, "r") as f:
|
|
for line in f:
|
|
words = line.split()
|
|
if len(words) >= 2 and line[0] != "#":
|
|
t = line.split()
|
|
id, name = int(t[0]), t[1]
|
|
# name = f'{util.get_username_online(id, default=name)}' # attempt to get updated name
|
|
talents[id] = name
|
|
_dict[id] = name
|
|
talents_company[id] = company
|
|
if len(words) > 2 and words[2] == "p":
|
|
privated_accounts[id] = name
|
|
|
|
|
|
def init():
|
|
global holo_en
|
|
global holo_id
|
|
global niji_en
|
|
global niji_exid
|
|
global test_talents
|
|
|
|
# holoEN
|
|
__create_dict(project_root(("lists",), "holoen.txt"), holo_en, "holoEN")
|
|
# holoID
|
|
__create_dict(project_root(("lists",), "holoid.txt"), holo_id, "holoID")
|
|
# nijiEN
|
|
__create_dict(project_root(("lists",), "nijien.txt"), niji_en, "nijiEN")
|
|
# nijiexID
|
|
__create_dict(project_root(("lists",), "nijiexid.txt"), niji_exid, "nijiex'ID")
|
|
# TODO: nijiex-KR
|
|
|
|
test_talents = holo_en
|
|
|
|
|
|
def is_niji(id: int) -> bool:
|
|
return id in niji_en or id in niji_exid
|
|
|
|
|
|
def is_holo(id: int) -> bool:
|
|
return id in holo_en or id in holo_id
|
|
|
|
|
|
def is_cross_company(id1: int, id2: int):
|
|
return (is_niji(id1) and is_holo(id2)) or (is_holo(id1) and is_niji(id2))
|
|
|
|
|
|
# For filtered stream
|
|
# DEPRECATED: thx elon
|
|
def get_twitter_rules():
|
|
global talents
|
|
rules = list()
|
|
|
|
names = list(talents.values())
|
|
curr_rule = f"from:{names}"
|
|
for name in list(talents.values())[1:]:
|
|
test_rule = curr_rule + f" OR from:{name}"
|
|
if len(test_rule) > 512:
|
|
rules.append(curr_rule)
|
|
curr_rule = f"from:{name}"
|
|
else:
|
|
curr_rule = test_rule
|
|
rules.append(curr_rule)
|
|
return rules
|