UPDATE fct update_to_sql

This commit is contained in:
Colas Geier 2021-12-01 14:43:18 +01:00
parent 0416810bf2
commit 359ac975e3

View File

@ -640,7 +640,8 @@ class ref_hydro:
##################################### #####################################
### Update ### ### 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 = [] a = []
b = [] b = []
table = table_name table = table_name
@ -652,6 +653,16 @@ def update_to_sql(df, con, table_name, schema_name, key_name):
b.append("t.{col}=f.{col}".format(col=col)) b.append("t.{col}=f.{col}".format(col=col))
else: else:
a.append("{col}=t.{col}".format(col=col)) a.append("{col}=t.{col}".format(col=col))
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( df.to_sql(
name = 'temp_table', name = 'temp_table',
con = con, con = con,