|
|
from flask import jsonify |
|
|
from database import get_db |
|
|
|
|
|
def get_users(): |
|
|
try: |
|
|
db = get_db() |
|
|
users = db.execute('SELECT * FROM users').fetchall() |
|
|
return jsonify({ |
|
|
"success": True, |
|
|
"data": { |
|
|
"users": [dict(user) for user in users] |
|
|
} |
|
|
}) |
|
|
except Exception as e: |
|
|
return jsonify({ |
|
|
"success": False, |
|
|
"error": str(e) |
|
|
}), 500 |
|
|
|
|
|
def get_user(udid): |
|
|
try: |
|
|
db = get_db() |
|
|
user = db.execute('SELECT * FROM users WHERE udid = ?', (udid,)).fetchone() |
|
|
|
|
|
if user: |
|
|
return jsonify({ |
|
|
"success": True, |
|
|
"data": { |
|
|
"user": dict(user) |
|
|
} |
|
|
}) |
|
|
|
|
|
|
|
|
db.execute( |
|
|
'INSERT INTO users (udid) VALUES (?)', |
|
|
(udid,) |
|
|
) |
|
|
db.commit() |
|
|
|
|
|
new_user = db.execute('SELECT * FROM users WHERE udid = ?', (udid,)).fetchone() |
|
|
return jsonify({ |
|
|
"success": True, |
|
|
"data": { |
|
|
"user": dict(new_user) |
|
|
} |
|
|
}), 201 |
|
|
|
|
|
except Exception as e: |
|
|
return jsonify({ |
|
|
"success": False, |
|
|
"error": str(e) |
|
|
}), 500 |