initial commit

movie_manager/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

movie_manager/apps.py

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

movie_manager/migrations/0001_initial.py

@@ -0,0 +1,48 @@
+# Generated by Django 5.1.4 on 2025-03-31 04:04
+
+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='Movie',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('title', models.CharField(max_length=100)),
+                ('imdb_id', models.CharField(max_length=100)),
+                ('year', models.IntegerField()),
+                ('critic_score', models.CharField(max_length=500)),
+                ('genre', models.CharField(max_length=100)),
+                ('director', models.CharField(max_length=500)),
+                ('actors', models.CharField(max_length=500)),
+                ('plot', models.CharField(max_length=500)),
+                ('poster', models.CharField(max_length=500)),
+                ('last_watched', models.DateTimeField()),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('deleted_at', models.DateTimeField(blank=True, null=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='MovieList',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=100)),
+                ('public', models.BooleanField(default=False)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('deleted_at', models.DateTimeField(blank=True, null=True)),
+                ('owner', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

movie_manager/migrations/0002_alter_movie_options_alter_movielist_options_and_more.py

@@ -0,0 +1,26 @@
+# Generated by Django 5.1.4 on 2025-04-07 05:02
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('movie_manager', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='movie',
+            options={'ordering': ['title']},
+        ),
+        migrations.AlterModelOptions(
+            name='movielist',
+            options={'ordering': ['name']},
+        ),
+        migrations.AddField(
+            model_name='movielist',
+            name='movies',
+            field=models.ManyToManyField(to='movie_manager.movie'),
+        ),
+    ]

movie_manager/models.py

@@ -0,0 +1,40 @@
+from django.db import models
+from django.contrib.auth.models import User
+
+# Create your models here.
+class Movie(models.Model):
+    class Meta:
+        ordering = ["title"]
+
+    title = models.CharField(max_length=100)
+    imdb_id = models.CharField(max_length=100)
+    year = models.IntegerField()
+    critic_score = models.CharField(max_length=500)
+    genre = models.CharField(max_length=100)
+    director = models.CharField(max_length=500)
+    actors = models.CharField(max_length=500)
+    plot = models.CharField(max_length=500)
+    poster = models.CharField(max_length=500)
+    last_watched = models.DateTimeField()
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    deleted_at = models.DateTimeField(null=True, blank=True)
+
+    def __str__(self):
+        return self.title
+
+
+class MovieList(models.Model):
+    class Meta:
+        ordering = ["name"]
+
+    name = models.CharField(max_length=100)
+    public = models.BooleanField(default=False)
+    owner = models.ForeignKey(User, on_delete=models.CASCADE)
+    movies = models.ManyToManyField(Movie)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    deleted_at = models.DateTimeField(null=True, blank=True)
+
+    def __str__(self):
+        return self.name

movie_manager/serializers.py

@@ -0,0 +1,21 @@
+from itertools import count
+
+from rest_framework import serializers
+from movie_manager.models import Movie, MovieList
+
+class MovieSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Movie
+        fields = '__all__'
+
+
+class MovieListSerializer(serializers.ModelSerializer):
+    movie_count = serializers.SerializerMethodField()
+
+    class Meta:
+        model = MovieList
+        fields = ["id","name","owner","public", "movie_count"]
+
+
+    def get_movie_count(self, obj):
+        return len(obj.movies.all())

movie_manager/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

movie_manager/views.py

@@ -0,0 +1,48 @@
+from django.http import HttpResponse, JsonResponse
+from django.contrib.auth.models import User
+from rest_framework import permissions, viewsets
+from knox.auth import TokenAuthentication
+from rest_framework.exceptions import NotFound
+
+from movie_manager.models import Movie, MovieList
+from movie_manager.serializers import MovieListSerializer, MovieSerializer
+
+
+# Create your views here.
+class MovieViewset(viewsets.ModelViewSet):
+    fields = '__all__'
+    queryset = Movie.objects.all().order_by("title")
+    authentication_classes = [TokenAuthentication]
+    permission_classes = [permissions.IsAuthenticated]
+
+    serializer_class = MovieSerializer
+
+class MovieListViewset(viewsets.ModelViewSet):
+    fields = '__all__'
+    queryset = MovieList.objects.all().order_by("name")
+    authentication_classes = [TokenAuthentication]
+    permission_classes = [permissions.IsAuthenticated]
+
+    serializer_class = MovieListSerializer
+
+    def update(self, request, pk=None, *args, **kwargs):
+        movie_list = MovieList.objects.get(pk=pk)
+        movie_list.name = request.data.get('name')
+        movie_list.owner = User.objects.get(pk=request.data.get("owner"))
+
+        if request.data.get('movies'):
+            movie_ids = request.data.get('movies')
+            for movie_id in movie_ids:
+                try:
+                    movie = Movie.objects.get(pk=movie_id)
+                    movie_list.movies.add(movie)
+                except Movie.DoesNotExist:
+                    raise NotFound(f"Movie {movie_id} does not exist")
+
+            removed_movies = Movie.objects.exclude(id__in=movie_ids)
+            for removed_movie in removed_movies:
+                removed_movie.delete()
+
+        movie_list.save()
+
+        return JsonResponse(MovieListSerializer(movie_list).data)