# ADMINISTRARIVE DB MANAGEMENT import time from datetime import timedelta from nostr_sdk import Keys, EventBuilder, PublicKey, Client, Filter, EventId from utils.database_utils import get_from_sql_table, list_db, delete_from_sql_table, update_sql_table, \ get_or_add_user, clean_db from utils.dvmconfig import DVMConfig from utils.nip89_utils import nip89_announce_tasks, NIP89Config, nip89_delete_announcement, \ fetch_nip89_paramters_for_deletion from utils.nostr_utils import send_event, update_profile class AdminConfig: REBROADCAST_NIP89: bool = False UPDATE_PROFILE: bool = False WHITELISTUSER: bool = False UNWHITELISTUSER: bool = False BLACKLISTUSER: bool = False DELETEUSER: bool = False LISTDATABASE: bool = False ClEANDB: bool = False USERNPUB: str = "" LUD16: str = "" def admin_make_database_updates(adminconfig: AdminConfig = None, dvmconfig: DVMConfig = None, client: Client = None): # This is called on start of Server, Admin function to manually whitelist/blacklist/add balance/delete users if adminconfig is None or dvmconfig is None: return if not isinstance(adminconfig, AdminConfig): return if (( adminconfig.WHITELISTUSER is True or adminconfig.UNWHITELISTUSER is True or adminconfig.BLACKLISTUSER is True or adminconfig.DELETEUSER is True) and adminconfig.USERNPUB == ""): return if adminconfig.UPDATE_PROFILE and (dvmconfig.NIP89 is None): return db = dvmconfig.DB if str(adminconfig.USERNPUB).startswith("npub"): publickey = PublicKey.from_bech32(adminconfig.USERNPUB).to_hex() else: publickey = adminconfig.USERNPUB if adminconfig.WHITELISTUSER: user = get_or_add_user(db, publickey, client=client, config=dvmconfig) update_sql_table(db, user.npub, user.balance, True, False, user.nip05, user.lud16, user.name, user.lastactive) user = get_from_sql_table(db, publickey) print(str(user.name) + " is whitelisted: " + str(user.iswhitelisted)) if adminconfig.UNWHITELISTUSER: user = get_from_sql_table(db, publickey) update_sql_table(db, user.npub, user.balance, False, False, user.nip05, user.lud16, user.name, user.lastactive) if adminconfig.BLACKLISTUSER: user = get_from_sql_table(db, publickey) update_sql_table(db, user.npub, user.balance, False, True, user.nip05, user.lud16, user.name, user.lastactive) if adminconfig.DELETEUSER: delete_from_sql_table(db, publickey) if adminconfig.ClEANDB: clean_db(db) if adminconfig.LISTDATABASE: list_db(db) if adminconfig.REBROADCAST_NIP89: nip89_announce_tasks(dvmconfig, client=client) # TODO make this callable delete_previous_announcement = False if delete_previous_announcement: eventid = "" #Id of Event to delete keys = Keys.from_sk_str("") # Private key from sender of Event (e.g. the key of an nip89 announcement you want to delete) fetch_nip89_paramters_for_deletion(keys, eventid, client, dvmconfig) if adminconfig.UPDATE_PROFILE: update_profile(dvmconfig, lud16=adminconfig.LUD16)