Commit b7af0cedf46a54d322be6758106464d984446e0c
1 parent
2373656b
Add additional classifications for all items
Showing
1 changed file
with
6 additions
and
1 deletions
... | ... | @@ -120,6 +120,7 @@ class OP_Provider(BaseProvider): |
120 | 120 | from a specific cpv group. Cpv group is three digits |
121 | 121 | in the beginning of each cpv id. |
122 | 122 | """ |
123 | + item_base_data = None | |
123 | 124 | if cpv_group is None: |
124 | 125 | item_base_data = self.random_element(self.items_base_data) |
125 | 126 | else: |
... | ... | @@ -133,12 +134,16 @@ class OP_Provider(BaseProvider): |
133 | 134 | if entity["cpv_id"] == cpv: |
134 | 135 | item_base_data = entity |
135 | 136 | break |
137 | + if not item_base_data: | |
138 | + raise ValueError('unable to find an item with CPV ' + cpv) | |
136 | 139 | |
137 | 140 | # choose appropriate additional classification for item_base_data's cpv |
138 | 141 | additional_class = [] |
139 | 142 | for entity in self.classifications: |
140 | 143 | if entity["classification"]["id"] == item_base_data["cpv_id"]: |
141 | 144 | additional_class.append(entity) |
145 | + if not additional_class: | |
146 | + raise ValueError('unable to find a matching additional classification for CPV ' + cpv) | |
142 | 147 | classification = self.random_element(additional_class) |
143 | 148 | |
144 | 149 | dk_descriptions = { |
... | ... | @@ -148,6 +153,7 @@ class OP_Provider(BaseProvider): |
148 | 153 | } |
149 | 154 | address = self.random_element(self.addresses) |
150 | 155 | item = { |
156 | + "additionalClassifications": classification["additionalClassifications"], | |
151 | 157 | "classification": classification["classification"], |
152 | 158 | "deliveryAddress": address["deliveryAddress"], |
153 | 159 | "deliveryLocation": address["deliveryLocation"], |
... | ... | @@ -157,7 +163,6 @@ class OP_Provider(BaseProvider): |
157 | 163 | if item_base_data["cpv_id"] == "99999999-9": |
158 | 164 | scheme = classification["additionalClassifications"][0]["scheme"] |
159 | 165 | item.update({ |
160 | - "additionalClassifications": classification["additionalClassifications"], | |
161 | 166 | "description": dk_descriptions[scheme][0], |
162 | 167 | "description_ru": dk_descriptions[scheme][1], |
163 | 168 | "description_en": dk_descriptions[scheme][2] | ... | ... |
Please
register
or
login
to post a comment