d23f32f08d
format: {time:YYYY-MM-DD HH:mm:ss} [{level}] | {name} | {message}
{name} is:
- Uvicorn: log from uvicorn server (#228B22)
- Service: log from class of `app.service` (blue)
- Fetcher: log from fetchers (magenta)
- Task: log from `app.tasks` (#FFD700)
- System: log from `system_logger` (red)
- Normal: log from `log(name)` (#FFC1C1)
- Default: the module name of caller
if you are writing services or tasks, you can just call `logger.`, we will pack it with name `Service` or `Task`
if you want to print fetcher logs, system-related logs, or normal logs, use `logger = (fetcher_logger / system_logger / log)(name)`
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Annotated
|
|
|
|
from app.config import settings
|
|
from app.dependencies.database import get_redis
|
|
from app.fetcher import Fetcher as OriginFetcher
|
|
from app.log import fetcher_logger
|
|
|
|
from fastapi import Depends
|
|
|
|
fetcher: OriginFetcher | None = None
|
|
|
|
|
|
async def get_fetcher() -> OriginFetcher:
|
|
global fetcher
|
|
if fetcher is None:
|
|
fetcher = OriginFetcher(
|
|
settings.fetcher_client_id,
|
|
settings.fetcher_client_secret,
|
|
settings.fetcher_scopes,
|
|
settings.fetcher_callback_url,
|
|
)
|
|
redis = get_redis()
|
|
access_token = await redis.get(f"fetcher:access_token:{fetcher.client_id}")
|
|
if access_token:
|
|
fetcher.access_token = str(access_token)
|
|
refresh_token = await redis.get(f"fetcher:refresh_token:{fetcher.client_id}")
|
|
if refresh_token:
|
|
fetcher.refresh_token = str(refresh_token)
|
|
if not fetcher.access_token or not fetcher.refresh_token:
|
|
fetcher_logger("Fetcher").opt(colors=True).info(
|
|
f"Login to initialize fetcher: <y>{fetcher.authorize_url}</y>"
|
|
)
|
|
return fetcher
|
|
|
|
|
|
Fetcher = Annotated[OriginFetcher, Depends(get_fetcher)]
|