diff --git a/lnbits/extensions/tpos/views.py b/lnbits/extensions/tpos/views.py index a1b1807..3ac8329 100644 --- a/lnbits/extensions/tpos/views.py +++ b/lnbits/extensions/tpos/views.py @@ -1,74 +1,27 @@ -import uuid -import json -import requests - -from flask import jsonify, render_template, request, redirect, url_for, g +from flask import g, abort, render_template from lnbits.decorators import check_user_exists, validate_uuids -from lnbits.db import open_db, open_ext_db from lnbits.extensions.tpos import tpos_ext +from lnbits.helpers import Status +import requests +import json + +from .crud import get_tpos -#add your endpoints here @tpos_ext.route("/") @validate_uuids(["usr"], required=True) @check_user_exists() def index(): - """Try to add descriptions for others.""" - usr = request.args.get("usr") - nme = request.args.get("nme") - wal = request.args.get("wal") - cur = request.args.get("cur") - - if usr: - if not len(usr) > 20: - return redirect(url_for("home")) - - # Get all the data - with open_db() as db: - user_wallets = db.fetchall("SELECT * FROM wallets WHERE user = ?", (usr,)) - user_wall = db.fetchall("SELECT * FROM wallets WHERE user = ? AND id = ?", (usr,wal)) - user_ext = db.fetchall("SELECT extension FROM extensions WHERE user = ? AND active = 1", (usr,)) - user_ext = [v[0] for v in user_ext] - - if nme: - uni = uuid.uuid4().hex - with open_ext_db("tpos") as pos_ext_db: - pos_ext_db.execute( - """ - INSERT OR IGNORE INTO tpos - (nme, uni, usr, invkey, cur) - VALUES (?, ?, ?, ?, ?) - """, - ( - nme, - uni, - usr, - user_wall[0][3], - cur, - - ), - ) - with open_ext_db("tpos") as pos_ext_dbb: - user_fau = pos_ext_dbb.fetchall("SELECT * FROM tpos WHERE usr = ?", (usr,)) - - return render_template( - "tpos/index.html", user_wallets=user_wallets, user_ext=user_ext, usr=usr, user_fau=user_fau, user=g.user - ) - -@tpos_ext.route("/tpos") -def tpos(): - """Try to add descriptions for others.""" - pos = request.args.get("pos") + return render_template("tpos/index.html", user=g.user) - with open_ext_db("tpos") as pos_ext_dbb: - user_fau = pos_ext_dbb.fetchall("SELECT * FROM tpos WHERE uni = ?", (pos,)) - if not user_fau: - return jsonify({"status": "ERROR", "reason":"NO POS"}), 400 +@tpos_ext.route("/") +def tpos(tpos_id): + tpos = get_tpos(tpos_id) or abort(Status.NOT_FOUND, "tpos does not exist.") + print(tpos) r = requests.get("https://api.opennode.co/v1/rates") r_json = r.json() + print(r_json["data"]["BTC" + tpos.currency][tpos.currency]) - return render_template( - "tpos/tpos.html", pos=pos, ratee=user_fau[0][5], exchange=int(r_json["data"]["BTC" + user_fau[0][5]][user_fau[0][5]]) - ) + return render_template("tpos/tpos.html", tpos=tpos, rate=r_json["data"]["BTC" + tpos.currency][tpos.currency])