Redis + Websocket

app.py

@@ -1,20 +1,31 @@
-from flask import Flask, request, jsonify
-from flask_cors import CORS
-from flask_socketio import SocketIO, emit
+from fastapi import FastAPI, HTTPException, Form, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from fastapi_socketio import SocketManager
+from typing import Optional
 import pymongo
 from bson.objectid import ObjectId
 import redis
 import json
 from redis.exceptions import RedisError
-
-# Khởi tạo Flask app và SocketIO
-app = Flask(__name__)
-CORS(app)
-socketio = SocketIO(app, cors_allowed_origins="*")
-
-# Khởi tạo Redis client
+from pydantic import BaseModel
+
+# Initialize FastAPI app
+app = FastAPI()
+socket_manager = SocketManager(app=app)
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Initialize Redis client
 redis_client = redis.Redis(host='localhost', port=6379, db=0)
-CACHE_EXPIRE_TIME = 300  # 5 phút
+CACHE_EXPIRE_TIME = 300  # 5 minutes
 
 # MongoDB connection
 mongo_url = "mongodb+srv://ip6ofme:[email protected]/"
@@ -23,125 +34,121 @@ db = client["test"]
 pdf_collection = db["PdfDetails"]
 voter_collection = db["Voters"]
 
-# API để upload file và lưu thông tin vào MongoDB
-@app.route('/upload-files', methods=['POST'])
-def upload_file():
-    # Chỉ lưu thông tin vào MongoDB
-    title = request.form.get('title')
-    group = request.form.get('group')
-    firebase_url = request.form.get('url')
+# Pydantic models for request validation
+class VoterRegistration(BaseModel):
+    name: str
+    group: str
+    role: str
+
+class VoteRequest(BaseModel):
+    voter_id: str
+    file_id: str
+    vote_count: Optional[int] = 1
+
+@app.post("/upload-files")
+async def upload_file(
+    title: str = Form(...),
+    group: str = Form(...),
+    url: str = Form(...)
+):
     new_pdf = {
         'title': title,
         'group': group,
-        'url': firebase_url,
+        'url': url,
         'votes': 0
     }
     result = pdf_collection.insert_one(new_pdf)
-    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
-
-# API để lấy số lượng votes của file theo ID
-@app.route('/get-votes', methods=['GET'])
-def get_votes():
-    file_id = request.args.get('id')
+    return {'status': 'ok', 'id': str(result.inserted_id)}
 
+@app.get("/get-votes")
+async def get_votes(file_id: str):
     try:
         file = pdf_collection.find_one({'_id': ObjectId(file_id)})
         if not file:
-            return jsonify({'status': 'error', 'message': 'File not found'}), 404
-
-        return jsonify({'status': 'ok', 'votes': file.get('votes', 0)})
+            raise HTTPException(status_code=404, detail="File not found")
+        return {'status': 'ok', 'votes': file.get('votes', 0)}
     except Exception as e:
-        return jsonify({'status': 'error', 'message': str(e)}), 500
+        raise HTTPException(status_code=500, detail=str(e))
 
-# API để lấy danh sách tất cả các file
-@app.route('/get-files', methods=['GET'])
-def get_files():
+@app.get("/get-files")
+async def get_files():
     try:
         files = pdf_collection.find({})
-        file_list = []
-        for file in files:
-            file_list.append({
+        file_list = [
+            {
                 'id': str(file['_id']),
                 'title': file['title'],
                 'group': file['group'],
                 'url': file['url'],
                 'votes': file.get('votes', 0)
-            })
-        return jsonify({'status': 'ok', 'data': file_list})
+            }
+            for file in files
+        ]
+        return {'status': 'ok', 'data': file_list}
     except Exception as e:
-        print(f"Error: {str(e)}")
-        return jsonify({'status': 'error', 'message': str(e)}), 500
-
-# API để đăng ký người bình chọn
-@app.route('/register-voter', methods=['POST'])
-def register_voter():
-    data = request.json
-    name = data.get('name')
-    group = data.get('group')
-    role = data.get('role')  # Thêm trường role
-    
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/register-voter")
+async def register_voter(voter: VoterRegistration):
     new_voter = {
-        'name': name,
-        'group': group,
-        'role': role,  # Lưu role vào MongoDB
+        'name': voter.name,
+        'group': voter.group,
+        'role': voter.role,
         'number_of_votes': 0
     }
     result = voter_collection.insert_one(new_voter)
