From b5117a4e1ceccdb563ce9d616a4f784233ca1fcc Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Wed, 20 Jul 2022 18:32:33 -0400
Subject: [PATCH] avoid double submission error

---
 books/views.py  |  5 +++++
 games/views.py  |  5 +++++
 movies/views.py |  5 +++++
 music/views.py  | 10 ++++++++++
 4 files changed, 25 insertions(+)

diff --git a/books/views.py b/books/views.py
index 4bf6b717..eeb91435 100644
--- a/books/views.py
+++ b/books/views.py
@@ -265,6 +265,11 @@ def create_update_mark(request):
         pk = request.POST.get('id')
         old_rating = None
         old_tags = None
+        if not pk:
+            book_id = request.POST.get('book')
+            mark = BookMark.objects.filter(book_id=book_id, owner=request.user).first()
+            if mark:
+                pk = mark.id
         if pk:
             mark = get_object_or_404(BookMark, pk=pk)
             if request.user != mark.owner:
diff --git a/games/views.py b/games/views.py
index 093d3723..ff171ad2 100644
--- a/games/views.py
+++ b/games/views.py
@@ -267,6 +267,11 @@ def create_update_mark(request):
         pk = request.POST.get('id')
         old_rating = None
         old_tags = None
+        if not pk:
+            game_id = request.POST.get('game')
+            mark = GameMark.objects.filter(game_id=game_id, owner=request.user).first()
+            if mark:
+                pk = mark.id
         if pk:
             mark = get_object_or_404(GameMark, pk=pk)
             if request.user != mark.owner:
diff --git a/movies/views.py b/movies/views.py
index 8afea850..50f8bdbc 100644
--- a/movies/views.py
+++ b/movies/views.py
@@ -266,6 +266,11 @@ def create_update_mark(request):
         pk = request.POST.get('id')
         old_rating = None
         old_tags = None
+        if not pk:
+            movie_id = request.POST.get('movie')
+            mark = MovieMark.objects.filter(movie_id=movie_id, owner=request.user).first()
+            if mark:
+                pk = mark.id
         if pk:
             mark = get_object_or_404(MovieMark, pk=pk)
             if request.user != mark.owner:
diff --git a/music/views.py b/music/views.py
index e0ac8495..f4188890 100644
--- a/music/views.py
+++ b/music/views.py
@@ -285,6 +285,11 @@ def create_update_song_mark(request):
         pk = request.POST.get('id')
         old_rating = None
         old_tags = None
+        if not pk:
+            song_id = request.POST.get('song')
+            mark = SongMark.objects.filter(song_id=song_id, owner=request.user).first()
+            if mark:
+                pk = mark.id
         if pk:
             mark = get_object_or_404(SongMark, pk=pk)
             if request.user != mark.owner:
@@ -828,6 +833,11 @@ def create_update_album_mark(request):
         pk = request.POST.get('id')
         old_rating = None
         old_tags = None
+        if not pk:
+            album_id = request.POST.get('album')
+            mark = AlbumMark.objects.filter(album_id=album_id, owner=request.user).first()
+            if mark:
+                pk = mark.id
         if pk:
             mark = get_object_or_404(AlbumMark, pk=pk)
             if request.user != mark.owner: