From 359ac975e3823fa917d21ca2ac567ff89a6b51cb Mon Sep 17 00:00:00 2001
From: Colas Geier <colas.geier@cen-isere.org>
Date: Wed, 1 Dec 2021 14:43:18 +0100
Subject: [PATCH] UPDATE fct update_to_sql

---
 pycen/zh.py | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/pycen/zh.py b/pycen/zh.py
index afdf17c..8741dc4 100644
--- a/pycen/zh.py
+++ b/pycen/zh.py
@@ -640,7 +640,8 @@ class ref_hydro:
 #####################################
 ###            Update             ###
 #####################################
-def update_to_sql(df, con, table_name, schema_name, key_name):
+def update_to_sql(df, con, table_name, schema_name, key_name,geom_col='geom'):
+  from geopandas import GeoDataFrame
   a = []
   b = []
   table = table_name
@@ -652,14 +653,24 @@ def update_to_sql(df, con, table_name, schema_name, key_name):
         b.append("t.{col}=f.{col}".format(col=col))
       else:
         a.append("{col}=t.{col}".format(col=col))
-  df.to_sql(
-    name = 'temp_table', 
-    con  = con,
-    schema = schema, 
-    if_exists = 'replace', 
-    index  = False, 
-    method = 'multi'
-  )
+  if isinstance(df,GeoDataFrame):
+    df.to_postgis(
+      name = 'temp_table', 
+      con  = con,
+      schema = schema, 
+      if_exists = 'replace', 
+      index  = False, 
+      geom_col=geom_col,
+    )
+  else:
+    df.to_sql(
+      name = 'temp_table', 
+      con  = con,
+      schema = schema, 
+      if_exists = 'replace', 
+      index  = False, 
+      method = 'multi'
+    )
   update_stmt_1 = "UPDATE {sch}.{final_table} f".format(sch=schema,final_table=table)
   update_stmt_2 = " FROM {sch}.temp_table t".format(sch=schema)
   update_stmt_6 = " WHERE %s"%' AND '.join(b)