aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--scripts/insert_data.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/scripts/insert_data.py b/scripts/insert_data.py
index 7bf20f1..f273c3a 100644
--- a/scripts/insert_data.py
+++ b/scripts/insert_data.py
@@ -114,12 +114,11 @@ def __format_query__(query, params):
def __param_str__(param):
return "', '".join(str(elt) for elt in param)
idx = query.find("VALUES")
+ idx = query.find("%")
fields = tuple(
- elt.strip().replace("%(", "").replace(")s", "")
- for elt in query[idx + 7:-1].split(","))
- values = (
- tuple(param[field] for field in fields)
- for param in params)
+ elt.replace("%(", "").replace(")s", "").replace(")", "").strip()
+ for elt in query[idx:-1].split(","))
+ values = (tuple(param[field] for field in fields) for param in params)
values_str = ", ".join(
f"('{__param_str__(value_tup)}')" for value_tup in values)
insert_str = query[:idx].replace(