99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
from flask import Flask, jsonify, request
|
|
import mysql.connector
|
|
import uuid
|
|
import base64
|
|
import json
|
|
import keyfile
|
|
|
|
|
|
API_KEY_NS = keyfile.API_KEY
|
|
|
|
app = Flask(__name__)
|
|
# Database connection
|
|
|
|
db_config = keyfile.requestDBCreds
|
|
|
|
def get_db_connection():
|
|
try:
|
|
connection = mysql.connector.connect(**db_config)
|
|
return connection
|
|
except mysql.connector.Error as err:
|
|
print(f"Error: {err}")
|
|
return None
|
|
|
|
|
|
|
|
|
|
# NEEDS FIXING
|
|
@app.route('/api/tiropitAddUser', methods=['POST'])
|
|
def tiropit_add_user():
|
|
data = request.json
|
|
|
|
if not data or "UserName" not in data or "UserPIN" not in data:
|
|
return jsonify({"error": "Account MUST include at least UserName & UserPIN"}), 400
|
|
|
|
user_pin = data.get("UserPIN")
|
|
friend_code = data.get("FriendCode")
|
|
|
|
if not friend_code or len(friend_code) != 5:
|
|
return jsonify({"error": "FirendCode must be 5 characters long"}), 400
|
|
if not user_pin or len(user_pin) != 4:
|
|
return jsonify({"error": "UserPIN must be 4 characters long"}), 400
|
|
|
|
|
|
|
|
connection = get_db_connection()
|
|
if connection is None:
|
|
return jsonify({"error": "Failed to contact db"}), 500
|
|
|
|
try:
|
|
cursor = connection.cursor(dictionary:true)
|
|
user_data = {
|
|
|
|
"UserID" : str(uuid.uuid4()),
|
|
"UserName" : data['UserName'],
|
|
"UserPIN" : user_pin,
|
|
"SteamName" : data.get('SteamName', ''),
|
|
"Level" : data.get('Level', 0),
|
|
"Awards" : json.dumps(data.get('Awards', '[]')),
|
|
"ProfilePicture" : data.get('ProfilePicture', ''),
|
|
"UserGames" : json.dumps(data.get('UserGames', '[]')),
|
|
"WarnLevel" : data.get('WarnLevel', 0),
|
|
"Banned" : data.get('Banned', False),
|
|
"AdminLevel" : 0,
|
|
"UserDiscord" : data.get('UserDiscord', ""),
|
|
"SteamID" : data.get('SteamID', '') ,
|
|
"FriendCode" : friend_code,
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
query = """
|
|
INSERT INTO users (UserID, UserName, SteamName, Level, Awards, UserGames, WarnLevel, Banned, AdminLevel, SteamID, UserDiscord, UserPIN, FriendCode)
|
|
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
|
"""
|
|
|
|
cursor.execute(query, tuple(user_data.values()))
|
|
connection.commit()
|
|
|
|
return jsonify(
|
|
{"message":"User Created",
|
|
"UserID": user_data['UserID']}, 201
|
|
)
|
|
except mysql.connector.Error as err:
|
|
connection.rollback()
|
|
if err.errno == 1062:
|
|
return jsonify({"error": "User already exists"}), 409
|
|
finally:
|
|
if connection.is_connected():
|
|
cursor.close()
|
|
connection.close()
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host="0.0.0.0", port=5000, debug=True)
|