hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-24 14:06:36 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Proof of concept for automated param removal

Browse Source
This commit is contained in:
tcely 2024-12-12 10:35:46 -05:00 committed by GitHub
parent 7b033d6e62
commit 32dead2126
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tubesync/tubesync/sqlite3/base.py
View File

@ -27,11 +27,24 @@ class DatabaseWrapper(base.DatabaseWrapper):
def get_new_connection(self, conn_params):
filter_map = {
"init_command": None,
"transaction_mode": ("isolation_level", "DEFERRED"),
}
filtered_params = {k: v for (k,v) in conn_params.items() if k not in filter_map}
filtered_params.update({v[0]: conn_params.get(k, v[1]) for (k,v) in filter_map.items() if v is not None})
return super().get_new_connection(filtered_params)
attempt = 0
connection = None
tries = len(filtered_params)
while connection is None and attempt < tries:
try:
attempt += 1
connection = super().get_new_connection(filtered_params)
except TypeError as e:
# remove unaccepted param
print(e, flush=True)
print('Exception args:', flush=True)
print(e.args, flush=True)
del filtered_params["init_command"]
return connection