lucky/GeminiKeyManagement
1
0
Fork 0
mirror of https://github.com/not-lucky/GeminiKeyManagement.git synced 2025-12-06 08:44:01 +05:30
Code Issues Packages Projects Releases Wiki Activity

save to database after processing an account

Browse Source
This commit is contained in:
not-lucky
2025-08-25 13:53:13 +05:30
parent c493d72187
commit 5311c68c58
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gemini_key_manager/actions.py
View File

@@ -255,7 +255,7 @@ def process_project_for_action(project, creds, action, dry_run, db_lock, account
logging.info(f"Completed processing {project_id}")
def process_account(email, creds, action, api_keys_data, dry_run=False, max_workers=5):
def process_account(email, creds, action, api_keys_data, schema, dry_run=False, max_workers=5):
"""Orchestrates account-level key management operations.
Args:
@@ -263,6 +263,7 @@ def process_account(email, creds, action, api_keys_data, dry_run=False, max_work
creds (Credentials): Authenticated credentials
action (str): 'create' or 'delete' action
api_keys_data (dict): Database structure
schema (dict): The database schema.
dry_run (bool): Simulation mode flag
max_workers (int): Max concurrent operations
"""
@@ -325,3 +326,6 @@ def process_account(email, creds, action, api_keys_data, dry_run=False, max_work
logging.error(f"Permission denied for {email}: {err}")
except google_exceptions.GoogleAPICallError as err:
logging.error(f"API error processing {email}: {err}")
if not dry_run:
database.save_keys_to_json(api_keys_data, config.API_KEYS_DATABASE_FILE, schema)