Redis + Websocket

app.py

@@ -1,20 +1,20 @@
-from fastapi import FastAPI, HTTPException, Request, Form, Depends
-from fastapi.responses import JSONResponse
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
-from bson import ObjectId
+from flask import Flask, request, jsonify
+from flask_cors import CORS
+from flask_socketio import SocketIO, emit
 import pymongo
+from bson.objectid import ObjectId
+import redis
+import json
+from redis.exceptions import RedisError
 
-app = FastAPI()
+# Khởi tạo Flask app và SocketIO
+app = Flask(__name__)
+CORS(app)
+socketio = SocketIO(app, cors_allowed_origins="*")
 
-# Enable CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# Khởi tạo Redis client
+redis_client = redis.Redis(host='localhost', port=6379, db=0)
+CACHE_EXPIRE_TIME = 300  # 5 phút
 
 # MongoDB connection
 mongo_url = "mongodb+srv://ip6ofme:[email protected]/"
@@ -23,111 +23,147 @@ db = client["test"]
 pdf_collection = db["PdfDetails"]
 voter_collection = db["Voters"]
 
-# Models
-class VoteRequest(BaseModel):
-    id: str
-
-class VoterVoteRequest(BaseModel):
-    voter_id: str
-    file_id: str
-
-class RegisterVoterRequest(BaseModel):
-    name: str
-    group: str
-    role: str
-
-# API to upload file information to MongoDB
-@app.post("/upload-files")
-async def upload_file(title: str = Form(...), group: str = Form(...), url: str = Form(...)):
+# 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')
     new_pdf = {
-        "title": title,
-        "group": group,
-        "url": url,
-        "votes": 0,
+        'title': title,
+        'group': group,
+        'url': firebase_url,
+        'votes': 0
     }
     result = pdf_collection.insert_one(new_pdf)
-    return {"status": "ok", "id": str(result.inserted_id)}
+    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
 
-# API to increment file vote count by ID
-@app.post("/vote")
-async def vote(request: VoteRequest):
-    file_id = request.id
-    try:
-        file = pdf_collection.find_one({"_id": ObjectId(file_id)})
-        if not file:
-            raise HTTPException(status_code=404, detail="File not found")
-        pdf_collection.update_one({"_id": ObjectId(file_id)}, {"$inc": {"votes": 1}})
-        return {"status": "ok"}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+# 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')
 
-# API to get vote count by file ID
-@app.get("/get-votes")
-async def get_votes(id: str):
     try:
-        file = pdf_collection.find_one({"_id": ObjectId(id)})
+        file = pdf_collection.find_one({'_id': ObjectId(file_id)})
         if not file:
-            raise HTTPException(status_code=404, detail="File not found")
-        return {"status": "ok", "votes": file.get("votes", 0)}
+            return jsonify({'status': 'error', 'message': 'File not found'}), 404
+
+        return jsonify({'status': 'ok', 'votes': file.get('votes', 0)})
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return jsonify({'status': 'error', 'message': str(e)}), 500
 
-# API to retrieve list of all files
-@app.get("/get-files")
-async def get_files():
+# API để lấy danh sách tất cả các file
+@app.route('/get-files', methods=['GET'])
+def get_files():
     try:
         files = pdf_collection.find({})
-        file_list = [
-            {
-                "id": str(file["_id"]),
-                "title": file["title"],
-                "group": file["group"],
-                "url": file["url"],
-                "votes": file.get("votes", 0),
-            }
-            for file in files
-        ]
-        return {"status": "ok", "data": file_list}
+        file_list = []
+        for file in files:
+            file_list.append({
+                '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})
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        print(f"Error: {str(e)}")
+        return jsonify({'status': 'error', 'message': str(e)}), 500
 
-# API to register a voter
-@app.post("/register-voter")
-async def register_voter(request: RegisterVoterRequest):
+# 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
+    
     new_voter = {
-        "name": request.name,
-        "group": request.group,
-        "role": request.role,
-        "number_of_votes": 0,
+        'name': name,
+        'group': group,
+        'role': role,  # Lưu role vào MongoDB
+        'number_of_votes': 0
     }
     result = voter_collection.insert_one(new_voter)
-    return {"status": "ok", "id": str(result.inserted_id)}
+    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)
 
-# API for voting by voter
-@app.post("/vote-by-voter")
-async def vote_by_voter(request: VoterVoteRequest):
-    voter = voter_collection.find_one({"_id": ObjectId(request.voter_id)})
+    if vote_count <= 0:
+        return jsonify({'status': 'error', 'message': 'Vote count must be greater than 0'}), 400
+
+    voter = voter_collection.find_one({'_id': ObjectId(voter_id)})
     if not voter:
-        raise HTTPException(status_code=404, detail="Voter not found")
-    
-    max_votes = 5 if voter["role"] == "judge" else 2
-    if voter["number_of_votes"] >= max_votes:
-        raise HTTPException(status_code=400, detail="Maximum votes reached")
+        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
+
+    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
+
+    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}})
+        
+        # Xóa cache
+        redis_client.delete('all_files')
+        redis_client.delete(f'voter:{voter_id}')
+        
+        # Emit socket event
+        updated_file = pdf_collection.find_one({'_id': ObjectId(file_id)})
+        socketio.emit('vote_update', {
+            'file_id': file_id,
+            'votes': updated_file.get('votes', 0)
+        })
+        
+        return jsonify({'status': 'ok', 'message': f'Vote recorded successfully with {vote_count} votes'})
+    except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)}), 500
+
+# 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')
     
-    voter_collection.update_one({"_id": ObjectId(request.voter_id)}, {"$inc": {"number_of_votes": 1}})
-    pdf_collection.update_one({"_id": ObjectId(request.file_id)}, {"$inc": {"votes": 1}})
-    return {"status": "ok", "message": "Vote recorded successfully"}
-
-# API to get voter information
-@app.get("/get-voter")
-async def get_voter(id: str):
-    voter = voter_collection.find_one({"_id": ObjectId(id)})
+    # Kiểm tra cache
+    cached_voter = redis_client.get(f'voter:{voter_id}')
+    if cached_voter:
+        return jsonify(json.loads(cached_voter))
+
+    voter = voter_collection.find_one({'_id': ObjectId(voter_id)})
     if not voter:
-        raise HTTPException(status_code=404, detail="Voter not found")
-    return {
-        "status": "ok",
-        "name": voter["name"],
-        "group": voter["group"],
-        "role": voter["role"],
-        "number_of_votes": voter["number_of_votes"],
+        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
+    
+    voter_data = {
+        'status': 'ok',
+        'name': voter['name'],
+        'group': voter['group'],
+        'role': voter['role'],
+        'number_of_votes': voter['number_of_votes']
     }
+    
+    # Lưu vào cache
+    redis_client.setex(f'voter:{voter_id}', CACHE_EXPIRE_TIME, json.dumps(voter_data))
+    return jsonify(voter_data)
+
+# Socket events
+@socketio.on('connect')
+def handle_connect():
+    print('Client connected')
+
+@socketio.on('disconnect')
+def handle_disconnect():
+    print('Client disconnected')
+
+# Khởi chạy server
+if __name__ == "__main__":
+    socketio.run(app, port=5000, debug=True)

requirements.txt

@@ -3,4 +3,8 @@ uvicorn[standard]
 firebase_admin
 uuid
 python-multipart
-pymongo
+pymongo
+redis
+flask
+flask-socketio 
+python-socketio