From bf88b8cfdc9b96707979e630418af2f99881bebe Mon Sep 17 00:00:00 2001 From: Josh Washburne Date: Mon, 24 Sep 2018 13:28:36 -0400 Subject: [PATCH] Add check to song request to make sure song exists. --- savepointradio/profiles/models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/savepointradio/profiles/models.py b/savepointradio/profiles/models.py index db417dc..b80d423 100644 --- a/savepointradio/profiles/models.py +++ b/savepointradio/profiles/models.py @@ -48,7 +48,10 @@ class RadioProfile(Disableable, Timestampable, models.Model): def make_request(self, song_requested): if isinstance(song_requested, int): - song = Song.objects.get(pk=song_requested) + try: + song = Song.objects.get(pk=song_requested) + except Song.DoesNotExist: + raise MakeRequestError('Song does not exist.') else: song = song_requested