-    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
-
-# API để bình chọn
-@app.route('/vote-by-voter', methods=['POST'])
-def vote_by_voter():
-    data = request.json
-    voter_id = data.get('voter_id')
-    file_id = data.get('file_id')
-    vote_count = data.get('vote_count', 1)
+    return {'status': 'ok', 'id': str(result.inserted_id)}
 
-    if vote_count <= 0:
-        return jsonify({'status': 'error', 'message': 'Vote count must be greater than 0'}), 400
+@app.post("/vote-by-voter")
+async def vote_by_voter(vote_request: VoteRequest):
+    if vote_request.vote_count <= 0:
+        raise HTTPException(status_code=400, detail="Vote count must be greater than 0")
 
-    voter = voter_collection.find_one({'_id': ObjectId(voter_id)})
+    voter = voter_collection.find_one({'_id': ObjectId(vote_request.voter_id)})
     if not voter:
-        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
+        raise HTTPException(status_code=404, detail="Voter not found")
 
     max_votes = 10 if voter['role'] == 'judge' else 2
 
-    if voter['number_of_votes'] + vote_count > max_votes:
-        return jsonify({'status': 'error', 'message': 'Maximum votes exceeded'}), 400
+    if voter['number_of_votes'] + vote_request.vote_count > max_votes:
+        raise HTTPException(status_code=400, detail="Maximum votes exceeded")
 
     try:
-        # Cập nhật MongoDB
-        voter_collection.update_one({'_id': ObjectId(voter_id)}, {'$inc': {'number_of_votes': vote_count}})
-        pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': vote_count}})
+        # Update MongoDB
+        voter_collection.update_one(
+            {'_id': ObjectId(vote_request.voter_id)}, 
+            {'$inc': {'number_of_votes': vote_request.vote_count}}
+        )
+        pdf_collection.update_one(
+            {'_id': ObjectId(vote_request.file_id)}, 
+            {'$inc': {'votes': vote_request.vote_count}}
+        )
         
-        # Xóa cache
+        # Clear cache
         redis_client.delete('all_files')
-        redis_client.delete(f'voter:{voter_id}')
+        redis_client.delete(f'voter:{vote_request.voter_id}')
         
         # Emit socket event
-        updated_file = pdf_collection.find_one({'_id': ObjectId(file_id)})
-        socketio.emit('vote_update', {
-            'file_id': file_id,
+        updated_file = pdf_collection.find_one({'_id': ObjectId(vote_request.file_id)})
+        await socket_manager.emit('vote_update', {
+            'file_id': vote_request.file_id,
             'votes': updated_file.get('votes', 0)
         })
         
-        return jsonify({'status': 'ok', 'message': f'Vote recorded successfully with {vote_count} votes'})
+        return {'status': 'ok', 'message': f'Vote recorded successfully with {vote_request.vote_count} votes'}
     except Exception as e:
-        return jsonify({'status': 'error', 'message': str(e)}), 500
+        raise HTTPException(status_code=500, detail=str(e))
 
-# API để lấy thông tin người bình chọn
-@app.route('/get-voter', methods=['GET'])
-def get_voter():
-    voter_id = request.args.get('id')
-    
-    # Kiểm tra cache
+@app.get("/get-voter")
+async def get_voter(voter_id: str):
+    # Check cache
     cached_voter = redis_client.get(f'voter:{voter_id}')
     if cached_voter:
-        return jsonify(json.loads(cached_voter))
+        return json.loads(cached_voter)
 
     voter = voter_collection.find_one({'_id': ObjectId(voter_id)})
     if not voter:
-        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
+        raise HTTPException(status_code=404, detail="Voter not found")
     
     voter_data = {
         'status': 'ok',
@@ -151,19 +158,18 @@ def get_voter():
         'number_of_votes': voter['number_of_votes']
     }
     
-    # Lưu vào cache
+    # Save to cache
     redis_client.setex(f'voter:{voter_id}', CACHE_EXPIRE_TIME, json.dumps(voter_data))
-    return jsonify(voter_data)
+    return voter_data
 
-# Socket events
-@socketio.on('connect')
-def handle_connect():
+@socket_manager.on('connect')
+async def handle_connect():
     print('Client connected')
 
-@socketio.on('disconnect')
-def handle_disconnect():
+@socket_manager.on('disconnect')
+async def handle_disconnect():
     print('Client disconnected')
 
-# Khởi chạy server
 if __name__ == "__main__":
-    socketio.run(app, port=5000, debug=True)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=5000)

requirements.txt

@@ -6,6 +6,6 @@ python-multipart
 pymongo
 redis
 flask
-flask-socketio 
+fastapi_socketio
 python-socketio
 flask-cors