x/stacktach
Code Issues Proposed changes

Merge pull request #201 from ramielrowe/master

Browse Source 
Handling case of no data in 0003 migration
This commit is contained in:
Andrew Melton
2013-09-13 11:58:05 -07:00
parent 8b986b6db8 eadf56642f
commit 44397a201f
1 changed files with 8 additions and 7 deletions
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. Note that the implementation of the iterator does not support ordered query sets.
''' '''
id = 0 id = 0
last_pk = queryset.order_by('-id')[0]['id'] if queryset.order_by('-id').count() > 0:
queryset = queryset.order_by('id') last_pk = queryset.order_by('-id')[0]['id']
while id < last_pk: queryset = queryset.order_by('id')
for row in queryset.filter(id__gt=id)[:chunksize]: while id < last_pk:
id = row['id'] for row in queryset.filter(id__gt=id)[:chunksize]:
yield row id = row['id']
gc.collect() yield row
gc.collect()
class Migration(DataMigration): class Migration(DataMigration):