Pleiades proxy store: SOCKS5 gateway + Flask storefront + BTCPay checkout

admin.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+"""Pleiades admin CLI — manage users, plans, balances."""
+import sys
+import time
+
+import db
+
+
+def create_user(username, web_pass, proxy_pass, country="", city="", region="",
+                sticky=1, balance_gb=0, active=1):
+    d = db.get_db()
+    try:
+        d.execute(
+            "INSERT INTO users (username,password_hash,proxy_password_hash,"
+            "proxy_password_plain,active,region,country,city,sticky,balance_gb,created_at) "
+            "VALUES (?,?,?,?,?,?,?,?,?,?,?)",
+            (username, db.scrypt_hash(web_pass), db.scrypt_hash(proxy_pass), proxy_pass,
+             int(active), region, country, city, int(sticky), float(balance_gb), int(time.time())))
+        d.commit()
+        print(f"OK created user '{username}'")
+    except Exception as e:
+        print(f"ERROR: {e}")
+    finally:
+        d.close()
+
+
+def set_country(username, country):
+    d = db.get_db()
+    d.execute("UPDATE users SET country=? WHERE username=?", (country, username))
+    d.commit()
+    d.close()
+    print(f"OK {username} country={country}")
+
+
+def set_field(username, field, value):
+    d = db.get_db()
+    d.execute(f"UPDATE users SET {field}=? WHERE username=?", (value, username))
+    d.commit()
+    d.close()
+    print(f"OK {username} {field}={value}")
+
+
+def add_balance(username, gb):
+    d = db.get_db()
+    d.execute("UPDATE users SET balance_gb=balance_gb+? WHERE username=?", (float(gb), username))
+    d.commit()
+    d.close()
+    print(f"OK {username} +{gb} GB")
+
+
+def list_users():
+    d = db.get_db()
+    for r in d.execute("SELECT id,username,active,country,city,sticky,balance_gb FROM users"):
+        print(f"#{r['id']} {r['username']} active={r['active']} geo={r['country'] or '-'}/{r['city'] or '-'} "
+              f"sticky={r['sticky']} bal={r['balance_gb']}GB")
+    d.close()
+
+
+def main():
+    db.init_db()
+    if len(sys.argv) < 2:
+        print(__doc__)
+        return
+    cmd = sys.argv[1]
+    a = sys.argv[2:]
+    if cmd == "create-user" and len(a) >= 3:
+        create_user(a[0], a[1], a[2], *a[3:])
+    elif cmd == "set-country" and len(a) == 2:
+        set_country(a[0], a[1])
+    elif cmd == "set" and len(a) == 3:
+        set_field(a[0], a[1], a[2])
+    elif cmd == "add-balance" and len(a) == 2:
+        add_balance(a[0], a[1])
+    elif cmd == "list":
+        list_users()
+    else:
+        print("commands: create-user <u> <webpass> <proxypass> [country city region sticky bal active] | "
+              "set-country <u> <CC> | set <u> <field> <val> | add-balance <u> <gb> | list")
+
+
+if __name__ == "__main__":
+    main()