Paytm-API / app.py
Bruhletme's picture
Upload folder using huggingface_hub
08e1a67 verified
Raw
History Blame Contribute Delete
908 Bytes
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)