Add op_faker.py

+# -*- coding: utf-8 -*-
+from faker import Faker
+from faker.providers import BaseProvider
+from munch import Munch
+from json import load
+import os
+
+
+def load_data_from_file(file_name):
+    if not os.path.exists(file_name):
+        file_name = os.path.join(os.path.dirname(__file__), 'data', file_name)
+    with open(file_name) as file_obj:
+        if file_name.endswith(".json"):
+            return Munch.fromDict(load(file_obj))
+        elif file_name.endswith(".yaml"):
+            return fromYAML(file_obj)
+
+
+class OP_Provider(BaseProvider):
+    __fake_data = load_data_from_file("faker_data.json")
+    word_list = __fake_data.words
+    procuringEntities = __fake_data.procuringEntities
+    addresses = __fake_data.addresses
+    classifications = __fake_data.classifications
+    units = __fake_data.units
+    cpvs = __fake_data.cpvs
+    items_base_data = __fake_data.items_base_data
+
+
+    @classmethod
+    def randomize_nb_elements(self, number=10, le=60, ge=140):
+        """
+        Returns a random value near number.
+
+        :param number: value to which the result must be near
+        :param le: lower limit of randomizing (percents). Default - 60
+        :param ge: upper limit of randomizing (percents). Default - 140
+        :returns: a random int in range [le * number / 100, ge * number / 100]
+            with minimum of 1
+        """
+        if le > ge:
+            raise Exception("Lower bound: {} is greater then upper: {}.".format(le, ge))
+        return int(number * self.random_int(min=le, max=ge) / 100) + 1
+
+
+    @classmethod
+    def word(self):
+        """
+        :example 'Курка'
+        """
+        return self.random_element(self.word_list)
+
+
+    @classmethod
+    def words(self, nb=3):
+        """
+        Generate an array of random words
+        :example: array('Надіньте', 'фуражка', 'зелено')
+        :param nb: how many words to return
+        """
+        return [self.word() for _ in range(0, nb)]
+
+
+    @classmethod
+    def sentence(self, nb_words=5, variable_nb_words=True):
+        """
+        Generate a random sentence
+        :example: 'Курка надіньте пречудовий зелено на.'
+        :param nb_words: how many words the sentence should contain
+        :param variable_nb_words: set to false if you want exactly $nbWords returned,
+            otherwise $nbWords may vary by +/-40% with a minimum of 1
+        """
+        if nb_words <= 0:
+            return ''
+
+        if variable_nb_words:
+            nb_words = self.randomize_nb_elements(number=nb_words)
+
+        words = self.words(nb_words)
+        words[0] = words[0].title()
+
+        return " ".join(words) + '.'
+
+
+    @classmethod
+    def title(self):
+        return self.sentence(nb_words = 3)
+
+
+    @classmethod
+    def description(self):
+        return self.sentence(nb_words = 10)
+
+
+    @classmethod
+    def procuringEntity(self):
+        return self.random_element(self.procuringEntities)
+
+
+    @classmethod
+    def cpv(self):
+        return self.random_element(self.cpvs)
+
+
+    @classmethod
+    def fake_item(self, cpv_group=None):
+        """
+        Generate a random item for openprocurement tenders
+
+        :param cpv_group: gives possibility to generate items
+            from one cpv group. Cpv group is three digits in the beginning
+            of each cpv id.
+        """
+        if cpv_group is None:
+            item_base_data = self.random_element(self.items_base_data)
+        else:
+            cpv_group = str(cpv_group)
+            similar_cpvs = []
+            for cpv_element in self.cpvs:
+                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
+
+        # choose appropriate dkpp classification for item_base_data's cpv
+        for entity in self.classifications:
+            if entity["classification"]["id"] == item_base_data["cpv_id"]:
+                classification = entity
+                break
+
+        address = self.random_element(self.addresses)
+        unit = self.random_element(self.units)
+        item = {
+            "description": item_base_data["description"],
+            "classification": classification["classification"],
+            "additionalClassifications": classification["additionalClassifications"],
+            "deliveryAddress": address["deliveryAddress"],
+            "deliveryLocation": address["deliveryLocation"],
+            "unit": item_base_data["unit"],
+            "quantity": self.randomize_nb_elements(number=item_base_data["quantity"], le=80, ge=120)
+        }
+        return item