update role

app.py

@@ -1,147 +1,139 @@
-from fastapi import FastAPI, HTTPException, Request, Depends
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
+from flask import Flask, request, jsonify
+from flask_cors import CORS
 import pymongo
 from bson.objectid import ObjectId
 
-# Khởi tạo FastAPI app và cho phép CORS
-app = FastAPI()
-
-origins = [
-    "*"  # Cho phép tất cả origin, bạn nên thay đổi trong môi trường production
-]
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=origins,
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# MongoDB connection   
+# Khởi tạo Flask app và cho phép CORS
+app = Flask(__name__)
+CORS(app)
 
+# MongoDB connection
 mongo_url = "mongodb+srv://ip6ofme:[email protected]/"
 client = pymongo.MongoClient(mongo_url)
 db = client["test"]
 pdf_collection = db["PdfDetails"]
 voter_collection = db["Voters"]
 
-# Pydantic models để validate dữ liệu
-class PDF(BaseModel):
-    title: str
-    group: str
-    url: str
-
-class Vote(BaseModel):
-    id: str
-
-class Voter(BaseModel):
-    name: str
-    group: str
-
-class VoteByVoter(BaseModel):
-    voter_id: str
-    file_id: str
-
 # API để upload file và lưu thông tin vào MongoDB
-@app.post("/upload-files")
-async def upload_file(pdf: PDF):
+@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": pdf.title,
-        "group": pdf.group,
-        "url": pdf.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 để tăng số lượng vote của file dựa trên ID
-@app.post("/vote")
-async def vote(vote: Vote):
+@app.route('/vote', methods=['POST'])
+def vote():
+    data = request.json
+    file_id = data.get('id')
+
     try:
-        file = pdf_collection.find_one({"_id": ObjectId(vote.id)})
+        file = pdf_collection.find_one({'_id': ObjectId(file_id)})
         if not file:
-            raise HTTPException(status_code=404, detail="File not found")
+            return jsonify({'status': 'error', 'message': 'File not found'}), 404
 
-        pdf_collection.update_one({"_id": ObjectId(vote.id)}, {"$inc": {"votes": 1}})
-        return {"status": "ok"}
+        pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': 1}})
+        return jsonify({'status': 'ok'})
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return jsonify({'status': 'error', 'message': str(e)}), 500
 
 # API để lấy số lượng votes của file theo ID
-@app.get("/get-votes")
-async def get_votes(id: str):
+@app.route('/get-votes', methods=['GET'])
+def get_votes():
+    file_id = request.args.get('id')
+
     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 jsonify({'status': 'error', 'message': 'File not found'}), 404
 
-        return {"status": "ok", "votes": file.get("votes", 0)}
+        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 để lấy danh sách tất cả các file
-@app.get("/get-files")
-async def get_files():
+@app.route('/get-files', methods=['GET'])
+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)
+                'id': str(file['_id']),
+                'title': file['title'],
+                'group': file['group'],
+                'url': file['url'],
+                'votes': file.get('votes', 0)
             })
-        return {"status": "ok", "data": file_list}
+        return jsonify({'status': 'ok', 'data': file_list})
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return jsonify({'status': 'error', 'message': str(e)}), 500
 
 # API mới để đăng ký người bình chọn
-@app.post("/register-voter")
-async def register_voter(voter: Voter):
+@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": voter.name,
-        "group": voter.group,
-        "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 mới để bình chọn
-@app.post("/vote-by-voter")
-async def vote_by_voter(vote: VoteByVoter):
-    voter = voter_collection.find_one({"_id": ObjectId(vote.voter_id)})
+@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)})
     if not voter:
-        raise HTTPException(status_code=404, detail="Voter not found")
-
-    if voter["number_of_votes"] >= 5:
-        raise HTTPException(status_code=400, detail="Maximum votes reached")
-
+        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
+    
     # Tăng số lượt bình chọn của người dùng
-    voter_collection.update_one({"_id": ObjectId(vote.voter_id)}, {"$inc": {"number_of_votes": 1}})
-
+    voter_collection.update_one({'_id': ObjectId(voter_id)}, {'$inc': {'number_of_votes': 1}})
+    
     # Tăng số lượt bình chọn cho file
-    pdf_collection.update_one({"_id": ObjectId(vote.file_id)}, {"$inc": {"votes": 1}})
-
-    return {"status": "ok", "message": "Vote recorded successfully"}
+    pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': 1}})
+    
+    return jsonify({'status': 'ok', 'message': 'Vote recorded successfully'})
 
 # API để lấy thông tin người bình chọn
-@app.get("/get-voter")
-async def get_voter(id: str):
-    voter = voter_collection.find_one({"_id": ObjectId(id)})
+@app.route('/get-voter', methods=['GET'])
+def get_voter():
+    voter_id = request.args.get('id')
+    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"],
-        "number_of_votes": voter["number_of_votes"]
-    }
+        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__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    app.run(port=5000, debug=True)