tpd94/CDRM-Project
1
0
Fork 1
Code Issues 1 Pull Requests 1 Actions Packages Projects Releases 3 Wiki Activity

Add search operators #10

Open
finnius wants to merge 2 commits from finnius/CDRM-Project:search-operators into main
pull from: finnius/CDRM-Project:search-operators
merge into: tpd94:main
Conversation 0 Commits 2 Files Changed 2 +16 -5
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 2dbedd7206 - Show all commits

21
custom_functions/database/cache_to_db_mariadb.py
View File

@ -70,13 +70,24 @@ def search_by_pssh_or_kid(search_filter):
try:
with mysql.connector.connect(**get_db_config()) as conn:
cursor = conn.cursor()
like_filter = f"%{search_filter}%"
if search_filter.lower().startswith("kid:"):
value = search_filter[4:]
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE PSSH LIKE %s', (like_filter,))
results.update(cursor.fetchall())
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE KID = %s', (value,))
results.update(cursor.fetchall())
elif search_filter.lower().startswith("pssh:"):
value = search_filter[5:]
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE KID LIKE %s', (like_filter,))
results.update(cursor.fetchall())
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE PSSH LIKE %s', (f"%{value}%",))
results.update(cursor.fetchall())
else:
like_filter = f"%{search_filter}%"
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE PSSH LIKE %s', (like_filter,))
results.update(cursor.fetchall())
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE KID LIKE %s', (like_filter,))
results.update(cursor.fetchall())
final_results = [{'PSSH': row[0], 'KID': row[1], 'Key': row[2]} for row in results]
return final_results[:20]