Merge pull request 'feat: add return type annotations to async DB functions (#20)' (#47) from ai/paper-dynasty-card-creation-20 into main

This commit is contained in:
cal 2026-03-23 19:52:02 +00:00
commit 8c00bacf59

View File

@ -4,7 +4,7 @@ import aiohttp
import pybaseball as pb
from dotenv import load_dotenv
from typing import Literal
from typing import Literal, Optional
from exceptions import logger
load_dotenv()
@ -62,7 +62,7 @@ async def db_get(
params: list = None,
none_okay: bool = True,
timeout: int = 3,
):
) -> Optional[dict]:
req_url = get_req_url(endpoint, api_ver=api_ver, object_id=object_id, params=params)
log_string = f"get:\n{endpoint} id: {object_id} params: {params}"
logger.info(log_string) if master_debug else logger.debug(log_string)
@ -86,7 +86,7 @@ async def db_get(
raise ValueError(f"DB: {e}")
async def url_get(url: str, timeout: int = 3):
async def url_get(url: str, timeout: int = 3) -> dict:
log_string = f"get:\n{url}"
logger.info(log_string) if master_debug else logger.debug(log_string)
@ -106,7 +106,7 @@ async def url_get(url: str, timeout: int = 3):
async def db_patch(
endpoint: str, object_id: int, params: list, api_ver: int = 2, timeout: int = 3
):
) -> dict:
req_url = get_req_url(endpoint, api_ver=api_ver, object_id=object_id, params=params)
log_string = f"patch:\n{endpoint} {params}"
logger.info(log_string) if master_debug else logger.debug(log_string)
@ -127,7 +127,7 @@ async def db_patch(
async def db_post(
endpoint: str, api_ver: int = 2, payload: dict = None, timeout: int = 3
):
) -> dict:
req_url = get_req_url(endpoint, api_ver=api_ver)
log_string = f"post:\n{endpoint} payload: {payload}\ntype: {type(payload)}"
logger.info(log_string) if master_debug else logger.debug(log_string)
@ -148,7 +148,7 @@ async def db_post(
async def db_put(
endpoint: str, api_ver: int = 2, payload: dict = None, timeout: int = 3
):
) -> dict:
req_url = get_req_url(endpoint, api_ver=api_ver)
log_string = f"put:\n{endpoint} payload: {payload}\ntype: {type(payload)}"
logger.info(log_string) if master_debug else logger.debug(log_string)
@ -167,7 +167,7 @@ async def db_put(
raise ValueError(f"DB: {e}")
async def db_delete(endpoint: str, object_id: int, api_ver: int = 2, timeout=3):
async def db_delete(endpoint: str, object_id: int, api_ver: int = 2, timeout=3) -> dict:
req_url = get_req_url(endpoint, api_ver=api_ver, object_id=object_id)
log_string = f"delete:\n{endpoint} {object_id}"
logger.info(log_string) if master_debug else logger.debug(log_string)