agenda/backend-collector
1
0
Fork 0
Code Issues Pull Requests Projects Releases 37 Wiki Activity

add redis storage

Browse Source
This commit is contained in:
Valentin CZERYBA 2023-10-15 16:48:58 +02:00
parent 3392d8720f
commit f2b7c63eee
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/dependencies/database.py
View File

@ -1,5 +1,7 @@
from pymongo import MongoClient from pymongo import MongoClient
import os import os, redis
client = MongoClient("mongodb+srv://{0}:{1}@{2}/?retryWrites=true&w=majority&appName=AtlasApp".format(os.environ["MONGO_USER"], os.environ["MONGO_PASSWORD"], os.environ["MONGO_HOST"])) client = MongoClient("mongodb+srv://{0}:{1}@{2}/?retryWrites=true&w=majority&appName=AtlasApp".format(os.environ["MONGO_USER"], os.environ["MONGO_PASSWORD"], os.environ["MONGO_HOST"]))
database = client[os.environ["MONGO_DATABASE"]] database = client[os.environ["MONGO_DATABASE"]]
connect_redis = redis.Redis(host=os.environ["REDIS_URL"], port=os.environ["REDIS_PORT"], decode_responses=True)

5
app/routers/mail.py
View File

@ -8,7 +8,7 @@ router = APIRouter()
@router.post("/mail",tags=["mail"]) @router.post("/mail",tags=["mail"])
async def read_users_me(userSingle: users.UserCreate | None = None): async def create_user(userSingle: users.UserCreate | None = None):
if userSingle is None: if userSingle is None:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, status_code=status.HTTP_400_BAD_REQUEST,
@ -37,4 +37,5 @@ async def read_users_me(userSingle: users.UserCreate | None = None):
await fm.send_message(message, template_name="mailer.html") await fm.send_message(message, template_name="mailer.html")
current_user = users.User(username=userSingle.username, password=users_token.get_password_hash(userSingle.password), email=userSingle.email) current_user = users.User(username=userSingle.username, password=users_token.get_password_hash(userSingle.password), email=userSingle.email)
user_repository.save(current_user) user_repository.save(current_user)
return JSONResponse(status_code=200, content={"message": "email has been sent"}) database.connect_redis.set(userSingle.username, key_hashed)
return JSONResponse(status_code=status.HTTP_200_OK, content={"message": "email has been sent"})