NijiHolo_EN_ID_Bot/src/talent_lists.py

import util

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()

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] != '#':
                id, name = line.split()
                # name = f'{util.get_username_online(id, default=name)}' # attempt to get updated name
                talents[int(id)] = name
                _dict[int(id)] = name
                talents_company[int(id)] = company
def init():
    global holo_en
    global holo_id
    global niji_en
    global niji_exid
    global test_talents

    # holoEN
    __create_dict(f'{util.get_project_dir()}/lists/holoen.txt', holo_en, 'holoEN')
    # holoID
    __create_dict(f'{util.get_project_dir()}/lists/holoid.txt', holo_id, 'holoID')
    # nijiEN
    __create_dict(f'{util.get_project_dir()}/lists/nijien.txt', niji_en, 'nijiEN')
    # nijiexID
    __create_dict(f'{util.get_project_dir()}/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