Merge pull request #1014 from ivanka12/master

change generation of items.classification

@@ -26,6 +26,7 @@ class OP_Provider(BaseProvider):
     addresses = _fake_data.addresses
     classifications = _fake_data.classifications
     cpvs = _fake_data.cpvs
+    moz_cpvs = _fake_data.moz_cpvs
     items_base_data = _fake_data.items_base_data
     rationale_types = _fake_data.rationale_types
 
@@ -121,8 +122,11 @@ class OP_Provider(BaseProvider):
             in the beginning of each cpv id.
         """
         item_base_data = None
+        cpv = None
         if cpv_group is None:
-            item_base_data = self.random_element(self.items_base_data)
+            cpv = self.random_element(self.cpvs)
+        elif cpv_group == 336:
+            cpv = self.random_element(self.moz_cpvs)
         else:
             cpv_group = str(cpv_group)
             similar_cpvs = []
@@ -130,12 +134,12 @@ class OP_Provider(BaseProvider):
                 if cpv_element.startswith(cpv_group):
                     similar_cpvs.append(cpv_element)
             cpv = self.random_element(similar_cpvs)
-            for entity in self.items_base_data:
-                if entity["cpv_id"] == cpv:
-                    item_base_data = entity
-                    break
-            if not item_base_data:
-                raise ValueError('unable to find an item with CPV ' + cpv)
+        for entity in self.items_base_data:
+            if entity["cpv_id"] == cpv:
+                item_base_data = entity
+                break
+        if not item_base_data:
+            raise ValueError('unable to find an item with CPV ' + cpv)
 
         # choose appropriate additional classification for item_base_data's cpv
         additional_class = []

@@ -7875,14 +7875,16 @@
         "30192130-1",
         "30197600-2",
         "30199000-0",
+        "99999999-9"
+    ],
+    "moz_cpvs": [
         "33615100-5",
         "33622200-8",
         "33632100-0",
         "33612000-3",
         "33692500-2",
         "33692100-8",
-        "33600000-6",
-        "99999999-9"
+        "33600000-6"
     ],
     "items_base_data": [
         {