nicolas
/
DCAv2Earn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor api-key handling refactor

This commit is contained in:
Nicolás Sánchez 2025-08-19 20:12:26 -03:00
parent 9ed2927c0b
commit b383cde1fe
1 changed files with 255 additions and 271 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

526
main.py
View File

@ -8,6 +8,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
from threading import Thread from threading import Thread
from flask import Flask, jsonify, request from flask import Flask, jsonify, request
from waitress import serve from waitress import serve
from functools import wraps
import time import time
import datetime import datetime
import json import json
@ -18,7 +19,7 @@ import os
def load_keys_from_db(file_name: str) -> list: def load_keys_from_db(file_name: str) -> list:
''' '''
Load valid API keys Load valid API keys to a set
Parameters Parameters
---------- ----------
@ -37,7 +38,7 @@ def load_keys_from_db(file_name: str) -> list:
data = database_cursor.fetchall() data = database_cursor.fetchall()
database_connection.close() database_connection.close()
valid_keys = [line[1] for line in data] valid_keys = {line[1] for line in data}
return valid_keys return valid_keys
@ -126,8 +127,21 @@ def main():
earn_api = Flask(__name__) earn_api = Flask(__name__)
#Helper functions
def require_api_key(func):
'''
Validates API key
'''
@wraps(func)
def wrapper(*args, **kwargs):
key = request.headers.get("X-API-KEY")
if not key or key not in valid_keys:
return jsonify({'Error': 'API key not valid'}), 401
return func(*args, **kwargs)
return wrapper
@earn_api.route("/get_global_status", methods=['GET']) @earn_api.route("/get_global_status", methods=['GET'])
@require_api_key
def get_global_status(): def get_global_status():
''' '''
GET request GET request
@ -136,26 +150,25 @@ def get_global_status():
None None
''' '''
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: response = {}
response = {} for item in earners:
for item in earners: response[str(item.connector)] = {"currency": item.get_currency(),
response[str(item.connector)] = {"currency": item.get_currency(), "trading_balance": item.get_trading_balance(),
"trading_balance": item.get_trading_balance(), "earning_balance": item.get_earning_balance(),
"earning_balance": item.get_earning_balance(), "is_paused": item.get_is_paused(),
"is_paused": item.get_is_paused(), "step_size": item.get_step_size(),
"step_size": item.get_step_size(), "percentage": item.get_percentage(),
"percentage": item.get_percentage(), "minimum_amount_in_trading_account": item.get_minimum_amount_in_trading_account(),
"minimum_amount_in_trading_account": item.get_minimum_amount_in_trading_account(), "time_between_subscriptions": item.get_time_between_subscriptions(),
"time_between_subscriptions": item.get_time_between_subscriptions(), "time_between_redemptions": item.get_time_between_redemptions(),
"time_between_redemptions": item.get_time_between_redemptions(), "last_subscription": item.get_last_subscription(),
"last_subscription": item.get_last_subscription(), "last_redemption": item.get_last_redemption()}
"last_redemption": item.get_last_redemption()} response["uptime"] = time.time() - start_time
response["uptime"] = time.time() - start_time return jsonify(response)
return jsonify(response)
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/toggle_pause", methods=['POST']) @earn_api.route("/toggle_pause", methods=['POST'])
@require_api_key
def toggle_pause(): def toggle_pause():
''' '''
GET request GET request
@ -163,25 +176,23 @@ def toggle_pause():
Parameters: Parameters:
broker: str broker: str
''' '''
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: if request.json is None:
valid_brokers = [str(item.connector) for item in earners] return jsonify({'Error': 'request.json is None'})
if request.json is None: broker = request.json["broker"]
return jsonify({'Error': 'request.json is None'}) if broker is None:
broker = request.json["broker"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: item.toggle_pause()
if str(item.connector)==broker: return jsonify({'Status': item.get_is_paused()})
item.toggle_pause() return jsonify({'Error': 'broker not found'})
return jsonify({'Status': item.get_is_paused()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_step_size", methods=["GET"]) @earn_api.route("/get_step_size", methods=["GET"])
@require_api_key
def get_step_size(): def get_step_size():
''' '''
GET request GET request
@ -189,242 +200,216 @@ def get_step_size():
Parameters: Parameters:
broker: str broker: str
''' '''
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'step_size': item.get_step_size()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'step_size': item.get_step_size()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/set_step_size", methods=["POST"]) @earn_api.route("/set_step_size", methods=["POST"])
def set_step_size(): @require_api_key
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: def set_step_size():
valid_brokers = [str(item.connector) for item in earners] if request.json is None:
if request.json is None: return jsonify({'Error': 'request.json is None'})
return jsonify({'Error': 'request.json is None'}) broker = request.json["broker"]
broker = request.json["broker"] new_step_size = request.json["new_step_size"]
new_step_size = request.json["new_step_size"] if broker is None:
if broker is None: return jsonify({'Error': 'broker is None'})
return jsonify({'Error': 'broker is None'}) if broker not in valid_broker_list:
if broker not in valid_brokers: return jsonify({'Error': 'broker not valid'})
return jsonify({'Error': 'broker not valid'}) if new_step_size is None:
if new_step_size is None: return jsonify({'Error': 'new_step_size is None'})
return jsonify({'Error': 'new_step_size is None'}) for item in earners:
for item in earners: if str(item.connector)==broker:
if str(item.connector)==broker: item.set_step_size(new_step_size)
item.set_step_size(new_step_size) return jsonify({'step_size': new_step_size})
return jsonify({'step_size': new_step_size}) return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_percentage", methods=["GET"]) @earn_api.route("/get_percentage", methods=["GET"])
@require_api_key
def get_percentage(): def get_percentage():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'percentage': item.get_percentage()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'percentage': item.get_percentage()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/set_percentage", methods=["POST"]) @earn_api.route("/set_percentage", methods=["POST"])
def set_percentage(): @require_api_key
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: def set_percentage():
valid_brokers = [str(item.connector) for item in earners] if request.json is None:
if request.json is None: return jsonify({'Error': 'request.json is None'})
return jsonify({'Error': 'request.json is None'}) broker = request.json["broker"]
broker = request.json["broker"] new_percentage = request.json["new_percentage"]
new_percentage = request.json["new_percentage"] if broker is None:
if broker is None: return jsonify({'Error': 'broker is None'})
return jsonify({'Error': 'broker is None'}) if broker not in valid_broker_list:
if broker not in valid_brokers: return jsonify({'Error': 'broker not valid'})
return jsonify({'Error': 'broker not valid'}) if new_percentage is None:
if new_percentage is None: return jsonify({'Error': 'new_step_size is None'})
return jsonify({'Error': 'new_step_size is None'}) for item in earners:
for item in earners: if str(item.connector)==broker:
if str(item.connector)==broker: item.set_percentage(new_percentage)
item.set_percentage(new_percentage) return jsonify({'percentage': new_percentage})
return jsonify({'percentage': new_percentage}) return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_time_between_subscriptions", methods=["GET"]) @earn_api.route("/get_time_between_subscriptions", methods=["GET"])
def get_time_between_subscriptions(): @require_api_key
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: def get_time_between_subscriptions():
valid_brokers = [str(item.connector) for item in earners] broker = request.args.get("broker")
broker = request.args.get("broker") if broker is None:
if broker is None: return jsonify({'Error': 'broker is None'})
return jsonify({'Error': 'broker is None'}) if broker not in valid_broker_list:
if broker not in valid_brokers: return jsonify({'Error': 'broker not valid'})
return jsonify({'Error': 'broker not valid'}) for item in earners:
for item in earners: if str(item.connector)==broker:
if str(item.connector)==broker: return jsonify({'time_between_subscriptions': item.get_time_between_subscriptions()})
return jsonify({'time_between_subscriptions': item.get_time_between_subscriptions()}) return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/set_time_between_subscriptions", methods=["POST"]) @earn_api.route("/set_time_between_subscriptions", methods=["POST"])
@require_api_key
def set_time_between_subscriptions(): def set_time_between_subscriptions():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: if request.json is None:
valid_brokers = [str(item.connector) for item in earners] return jsonify({'Error': 'request.json is None'})
if request.json is None: broker = request.json["broker"]
return jsonify({'Error': 'request.json is None'}) new_time_between_subscriptions = request.json["new_time_between_subscriptions"]
broker = request.json["broker"] if broker is None:
new_time_between_subscriptions = request.json["new_time_between_subscriptions"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: if new_time_between_subscriptions is None:
return jsonify({'Error': 'broker not valid'}) return jsonify({'Error': 'new_step_size is None'})
if new_time_between_subscriptions is None: for item in earners:
return jsonify({'Error': 'new_step_size is None'}) if str(item.connector)==broker:
for item in earners: item.set_time_between_subscriptions(new_time_between_subscriptions)
if str(item.connector)==broker: return jsonify({'time_between_subscriptions': new_time_between_subscriptions})
item.set_time_between_subscriptions(new_time_between_subscriptions) return jsonify({'Error': 'broker not found'})
return jsonify({'time_between_subscriptions': new_time_between_subscriptions})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_time_between_redemptions", methods=["GET"]) @earn_api.route("/get_time_between_redemptions", methods=["GET"])
@require_api_key
def get_time_between_redemptions(): def get_time_between_redemptions():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'time_between_redemptions': item.get_time_between_redemptions()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'time_between_redemptions': item.get_time_between_redemptions()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/set_time_between_redemptions", methods=["POST"]) @earn_api.route("/set_time_between_redemptions", methods=["POST"])
@require_api_key
def set_time_between_redemptions(): def set_time_between_redemptions():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: if request.json is None:
valid_brokers = [str(item.connector) for item in earners] return jsonify({'Error': 'request.json is None'})
if request.json is None: broker = request.json["broker"]
return jsonify({'Error': 'request.json is None'}) new_time_between_redemptions = request.json["new_time_between_redemptions"]
broker = request.json["broker"] if broker is None:
new_time_between_redemptions = request.json["new_time_between_redemptions"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: if new_time_between_redemptions is None:
return jsonify({'Error': 'broker not valid'}) return jsonify({'Error': 'new_step_size is None'})
if new_time_between_redemptions is None: for item in earners:
return jsonify({'Error': 'new_step_size is None'}) if str(item.connector)==broker:
for item in earners: item.set_time_between_redemptions(new_time_between_redemptions)
if str(item.connector)==broker: return jsonify({'time_between_redemptions': new_time_between_redemptions})
item.set_time_between_redemptions(new_time_between_redemptions) return jsonify({'Error': 'broker not found'})
return jsonify({'time_between_redemptions': new_time_between_redemptions})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_minimum_amount_in_trading_account", methods=["GET"]) @earn_api.route("/get_minimum_amount_in_trading_account", methods=["GET"])
@require_api_key
def get_minimum_amount_in_trading_account(): def get_minimum_amount_in_trading_account():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'minimum_amount_in_trading_account': item.get_minimum_amount_in_trading_account()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'minimum_amount_in_trading_account': item.get_minimum_amount_in_trading_account()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/set_minimum_amount_in_trading_account", methods=["POST"]) @earn_api.route("/set_minimum_amount_in_trading_account", methods=["POST"])
@require_api_key
def set_minimum_amount_in_trading_account(): def set_minimum_amount_in_trading_account():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.json["broker"]
valid_brokers = [str(item.connector) for item in earners] new_minimum_amount_in_trading_account = request.json["new_minimum_amount_in_trading_account"]
broker = request.json["broker"] if broker is None:
new_minimum_amount_in_trading_account = request.json["new_minimum_amount_in_trading_account"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: if new_minimum_amount_in_trading_account is None:
return jsonify({'Error': 'broker not valid'}) return jsonify({'Error': 'new_step_size is None'})
if new_minimum_amount_in_trading_account is None: for item in earners:
return jsonify({'Error': 'new_step_size is None'}) if str(item.connector)==broker:
for item in earners: item.set_minimum_amount_in_trading_account(new_minimum_amount_in_trading_account)
if str(item.connector)==broker: return jsonify({'minimum_amount_in_trading_account': new_minimum_amount_in_trading_account})
item.set_minimum_amount_in_trading_account(new_minimum_amount_in_trading_account) return jsonify({'Error': 'broker not found'})
return jsonify({'minimum_amount_in_trading_account': new_minimum_amount_in_trading_account})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_last_subscription", methods=["GET"]) @earn_api.route("/get_last_subscription", methods=["GET"])
@require_api_key
def get_last_subscription(): def get_last_subscription():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'last_subscription': item.get_last_subscription()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'last_subscription': item.get_last_subscription()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_last_redemption", methods=["GET"]) @earn_api.route("/get_last_redemption", methods=["GET"])
@require_api_key
def get_last_redemption(): def get_last_redemption():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'last_redemption': item.get_last_redemption()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'last_redemption': item.get_last_redemption()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/get_total_balance", methods=["GET"]) @earn_api.route("/get_total_balance", methods=["GET"])
@require_api_key
def get_total_balance(): def get_total_balance():
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: broker = request.args.get("broker")
valid_brokers = [str(item.connector) for item in earners] if broker is None:
broker = request.args.get("broker") return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: for item in earners:
return jsonify({'Error': 'broker not valid'}) if str(item.connector)==broker:
for item in earners: return jsonify({'trading_balance': item.get_trading_balance(), 'earning_balance': item.get_earning_balance()})
if str(item.connector)==broker: return jsonify({'Error': 'broker not found'})
return jsonify({'trading_balance': item.get_trading_balance(), 'earning_balance': item.get_earning_balance()})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/subscribe", methods=["POST"]) @earn_api.route("/subscribe", methods=["POST"])
@require_api_key
def subscribe(): def subscribe():
''' '''
args: args:
@ -432,32 +417,30 @@ def subscribe():
amount: amount to subscribe amount: amount to subscribe
force_pause: True or False force_pause: True or False
''' '''
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: if request.json is None:
valid_brokers = [str(item.connector) for item in earners] return jsonify({'Error': 'request.json is None'})
if request.json is None: broker = request.json["broker"]
return jsonify({'Error': 'request.json is None'}) amount = request.json["amount"]
broker = request.json["broker"] force_pause = False
amount = request.json["amount"] if "force_pause" in request.json:
force_pause = False force_pause = request.json["force_pause"]
if "force_pause" in request.json: if broker is None:
force_pause = request.json["force_pause"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: if amount is None:
return jsonify({'Error': 'broker not valid'}) return jsonify({'Error': 'amount is None'})
if amount is None: if force_pause is None:
return jsonify({'Error': 'amount is None'}) return jsonify({'Error': 'force_pause is None'})
if force_pause is None: for item in earners:
return jsonify({'Error': 'force_pause is None'}) if str(item.connector)==broker:
for item in earners: subscription = item.subscribe(amount, force_pause=force_pause)
if str(item.connector)==broker: return jsonify({'Success': str(subscription)})
subscription = item.subscribe(amount, force_pause=force_pause) return jsonify({'Error': 'broker not found'})
return jsonify({'Success': str(subscription)})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
@earn_api.route("/redeem", methods=["POST"]) @earn_api.route("/redeem", methods=["POST"])
@require_api_key
def redeem(): def redeem():
''' '''
args: args:
@ -465,29 +448,27 @@ def redeem():
amount: amount to redeem amount: amount to redeem
force_pause: True or False force_pause: True or False
''' '''
if "X-API-KEY" in request.headers and request.headers.get("X-API-KEY") in valid_keys: if request.json is None:
valid_brokers = [str(item.connector) for item in earners] return jsonify({'Error': 'request.json is None'})
if request.json is None: broker = request.json["broker"]
return jsonify({'Error': 'request.json is None'}) amount = request.json["amount"]
broker = request.json["broker"] force_pause = False
amount = request.json["amount"] if "force_pause" in request.json:
force_pause = False force_pause = request.json["force_pause"]
if "force_pause" in request.json: if broker is None:
force_pause = request.json["force_pause"] return jsonify({'Error': 'broker is None'})
if broker is None: if broker not in valid_broker_list:
return jsonify({'Error': 'broker is None'}) return jsonify({'Error': 'broker not valid'})
if broker not in valid_brokers: if amount is None:
return jsonify({'Error': 'broker not valid'}) return jsonify({'Error': 'amount is None'})
if amount is None: if force_pause is None:
return jsonify({'Error': 'amount is None'}) return jsonify({'Error': 'force_pause is None'})
if force_pause is None: for item in earners:
return jsonify({'Error': 'force_pause is None'}) if str(item.connector)==broker:
for item in earners: redemption = item.redeem(amount, force_pause=force_pause)
if str(item.connector)==broker: return jsonify({'Success': str(redemption)})
redemption = item.redeem(amount, force_pause=force_pause) return jsonify({'Error': 'broker not found'})
return jsonify({'Success': str(redemption)})
return jsonify({'Error': 'broker not found'})
return jsonify({'Error': 'API key not valid'}), 401
def run_API(port): def run_API(port):
@ -525,6 +506,9 @@ if __name__=="__main__":
for item in config["exchanges"]: for item in config["exchanges"]:
earners.append(earner(connectors[item], config["exchanges"][item])) earners.append(earner(connectors[item], config["exchanges"][item]))
#Valid broker list
valid_broker_list = [str(item.connector) for item in earners]
#Load valid API keys #Load valid API keys
valid_keys = load_keys_from_db("keys/api_credentials.db") valid_keys = load_keys_from_db("keys/api_credentials.db")