Compare commits

...

5 Commits

Author SHA1 Message Date
aa92220756
Merge branch 'dev'
All checks were successful
Build Docker / Build Image (push) Successful in 41s
2024-11-21 19:35:56 +11:00
3290240a25
fix: Cleanup plugin.py
All checks were successful
Build Docker / Build Image (push) Successful in 1m13s
2024-11-21 19:35:33 +11:00
40f520ba5e
feat: Updated automations blacklist file to make it clearer
All checks were successful
Build Docker / Build Image (push) Successful in 44s
2024-11-21 19:33:10 +11:00
9f1f7fb18e
feat: Add wallet blacklisting to automations plugin
All checks were successful
Build Docker / Build Image (push) Successful in 46s
2024-11-21 19:30:19 +11:00
3e78732333
fix: Check if .git dir exists before trying to read it 2024-11-21 16:30:44 +11:00
4 changed files with 95 additions and 19 deletions

2
.gitignore vendored
View File

@ -10,3 +10,5 @@ ignore/
plugins/signatures.json
.venv/
user_data/

View File

@ -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":

View File

@ -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
return dashboardFunctions

View File

@ -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