Update app.py

app.py

@@ -1,69 +1,133 @@
 from flask import Flask, request, jsonify
 import requests
-from bs4 import BeautifulSoup
+import json
+import base64
 
 app = Flask(__name__)
 
-# Example URL, modify based on the search page you're targeting
-url = "https://html.duckduckgo.com/html/"
+# GitHub credentials from environment variables
+GITHUB_TOKEN = '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'  # Set your token in environment variables
+REPO_OWNER = 'hussein2000-oo'
+REPO_NAME = 'dbailloolloloolollhrthlnewrgnk'
+USER_FILE_NAME = 'user.json'
 
-# Function to fetch results from a specific page (1st page: page_num=1, 2nd page: page_num=2, etc.)
-def fetch_duckduckgo_results(search_query, page_num=1):
-    headers = {
-        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/85.0.4183.102 Safari/537.36'
-    }
+# Function to fetch user data from GitHub
+def fetch_user_data():
+    url = f'https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/contents/{USER_FILE_NAME}'
+    headers = {'Authorization': f'token {GITHUB_TOKEN}'}
+    response = requests.get(url, headers=headers)
     
-    # DuckDuckGo uses 's' parameter for result offset, calculate offset for pages (e.g., 0 for first, 50 for second)
-    offset = (page_num - 1) * 50
+    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}'
+    headers = {'Authorization': f'token {GITHUB_TOKEN}'}
     
-    # Search query parameters (assuming DuckDuckGo), 's' is for pagination
-    params = {'q': search_query, 's': offset}
+    updated_content = base64.b64encode(json.dumps(user_data).encode('utf-8')).decode('utf-8')
     
-    # Send request to DuckDuckGo or relevant HTML page
-    response = requests.get(url, headers=headers, params=params)
+    payload = {
+        "message": "Update user.json with new user",
+        "content": updated_content,
+        "sha": sha
+    }
     
-    # Check if request was successful
-    if response.status_code != 200:
-        return {"error": f"Failed to retrieve data: {response.status_code}"}
+    response = requests.put(url, headers=headers, json=payload)
     
-    # Parse HTML content
-    soup = BeautifulSoup(response.text, 'html.parser')
+    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()
     
-    # Find all result blocks
-    results = []
-    for result in soup.find_all('div', class_='result__body'):
-        title_tag = result.find('h2', class_='result__title')
-        snippet_tag = result.find('a', class_='result__snippet')
-        icon_tag = result.find('img', class_='result__icon__img')
-        url_tag = result.find('a', class_='result__a')
-        
-        if title_tag and snippet_tag and icon_tag and url_tag:
-            result_data = {
-                'title': title_tag.text.strip(),
-                'description': snippet_tag.text.strip(),
-                'icon_url': "https:" + icon_tag['src'] if icon_tag else None,
-                'url': url_tag['href']
+    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
             }
-            results.append(result_data)
-    
-    return results
-
-# API endpoint for fetching search results
-@app.route('/search', methods=['GET'])
-def search():
-    # Get query parameters
-    search_query = request.args.get('query', default='', type=str)
-    page_num = request.args.get('page', default=1, type=int)
-    
-    if not search_query:
-        return jsonify({"error": "Search query is required"}), 400
+            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
+
+
+# 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()
     
-    # Fetch results from DuckDuckGo
-    results = fetch_duckduckgo_results(search_query, page_num)
+    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'])
+def reset_password():
+    data = request.json
+    username = data.get('username')
+    answers = data.get('answers')
+
+    user_data, _ = fetch_user_data()
     
-    # Return results as JSON
-    return jsonify(results)
+    if username in user_data:
+        questions = user_data[username]['security_questions ']
+        
+        # Check if answers match
+        if all(user_data[username]['security_questions'][q] == answers[q] for q in questions):
+            new_password = data.get('new_password')
+            user_data[username]['password'] = new_password
+            update_user_data(user_data, _)
+            return jsonify({"message": "Password reset successfully."}), 200
+        else:
+            return jsonify({"message": "Security answers do not match."}), 401
+    else:
+        return jsonify({"message": "User not found."}), 404
+
+
 
-# Run the Flask application
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)