init depot renew
This commit is contained in:
@@ -0,0 +1,173 @@
|
||||
from sqlalchemy import create_engine, text,PrimaryKeyConstraint
|
||||
from geoalchemy2 import Geometry
|
||||
import geopandas as gpd
|
||||
from os import listdir,path
|
||||
from pycen import con_rho as con
|
||||
|
||||
# user_cad = 'xxxxxxx' # utilisateur de connexion à la bdd
|
||||
# pwd_cad = 'xxxxxxx' # mot de passe de la bdd
|
||||
# adr_cad = '91.134.194.221' # adresse ip de la bdd
|
||||
# port_cad = '5432' # port de la bdd
|
||||
# base_cad = 'rhomeo' # nom de la bdd
|
||||
schema_cad = 'bdtopo3'
|
||||
# con = create_engine('postgresql+psycopg2://{0}:{1}@{2}:{3}/{4}'.format(user_cad,pwd_cad,adr_cad,port_cad,base_cad), echo=False)
|
||||
|
||||
|
||||
def get_list_table(con,sch="public"):
|
||||
with con.begin() as cnx:
|
||||
return cnx.dialect.get_table_names(cnx,schema=sch)
|
||||
|
||||
def get_columns(con,sch,tab):
|
||||
with con.begin() as cnx:
|
||||
return cnx.dialect.get_columns(cnx,tab,sch)
|
||||
|
||||
def get_table_pkey(con,sch,tab):
|
||||
with con.begin() as cnx:
|
||||
return cnx.dialect.get_foreign_keys(cnx,tab,sch)
|
||||
|
||||
|
||||
# for tab in ['spatial_ref_sys',#'info_metadonnees','metadonnees_lot','metadonnees_theme',
|
||||
# 'layer_styles']:
|
||||
# sql = """ALTER TABLE bdtopo3.%s
|
||||
# SET SCHEMA public;"""%tab
|
||||
# with con.begin() as cnx:
|
||||
# cnx.execute(sql)
|
||||
# print('Table %s SET TO SCHEMA bdtopo3'%(tab))
|
||||
|
||||
lst_zip = [
|
||||
'/vsizip//home/colas/Craig/CEN_38/BD Vectorielles/IGN - BDTOPO - 202309/bdtopo_dep07_202309_geopackage.zip/BDT_3-3_GPKG_LAMB93_D007-ED2023-09-15/BDT_3-3_GPKG_LAMB93_D007-ED2023-09-15.gpkg',
|
||||
'/vsizip//home/colas/Craig/CEN_38/BD Vectorielles/IGN - BDTOPO - 202309/bdtopo_dep26_202309_geopackage.zip/BDT_3-3_GPKG_LAMB93_D026-ED2023-09-15/BDT_3-3_GPKG_LAMB93_D026-ED2023-09-15.gpkg',
|
||||
'/vsizip//home/colas/Craig/CEN_38/BD Vectorielles/IGN - BDTOPO - 202309/bdtopo_dep42_202309_geopackage.zip/BDT_3-3_GPKG_LAMB93_D042-ED2023-09-15/BDT_3-3_GPKG_LAMB93_D042-ED2023-09-15.gpkg'
|
||||
]
|
||||
|
||||
lst_tab = [
|
||||
'cimetiere', # BATI
|
||||
'construction_surfacique', # BATI
|
||||
'reservoir',
|
||||
'batiment',
|
||||
'troncon_de_route',
|
||||
'piste_d_aerodrome',
|
||||
'troncon_de_voie_ferree',
|
||||
'terrain_de_sport',
|
||||
]
|
||||
# cimetiere
|
||||
# construction_surfacique
|
||||
# reservoir
|
||||
# batiment
|
||||
# terrain_de_sport
|
||||
|
||||
|
||||
dict_cols = {
|
||||
'id':'cleabs'
|
||||
}
|
||||
|
||||
|
||||
PATH = "/home/colas/Documents/5_BDD/6_IGN/BDTOPO/BDTOPO_3-3_TOUSTHEMES_SHP_LAMB93_D074_2023-09-15/BDTOPO/1_DONNEES_LIVRAISON_2023-09-00196/BDT_3-3_SHP_LAMB93_D074-ED2023-09-15/BATI"
|
||||
PATH2 = '../TRANSPORT'
|
||||
for tab in lst_tab:
|
||||
file = tab.upper()+'.shp'
|
||||
if file in listdir(path.join(PATH)):
|
||||
df = gpd.read_file(path.join(PATH,file))
|
||||
df.columns = df.columns.str.lower()
|
||||
df.set_index('id',inplace=True)
|
||||
# df.rename(columns=dict_cols,inplace=True)
|
||||
print('Load %s'%tab)
|
||||
else: continue
|
||||
|
||||
# columns_type = get_columns(con,schema_cad,tab)
|
||||
# geom_type = [x for x in columns_type if x['name']=='geometrie']
|
||||
# print('Ident geom_type OK')
|
||||
|
||||
# if geom_type:
|
||||
# geom_type = geom_type[0]
|
||||
|
||||
# if df.geometry.name != geom_type['name']:
|
||||
# df.rename_geometry(geom_type['name'],inplace=True)
|
||||
if df.geometry.name != 'geom':
|
||||
df.rename_geometry('geom',inplace=True)
|
||||
|
||||
has_z = True if df.has_z.all() else False
|
||||
unique_geom = df.geom_type.unique()
|
||||
geom_type = df.geom_type[0].upper() if len(unique_geom) == 1 else 'GEOMETRY'
|
||||
geom_type = geom_type+'Z'if has_z else geom_type
|
||||
|
||||
(df
|
||||
.to_wkt()
|
||||
.to_sql(
|
||||
tab,
|
||||
con,
|
||||
schema_cad,
|
||||
if_exists='append',
|
||||
index=True,
|
||||
dtype={
|
||||
'geom':Geometry(geometry_type=geom_type,srid=2154)
|
||||
}
|
||||
)
|
||||
)
|
||||
print(tab)
|
||||
|
||||
|
||||
# get_table_pkey(con,schema_cad,tab)
|
||||
# get_columns(con,schema_cad,tab)
|
||||
|
||||
## 38
|
||||
# Load troncon_de_voie_ferree
|
||||
# Ident geom_type OK
|
||||
# troncon_de_voie_ferree
|
||||
# Load terrain_de_sport
|
||||
# Ident geom_type OK
|
||||
# terrain_de_sport
|
||||
## 26
|
||||
# Load cimetiere
|
||||
# Ident geom_type OK
|
||||
# cimetiere
|
||||
# Load construction_surfacique
|
||||
# Ident geom_type OK
|
||||
# construction_surfacique
|
||||
# Load reservoir
|
||||
# Ident geom_type OK
|
||||
# reservoir
|
||||
# Load batiment
|
||||
# Ident geom_type OK
|
||||
# batiment
|
||||
|
||||
## NO INSERT
|
||||
# Load troncon_de_route
|
||||
# Ident geom_type OK
|
||||
|
||||
|
||||
for Zip in lst_zip:
|
||||
print('\n'+Zip+'\n')
|
||||
for tab in get_list_table(con,schema_cad):
|
||||
if tab not in lst_tab:
|
||||
continue
|
||||
if Zip in lst_zip[:2] or (
|
||||
Zip==lst_zip[2] and tab in [
|
||||
'cimetiere',
|
||||
'construction_surfacique',
|
||||
'reservoir',
|
||||
'batiment',
|
||||
]):
|
||||
continue
|
||||
df = gpd.read_file(Zip,layer=tab)
|
||||
print('Load %s'%tab)
|
||||
|
||||
columns_type = get_columns(con,schema_cad,tab)
|
||||
geom_type = [x for x in columns_type if x['name']=='geometrie']
|
||||
print('Ident geom_type OK')
|
||||
|
||||
if geom_type:
|
||||
geom_type = geom_type[0]
|
||||
|
||||
if df.geometry.name != geom_type['name']:
|
||||
df.rename_geometry(geom_type['name'],inplace=True)
|
||||
|
||||
df.to_wkt().to_sql(
|
||||
tab,
|
||||
con,
|
||||
schema_cad,
|
||||
if_exists='append',
|
||||
index=False,
|
||||
dtype={geom_type['name']:geom_type['type']} if geom_type else None
|
||||
)
|
||||
print(tab)
|
||||
Reference in New Issue
Block a user