Add funders functions

@@ -118,6 +118,31 @@ def get_document_by_id(data, doc_id):
     raise Exception('Document with id {} not found'.format(doc_id))
     raise Exception('Document with id {} not found'.format(doc_id))
 
 
 
 
+def get_tenders_by_funder_id(client,
+                             funder_id=None,
+                             descending=True,
+                             tender_id_field='tenderID',
+                             opt_fields=('funders',)):
+    params = {'offset': '',
+              'opt_fields': ','.join((tender_id_field,) + opt_fields),
+              'descending': descending}
+    tender_list = True
+    client._update_params(params)
+    tenders_with_funder = {}
+    while tender_list and not tenders_with_funder:
+        tender_list = client.get_tenders()
+        for tender in tender_list:
+            if 'funders' in tender:
+                tenders_with_funder[tender[tender_id_field]] = [el['identifier']['id'] for el in tender['funders']]
+    # In case we are looking for a specific funder
+    if funder_id:
+        tenders_with_funder = {k: v for k, v in tenders_with_funder.items() if funder_id in v}
+    if not tenders_with_funder:
+        raise IdNotFound
+    else:
+        return tenders_with_funder
+
+
 def download_file_from_url(url, path_to_save_file):
 def download_file_from_url(url, path_to_save_file):
     f = open(path_to_save_file, 'wb')
     f = open(path_to_save_file, 'wb')
     f.write(urllib.urlopen(url).read())
     f.write(urllib.urlopen(url).read())

@@ -37,10 +37,18 @@ def create_fake_funder():
     return fake.funders_data()
     return fake.funders_data()
 
 
 
 
+def get_fake_funder_scheme():
+    return fake.funder_scheme()
+
+
 def create_fake_amount(award_amount):
 def create_fake_amount(award_amount):
     return round(random.uniform(1, award_amount), 2)
     return round(random.uniform(1, award_amount), 2)
 
 
 
 
+def create_fake_number(min_number, max_number):
+    return random.randint(int(min_number), int(max_number))
+
+
 def create_fake_title():
 def create_fake_title():
     return u"[ТЕСТУВАННЯ] {}".format(fake.title())
     return u"[ТЕСТУВАННЯ] {}".format(fake.title())
 
 

@@ -22,6 +22,7 @@ class OP_Provider(BaseProvider):
     word_list = _fake_data.words
     word_list = _fake_data.words
     procuringEntities = _fake_data.procuringEntities
     procuringEntities = _fake_data.procuringEntities
     funders = _fake_data.funders
     funders = _fake_data.funders
+    funders_scheme_list = _fake_data.funders_scheme
     addresses = _fake_data.addresses
     addresses = _fake_data.addresses
     classifications = _fake_data.classifications
     classifications = _fake_data.classifications
     cpvs = _fake_data.cpvs
     cpvs = _fake_data.cpvs
@@ -95,6 +96,9 @@ class OP_Provider(BaseProvider):
     def funders_data(self):
     def funders_data(self):
         return self.random_element(self.funders)
         return self.random_element(self.funders)
 
 
+    @classmethod
+    def funder_scheme(self):
+        return self.random_element(self.funders_scheme_list)
 
 
     @classmethod
     @classmethod
     def cpv(self):
     def cpv(self):

@@ -532,6 +532,130 @@
             }
             }
         }
         }
     ],
     ],
+    "funders_scheme": [
+        "AE-ACCI",
+        "AE-ADCD",
+        "AE-AFZ",
+        "AE-DCCI",
+        "AE-DFSA",
+        "AE-DIFC",
+        "AE-FFZ",
+        "AT-FN",
+        "AE-FUJCCI",
+        "AE-HFZA",
+        "AE-RAKIA",
+        "AE-SAIF",
+        "AE-SCCI",
+        "AE-UAQCCI",
+        "AF-CBR",
+        "AF-MOE",
+        "AU-ABN",
+        "AU-ACNC",
+        "BD-NAB",
+        "BE-BCE_KBO",
+        "BE-GTCF",
+        "BG-EIK",
+        "BY-UNP",
+        "CA-CRA_ACR",
+        "CH-ZEFIX",
+        "CN-SAIC",
+        "CO-CCB",
+        "CY-DRCOR",
+        "CZ-ICO",
+        "DE-CRP",
+        "DK-CVR",
+        "EE-RIK",
+        "ES-DIR3",
+        "ES-RMS",
+        "ET-MFA",
+        "FI-PRO",
+        "FR-INSEE",
+        "FR-RCS",
+        "GB-CHC",
+        "GB-COH",
+        "GB-EDU",
+        "GB-GOV",
+        "GB-GOVUK",
+        "GB-NIC",
+        "GB-REV",
+        "GB-SC",
+        "GB-UKPRN",
+        "GG-RCE",
+        "GH-DSW",
+        "HU-VAT",
+        "ID-KDN",
+        "ID-KHH",
+        "ID-KLN",
+        "ID-PRO",
+        "IE-CHY",
+        "IE-CRO",
+        "IL-ROC",
+        "IM-CR",
+        "IM-GR",
+        "IN-MCA",
+        "IT-RI",
+        "JE-CR",
+        "JE-OAC",
+        "JP-JCN",
+        "KE-NCB",
+        "KE-RCO",
+        "KE-RSO",
+        "KZ-BIN",
+        "LS-LCN",
+        "LT-PVM",
+        "LT-RC",
+        "LV-RE",
+        "MD-IDNO",
+        "MM-MHA",
+        "MW-CNM",
+        "MW-MRA",
+        "MW-NBM",
+        "MW-RG",
+        "MY-SSM",
+        "MZ-MOJ",
+        "NG-CAC",
+        "NL-KVK",
+        "NO-BRC",
+        "NP-CRO",
+        "NP-SWC",
+        "PK-PCP",
+        "PK-VSWA",
+        "PL-REGON",
+        "PL-NIP",
+        "PL-KRS",
+        "PT-NIPPC",
+        "RO-CUI",
+        "RS-APR",
+        "RU-INN",
+        "RU-OGRN",
+        "SE-BLV",
+        "SG-ACRA",
+        "SI-PRS",
+        "SI-TIN",
+        "SK-ZRSR",
+        "TR-MERSIS",
+        "TR-VAT",
+        "TZ-BRLA",
+        "UA-EDR",
+        "UG-NGB",
+        "UG-RSB",
+        "US-DOS",
+        "US-EIN",
+        "US-USAGOV",
+        "UZ-KTUT",
+        "XI-IATI",
+        "XI-PB",
+        "XM-DAC",
+        "XM-EORI",
+        "XM-OCHA",
+        "ZA-CIP",
+        "ZA-NPO",
+        "ZA-PBO",
+        "ZM-NRB",
+        "ZM-PCR",
+        "ZW-PVO",
+        "ZW-ROD"
+    ],
     "procuringEntities": [
     "procuringEntities": [
         {
         {
             "contactPoint": {
             "contactPoint": {

@@ -20,8 +20,10 @@ from .initial_data import (
     create_fake_doc,
     create_fake_doc,
     create_fake_sentence,
     create_fake_sentence,
     create_fake_amount,
     create_fake_amount,
+    create_fake_number,
     create_fake_date,
     create_fake_date,
     create_fake_funder,
     create_fake_funder,
+    get_fake_funder_scheme,
     fake,
     fake,
     subtraction,
     subtraction,
     field_with_id,
     field_with_id,