Done

Dockerfile

@@ -10,4 +10,4 @@ COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
-CMD ["python", "app.py"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,132 +1,147 @@
-from flask import Flask, request, jsonify
-from flask_cors import CORS
+from fastapi import FastAPI, HTTPException, Request, Depends
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 import pymongo
 from bson.objectid import ObjectId
 
-# Khởi tạo Flask app và cho phép CORS
-app = Flask(__name__)
-CORS(app)
+# 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   
 
-# 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.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')
+@app.post("/upload-files")
+async def upload_file(pdf: PDF):
     new_pdf = {
-        'title': title,
-        'group': group,
-        'url': firebase_url,
-        'votes': 0
+        "title": pdf.title,
+        "group": pdf.group,
+        "url": pdf.url,
+        "votes": 0
     }
     result = pdf_collection.insert_one(new_pdf)
-    return jsonify({'status': 'ok', 'id': str(result.inserted_id)})
+    return {"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')
-
+@app.post("/vote")
+async def vote(vote: Vote):
     try:
-        file = pdf_collection.find_one({'_id': ObjectId(file_id)})
+        file = pdf_collection.find_one({"_id": ObjectId(vote.id)})
         if not file:
-            return jsonify({'status': 'error', 'message': 'File not found'}), 404
+            raise HTTPException(status_code=404, detail="File not found")
 
-        pdf_collection.update_one({'_id': ObjectId(file_id)}, {'$inc': {'votes': 1}})
-        return jsonify({'status': 'ok'})
+        pdf_collection.update_one({"_id": ObjectId(vote.id)}, {"$inc": {"votes": 1}})
+        return {"status": "ok"}
     except Exception as e:
-        return jsonify({'status': 'error', 'message': str(e)}), 500
+        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')
-
+@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
+            raise HTTPException(status_code=404, detail="File not found")
 
-        return jsonify({'status': 'ok', 'votes': file.get('votes', 0)})
+        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({
-                '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 jsonify({'status': 'ok', 'data': file_list})
+        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')
-    
+@app.post("/register-voter")
+async def register_voter(voter: Voter):
     new_voter = {
-        'name': name,
-        'group': group,
-        'number_of_votes': 0
+        "name": voter.name,
+        "group": voter.group,
+        "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)})
+@app.post("/vote-by-voter")
+async def vote_by_voter(vote: VoteByVoter):
+    voter = voter_collection.find_one({"_id": ObjectId(vote.voter_id)})
     if not voter:
-        return jsonify({'status': 'error', 'message': 'Voter not found'}), 404
-    
-    if voter['number_of_votes'] >= 5:
-        return jsonify({'status': 'error', 'message': 'Maximum votes reached'}), 400
-    
+        raise HTTPException(status_code=404, detail="Voter not found")
+
+    if voter["number_of_votes"] >= 5:
+        raise HTTPException(status_code=400, detail="Maximum votes reached")
+
     # 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}})
-    
+    voter_collection.update_one({"_id": ObjectId(vote.voter_id)}, {"$inc": {"number_of_votes": 1}})
+
     # 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'})
+    pdf_collection.update_one({"_id": ObjectId(vote.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)})
+@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'],
-        'number_of_votes': voter['number_of_votes']
-    })
-
-app.run(port=7860, debug=True)
+        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"]
+    }
+
+# Khởi chạy server
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -1,5 +1,5 @@
-flask
-python-multipart
-flask_cors
-pymongo
-uvicorn
+fastapi
+uvicorn[standard]
+firebase_admin
+uuid
+python-multipart