File size: 908 Bytes
08e1a67 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | import sqlite3, re, os
from flask import Flask, request, jsonify
from huggingface_hub import hf_hub_download
app = Flask(__name__)
DB_PATH = hf_hub_download(
repo_id="Bruhletme/Paytm",
filename="paytm.db",
repo_type="dataset",
token=os.environ.get("HF_TOKEN")
)
def query_db(q, a=()):
conn = sqlite3.connect(DB_PATH)
conn.row_factory = sqlite3.Row
cur = conn.cursor()
cur.execute(q, a)
r = cur.fetchall()
conn.close()
return [dict(x) for x in r]
@app.route("/num")
def num():
num = (request.args.get("info") or "").strip()
num = re.sub(r"\D","",num)[-10:]
if len(num)!=10:
return jsonify({"error":"invalid"}),400
d = query_db("SELECT * FROM users WHERE mobile=? LIMIT 1",(num,))
return jsonify({"found":bool(d),"data":d[0] if d else None})
@app.route("/")
def home():
return {"status":"ok"}
app.run(host="0.0.0.0", port=7860)
|