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}") 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. """Orchestrates account-level key management operations.
Args: Args:
@@ -263,6 +263,7 @@ def process_account(email, creds, action, api_keys_data, dry_run=False, max_work
creds (Credentials): Authenticated credentials creds (Credentials): Authenticated credentials
action (str): 'create' or 'delete' action action (str): 'create' or 'delete' action
api_keys_data (dict): Database structure api_keys_data (dict): Database structure
schema (dict): The database schema.
dry_run (bool): Simulation mode flag dry_run (bool): Simulation mode flag
max_workers (int): Max concurrent operations 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}") logging.error(f"Permission denied for {email}: {err}")
except google_exceptions.GoogleAPICallError as err: except google_exceptions.GoogleAPICallError as err:
logging.error(f"API error processing {email}: {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)

5
gemini_key_manager/main.py
View File

@@ -91,9 +91,6 @@ def main():
for email in emails_to_process: for email in emails_to_process:
if email in creds_map: if email in creds_map:
actions.process_account(email, creds_map[email], args.action, api_keys_data, dry_run=args.dry_run, max_workers=args.max_workers) actions.process_account(email, creds_map[email], args.action, api_keys_data, schema, dry_run=args.dry_run, max_workers=args.max_workers)
else: else:
logging.warning(f"Skipping account {email} because authentication was not successful.") logging.warning(f"Skipping account {email} because authentication was not successful.")
if not args.dry_run:
database.save_keys_to_json(api_keys_data, config.API_KEYS_DATABASE_FILE, schema)