CEN38/pycen
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updt fonction update_to_sql - Mise à jour des données avec un champ enum

Browse Source
This commit is contained in:
Colas Geier 2022-06-17 13:20:15 +02:00
parent 9d7ffb83a5
commit 52293d614b
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pycen/update.py
View File

@ -21,6 +21,13 @@ def __get_pkey__(engine,table_name,schema):
pk = engine.dialect.get_pk_constraint(engine,table_name=table_name,schema=schema)
return pk
def __get_dtype__(engine,table_name,schema):
cols = engine.dialect.get_columns(engine,table_name=table_name,schema=schema)
# name_cols = [i['name'] for i in cols]
# type_cols = [i['type'] for i in cols]
type_cols = {i['name']:i['type'] for i in cols}
return type_cols
def _get_dic(self):
if self._table:
select_cols = DIC_REF_HAB[self._table]
@ -238,9 +245,18 @@ class update:
primary_key = key_name
pkey = __get_pkey__(
con,table_name=table,schema=schema)
type_cols = __get_dtype__(
con,table_name=table,schema=schema)
if pkey not in df.columns:
exit('Le champs clé primaire "%s" ne figure pas dans le DataFrame'%pkey)
if not all(item in pkey['constrained_columns'] for item in df.columns):
print('Le(s) champs clé primaire "%s" ne figure pas dans le DataFrame'%pkey['constrained_columns'])
Q = input('Voulez-vous continuer la mise à jour ? (y/n) ')
if Q.lower() == 'y':
pass
else :
return print('Données non mise à jour')
# if pkey not in df.columns:
# exit('Le champs clé primaire "%s" ne figure pas dans le DataFrame'%pkey)
if isinstance(primary_key, str):
primary_key = [primary_key]
@ -249,7 +265,12 @@ class update:
if col in primary_key:
b.append("t.{col}=f.{col}".format(col=col))
else:
a.append("{col}=t.{col}".format(col=col))
dtype = type_cols[col]
if hasattr(dtype,'enums') :
dty = '.'.join([dtype.schema,dtype.name])
a.append("{col}=t.{col}::{typ}".format(col=col,typ=dty))
else:
a.append("{col}=t.{col}".format(col=col))
if isinstance(df, gpd.GeoDataFrame):
df.to_postgis(