52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
|
from abc import ABC, abstractmethod
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from updater.site.enum import GameVersion
|
||
|
|
|
||
|
|
|
||
|
|
class SiteError(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class AbstractSite(ABC):
|
||
|
|
# each implementation should declare a static _URLS list of
|
||
|
|
_URLS: [str] = None
|
||
|
|
# each implementation should create a static session for itself
|
||
|
|
session: requests.Session = None
|
||
|
|
|
||
|
|
def __init__(self, url: str, game_version: GameVersion):
|
||
|
|
self.url = url
|
||
|
|
self.game_version = game_version
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def handles(cls, url: str) -> bool:
|
||
|
|
return any(supported_url in url for supported_url in cls.get_supported_urls())
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_supported_urls(cls) -> [str]:
|
||
|
|
if not cls._URLS:
|
||
|
|
raise NotImplementedError(f"Can't instantiate class {cls.__name__}"
|
||
|
|
" without list of supported URLs cls._URLS")
|
||
|
|
return cls._URLS
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def find_zip_url(self) -> str:
|
||
|
|
pass
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def get_latest_version(self) -> str:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_addon_name(self) -> str:
|
||
|
|
name = self.url
|
||
|
|
for url in self.get_supported_urls():
|
||
|
|
name = name.replace(url, '')
|
||
|
|
return name
|
||
|
|
|
||
|
|
def download_error(self) -> SiteError:
|
||
|
|
return SiteError(f"Failed to find downloadable file for game version: {self.game_version}, {self.url}")
|
||
|
|
|
||
|
|
def version_error(self) -> SiteError:
|
||
|
|
return SiteError(f"Failed to find addon version number for game version: {self.game_version}, {self.url}")
|