Skip to content

Backend #60

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
EscGit merged 2 commits into from
May 5, 2025
Merged

Backend #60

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3e7e7a6
hold on
AlphaIru May 5, 2025
f9bdf6b
image support done
AlphaIru May 5, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions backend/controller_group.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -223,7 +223,6 @@ def get_group_members_control(
raise BackendError("Backend Error: Group does not exist", "306")
if not Validator().check_user_in_group(user_id, group_id):
raise BackendError("Backend Error: User not in group", "308")

with db_operation() as data_cursor:
# Get all users in the group
data_cursor.execute(
Expand All @@ -235,10 +234,10 @@ def get_group_members_control(
(group_id,),
)
user_ids = [row[0] for row in data_cursor.fetchall()]

if not user_ids:
return []

# Get user information for each member
placeholders = ",".join("?" for _ in user_ids)
data_cursor.execute(
Expand All @@ -250,13 +249,13 @@ def get_group_members_control(
user_ids,
)
members_data = data_cursor.fetchall()

# Create a list of member dictionaries
members = [
{"user_id": member_id, "username": member_name, "email": member_email}
for member_id, member_name, member_email in members_data
]

return members


Expand Down
267 changes: 267 additions & 0 deletions backend/controller_image.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,267 @@
# coding: utf-8
"""This module handles image processing and storage."""

from os import remove
from os.path import exists, join
from typing import Any

from werkzeug.utils import secure_filename

from error import BackendError
from utils import db_operation
from validator import Validator, UPLOAD_FOLDER, ALLOWED_EXTENSIONS


class ImageController:
"""This class will handle the image."""

def __init__(self) -> None:
"""Initialize the image controller."""
return

def allowed_file(self, filename: str) -> bool:
"""Check if the file has an allowed extension."""
return (
"." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXTENSIONS
)

def get_user_image_control(
self,
image_url: str,
user_id: str,
password: str,
) -> str:
"""Gets the user image."""
if not Validator().check_user_exists(user_id=user_id):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(user_id=user_id, password=password):
raise BackendError("Backend Error: Password is incorrect", "305")
if not exists(image_url):
raise BackendError("Backend Error: Image does not exist", "313")
return image_url

def get_task_image_control(
self,
request_data: dict[str, Any],
) -> str:
"""Gets the task image."""
if not Validator().check_user_exists(user_id=request_data["user_id"]):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(
user_id=request_data["user_id"], password=request_data["password"]
):
raise BackendError("Backend Error: Password is incorrect", "305")
if not Validator().check_task_exists(task_id=request_data["task_id"]):
raise BackendError("Backend Error: Task does not exist", "309")
if (request_data["group_id"] != "0") and not Validator().check_group_exists(
group_id=request_data["group_id"]
):
raise BackendError("Backend Error: Group does not exist", "306")
if not exists(request_data["image_url"]):
raise BackendError("Backend Error: Image does not exist", "313")
return request_data["image_url"]

def upload_user_image_control(
self,
user_id: str,
password: str,
file,
) -> str:
"""Uploads the user image."""
if not Validator().check_user_exists(user_id=user_id):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(user_id=user_id, password=password):
raise BackendError("Backend Error: Password is incorrect", "305")

if not file or not self.allowed_file(file.filename):
raise BackendError("Backend Error: Invalid file type", "312")

filename = secure_filename(file.filename)
file_path = join(UPLOAD_FOLDER, filename)
file.save(file_path)

with db_operation() as data_cursor:
data_cursor.execute(
"UPDATE user SET image_path = ? WHERE uuid = ?;",
(file_path, user_id),
)

return file_path

def upload_task_image_control(
self,
request_data: dict[str, Any],
file,
) -> str:
"""Uploads the task image."""
if not Validator().check_user_exists(user_id=request_data["user_id"]):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(
user_id=request_data["user_id"], password=request_data["password"]
):
raise BackendError("Backend Error: Password is incorrect", "305")
if not Validator().check_task_exists(task_id=request_data["task_id"]):
raise BackendError("Backend Error: Task does not exist", "309")
if (request_data["group_id"] != "0") and not Validator().check_group_exists(
group_id=request_data["group_id"]
):
raise BackendError("Backend Error: Group does not exist", "306")

if not file or not self.allowed_file(file.filename):
raise BackendError("Backend Error: Invalid file type", "312")

filename = secure_filename(file.filename)
file_path = join(UPLOAD_FOLDER, filename)
file.save(file_path)

with db_operation() as data_cursor:
data_cursor.execute(
"UPDATE task SET image_path = ? WHERE uuid = ?;",
(file_path, request_data["task_id"]),
)

return file_path

def edit_user_image_control(
self,
user_id: str,
password: str,
file,
) -> str:
"""Edits the user image."""
if not Validator().check_user_exists(user_id=user_id):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(user_id=user_id, password=password):
raise BackendError("Backend Error: Password is incorrect", "305")

if not file or not self.allowed_file(file.filename):
raise BackendError("Backend Error: Invalid file type", "312")

filename = secure_filename(file.filename)
file_path = join(UPLOAD_FOLDER, filename)
file.save(file_path)

with db_operation() as data_cursor:
data_cursor.execute(
"UPDATE user SET image_path = ? WHERE uuid = ?;",
(file_path, user_id),
)

return file_path

def edit_task_image_control(
self,
request_data: dict[str, Any],
file,
) -> str:
"""Edits the task image."""
if not Validator().check_user_exists(user_id=request_data["user_id"]):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(
user_id=request_data["user_id"], password=request_data["password"]
):
raise BackendError("Backend Error: Password is incorrect", "305")
if not Validator().check_task_exists(task_id=request_data["task_id"]):
raise BackendError("Backend Error: Task does not exist", "309")
if (request_data["group_id"] != "0") and not Validator().check_group_exists(
group_id=request_data["group_id"]
):
raise BackendError("Backend Error: Group does not exist", "306")

if not file or not self.allowed_file(file.filename):
raise BackendError("Backend Error: Invalid file type", "312")

filename = secure_filename(file.filename)
file_path = join(UPLOAD_FOLDER, filename)
file.save(file_path)

with db_operation() as data_cursor:
data_cursor.execute(
"UPDATE task SET image_path = ? WHERE uuid = ?;",
(file_path, request_data["task_id"]),
)

return file_path

def delete_user_image_control(
self,
user_id: str,
password: str,
) -> None:
"""Deletes the user image."""
if not Validator().check_user_exists(user_id=user_id):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(user_id=user_id, password=password):
raise BackendError("Backend Error: Password is incorrect", "305")

with db_operation() as data_cursor:
data_cursor.execute(
"SELECT image_path FROM user WHERE uuid = ?;",
(user_id,),
)
result = data_cursor.fetchone()
if not result:
raise BackendError("Backend Error: Image does not exist", "313")
remove(result[0])
try:
remove(result[0])
except FileNotFoundError as err:
raise BackendError(
"Backend Error: Image does not exist", "313"
) from err
except Exception as err:
raise BackendError(
"Backend Error: Failed to delete image", "203"
) from err

data_cursor.execute(
"UPDATE user SET image_path = NULL WHERE uuid = ?;",
(user_id,),
)

def delete_task_image_control(
self,
request_data: dict[str, Any],
) -> None:
"""Deletes the task image."""
if not Validator().check_user_exists(user_id=request_data["user_id"]):
raise BackendError("Backend Error: User does not exist", "304")
if not Validator().check_password(
user_id=request_data["user_id"], password=request_data["password"]
):
raise BackendError("Backend Error: Password is incorrect", "305")
if not Validator().check_task_exists(task_id=request_data["task_id"]):
raise BackendError("Backend Error: Task does not exist", "309")
if (request_data["group_id"] != "0") and not Validator().check_group_exists(
group_id=request_data["group_id"]
):
raise BackendError("Backend Error: Group does not exist", "306")

with db_operation() as data_cursor:
data_cursor.execute(
"SELECT image_path FROM task WHERE uuid = ?;",
(request_data["task_id"],),
)
result = data_cursor.fetchone()
if not result:
raise BackendError("Backend Error: Image does not exist", "313")
remove(result[0])
try:
remove(result[0])
except FileNotFoundError as err:
raise BackendError(
"Backend Error: Image does not exist", "313"
) from err
except Exception as err:
raise BackendError(
"Backend Error: Failed to delete image", "203"
) from err

data_cursor.execute(
"UPDATE task SET image_path = NULL WHERE uuid = ?;",
(request_data["task_id"],),
)


if __name__ == "__main__":
print("This module is not meant to be run directly.")
3 changes: 2 additions & 1 deletion backend/controller_invite.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# coding: utf-8
"""This module handles the functions related toi invites for database."""

from datetime import datetime, timezone
Expand Down Expand Up @@ -68,7 +69,7 @@ def get_pending_control(
invites_data: list[tuple] = data_cursor.fetchall()

if not invites_data:
return {} # Return early if no invites
return {}

# 2. Collect unique IDs
group_ids = {item[2] for item in invites_data}
Expand Down
Loading
Loading