mirror of
https://gitlab.com/chicken-riders/RcGcDb.git
synced 2025-02-23 00:54:09 +00:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
from typing import TYPE_CHECKING, Optional
|
||
|
|
from urllib.parse import urlparse, urlunparse
|
||
|
|
from src.config import settings
|
||
|
|
from src.domain import Domain
|
||
|
|
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from src.wiki import Wiki
|
||
|
|
|
||
|
|
|
||
|
|
class DomainManager:
|
||
|
|
def __init__(self):
|
||
|
|
self.domains: dict[str, Domain] = {}
|
||
|
|
|
||
|
|
def new_wiki(self, wiki: Wiki):
|
||
|
|
"""Finds a domain for the wiki and adds a wiki to the domain object.
|
||
|
|
|
||
|
|
:parameter wiki - Wiki object to be added"""
|
||
|
|
wiki_domain = self.get_domain(wiki.script_url)
|
||
|
|
try:
|
||
|
|
self.domains[wiki_domain].add_wiki(wiki)
|
||
|
|
except KeyError:
|
||
|
|
self.new_domain(wiki_domain).add_wiki(wiki)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_domain(url: str) -> str:
|
||
|
|
"""Returns a domain for given URL (for example fandom.com, wikipedia.org)"""
|
||
|
|
parsed_url = urlparse(url)
|
||
|
|
return ".".join(urlunparse((*parsed_url[0:2], "", "", "", "")).split(".")[-2:])
|
||
|
|
|
||
|
|
def new_domain(self, name: str) -> Domain:
|
||
|
|
# TODO IRC Part
|
||
|
|
self.domains[name] = Domain(name, irc)
|
||
|
|
return self.domains[name]
|
||
|
|
|
||
|
|
|
||
|
|
domains = DomainManager()
|