x/stacktach
Code Issues Proposed changes

Handling case of no data in 0003 migration

Browse Source
This commit is contained in:
Andrew Melton 2013-09-13 14:56:30 -04:00
parent 9ce61c8e55
commit eadf56642f
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
stacktach/migrations/0003_populate_usage_related_fields_in_rawdataimagemeta_instanceexists_and_instanceusages_from_rawdata.py
View File

@ -41,13 +41,14 @@ def queryset_iterator(queryset, chunksize=1000):
Note that the implementation of the iterator does not support ordered query sets.
'''
id = 0
last_pk = queryset.order_by('-id')[0]['id']
queryset = queryset.order_by('id')
while id < last_pk:
for row in queryset.filter(id__gt=id)[:chunksize]:
id = row['id']
yield row
gc.collect()
if queryset.order_by('-id').count() > 0:
last_pk = queryset.order_by('-id')[0]['id']
queryset = queryset.order_by('id')
while id < last_pk:
for row in queryset.filter(id__gt=id)[:chunksize]:
id = row['id']
yield row
gc.collect()
class Migration(DataMigration):