From 061b361f8e6bb384a3b6113445cf270cb813f432 Mon Sep 17 00:00:00 2001 From: Rob Savoye Date: Tue, 19 Dec 2023 20:41:51 -0700 Subject: [PATCH] fix: Escape strings to remove embedded single quotes --- tm_admin/tmdb.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tm_admin/tmdb.py b/tm_admin/tmdb.py index d344c707..528c9a11 100755 --- 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}', "