voussoir/ycdl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Improve normalize_db_row by making it more strict on types.

Browse source
This commit is contained in:
voussoir 2021-08-26 21:32:44 -07:00
parent 8af64598a2
commit 46be4f8cff
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
ycdl/objects.py
View file

@ -4,10 +4,19 @@ from . import constants
from . import exceptions from . import exceptions
from . import ytrss from . import ytrss
def normalize_db_row(db_row, table): def normalize_db_row(db_row, table) -> dict:
'''
Raises KeyError if table is not one of the recognized tables.
Raises TypeError if db_row is not the right type.
'''
if isinstance(db_row, dict):
return db_row
if isinstance(db_row, (list, tuple)): if isinstance(db_row, (list, tuple)):
db_row = dict(zip(constants.SQL_COLUMNS[table], db_row)) return dict(zip(constants.SQL_COLUMNS[table], db_row))
return db_row
raise TypeError(f'db_row should be {dict}, {list}, or {tuple}, not {type(db_row)}.')
class Base: class Base:
def __init__(self, ycdldb): def __init__(self, ycdldb):