stablediffusionapi / post_methods.py
yusufgundogdu's picture
Update post_methods.py
d81c588 verified
raw
history blame
1.24 kB
from flask import jsonify, request
from database import get_db
def add_user():
"""Yeni kullanıcı ekler veya var olanı döndürür"""
try:
if not request.json or "udid" not in request.json:
return jsonify({"error": "UDID zorunlu alan"}), 400
db = get_db()
cursor = db.cursor()
user = cursor.execute(
'SELECT * FROM users WHERE udid = ?',
(request.json["udid"],)
).fetchone()
if user:
return jsonify(dict(user))
cursor.execute(
'INSERT INTO users (udid, name, premium_account, usage_limit, invoice) VALUES (?, ?, ?, ?, ?)',
(request.json["udid"],
request.json.get("name"),
request.json.get("premium_account", False),
request.json.get("usage_limit", 4),
request.json.get("invoice"))
)
db.commit()
new_user = cursor.execute(
'SELECT * FROM users WHERE udid = ?',
(request.json["udid"],)
).fetchone()
return jsonify(dict(new_user)), 201
except Exception as e:
return jsonify({"error": str(e)}), 500