102 lines
3.8 KiB
Python
Executable File
102 lines
3.8 KiB
Python
Executable File
|
|
import os
|
|
from flask import Flask, render_template, redirect, url_for, request, session, jsonify
|
|
import requests
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'super-secret')
|
|
|
|
API_URL = os.environ.get('API_URL', 'http://localhost:5000')
|
|
|
|
@app.route('/')
|
|
def index():
|
|
if 'access_token' in session:
|
|
return redirect(url_for('dashboard'))
|
|
return render_template('index.html')
|
|
|
|
@app.route('/register', methods=['GET', 'POST'])
|
|
def register():
|
|
if request.method == 'POST':
|
|
data = {
|
|
'name': request.form['name'],
|
|
'email': request.form['email'],
|
|
'age': request.form['age'],
|
|
'location': request.form['location'],
|
|
'gender': request.form['gender'],
|
|
'password': request.form['password']
|
|
}
|
|
response = requests.post(f"{API_URL}/users", json=data)
|
|
if response.status_code == 201:
|
|
return redirect(url_for('login'))
|
|
else:
|
|
return render_template('register.html', error=response.json().get('message', 'Error during registration'))
|
|
return render_template('register.html')
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
email = request.form['email']
|
|
password = request.form['password']
|
|
response = requests.post(f"{API_URL}/auth", json={'email': email, 'password': password})
|
|
if response.status_code == 200:
|
|
session['access_token'] = response.json()['access_token']
|
|
# Save the user ID from the access token or fetch it from the API
|
|
session['user_id'] = email # Or any unique identifier like user ID
|
|
return redirect(url_for('dashboard'))
|
|
else:
|
|
return render_template('login.html', error=response.json().get('message', 'Invalid credentials'))
|
|
return render_template('login.html')
|
|
|
|
@app.route('/dashboard')
|
|
def dashboard():
|
|
if 'access_token' not in session:
|
|
return redirect(url_for('login'))
|
|
|
|
# Fetch the current user's details
|
|
headers = {'Authorization': f'Bearer {session["access_token"]}'}
|
|
user_response = requests.get(f"{API_URL}/users/me", headers=headers) # Adjust the API URL as necessary
|
|
user = user_response.json() if user_response.status_code == 200 else None
|
|
|
|
# Handle search filters
|
|
location = request.args.get('location', '')
|
|
age = request.args.get('age', '')
|
|
gender = request.args.get('gender', 'Any')
|
|
|
|
filters = {}
|
|
if location:
|
|
filters['location'] = location
|
|
if age:
|
|
filters['age'] = age
|
|
if gender and gender != "Any":
|
|
filters['gender'] = gender
|
|
|
|
# Fetch users based on filters
|
|
response = requests.get(f"{API_URL}/users", params=filters, headers=headers)
|
|
users = response.json() if response.status_code == 200 else []
|
|
|
|
# Fetch messages for the current user
|
|
messages_response = requests.get(f"{API_URL}/messages", headers=headers)
|
|
messages = messages_response.json() if messages_response.status_code == 200 else []
|
|
|
|
return render_template('dashboard.html', user=user, users=users, messages=messages)
|
|
|
|
@app.route('/messages', methods=['POST'])
|
|
def send_message():
|
|
if 'access_token' not in session:
|
|
return redirect(url_for('login'))
|
|
|
|
data = {
|
|
'receiver_id': request.form['receiver_id'],
|
|
'content': request.form['message_content']
|
|
}
|
|
headers = {'Authorization': f'Bearer {session["access_token"]}', 'Content-Type': 'application/json'}
|
|
response = requests.post(f"{API_URL}/messages", headers=headers, json=data)
|
|
|
|
if response.status_code == 201:
|
|
return redirect(url_for('dashboard'))
|
|
else:
|
|
return jsonify({'message': 'Error sending message'}), 500
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=8000, debug=True)
|