init archive back app

Author: githubering182

backend-app/archive/__init__.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ArchiveConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'archive'

backend-app/archive/apps.py

@@ -0,0 +1,28 @@
+from django.db.models import (
+    Model,
+    CharField,
+    ForeignKey,
+    ManyToManyField,
+    DO_NOTHING,
+    DateTimeField,
+)
+
+
+class Archive(Model):
+    STATUSES: tuple = (
+        ("s", "SUCCESS"),
+        ("f", "FAILURE"),
+        ("p", "PENGING"),
+    )
+
+    id = CharField(max_length=24, primary_key=True, unique=True)
+    result_id = CharField(max_length=24, unique=True)
+    status = CharField(max_length=1, choices=STATUSES, default="p")
+    create_date = DateTimeField(auto_now_add=True)
+
+    project = ForeignKey("project.Project", on_delete=DO_NOTHING)
+    author = ForeignKey("user.CustomUser", on_delete=DO_NOTHING)
+    file = ManyToManyField("file.File")
+
+    class Meta:
+        db_table = "archive"

backend-app/archive/models.py

@@ -0,0 +1,15 @@
+from rest_framework.permissions import BasePermission
+from rest_framework.views import Request, APIView
+
+
+class ArchivesPermission(BasePermission):
+    def has_permission(self, request: Request, view: APIView) -> bool | None:
+        project_perm = bool(request.user.project_download.filter(id=view.kwargs["p_pk"]))
+        return request.user.is_superuser or project_perm
+
+
+class ArchivePermission(BasePermission):
+    def has_permission(self, request: Request, view: APIView) -> bool | None:
+        project_perm = bool(request.user.project_download.filter(id=view.kwargs["p_pk"]))
+        arch_perm = bool(request.user.archive_set.filter(id=view.kwargs["pk"]))
+        return request.user.is_superuser or (project_perm and arch_perm)

backend-app/archive/permissions.py

@@ -0,0 +1,9 @@
+from rest_framework.serializers import ModelSerializer
+from typing import Any
+from .models import Archive
+
+
+class ArchiveSerializer(ModelSerializer):
+    class Meta:
+        model = Archive
+        fields = ("id", "result_id", "status",)

backend-app/archive/serializers.py

@@ -0,0 +1,76 @@
+from rest_framework.status import (
+    HTTP_400_BAD_REQUEST,
+    HTTP_201_CREATED,
+    HTTP_200_OK,
+    HTTP_202_ACCEPTED
+)
+from django.db.models.aggregates import Count
+from typing import Any
+from project.models import Project
+from api.mixins import with_model_assertion
+from .serializers import ArchiveSerializer
+
+
+@with_model_assertion(Project, "id", filter={"visible": True},)
+def _get_archives(project: Project) -> tuple[dict[str, Any], int]:
+    query = project.archive_set \
+        .annotate(file_count=Count("file")) \
+        .order_by("create_date")
+    return ArchiveSerializer(query, many=True).data, HTTP_200_OK
+
+
+@with_model_assertion(Project, "id", filter={"visible": True})
+def _make_archive(
+    project: Project,
+    request_data: dict[str, Any]
+) -> tuple[dict[str, Any], int]:
+    ...
+#     response: dict[str, Any]
+#     status: int
+
+#     try:
+#         attribute = project.attribute_set.get(id=request_data["attribute_id"])
+
+#         assert not attribute.projectgoal_set.count(), "Goal for Attribute already exists"
+
+#         goal_attribute_names = [
+#             f"{name} ({level_name})"
+#             for name, level_name in
+#             attribute
+#             .ancestors(include_self=True)
+#             .values_list("name", "level__name")
+#         ]
+
+#         project.projectgoal_set.create(
+#             attribute=attribute,
+#             amount=int(request_data["amount"]),
+#             name=" > ".join(goal_attribute_names),
+#             image_mod=int(request_data.get("image_mod", 1)),
+#             video_mod=int(request_data.get("video_mod", 1)),
+#         )
+
+#         response, status = {"ok": True}, HTTP_201_CREATED
+
+#     except AssertionError as e:
+#         if int(request_data.get("update", 0)) == 1:
+#             goal = attribute.projectgoal_set.first()
+#             goal.amount = int(request_data["amount"])
+#             goal.image_mod = int(request_data.get("image_mod", 1))
+#             goal.video_mod = int(request_data.get("video_mod", 1))
+#             goal.save()
+
+#             response, status = {"ok": True}, HTTP_202_ACCEPTED
+
+#         else: response, status = {"errors": str(e)}, HTTP_400_BAD_REQUEST
+
+#     except Exception as e: response, status = {"errors": str(e)}, HTTP_400_BAD_REQUEST
+
+#     return response, status
+
+# @with_model_assertion(ProjectGoal, "id")
+# def _delete_goal(self, goal: ProjectGoal):
+#     try:
+#         goal.delete()
+#         return {"ok": True}, HTTP_202_ACCEPTED
+
+#     except Exception as e: return {"errors": str(e)}, HTTP_400_BAD_REQUEST

backend-app/archive/services.py

@@ -0,0 +1 @@
+

backend-app/archive/tests.py

@@ -0,0 +1,32 @@
+from rest_framework.views import Response, APIView, Request
+from rest_framework.permissions import IsAuthenticated
+from .permissions import ArchivePermission, ArchivesPermission
+from .services import _get_archives, _make_archive
+
+
+class ArchivesViewSet(APIView):
+    http_method_names = ("post", "get")
+    permission_classes = (IsAuthenticated, ArchivesPermission)
+
+    def get(self, request: Request, p_pk: int) -> Response:
+        response, status = _get_archives(p_pk)
+        return Response(response, status=status)
+
+    def post(self, request: Request, p_pk: int) -> Response:
+        response, status = _make_archive(p_pk, request.data)
+        return Response(response, status=status)
+
+
+class ArchiveViewSet(APIView):
+    http_method_names = ("get", "patch")
+    permission_classes = (IsAuthenticated, ArchivePermission)
+
+    def get(self, request: Request, pk: int) -> Response:
+        ...
+        # response, status = _get_project(pk, request)
+        # return Response(response, status=status)
+
+    def patch(self, request: Request, pk: int) -> Response:
+        ...
+        # response, status = _patch_project(pk, request.data)
+        # return Response(response, status=status)

backend-app/archive/views.py

@@ -56,7 +56,8 @@
     'api',
     'project',
     'attribute',
-    'file'
+    'file',
+    "archive"
 ]
 
 MIDDLEWARE = [

backend-app/proj_back/settings.py

@@ -1,8 +1,11 @@
 from django.urls import path
 from .views import ProjectsViewSet, ProjectViewSet, GoalViewSet
+from archive.views import ArchivesViewSet, ArchiveViewSet
 
 urlpatterns: tuple = (
     path("", ProjectsViewSet.as_view()),
     path("goals/<int:pk>/", GoalViewSet.as_view()),
+    path("archive/<int:p_pk>/<str:pk>/", ArchiveViewSet.as_view()),
+    path("archive/<int:p_pk>/", ArchivesViewSet.as_view()),
     path("<int:pk>/", ProjectViewSet.as_view())
 )