Merge pull request #23 from Kalgoc/feat/piggies

Piggies

Author: DaHunterTime

Diff for: piggies/__init__.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class UserExpenseTypeConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "piggies"

Diff for: piggies/admin.py

@@ -0,0 +1,41 @@
+# Generated by Django 5.0.6 on 2024-06-20 20:00
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Piggies",
+            fields=[
+                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                (
+                    "piggy",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="friend",
+                        to=settings.AUTH_USER_MODEL,
+                        to_field="username",
+                    ),
+                ),
+                (
+                    "username",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="origin",
+                        to=settings.AUTH_USER_MODEL,
+                        to_field="username",
+                    ),
+                ),
+            ],
+        ),
+    ]

Diff for: piggies/apps.py

@@ -0,0 +1,26 @@
+# Generated by Django 5.0.6 on 2024-06-20 20:37
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("piggies", "0001_initial"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="piggies",
+            name="piggy",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="friend",
+                to=settings.AUTH_USER_MODEL,
+                to_field="user_id",
+            ),
+        ),
+    ]

Diff for: piggies/migrations/0001_initial.py

@@ -0,0 +1,26 @@
+# Generated by Django 5.0.6 on 2024-06-20 20:58
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("piggies", "0002_alter_piggies_piggy"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="piggies",
+            name="username",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="origin",
+                to=settings.AUTH_USER_MODEL,
+                to_field="user_id",
+            ),
+        ),
+    ]

Diff for: piggies/migrations/0002_alter_piggies_piggy.py

@@ -0,0 +1,11 @@
+from django.db import models
+from django.conf import settings
+
+
+class Piggies(models.Model):
+    username = models.ForeignKey(
+        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, to_field="user_id", related_name="origin"
+    )
+    piggy = models.ForeignKey(
+        settings.AUTH_USER_MODEL, on_delete=models.CASCADE, to_field="user_id", related_name="friend"
+    )

Diff for: piggies/migrations/0003_alter_piggies_username.py

@@ -0,0 +1,13 @@
+from rest_framework import serializers
+from .models import Piggies
+
+
+class PiggiesSerializer(serializers.ModelSerializer):
+    id = serializers.ReadOnlyField()
+
+    class Meta:
+        model = Piggies
+        fields = "__all__"
+
+    def create(self, validated_data):
+        return Piggies.objects.create(**validated_data)

Diff for: piggies/migrations/__init__.py

@@ -0,0 +1,15 @@
+from django.urls import path
+from .views import PiggiesViewSet, NotPiggiesViewSet
+
+urlpatterns = [
+    path(
+        "",
+        PiggiesViewSet.as_view({"get": "list", "post": "create"}),
+        name="piggies",
+    ),
+    path(
+        "users/",
+        NotPiggiesViewSet.as_view({"get": "users"}),
+        name="not_piggies",
+    ),
+]

Diff for: piggies/models.py

@@ -0,0 +1,108 @@
+from rest_framework import viewsets, status
+from rest_framework.response import Response
+import jwt
+from django.contrib.auth import get_user_model
+
+from authentication.decorators import cognito_authenticated
+
+from .models import Piggies
+from .serializers import PiggiesSerializer
+
+
+class PiggiesViewSet(viewsets.ViewSet):
+    def get_user_id_from_token(self, request):
+        try:
+            authorization_header = request.headers.get("Authorization")
+            if not authorization_header:
+                raise Exception("Authorization header not found")
+
+            token = authorization_header.split()[1]
+            decoded_token = jwt.decode(token, options={"verify_signature": False})
+            username = decoded_token.get("username")
+            if not username:
+                raise Exception("User ID not found in token")
+            return username
+        except jwt.DecodeError:
+            raise Exception("Invalid token")
+        except jwt.ExpiredSignatureError:
+            raise Exception("Expired token")
+        except Exception as e:
+            raise Exception(f"Error decoding token: {e}")
+
+    @cognito_authenticated
+    def list(self, request):
+        try:
+            username = self.get_user_id_from_token(request)
+            piggies = Piggies.objects.filter(username=username)
+            serializer = PiggiesSerializer(piggies, many=True)
+            return Response(data=serializer.data)
+        except Piggies.DoesNotExist:
+            return Response({"error": "Piggies not found"}, status=status.HTTP_404_NOT_FOUND)
+        except Exception as e:
+            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+    @cognito_authenticated
+    def create(self, request):
+        try:
+            username = self.get_user_id_from_token(request)
+            data = request.data.copy()
+            data["username"] = username
+
+            serializer = PiggiesSerializer(data=data)
+            if serializer.is_valid():
+                serializer.save()
+                response = {
+                    "username": serializer.data["username"],
+                    "piggy": serializer.data["piggy"],
+                }
+                return Response(data=response, status=status.HTTP_201_CREATED)
+            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+        except Exception as e:
+            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
+class NotPiggiesViewSet(viewsets.ViewSet):
+    def get_user_id_from_token(self, request):
+        try:
+            authorization_header = request.headers.get("Authorization")
+            if not authorization_header:
+                raise Exception("Authorization header not found")
+
+            token = authorization_header.split()[1]
+            decoded_token = jwt.decode(token, options={"verify_signature": False})
+            username = decoded_token.get("username")
+            if not username:
+                raise Exception("User ID not found in token")
+            return username
+        except jwt.DecodeError:
+            raise Exception("Invalid token")
+        except jwt.ExpiredSignatureError:
+            raise Exception("Expired token")
+        except Exception as e:
+            raise Exception(f"Error decoding token: {e}")
+
+    def users(self, request):
+        try:
+            username = self.get_user_id_from_token(request)
+            User = get_user_model()
+            users = [{"user_id": str(x.user_id), "first_name": x.first_name} for x in User.objects.all()]
+
+            filtered_users = []
+
+            for user in users:
+                if user["user_id"] != username:
+                    filtered_users.append(user)
+
+            piggies = Piggies.objects.filter(username=username)
+            serializer = PiggiesSerializer(piggies, many=True)
+
+            final_users = []
+
+            for pig in serializer.data:
+                for user in filtered_users:
+                    if str(pig["piggy"]) != user["user_id"]:
+                        final_users.append(user)
+
+            return Response(data=final_users)
+        except Exception as e:
+            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)

Diff for: piggies/serializers.py

@@ -49,6 +49,7 @@
     "debt",
     "user_expense_type",
     "categories",
+    "piggies",
 ]
 
 REST_FRAMEWORK = {

Diff for: piggies/tests.py

@@ -26,4 +26,5 @@
     path("debt/", include("debt.urls")),
     path("user_expense_type/", include("user_expense_type.urls")),
     path("categories/", include("categories.urls")),
+    path("piggies/", include("piggies.urls")),
 ]