fix: Escape strings to remove embedded single quotes

hotosm · Dec 20, 2023 · 061b361 · 061b361
1 parent 30d521f
commit 061b361
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tm_admin/tmdb.py b/tm_admin/tmdb.py
@@ -227,8 +227,11 @@ def writeAllData(self,
                                 continue
                             values += "ARRAY["
                             for item in val:
-                                esc = item.replace("'", "")
-                                values += f"'{esc}', "
+                                if type(item) == str:
+                                    esc = item.replace("'", "")
+                                    values += f"'{esc}', "
+                                elif type(item) == int:
+                                    values += f"{item}, "
                             values = values[:-2]
                             values += "], "
                             continue
@@ -241,7 +244,10 @@ def writeAllData(self,
                             values += f"{val}, "
                     else:
                         if val is None:
-                            values += f"NULL, "
+                            if self.config[key]['required']:
+                                values += f"'', "
+                            else:
+                                values += f"NULL, "
                         else:
                             esc = val.replace("'", "")
                             values += f"'{esc}', "