Add spanish names

colombia/dataset_tools.py

@@ -12,6 +12,12 @@ def classification_to_models(classification, model):
         m.id = index.item()
         m.code = row["code"]
         m.name_en = row["name"]
+        if "name_es" in row:
+            m.name_es = row["name_es"]
+        if "name_short_en" in row:
+            m.name_short_en = row["name_short_en"]
+        if "name_short_es" in row:
+            m.name_short_es = row["name_short_es"]
         m.level = row["level"]
         m.parent_id = row["parent_id"]
         models.append(m)

colombia/import.py

@@ -5,6 +5,8 @@
 from datasets import (trade4digit_department, industry4digit_department,
                       industry4digit_municipality, population, gdp)
 
+import pandas as pd
+
 if __name__ == "__main__":
 
         app = create_app()
@@ -15,6 +17,12 @@
                                   industry_classification,
                                   location_classification)
 
+            df = pd.read_excel("/Users/makmana/classifications/product/HS/Mexico/in/Mexico Short Names Final.xlsx")
+            df = df[["code", "hs4_name_spanish", "Spanish Short Name", "hs4_name_en (original short name)"]]
+            df.columns = ["code", "name_es", "name_short_es", "name_short_en"]
+            df.code = df.code.astype(str).str.zfill(4)
+            product_classification.table = product_classification.table.merge(df, on="code", how="left")
+
             products = classification_to_models(product_classification,
                                                 models.HSProduct)
             db.session.add_all(products)