API upload
This commit is contained in:
83
API/Tiri.py
Normal file
83
API/Tiri.py
Normal file
@@ -0,0 +1,83 @@
|
||||
from flask import Flask, jsonify, request
|
||||
import mysql.connector
|
||||
import uuid
|
||||
import base64
|
||||
import json
|
||||
|
||||
API_KEY_NS = "75834657346587346587634865786348658"
|
||||
|
||||
app = Flask(__name__)
|
||||
# Database connection
|
||||
|
||||
db_config = {
|
||||
'host': 'localhost',
|
||||
'user': 'kevin', # Replace with your MySQL username
|
||||
'password': '0011', # Replace with your MySQL password
|
||||
'database': 'TiropitAppDatabase' # Replace with your database name
|
||||
}
|
||||
|
||||
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.form
|
||||
picture = request.files.get('ProfilePicture')
|
||||
user_id = str(uuid.uuid4())
|
||||
user_name = data.get('UserName')
|
||||
steam_name = data.get('SteamName')
|
||||
steam_id = data.get('SteamID') # treat this as string
|
||||
level = int(data.get('Level', 0))
|
||||
awards = data.get('Awards', '[]')
|
||||
user_games = data.get('UserGames', '[]')
|
||||
warn_level = int(data.get('WarnLevel', 1))
|
||||
banned = data.get('Banned')
|
||||
admin_level = int(data.get('AdminLevel', 1))
|
||||
user_discord = data.get('UserDiscord')
|
||||
user_PIN = data.get('UserPIN')
|
||||
friend_code = data.get('FirendCode')
|
||||
|
||||
if not picture:
|
||||
return jsonify({"error": "No profile picture provided"}), 400
|
||||
if not user_name or not steam_name or not steam_id:
|
||||
return jsonify({"error": "UserName, SteamName, and SteamID are required"}), 400
|
||||
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
|
||||
|
||||
|
||||
|
||||
conn = get_db_connection()
|
||||
Cursor = conn.cursor()
|
||||
try:
|
||||
Cursor.execute("SELECT 1 FROM users WHERE FriendCode = %s", (friend_code,))
|
||||
if Cursor.fetchone():
|
||||
return jsonify({"error": "FriendCode already exists"}), 409
|
||||
|
||||
profile_pic = picture.read()
|
||||
|
||||
|
||||
Cursor.execute("""
|
||||
INSERT INTO users (UserID, UserName, SteamName, Level, Awards, ProfilePicture, UserGames, WarnLevel, Banned, AdminLevel, SteamID, UserDiscord, UserPIN, FriendCode)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
""", (
|
||||
user_id, user_name, steam_name, level, awards, profile_pic,
|
||||
user_games, warn_level, banned, admin_level, steam_id, user_discord, user_PIN, friend_code
|
||||
))
|
||||
conn.commit()
|
||||
except mysql.connector.Error as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
finally:
|
||||
Cursor.close()
|
||||
conn.close()
|
||||
|
||||
return jsonify({"message": "User added successfully", "UserID": user_id}), 201
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user