Jukebox/jukebox-software
10
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases 2 Wiki Activity

get column parity implemented

Browse Source
This commit is contained in:
Camryn Thomas 2024-01-14 15:36:15 -06:00
parent bb19158aa0
commit 75fad013c0
Signed by: cptlobster
GPG Key ID: 6D341D688163A176
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
database.py
View File

@ -60,13 +60,18 @@ class DBConnector:
return rows return rows
def _column_parity(self, columns: list[str] | set[str]) -> set[str]: def _column_parity(self, columns: list[str] | set[str]) -> set[str]:
"""If the listed columns are not in the database, add them.""" """If the listed columns are not in the database, add them.
:param columns: The columns we expect are in the database.
:return: The list of columns in the database after querying."""
cols = set(columns) cols = set(columns)
existing = self._get_cols() existing = self._get_cols()
needs = cols.difference(existing.intersection(cols)) needs = cols.difference(existing.intersection(cols))
query = f"ALTER TABLE {DB_TABLE} {', '.join([f'ADD COLUMN {c}' for c in needs])}" if len(needs) > 0:
self._query(query) query = f"ALTER TABLE {DB_TABLE} {', '.join([f'ADD COLUMN {c}' for c in needs])}"
return self._get_cols() self._query(query)
existing = self._get_cols()
return existing
def _query(self, sql) -> list[dict]: def _query(self, sql) -> list[dict]:
"""Basic function for running queries. """Basic function for running queries.
@ -114,6 +119,7 @@ class DBConnector:
:param kwargs: Values to write for each database; specify each column separately! :param kwargs: Values to write for each database; specify each column separately!
:returns: The row you just added.""" :returns: The row you just added."""
self._column_parity(set(kwargs.keys()))
values = [] values = []
for val in kwargs.keys(): for val in kwargs.keys():
values.append(kwargs[val]) values.append(kwargs[val])