update role

app.py

@@ -1,11 +1,20 @@
-from flask import Flask, request, jsonify
-from flask_cors import CORS
+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
 import pymongo
-from bson.objectid import ObjectId
 
-# Khởi tạo Flask app và cho phép CORS
-app = Flask(__name__)
-CORS(app)
+app = FastAPI()
+
+# Enable CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 # MongoDB connection
 mongo_url = "mongodb+srv://ip6ofme:[email protected]/"
@@ -14,126 +23,111 @@ 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')
+# 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(...)):
     new_pdf = {
-        'title': title,
-        'group': group,
-        'url': firebase_url,
-        'votes': 0
+        "title": title,
+        "group": group,
+        "url": url,
+        "votes": 0,
     }
     result = pdf_collection.insert_one(new_pdf)
-    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
-
-# API để tăng số lượng vote của file dựa trên ID
-@app.route('/vote', methods=['POST'])
-def vote():
-    data = request.json
-    file_id = data.get('id')
+    return {"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)})
+        file = pdf_collection.find_one({"_id": ObjectId(file_id)})
         if not file:
-            return jsonify({'status': 'error', 'message': 'File not found'}), 404
-
-        pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': 1}})
-        return jsonify({'status': 'ok'})
+            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:
-        return jsonify({'status': 'error', 'message': str(e)}), 500
-
-# 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')
+        raise HTTPException(status_code=500, detail=str(e))
 
+# 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(file_id)})
+        file = pdf_collection.find_one({"_id": ObjectId(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():
+# API to retrieve list of all files
+@app.get("/get-files")
+async def get_files():
     try:
         files = pdf_collection.find({})
-        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})
+        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}
     except Exception as e:
-        return jsonify({'status': 'error', 'message': str(e)}), 500
+        raise HTTPException(status_code=500, detail=str(e))
 
-# API mới để đă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
-    
+# API to register a voter
+@app.post("/register-voter")
+async def register_voter(request: RegisterVoterRequest):
     new_voter = {
-        'name': name,
-        'group': group,
-        'role': role,  # Lưu role vào MongoDB
-        'number_of_votes': 0
+        "name": request.name,
+        "group": request.group,
+        "role": request.role,
+        "number_of_votes": 0,
     }
     result = voter_collection.insert_one(new_voter)
-    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
+    return {"status": "ok", "id": str(result.inserted_id)}
 
-# API mới để 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')
-    
-    voter = voter_collection.find_one({'_id': ObjectId(voter_id)})
+# 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 not voter:
-        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
-    
-    max_votes = 5 if voter['role'] == 'judge' else 2
-    
-    if voter['number_of_votes'] >= max_votes:
-        return jsonify({'status': 'error', 'message': 'Maximum votes reached'}), 400
+        raise HTTPException(status_code=404, detail="Voter not found")
     
-    # Tăng số lượt bình chọn của người dùng
-    voter_collection.update_one({'_id': ObjectId(voter_id)}, {'$inc': {'number_of_votes': 1}})
+    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")
     
-    # Tăng số lượt bình chọn cho file
-    pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': 1}})
-    
-    return jsonify({'status': 'ok', 'message': 'Vote recorded successfully'})
+    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 để 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 = voter_collection.find_one({'_id': ObjectId(voter_id)})
+# API to get voter information
+@app.get("/get-voter")
+async def get_voter(id: str):
+    voter = voter_collection.find_one({"_id": ObjectId(id)})
     if not voter:
-        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
-    return jsonify({
-        'status': 'ok',
-        'name': voter['name'],
-        'group': voter['group'],
-        'role': voter['role'],  # Thêm role vào phản hồi
-        'number_of_votes': voter['number_of_votes']
-    })
-
-# Khởi chạy server
-if __name__ == "__main__":
-    app.run(port=5000, debug=True)
+        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"],
+    }