RcGcDb/src/domain.py

from __future__ import annotations
import asyncio
from collections import OrderedDict
from typing import TYPE_CHECKING, Optional


if TYPE_CHECKING:
    import src.wiki
    import src.wiki_ratelimiter
    import irc.client_aio


class Domain:
    def __init__(self, name: str):
        self.name = name  # This should be always in format of topname.extension for example fandom.com
        self.task: asyncio.Task = self.create_task()
        self.wikis: OrderedDict[str, src.wiki.Wiki] = OrderedDict()
        self.rate_limiter: src.wiki_ratelimiter = src.wiki_ratelimiter.RateLimiter()
        self.irc = None

    def __iter__(self):
        return iter(self.wikis)

    def __getitem__(self, item):
        return

    def set_irc(self, irc_client: irc.client_aio.AioSimpleIRCClient):
        self.irc = irc_client

    def add_wiki(self, wiki: src.wiki.Wiki, first=False):
        """Adds a wiki to domain list.

        :parameter wiki - Wiki object
        :parameter first (optional) - bool indicating if wikis should be added as first or last in the ordered dict"""
        self.wikis[wiki.script_url] = wiki
        if first:
            self.wikis.move_to_end(wiki.script_url, last=False)

    def create_task(self) -> asyncio.Task:
        return asyncio.create_task(self.run_wiki_check())

    async def run_wiki_check(self):
        raise NotImplementedError