update script to populate data for Colombian companies

demo/company.py

@@ -11,9 +11,8 @@ path = "".join([dir, "/", file_])
 
 
 def setup(config, modules, company_config):
-    data, = open_file(path)
 
-    Address = Model.get('party.address')
+    Identifier = Model.get('party.identifier')
     Company = Model.get('company.company')
     Country = Model.get('country.country')
     Currency = Model.get('currency.currency')
@@ -27,32 +26,49 @@ def setup(config, modules, company_config):
     rate = company_currency.rates.new()
     rate.date = dt.date(dt.date.today().year, 1, 1)
     rate.rate = Decimal('1')
-    
+
     company_currency.save()
 
     try:
-        company_country, = Country.find([('code', '=', company_config["company_country"])])
+        company_country, = Country.find(
+            [('code', '=', company_config["company_country"])])
     except ValueError:
         company_country = None
 
-    
+    try:
+        company_subdivision, = Subdivision.find(
+            [('code', '=', company_config["company_subdivision"])]
+        )
+    except ValueError:
+        company_subdivision = None
+
+    try:
+        company_municipality, = Subdivision.find(
+            [('code', '=', company_config["company_municipality"])]
+        )
+    except ValueError:
+        company_municipality = None
+
     CompanyWizard = Wizard('company.company.config')
     CompanyWizard.execute('company')
 
     company = CompanyWizard.form
-    party = Party(name=data.get("name"))
-    
-    address = Address()
-    address.street = data.get("street")
-    address.city = data.get("city")
-    address.country = company_country
+    party = Party(
+        name=company_config["company_name"],
+        type_person=company_config["type_person"]
+    )
 
-    try:
-        address.subdivision, = Subdivision.find([
-            ('code', '=', company_config["company_subdivision"])
-        ])
-    except ValueError:
-        pass
+    identifier = Identifier()
+    identifier.type = company_config["company_identifier_type"]
+    identifier.code = company_config["company_identifier"]
+
+    party.identifiers.append(identifier)
+
+    address = party.addresses[0]
+    address.street = company_config["company_street"]
+    address.country = company_country
+    address.subdivision = company_subdivision
+    address.subdivision_municipality = company_municipality
 
     party.addresses.append(address)
     party.save()
@@ -71,11 +87,12 @@ def setup(config, modules, company_config):
     return company
 
 
-def get():
+def get(company_config):
 
-    data, = open_file(path)
     Company = Model.get('company.company')
 
-    if company := Company.find([('party.name', '=', data.get("name"))]):
+    if company := Company.find([
+        ('party.name', '=', company_config["company_name"])
+    ]):
         return company
-    return 
+    return