Update views.py

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("/<tpos_id>")
+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])