Provide a default value to next() function in case of empty errors array

This commit is contained in:
David-Wobrock committed 2018-09-06 21:47:28 +02:00
1 parent 6174c76c36
commit c26bd2b628
1 file changed
+6 -4
+6 -4
View File
@@ -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