stablediffusionapi / get_methods.py
yusufgundogdu's picture
Update get_methods.py
18bb9de verified
raw
history blame
1.21 kB
from flask import jsonify
from database import get_db
def get_users():
"""Tüm kullanıcıları getirir"""
try:
db = get_db()
users = db.execute('SELECT * FROM users').fetchall()
return jsonify([dict(user) for user in users])
except Exception as e:
return jsonify({"error": str(e)}), 500
def get_user(udid):
"""UDID ile kullanıcıyı getirir (yoksa oluşturur)"""
try:
db = get_db()
cursor = db.cursor()
user = cursor.execute(
'SELECT * FROM users WHERE udid = ?',
(udid,)
).fetchone()
if user:
return jsonify(dict(user))
# Kullanıcı yoksa otomatik oluştur
cursor.execute(
'INSERT INTO users (udid, name, premium_account, usage_limit, invoice) VALUES (?, ?, ?, ?, ?)',
(udid, 'guest', False, 3, 'empty')
)
db.commit()
new_user = cursor.execute(
'SELECT * FROM users WHERE udid = ?',
(udid,)
).fetchone()
return jsonify(dict(new_user)), 201
except Exception as e:
return jsonify({"error": str(e)}), 500