mirror of https://github.com/jerome158/LIVETV.git
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
import os
|
|
|
|
import utils.constants as constants
|
|
from utils.tools import get_real_path, resource_path, format_name
|
|
|
|
|
|
class Alias:
|
|
def __init__(self):
|
|
self.primary_to_aliases: dict[str, set[str]] = {}
|
|
self.alias_to_primary: dict[str, str] = {}
|
|
|
|
real_path = get_real_path(resource_path(constants.alias_path))
|
|
if os.path.exists(real_path):
|
|
with open(real_path, "r", encoding="utf-8") as f:
|
|
for line in f:
|
|
if line.strip() and not line.startswith("#") and "," in line:
|
|
parts = [p.strip() for p in line.split(",")]
|
|
primary = parts[0]
|
|
aliases = set(parts[1:])
|
|
aliases.add(format_name(primary))
|
|
self.primary_to_aliases[primary] = aliases
|
|
for alias in aliases:
|
|
self.alias_to_primary[alias] = primary
|
|
self.alias_to_primary[primary] = primary
|
|
|
|
def get(self, name: str):
|
|
"""
|
|
Get the alias by name
|
|
"""
|
|
return self.primary_to_aliases.get(name, set())
|
|
|
|
def get_primary(self, name: str):
|
|
"""
|
|
Get the primary name by alias
|
|
"""
|
|
primary_name = self.alias_to_primary.get(name, None)
|
|
if primary_name is None:
|
|
alias_format_name = format_name(name)
|
|
primary_name = self.alias_to_primary.get(alias_format_name, name)
|
|
return primary_name
|
|
|
|
def set(self, name: str, aliases: set[str]):
|
|
"""
|
|
Set the aliases by name
|
|
"""
|
|
if name in self.primary_to_aliases:
|
|
for alias in self.primary_to_aliases[name]:
|
|
self.alias_to_primary.pop(alias, None)
|
|
self.primary_to_aliases[name] = set(aliases)
|
|
for alias in aliases:
|
|
self.alias_to_primary[alias] = name
|
|
self.alias_to_primary[name] = name
|