finnius/CDRM-Project
1
0
Fork 0
forked from tpd94/CDRM-Project
Code Pull Requests Activity

Add 'kid' and 'pssh' search operators to /api/cache/search endpoint sqlite

Browse Source
This commit is contained in:
finnius 2025-06-13 01:50:07 +00:00
parent e4fb07348c
commit 4f789dfa7a
1 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
custom_functions/database/cache_to_db_sqlite.py
View File

@ -43,21 +43,28 @@ def search_by_pssh_or_kid(search_filter):
# Initialize a set to store unique matching records # Initialize a set to store unique matching records
results = set() results = set()
if search_filter.lower().startswith("kid:"):
value = search_filter[4:]
# Search for records where KID matches the operator value
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE KID = ?', (value,))
results.update(cursor.fetchall())
elif search_filter.lower().startswith("pssh:"):
value = search_filter[5:]
# Search for records where PSSH contains the operator
cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE PSSH LIKE ?', (f"%{value}%",))
results.update(cursor.fetchall())
else:
like_filter = f"%{search_filter}%"
# Search for records where PSSH contains the search_filter # Search for records where PSSH contains the search_filter
cursor.execute(''' cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE PSSH LIKE ?', (like_filter,))
SELECT * FROM licenses WHERE PSSH LIKE ? results.update(cursor.fetchall())
''', ('%' + search_filter + '%',))
rows = cursor.fetchall()
for row in rows:
results.add((row[1], row[2], row[3])) # (PSSH, KID, Key)
# Search for records where KID contains the search_filter # Search for records where KID contains the search_filter
cursor.execute(''' cursor.execute('SELECT PSSH, KID, `Key` FROM licenses WHERE KID LIKE ?', (like_filter,))
SELECT * FROM licenses WHERE KID LIKE ? results.update(cursor.fetchall())
''', ('%' + search_filter + '%',))
rows = cursor.fetchall()
for row in rows:
results.add((row[1], row[2], row[3])) # (PSSH, KID, Key)
# Convert the set of results to a list of dictionaries for output # Convert the set of results to a list of dictionaries for output
final_results = [{'PSSH': result[0], 'KID': result[1], 'Key': result[2]} for result in results] final_results = [{'PSSH': result[0], 'KID': result[1], 'Key': result[2]} for result in results]