cgiacofei/TandoorRecipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

name uniquness checking for food import

Browse Source
This commit is contained in:
vabene1111 2024-02-18 18:01:59 +01:00
parent 596ec9134e
commit 7bc567ab95
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cookbook/helper/open_data_importer.py
View File

@ -185,8 +185,11 @@ class OpenDataImporter:
unit_g = Unit.objects.filter(space=self.request.space, base_unit__iexact='g').first() unit_g = Unit.objects.filter(space=self.request.space, base_unit__iexact='g').first()
existing_data = {} existing_data = {}
for obj in Food.objects.filter(space=self.request.space, open_data_slug__isnull=False).values('pk', 'name', 'open_data_slug'): existing_data_names = {}
existing_data[obj['open_data_slug']] = obj for obj in Food.objects.filter(space=self.request.space).values('pk', 'name', 'open_data_slug'):
if obj.open_data_slug:
existing_data[obj['open_data_slug']] = obj
existing_data_names[obj['name']] = obj
update_list = [] update_list = []
create_list = [] create_list = []
@ -203,9 +206,13 @@ class OpenDataImporter:
'space': self.request.space.id, 'space': self.request.space.id,
} }
if obj['open_data_slug'] in existing_data: if obj['open_data_slug'] in existing_data or obj['name'] in existing_data_names:
if obj['open_data_slug'] in existing_data:
obj['pk'] = existing_data[obj['open_data_slug']]['pk']
elif obj['name'] in existing_data_names:
obj['pk'] = existing_data_names[obj['name']]['pk']
obj['space'] = self.request.space obj['space'] = self.request.space
obj['pk'] = existing_data[obj['open_data_slug']]['pk']
obj = Food(**obj) obj = Food(**obj)
update_list.append(obj) update_list.append(obj)
else: else: