Juartaurus's picture
Upload folder using huggingface_hub
1865436
raw
history blame contribute delete
No virus
1.17 kB
import motor.motor_asyncio
from pymongo import MongoClient
from fastapi.responses import JSONResponse, HTMLResponse
# Mongodb setting
client = MongoClient("mongodb://localhost:27017/")
db = client.get_database("database")
user_collection = db.get_collection("users")
def login_user(username, password):
query = {"username": {"$eq": username}}
users = list(user_collection.find(query))
if len(users) > 0:
if password != str(users[0]['password']):
response = "Incorrect password"
else:
response = "Login Successfully"
return JSONResponse(content={"name": users[0]['username']}, status_code=200)
else:
response = "Incorrect username"
return JSONResponse(status_code=404, content={"message": response})
def get_project_user(username):
query = {"username": {"$eq": username}}
users = list(user_collection.find(query))
if len(users) > 0:
return JSONResponse(content={"name": users[0]['projectname'], "path":users[0]["projectpath"]}, status_code=200)
else:
response = "Incorrect username"
return JSONResponse(status_code=404, content={"message": response})