diff --git a/oxdata/movie/models.py b/oxdata/movie/models.py index 14d5ef6..8ee6195 100644 --- a/oxdata/movie/models.py +++ b/oxdata/movie/models.py @@ -53,7 +53,7 @@ def find(info, guess=True): q = Imdb.objects.all() if 'id' in info: q = q.filter(imdb=info['id']) - if q.count() == 0 and len(info['id']) == 7: + if q.count() == 0 and len(info['id']) >= 7 and str(info['id']).isdigit(): i = Imdb(imdb=info['id']) i.save() q = Imdb.objects.filter(imdb=info['id']) diff --git a/oxdata/movie/views.py b/oxdata/movie/views.py index 171517a..db017ea 100644 --- a/oxdata/movie/views.py +++ b/oxdata/movie/views.py @@ -29,7 +29,7 @@ def getData(request, data): id = data['id'] if isinstance(id, int): id = str(id) - if len(id) == 7: + if len(id) >= 7 and id.isdigit(): i, created = models.Imdb.objects.get_or_create(imdb=id) if created: i.update()