FazziCLAY
7495bd949b
All checks were successful
Build and Run VSTU Public TG bot / build_and_run (push) Successful in 27s
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import aiohttp
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class DataManager:
|
|
def __init__(self, facultets_url, groups_url, parser_url):
|
|
self.facultets_url = facultets_url
|
|
self.groups_url = groups_url
|
|
self.parser_url = parser_url
|
|
self.facs = {} # {id: {name: ...}}
|
|
self.groups = {} # {id: {real_name: ..., facultet_tech: ..., ...}}
|
|
self.parser = {}
|
|
|
|
async def refresh_cache(self, connector):
|
|
async with aiohttp.ClientSession(connector=connector) as session:
|
|
async def fetch(url):
|
|
async with session.get(url) as r: return await r.json()
|
|
try:
|
|
self.facs = await fetch(self.facultets_url)
|
|
self.groups = (await fetch(self.groups_url))['groups']
|
|
self.parser = (await fetch(self.parser_url))
|
|
|
|
logger.info(f"Loaded {len(self.facs)} facs and {len(self.groups)} groups")
|
|
except Exception as e:
|
|
logger.error(f"Cache refresh failed: {e}")
|
|
|
|
def get_groups_by_fac(self, fac_id):
|
|
return [
|
|
(gid, g['real_name']) for gid, g in self.groups.items()
|
|
if g.get('facultet_tech') == fac_id or g.get('facultet_recognized') == fac_id
|
|
]
|
|
|
|
def get_files_by_fac(self, fac_id):
|
|
files = {} # {uniqpath*: display_name}
|
|
i = -1
|
|
for ex in self.parser['all_files']:
|
|
i += 1
|
|
if ex.get('facultet') == fac_id:
|
|
files[i] = ex['display_filename']
|
|
|
|
return list(files.items())
|
|
|
|
def get_excel_by_uniqpath(self, uniqid):
|
|
for g in self.groups.values():
|
|
for ex in g.get('excels', []):
|
|
if uniqid == ex['uniqpath']:
|
|
return ex
|
|
|
|
return None
|
|
|
|
def find_group_by_name(self, name):
|
|
name = name.lower()
|
|
for gid, g in self.groups.items():
|
|
if g['real_name'].lower() == name:
|
|
return gid, g['real_name']
|
|
return None, None |