main.py

from typing import Optional, Type, Union

from py2neo.integration import Table
from telethon.sync import TelegramClient

from core.tg_api_connector import ChannelParser
from db.dal import GraphManager
from models import FetchedChannel


async def entry(client: TelegramClient, channel: Union[str, int]) -> FetchedChannel:
    parser = ChannelParser(client)
    await parser.start()
    await parser.join_channel(channel)
    channel_instance = await parser.get_all_participants(channel)
    return channel_instance


class DataManager:

    @staticmethod
    async def load_data(client: TelegramClient, channel: Union[str, int]) -> None:
        fetched_channel = await entry(client, channel)
        GraphManager.add_fetched_channel(fetched_channel)

    @staticmethod
    async def add_user(user: tuple, groups: list) -> None:
        GraphManager.add_user(user, groups)

    # @staticmethod
    # async def add_user(user: tuple) -> None:
    #    GraphManager.add_user(user)  # TODO return to previous params

    @staticmethod
    async def create_relationships() -> None:
        GraphManager.create_relationships()

    @staticmethod
    async def get_data(
        query: str, n: Optional[int] = None
    ) -> Union[Table, dict, Type["DataFrame"]]:
        return GraphManager.fetch_data(query, n)