Update app.py

app.py

@@ -6,7 +6,7 @@ import base64
 app = Flask(__name__)
 
 # GitHub credentials from environment variables
-GITHUB_TOKEN = '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'  # Set your token in environment variables
+GITHUB_TOKEN = 'your_github_token'  # Replace with your GitHub token
 REPO_OWNER = 'hussein2000-oo'
 REPO_NAME = 'dbailloolloloolollhrthlnewrgnk'
 USER_FILE_NAME = 'user.json'
@@ -20,18 +20,10 @@ def fetch_user_data():
     if response.status_code == 200:
         content = response.json()
         user_data = json.loads(base64.b64decode(content['content']).decode('utf-8'))
-        
-        # Ensure user_data is a dictionary
-        if not isinstance(user_data, dict):
-            print("User  data is not in the expected format. Initializing empty user data.")
-            user_data = {}
-        
         return user_data, content['sha']  # Return the SHA for updating the file
     else:
-        print("Failed to fetch user data:", response.status_code, response.json())
         return {}, None  # Return an empty dict if fetching fails
 
-
 # Function to update user data on GitHub
 def update_user_data(user_data, sha):
     url = f'https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/contents/{USER_FILE_NAME}'
@@ -40,93 +32,41 @@ def update_user_data(user_data, sha):
     updated_content = base64.b64encode(json.dumps(user_data).encode('utf-8')).decode('utf-8')
     
     payload = {
-        "message": "Update user.json with new user",
+        "message": "Update user.json with new password",
         "content": updated_content,
         "sha": sha
     }
     
     response = requests.put(url, headers=headers, json=payload)
-    
-    if response.status_code == 200:
-        print("User  data updated successfully.")
-    else:
-        print("Failed to update user data:", response.status_code, response.json())
-
-
-# API endpoint to create a user account
-@app.route('/api/create_user', methods=['POST'])
-def create_user():
-    data = request.json
-    username = data.get('username')
-    password = data.get('password')
-    first_name = data.get('first_name')
-    last_name = data.get('last_name')
-    birthday = data.get('birthday')
-    security_questions = data.get('security_questions')
-
-    user_data, sha = fetch_user_data()
-    
-    if user_data is not None:
-        if username in user_data:
-            return jsonify({"message": "User  already exists."}), 400
-        else:
-            user_data[username] = {
-                "password": password,
-                "first_name": first_name,
-                "last_name": last_name,
-                "birthday": birthday,
-                "security_questions": security_questions  # Store security questions
-            }
-            update_user_data(user_data, sha)
-            return jsonify({"message": f"User  {username} created successfully."}), 201
-    else:
-        return jsonify({"message": "Could not create user. User data fetch failed."}), 500
+    return response.status_code == 200
 
-
-# API endpoint to sign in
-@app.route('/api/sign_in', methods=['POST'])
-def sign_in():
-    data = request.json
-    username = data.get('username')
-    password = data.get('password')
-
-    user_data, _ = fetch_user_data()
-    
-    if user_data is not None and isinstance(user_data, dict):
-        if username in user_data:
-            if user_data[username]['password'] == password:  # Corrected password check
-                return jsonify({"message": "Signed in successfully!"}), 200
-            else:
-                return jsonify({"message": "Sign in failed."}), 401
-        else:
-            return jsonify({"message": "User  not found."}), 404
-    else:
-        return jsonify({"message": "Unexpected data format in user.json."}), 500
-
-
-# API endpoint to reset password using security questions
-@app.route('/api/reset_password', methods=['POST'])
+@app.route('/reset_password', methods=['POST'])
 def reset_password():
     data = request.json
     username = data.get('username')
-    answers = data.get('answers')
-
-    user_data, _ = fetch_user_data()
+    
+    user_data, sha = fetch_user_data()
     
     if username in user_data:
-        questions = user_data[username]['security_questions ']
+        questions = user_data[username]['security_questions']
+        answers = {}
         
+        for question in questions:
+            answer = input(f"{question}: ")
+            answers[question] = answer
+            
         # Check if answers match
         if all(user_data[username]['security_questions'][q] == answers[q] for q in questions):
-            new_password = data.get('new_password')
+            new_password = input("Enter your new password: ")
             user_data[username]['password'] = new_password
-            update_user_data(user_data, _)
-            return jsonify({"message": "Password reset successfully."}), 200
+            if update_user_data(user_data, sha):
+                return jsonify({"message": "Password reset successfully."}), 200
+            else:
+                return jsonify({"message": "Failed to update user data."}), 500
         else:
-            return jsonify({"message": "Security answers do not match."}), 401
+            return jsonify({"message": "Security answers do not match."}), 403
     else:
-        return jsonify({"message": "User not found."}), 404
-
+        return jsonify({"message": "User  not found."}), 404
 
 
 if __name__ == '__main__':