LobotomyLabs/classquiz-ai
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🚑 image-fix (4th time)

Browse Source
This commit is contained in:
Mawoka
2022-04-11 17:58:13 +02:00
parent 072791c5ed
commit 5b08c382a0
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classquiz/routers/quiz.py
+10 -5
View File
@@ -14,17 +14,18 @@ from classquiz.db.models import Quiz, QuizInput, User, PlayGame
from classquiz.kahoot_importer.import_quiz import import_quiz from classquiz.kahoot_importer.import_quiz import import_quiz
settings = settings() settings = settings()
imgur_regex = r"^https://i\.imgur\.com\/.{7}.(jpg|png|gif)$"
server_regex = rf"^{settings.root_address}/api/v1/storage/download/.{36}--.{36}$"
router = APIRouter() router = APIRouter()
@router.post("/create") @router.post("/create")
async def create_quiz_lol(quiz_input: QuizInput, user: User = Depends(get_current_user)): async def create_quiz_lol(quiz_input: QuizInput, user: User = Depends(get_current_user)):
imgur_regex = r"^https://i\.imgur\.com\/.{7}.(jpg|png|gif)$"
server_regex = rf"^{settings.root_address}/api/v1/storage/download/.{36}--.{36}$"
for question in quiz_input.questions: for question in quiz_input.questions:
print(question.image) if question.image == "":
question.image = None
if question.image is not None or question.image != "": if question.image is not None or question.image != "":
if not re.match(imgur_regex, question.image) or not re.match(server_regex, question.image): if not re.match(imgur_regex, question.image) or not re.match(server_regex, question.image):
raise HTTPException(status_code=400, detail="image url is not valid") raise HTTPException(status_code=400, detail="image url is not valid")
@@ -92,8 +93,12 @@ async def get_quiz_list(user: User = Depends(get_current_user)):
@router.put("/update/{quiz_id}") @router.put("/update/{quiz_id}")
async def update_quiz(quiz_id: str, quiz_input: QuizInput, user: User = Depends(get_current_user)): async def update_quiz(quiz_id: str, quiz_input: QuizInput, user: User = Depends(get_current_user)):
imgur_regex = r"^https://i\.imgur\.com\/.{7}.(jpg|png|gif)$"
server_regex = rf"^{settings.root_address}/api/v1/storage/download/.{36}--.{36}$"
for question in quiz_input.questions: for question in quiz_input.questions:
if question.image is not None or question.image != "": if question.image == "":
question.image = None
if question.image is not None:
if not re.match(imgur_regex, question.image) or not re.match(server_regex, question.image): if not re.match(imgur_regex, question.image) or not re.match(server_regex, question.image):
raise HTTPException(status_code=400, detail="image url is not valid") raise HTTPException(status_code=400, detail="image url is not valid")
try: try: