From c26bd2b62809b6253c16af162e339d0284582a6a Mon Sep 17 00:00:00 2001 From: David-Wobrock Date: Thu, 6 Sep 2018 21:47:28 +0200 Subject: [PATCH] Provide a default value to next() function in case of empty errors array --- django_migration_linter/sql_analyser.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/django_migration_linter/sql_analyser.py b/django_migration_linter/sql_analyser.py index aa06ec2..0fe3c1b 100644 --- a/django_migration_linter/sql_analyser.py +++ b/django_migration_linter/sql_analyser.py @@ -20,13 +20,15 @@ logger = logging.getLogger(__name__) def has_default(sql, **kwargs): if re.search('SET DEFAULT', sql) and kwargs['errors']: - err = next(err - for err in kwargs['errors'] - if err['code'] == 'NOT_NULL') + err = next( + (err for err in kwargs['errors'] if err['code'] == 'NOT_NULL'), + None, + ) if err: logger.info( ('Found a NOT_NULL error in migration, ' - 'but it has a default value added: {}').format(err)) + 'but it has a default value added: {}').format(err) + ) kwargs['errors'].remove(err) return False # Never fails