feat: add function to retrieve unique import settings for multiple arr IDs

This commit is contained in:
Sam Chau
2025-01-10 00:44:22 +10:30
parent 54e3e7499a
commit f39b693b3d

View File

@@ -156,3 +156,34 @@ def save_settings(settings_dict):
updated_at = CURRENT_TIMESTAMP
''', (key, value))
conn.commit()
def get_unique_arrs(arr_ids):
"""
Get import_as_unique settings for a list of arr IDs.
Args:
arr_ids (list): List of arr configuration IDs
Returns:
dict: Dictionary mapping arr IDs to their import_as_unique settings and names
"""
if not arr_ids:
return {}
with get_db() as conn:
placeholders = ','.join('?' * len(arr_ids))
query = f'''
SELECT id, name, import_as_unique
FROM arr_config
WHERE id IN ({placeholders})
'''
results = conn.execute(query, arr_ids).fetchall()
return {
row['id']: {
'import_as_unique': bool(row['import_as_unique']),
'name': row['name']
}
for row in results
}