diff --git a/.gitignore b/.gitignore index 397f345..33a7451 100644 --- a/.gitignore +++ b/.gitignore @@ -10,3 +10,5 @@ ignore/ plugins/signatures.json .venv/ + +user_data/ \ No newline at end of file diff --git a/main.py b/main.py index f039e08..18abf79 100644 --- a/main.py +++ b/main.py @@ -1040,6 +1040,9 @@ def settings(): if success == None: success = "" + if not os.path.exists(".git"): + return render_template("settings.html", account=account,sync=account_module.getNodeSync(), + error=error,success=success,version="Error") info = gitinfo.get_git_info() branch = info['refs'] if branch != "main": diff --git a/plugin.py b/plugin.py index 659c43e..c2d5edc 100644 --- a/plugin.py +++ b/plugin.py @@ -5,7 +5,6 @@ import sys import hashlib - def listPlugins(): plugins = [] for file in os.listdir("plugins"): @@ -27,7 +26,7 @@ def listPlugins(): # Write a new signatures file with open("plugins/signatures.json", "w") as f: json.dump(signatures, f) - + for plugin in plugins: # Hash the plugin file pluginHash = hashPlugin(plugin["link"]) @@ -45,6 +44,7 @@ def pluginExists(plugin: str): return True return False + def verifyPlugin(plugin: str): signatures = [] try: @@ -73,12 +73,6 @@ def hashPlugin(plugin: str): break sha256.update(data) return sha256.hexdigest() - - - - - - def getPluginData(pluginStr: str): @@ -101,18 +95,20 @@ def getPluginData(pluginStr: str): info["verified"] = False else: info["verified"] = True - - return info + + return info + def getPluginFunctions(plugin: str): plugin = importlib.import_module("plugins."+plugin) return plugin.functions + def runPluginFunction(plugin: str, function: str, params: dict, authentication: str): plugin_module = importlib.import_module("plugins."+plugin) if function not in plugin_module.functions: return {"error": "Function not found"} - + if not hasattr(plugin_module, function): return {"error": "Function not found"} @@ -134,7 +130,6 @@ def runPluginFunction(plugin: str, function: str, params: dict, authentication: if pluginHash not in signatures: return {"error": "Plugin not verified"} - # Call the function with provided parameters try: result = plugin_function(params, authentication) @@ -144,14 +139,17 @@ def runPluginFunction(plugin: str, function: str, params: dict, authentication: return {"error": str(e)} # return plugin.runFunction(function, params, authentication) + def getPluginFunctionInputs(plugin: str, function: str): plugin = importlib.import_module("plugins."+plugin) return plugin.functions[function]["params"] + def getPluginFunctionReturns(plugin: str, function: str): plugin = importlib.import_module("plugins."+plugin) return plugin.functions[function]["returns"] + def getDomainFunctions(): plugins = listPlugins() domainFunctions = [] @@ -166,6 +164,7 @@ def getDomainFunctions(): }) return domainFunctions + def getSearchFunctions(): plugins = listPlugins() searchFunctions = [] @@ -180,6 +179,7 @@ def getSearchFunctions(): }) return searchFunctions + def getDashboardFunctions(): plugins = listPlugins() dashboardFunctions = [] @@ -192,4 +192,4 @@ def getDashboardFunctions(): "function": function, "description": functions[function]["description"] }) - return dashboardFunctions \ No newline at end of file + return dashboardFunctions diff --git a/plugins/automations.py b/plugins/automations.py index 9324224..a85e9b5 100644 --- a/plugins/automations.py +++ b/plugins/automations.py @@ -34,6 +34,45 @@ functions = { "type": "text" } } + }, + "disable":{ + "name": "Disable Automations", + "type": "default", + "description": "Disable Automations for this wallet", + "params": {}, + "returns": { + "Status": + { + "name": "Status", + "type": "text" + } + } + }, + "enable":{ + "name": "Enable Automations", + "type": "default", + "description": "Enable Automations for this wallet", + "params": {}, + "returns": { + "Status": + { + "name": "Status", + "type": "text" + } + } + }, + "list":{ + "name": "List Disabled Wallets", + "type": "default", + "description": "List wallets with automations disabled", + "params": {}, + "returns": { + "wallets": + { + "name": "List of wallets", + "type": "list" + } + } } } @@ -43,16 +82,43 @@ started = False def automation(params, authentication): global started + wallet = authentication.split(":")[0] + if os.path.exists(f"user_data/{wallet}.autoRenew"): + return {"Status": "Automations disabled"} + if started: - return {"Status": "Auto Renews running"} - started = True + return {"Status": "Automations running"} + started = True + threading.Thread(target=automations_background, args=(authentication,)).start() - return {"Status": "Started Auto Renews"} + return {"Status": "Starting Automations..."} + +def disable(params, authentication): + # Create walletname file in user_data + wallet = authentication.split(":")[0] + with open(f"user_data/{wallet}.autoRenew", "w") as f: + f.write(f"This file is used to disable automations for '{wallet}' wallet.\nDelete this file to enable automations.") + return {"Status": "Disabled Automations"} + +def enable(params, authentication): + # Delete walletname file in user_data + wallet = authentication.split(":")[0] + if os.path.exists(f"user_data/{wallet}.autoRenew"): + os.remove(f"user_data/{wallet}.autoRenew") + + return {"Status": "Enabled Automations"} + +def list(params, authentication): + wallets = [] + for file in os.listdir("user_data"): + if file.endswith(".autoRenew"): + wallets.append(file[:-10]) + return {"wallets": wallets} # Background function to run the automations def automations_background(authentication): + while True: - print("Running automations") # Get account details account_name = account.check_account(authentication) password = ":".join(authentication.split(":")[1:]) @@ -63,13 +129,18 @@ def automations_background(authentication): "message": "Invalid account" } } - + + if os.path.exists(f"user_data/{account_name}.autoRenew"): + print("Skipping Automations") + time.sleep(300) + continue + print("Running automations") try: # Try to select and login to the wallet response = account.hsw.rpc_selectWallet(account_name) if response['error'] is not None: return - response = account.hsw.rpc_walletPassphrase(password,10) + response = account.hsw.rpc_walletPassphrase(password,30) if response['error'] is not None: return # Try to send the batch of all renew, reveal and redeem actions