Mawoka
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
import html
|
|
import json
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from aiohttp import ClientSession
|
|
|
|
from classquiz.config import settings, storage
|
|
from classquiz.db.models import Quiz, QuizAnswer, QuizQuestion, User
|
|
from classquiz.kahoot_importer.get import get as get_quiz
|
|
|
|
settings = settings()
|
|
|
|
|
|
async def _download_image(url: str) -> bytes:
|
|
async with ClientSession() as session:
|
|
async with session.get(url) as resp:
|
|
return await resp.read()
|
|
|
|
|
|
# async def _upload_to_imgur(image: bytes, album_id: str | None) -> str:
|
|
# """
|
|
# Uploads an image to Imgur.
|
|
# :param image: The image to upload.
|
|
# :return: The URL of the image on Imgur.
|
|
# """
|
|
#
|
|
# formdata = FormData()
|
|
# formdata.add_field('image', image, filename='image.png')
|
|
# if album_id:
|
|
# formdata.add_field('album', album_id)
|
|
# async with ClientSession(headers={"Authorization": f"Client-ID {settings.imgur_client_id}"}) as session:
|
|
# print(settings.imgur_client_id)
|
|
# async with session.post("https://api.imgur.com/3/image", data=image) as resp:
|
|
# data = await resp.json()
|
|
# print(resp.headers)
|
|
# try:
|
|
# return data["data"]["link"]
|
|
# except KeyError:
|
|
# raise Exception(data)
|
|
#
|
|
#
|
|
# async def _generate_imgur_album(quiz_id: str) -> str:
|
|
# # """
|
|
# # Generates an album on Imgur.
|
|
# # :param quiz_id: The ID of the quiz.
|
|
# # :return: The URL of the album on Imgur.
|
|
# # """
|
|
# formdata = FormData()
|
|
# formdata.add_field('title', quiz_id)
|
|
# async with ClientSession(headers={"Authorization": f"Client-ID {settings.imgur_client_id}"}) as session:
|
|
# async with session.post("https://api.imgur.com/3/album", data={"title": "Quiz: " + quiz_id}, ) as resp:
|
|
# data = await resp.json()
|
|
# return data["data"]["id"]
|
|
|
|
|
|
async def import_quiz(quiz_id: str, user: User) -> Quiz | str:
|
|
"""
|
|
Imports a quiz from Kahoot.
|
|
:param user: The user object
|
|
:param quiz_id: The ID of the quiz to import.
|
|
:return: True if the import was successful, False otherwise.
|
|
"""
|
|
quiz = await get_quiz(quiz_id)
|
|
if quiz is None:
|
|
return "quiz not found"
|
|
quiz_questions: list[dict] = []
|
|
quiz_id = uuid.uuid4()
|
|
|
|
for q in quiz.kahoot.questions:
|
|
answers: list[QuizAnswer] = []
|
|
image = None
|
|
if q.image is not None and q.image != "":
|
|
image_bytes = await _download_image(q.image)
|
|
image_name = f"{quiz_id}--{uuid.uuid4()}"
|
|
image = await storage.upload(file_name=image_name, file_data=image_bytes)
|
|
image = f"{settings.root_address}/api/v1/storage/download/{image_name}"
|
|
# image = q.image
|
|
for a in q.choices:
|
|
answers.append((QuizAnswer(right=a.correct, answer=html.unescape(a.answer))))
|
|
quiz_questions.append(
|
|
QuizQuestion(
|
|
question=q.question,
|
|
answers=answers,
|
|
time=str(q.time / 1000),
|
|
image=image,
|
|
).dict()
|
|
)
|
|
quiz_data = Quiz(
|
|
id=quiz_id,
|
|
public=False,
|
|
title=quiz.kahoot.title,
|
|
description=quiz.kahoot.description,
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now(),
|
|
user_id=user.id,
|
|
questions=json.dumps(quiz_questions),
|
|
)
|
|
return await quiz_data.save()
|