Polishing

This commit is contained in:
Taras Terletskyi 2022-10-18 02:37:39 +03:00
parent 397d6921ac
commit 415a2025be
3 changed files with 5 additions and 5 deletions

View file

@ -3,14 +3,14 @@ from sqlalchemy.ext.asyncio import AsyncSession
from api.api_v1.schemas.ytdlp import YTDLPLatestVersion
from yt_shared.db import get_db
from yt_shared.ytdlp.version_checker import VersionChecker
from yt_shared.ytdlp.version_checker import YtdlpVersionChecker
router = APIRouter()
@router.get('/', response_model=YTDLPLatestVersion, response_model_by_alias=False)
async def yt_dlp_version(db: AsyncSession = Depends(get_db)):
version_checker = VersionChecker()
version_checker = YtdlpVersionChecker()
ctx = await version_checker.get_version_context(db)
return YTDLPLatestVersion(
current=ctx.current, latest=ctx.latest, need_upgrade=ctx.has_new_version

View file

@ -9,7 +9,7 @@ from yt_shared.db import get_db
from yt_shared.emoji import INFORMATION_EMOJI
from yt_shared.schemas.ytdlp import VersionContext
from yt_shared.task_utils.abstract import AbstractTask
from yt_shared.ytdlp.version_checker import VersionChecker
from yt_shared.ytdlp.version_checker import YtdlpVersionChecker
if TYPE_CHECKING:
from core.bot import VideoBot
@ -19,7 +19,7 @@ class YtdlpNewVersionNotifyTask(AbstractTask):
def __init__(self, bot: 'VideoBot') -> None:
super().__init__()
self._bot = bot
self._version_checker = VersionChecker()
self._version_checker = YtdlpVersionChecker()
self._startup_message_sent = False
async def run(self) -> None:

View file

@ -13,7 +13,7 @@ from yt_shared.schemas.ytdlp import (
)
class VersionChecker:
class YtdlpVersionChecker:
"""yt-dlp version number checker."""
LATEST_TAG_URL = 'https://github.com/yt-dlp/yt-dlp/releases/latest'