image test

app/api/classes.py

@@ -23,6 +23,8 @@ from tortoise.queryset import Q
 
 classes = APIRouter()
 
+IMAGEDIR = "/var/www/ntcri/app/api/images/"
+
 # SECRET: str = secrets.token_urlsafe(32)
 # manager = LoginManager(SECRET, '/login',default_expiry=timedelta(hours=72))
 
@@ -52,10 +54,23 @@ async def insert_class(
     contact: str = Form(default=''),
     introduction: str = Form(default=''),
     content: str = Form(default=''),
-    cover_img: str = Form(default='')
+    cover_img: str = Form(default=''),
+    cover_img_file:UploadFile = File(default='')
 
 ):
     try:
+        contents = await cover_img_file.read()
+
+        if cover_img.strip() != '':
+            cover_img = f"{IMAGEDIR}{cover_img}"
+        else:
+            cover_img = f"{IMAGEDIR}{cover_img_file.filename}"
+
+        #save the file
+        with open(cover_img, "wb") as f:
+            f.write(contents)
+        
+
         new_class = await Class_list.create(
             id=id,
             name=name,
@@ -115,6 +130,7 @@ async def update_class(
     introduction: str = Form(default=''),
     content: str = Form(default=''),
     cover_img: str = Form(default=''),
+    cover_img_file:UploadFile = File(default='')
 ):
     try:
         class_obj = await Class_list.get(id=id)
@@ -146,8 +162,30 @@ async def update_class(
         if content.strip() != '':
             class_obj.content = content
 
+        
+
         if cover_img.strip() != '':
-            class_obj.cover_img = cover_img
+            if cover_img_file.strip() != '':
+                class_obj.cover_img = f"{IMAGEDIR}{cover_img}"
+
+                contents = await cover_img_file.read()
+                with open(class_obj.cover_img, "wb") as f:
+                    f.write(contents)
+            else:
+                os.rename(class_obj.cover_img, f"{IMAGEDIR}{cover_img}")
+                class_obj.cover_img = f"{IMAGEDIR}{cover_img}"
+
+        else:
+            if cover_img_file.strip() != '':
+                class_obj.cover_img =f"{IMAGEDIR}{cover_img_file.filename}"
+
+                contents = await cover_img_file.read()
+                with open(class_obj.cover_img, "wb") as f:
+                    f.write(contents)
+
+
+        
+
 
 
         await class_obj.save()
@@ -239,7 +277,7 @@ Q(name__icontains=keyword) | Q(lecturer__icontains=keyword)
     except Exception as e:
         return {"msg": str(e), "code": 500}
 
-IMAGEDIR = "/var/www/ntcri/app/api/images/"
+
 
 @classes.post("/upload/")
 async def create_upload_file(file: UploadFile = File(...)):
@@ -254,8 +292,8 @@ async def create_upload_file(file: UploadFile = File(...)):
     return {"filename": file.filename}
  #"/var/www/ntcri/app/api/images/test.jpeg"
  
-@classes.get("/show/")
-async def read_random_file():
+
+async def read_image_file():
  
     # get random file from the image directory
     files = os.listdir(IMAGEDIR)
@@ -264,115 +302,3 @@ async def read_random_file():
     path = f"{IMAGEDIR}{files[random_index]}"
      
     return FileResponse(path)
-# @classes.post("/login")
-# async def login(data: OAuth2PasswordRequestForm = Depends()):
-#     username = data.username
-#     password = data.password
-
-#     user = await query_user(username)
-#     print(user)
-
-#     if not user:
-#         # you can return any response or error of your choice
-#         raise InvalidCredentialsException
-#     elif password != user.password:
-#         raise InvalidCredentialsException
-#     access_token = manager.create_access_token(
-#         data={'sub': username}
-#     )
-#     return {'access_token': access_token}
-
-
-# @classes.post("/logout")
-# async def logout():
-
-#     return {"msg":"logout success","code":200}
-
-# @classes.post("/add")
-# async def add(username: str = Form(default=''), password: str = Form(default=''), email: str = Form(default='')):
-#     if username and password and email:
-#         u = await User.create(username=username, password=password, email=email)
-#         if u:
-#             send_email()
-#             return {"msg": "已寄送認證信", "code": 200}
-
-#     return {"msg": "create user failed", "code": 403}
-
-
-# def generate_password_reset_token(email: str) -> str:
-#     delta = timedelta(hours=settings.EMAIL_RESET_TOKEN_EXPIRE_HOURS)
-#     now = datetime.utcnow()
-#     expires = now + delta
-#     exp = expires.timestamp()
-#     encoded_jwt = jwt.encode(
-#         {"exp": exp, "nbf": now, "sub": email}, settings.SECRET_KEY, algorithm="HS256",
-#     )
-#     return encoded_jwt
-
-
-# def send_email(
-#     email_to: str,
-#     subject_template: str = "",
-#     html_template: str = "",
-#     environment: Dict[str, Any] = {},
-# ) -> None:
-#     # assert settings.EMAILS_ENABLED, "no provided configuration for email variables"
-#     message = emails.Message(
-#         subject=JinjaTemplate(subject_template),
-#         html=JinjaTemplate(html_template),
-#         mail_from=(settings.EMAILS_FROM_NAME, settings.EMAILS_FROM_EMAIL),
-#     )
-#     smtp_options = {"host": settings.SMTP_HOST, "port": settings.SMTP_PORT}
-#     if settings.SMTP_TLS:
-#         smtp_options["tls"] = True
-#     if settings.SMTP_USER:
-#         smtp_options["user"] = settings.SMTP_USER
-#     if settings.SMTP_PASSWORD:
-#         smtp_options["password"] = settings.SMTP_PASSWORD
-#     response = message.send(to=email_to, render=environment, smtp=smtp_options)
-#     logging.info(f"send email result: {response}")
-
-
-# def send_reset_password_email(email_to: str, email: str, token: str) -> None:
-#     subject = f"Password recovery for user {email}"
-#     with open(Path(settings.EMAIL_TEMPLATES_DIR) / "reset_password.html") as f:
-#         template_str = f.read()
-#     server_host = settings.SERVER_HOST
-#     link = f"{server_host}/reset-password?token={token}"
-#     send_email(
-#         email_to=email_to,
-#         subject_template=subject,
-#         html_template=template_str,
-#         environment={
-#             # "project_name": settings.PROJECT_NAME,
-#             "username": email,
-#             "email": email_to,
-#             "valid_hours": settings.EMAIL_RESET_TOKEN_EXPIRE_HOURS,
-#             "link": link,
-#         },
-#     )
-
-
-# @users.post("/password-recovery/{email}")
-# async def recover_password(email:str):
-#     user = await User.filter(email=email).first()
-#     if not user:
-#         raise HTTPException(
-#             status_code=404,
-#             detail="The user with this username does not exist in the system.",
-#         )
-#     password_reset_token = generate_password_reset_token(email=email)
-#     send_reset_password_email(
-#         email_to=user.email, email=email, token=password_reset_token
-#     )
-#     return {"msg": "Password recovery email sent"}
-
-
-
-# @users.get("/delete_user/{id}")
-# async def delete(id: int):
-#     if id:
-#         await User.filter(id=id).delete()
-#         return {"msg": "success", "code": 200}
-
-#     return {"msg": "failed", "code": 400}