Update app.py

app.py

@@ -1,84 +1,71 @@
 from flask import Flask, request, jsonify
-import numpy as np
-import random
 import time
 
 app = Flask(__name__)
 
-# Sample database simulation
-shared_data = {
-    "users": [
-        # Sample user data structure
-        # {
-        #     "id": 1,
-        #     "timestamp": int(time.time()),
-        #     "udid": "sample_udid_123",
-        #     "platform": "android",
-        #     "credit": 100
-        # }
-    ]
-}
-
-# Account model structure
-account_model = {
-    "id": None,  # int (auto-incremented)
-    "timestamp": int(time.time()),  # int (current timestamp)
-    "udid": None,  # string
-    "platform": None,  # string
-    "credit": 100  # int (default 100)
+# Fake database simulation - more generic name
+fake_db = {
+    "accounts": []  # Changed from "users" to "accounts" to better reflect the data
 }
 
 # Helper function to generate a new ID
 def generate_new_id():
-    if not shared_data["users"]:
+    if not fake_db["accounts"]:
         return 1
-    return max(user["id"] for user in shared_data["users"]) + 1
+    return max(account["id"] for account in fake_db["accounts"]) + 1
 
-# Ana sayfa
+# Home page
 @app.route('/')
 def home():
     return "stablediffusionapi test"
 
-# GET Endpoint'leri
-@app.route('/users', methods=['GET'])
-def get_users():
-    return jsonify({"users": shared_data["users"]})
+# GET all accounts
+@app.route('/accounts', methods=['GET'])
+def get_all_accounts():
+    return jsonify({"accounts": fake_db["accounts"]})
 
-# GET user by UDID
-@app.route('/user/<udid>', methods=['GET'])
-def get_user_by_udid(udid):
-    # Search for user in the database
-    user = next((u for u in shared_data["users"] if u["udid"] == udid), None)
+# GET or CREATE account by platform and UDID
+@app.route('/account/<platform>/<udid>', methods=['GET'])
+def get_or_create_account(platform, udid):
+    # Validate platform (should be "IOS" or "AND")
+    platform = platform.upper()
+    if platform not in ["IOS", "AND"]:
+        return jsonify({
+            "error": "Invalid platform. Use 'IOS' or 'AND'."
+        }), 400
+    
+    # Search for account in the database
+    account = next((acc for acc in fake_db["accounts"] if acc["udid"] == udid), None)
     
-    if user:
-        # User found, return the user data
+    if account:
+        # Account found, return the account data
         return jsonify({
-            "id": user["id"],
-            "timestamp": user["timestamp"],
-            "udid": user["udid"],
-            "platform": user["platform"],
-            "credit": user["credit"]
+            "id": account["id"],
+            "timestamp": account["timestamp"],
+            "udid": account["udid"],
+            "platform": account["platform"],
+            "credit": account["credit"]
         })
     else:
-        # User not found, create new user with 100 credits
-        new_user = {
+        # Account not found, create new account with 100 credits
+        new_account = {
             "id": generate_new_id(),
             "timestamp": int(time.time()),
             "udid": udid,
-            "platform": request.headers.get('User-Agent', 'unknown'),  # Get platform from User-Agent
+            "platform": platform,
             "credit": 100  # Default credit
         }
         
         # Add to database
-        shared_data["users"].append(new_user)
+        fake_db["accounts"].append(new_account)
         
-        # Return the newly created user
+        # Return the newly created account
         return jsonify({
-            "id": new_user["id"],
-            "timestamp": new_user["timestamp"],
-            "udid": new_user["udid"],
-            "platform": new_user["platform"],
-            "credit": new_user["credit"]
+            "id": new_account["id"],
+            "timestamp": new_account["timestamp"],
+            "udid": new_account["udid"],
+            "platform": new_account["platform"],
+            "credit": new_account["credit"]
         }), 201  # 201 Created status code
 
 if __name__ == '__main__':