Redis + Websocket

app.py

@@ -1,3 +1,4 @@
+Nguyễn Xuân Cường
 from fastapi import FastAPI, HTTPException, Form, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
@@ -106,11 +107,14 @@ async def get_files():
 
 @app.post("/register-voter")
 async def register_voter(voter: VoterRegistration):
+    # Xác định số lượng vote tối đa dựa vào role
+    max_votes = 10 if voter.role == 'judge' else 2
+    
     new_voter = {
         'name': voter.name,
         'group': voter.group,
         'role': voter.role,
-        'number_of_votes': 0
+        'number_of_votes': max_votes  # Khởi tạo với số vote tối đa
     }
     result = voter_collection.insert_one(new_voter)
     return {'status': 'ok', 'id': str(result.inserted_id)}
@@ -124,16 +128,16 @@ async def vote_by_voter(vote_request: VoteRequest):
     if not voter:
         raise HTTPException(status_code=404, detail="Voter not found")
 
-    max_votes = 10 if voter['role'] == 'judge' else 2
-
-    if voter['number_of_votes'] + vote_request.vote_count > max_votes:
-        raise HTTPException(status_code=400, detail="Maximum votes exceeded")
+    # Kiểm tra số vote còn lại
+    remaining_votes = voter['number_of_votes']  # Số vote còn lại
+    if vote_request.vote_count > remaining_votes:
+        raise HTTPException(status_code=400, detail=f"Not enough votes remaining. You have {remaining_votes} votes left")
 
     try:
-        # Update MongoDB
+        # Cập nhật MongoDB - giảm số vote còn lại
         voter_collection.update_one(
             {'_id': ObjectId(vote_request.voter_id)}, 
-            {'$inc': {'number_of_votes': vote_request.vote_count}}
+            {'$inc': {'number_of_votes': -vote_request.vote_count}}  # Giảm số vote còn lại
         )
         pdf_collection.update_one(
             {'_id': ObjectId(vote_request.file_id)}, 
@@ -165,7 +169,6 @@ async def get_voter(id: str):
                 return json.loads(cached_voter)
         except RedisError as redis_err:
             print(f"Redis error: {redis_err}")
-            # Continue without cache if Redis fails
             pass
 
         voter = voter_collection.find_one({'_id': ObjectId(id)})
@@ -177,15 +180,13 @@ async def get_voter(id: str):
             'name': voter['name'],
             'group': voter['group'],
             'role': voter['role'],
-            'number_of_votes': voter['number_of_votes']
+            'number_of_votes': voter['number_of_votes']  # Số này giờ là số vote còn lại
         }
         
-        # Try to save to cache
         try:
             redis_client.setex(f'voter:{id}', CACHE_EXPIRE_TIME, json.dumps(voter_data))
         except RedisError as redis_err:
             print(f"Redis caching error: {redis_err}")
-            # Continue even if caching fails
             pass
 
         return voter_data
@@ -212,6 +213,6 @@ async def shutdown_event():
     # Close Redis connection
     redis_client.close()
 
-if __name__ == "__main__":
+if _name_ == "_main_":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=5000)