Create update_pk_infos.py

maj des infos sur les voies OSM

scripts_v2/update_pk_infos.py

@@ -0,0 +1,193 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+#
+# Libraries
+#
+
+import os
+import sys
+import datetime
+import time
+import configparser
+import psycopg2
+#from logguer import log_verif
+
+
+# ==============================================================================
+
+# répertoire courant
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
+# jour courant
+date = datetime.datetime.now()
+
+# lecture du fichier de configuration qui contient les infos de connection
+config = configparser.ConfigParser()
+config.read( script_dir + '/config.ini')
+
+
+
+#fichier_log = str(date.year)+str(date.month)+str(date.day)+"_"+str(date.hour)+str(date.minute)
+fichier_log = str(date.year)+str(date.month)+str(date.day)
+
+
+# ==============================================================================
+
+def initConnRedadegDB():
+
+  # connexion à la base postgresql
+  try:
+    global db_redadeg_pg_conn
+    db_redadeg_pg_conn = psycopg2.connect(db_redadeg_conn_str)
+    db_redadeg_pg_conn.set_session(autocommit=True)
+
+    print("  Connexion à la base "+db_redadeg_db+" sur "+db_redadeg_host+" réussie ")
+
+
+  except Exception as err:
+    print("  Connexion à la base "+db_redadeg_db+" sur "+db_redadeg_host+ " impossible ")
+    try:
+      err.pgcode
+      print("  PostgreSQL error code : " + err.pgcode)
+      sys.exit()
+    except:
+      print("  " + str(err),0)
+      sys.exit()
+
+
+# ==============================================================================
+
+def closeConnRedadegDB():
+
+  try:
+    db_redadeg_pg_conn.close()
+    print("  Fermeture de la connexion à la base "+db_redadeg_db+" sur "+db_redadeg_host)
+  except:
+    pass
+
+
+# ==============================================================================
+
+
+# ==============================================================================
+# ==============================================================================
+# ==============================================================================
+# Start processing
+#
+
+startTime = time.perf_counter()
+
+# on récupère les arguments passés
+list_of_args = sys.argv
+
+millesime = ""
+secteur = ""
+typemaj = ""
+
+# et on fait des tests
+
+try:
+    if len(list_of_args[1]) != 4:
+        print("Pas de millésime en argument")
+        sys.exit()
+    else:
+        millesime = list_of_args[1]
+
+
+except SystemExit:
+    print("Erreur dans les arguments --> stop")
+    sys.exit()
+except:
+    print("oups : vérifiez vos arguments passés au script !")
+    print("stop")
+    sys.exit()
+
+print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
+print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
+print("")
+print(" Mise à jour des informations OSM des PK pour le millésime " + str(millesime))
+print("")
+
+print("  Lecture du fichier de configuration ")
+print("")
+
+# BD Ar Redadeg
+global db_redadeg_host
+global db_redadeg_db
+db_redadeg_host = config.get('redadeg_database', 'host')
+db_redadeg_port = config.get('redadeg_database', 'port')
+db_redadeg_db = config.get('redadeg_database', 'db') + "_" + str(millesime)
+db_redadeg_user = config.get('redadeg_database', 'user')
+db_redadeg_passwd = config.get('redadeg_database', 'passwd')
+# chaîne de connection
+global db_redadeg_conn_str
+db_redadeg_conn_str = "host=" + db_redadeg_host + " port=" + db_redadeg_port + " dbname=" + db_redadeg_db + " user=" + db_redadeg_user + " password=" + db_redadeg_passwd
+
+initConnRedadegDB()
+
+# le cursor
+db_redadeg_cursor = db_redadeg_pg_conn.cursor()
+
+print("")
+
+# ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+try:
+
+    print("  Mise à jour des informations sur les voies")
+
+    sql_update_infos_ways = """
+UPDATE phase_3_pk
+SET
+  way_osm_id = sub.osm_id ,
+  way_highway = sub.highway ,
+  way_type = sub."type" ,
+  way_oneway = sub.oneway ,
+  way_ref = sub."ref" ,
+  way_name_fr = sub.name_fr ,
+  way_name_br = sub.name_br
+FROM (
+  SELECT
+   pk.pk_id,
+   t.osm_id, t.highway, t."type", t.oneway, t."ref", t.name_fr, t.name_br 
+  FROM phase_3_pk pk, phase_3_troncons_pgr t
+  WHERE ST_INTERSECTS(pk.the_geom, t.the_geom)
+  ORDER BY pk_id 
+) sub
+WHERE phase_3_pk.pk_id = sub.pk_id;"""
+
+    db_redadeg_cursor.execute(sql_update_infos_ways)
+    print("  fait")
+
+
+    print("")
+    print("")
+
+except Exception as err:
+    print("  ERREUR : " + str(err))
+    closeConnRedadegDB()
+    sys.exit()
+
+    # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+db_redadeg_cursor.close()
+
+closeConnRedadegDB()
+
+# pour connaître le temps d'exécution
+print("")
+print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
+print("F I N")
+print("")
+
+stopTime = time.perf_counter()
+
+# version simple en secondes
+# print(f"Exécuté en {stopTime - startTime:0.4f} secondes")
+
+# version en h min s
+hours, rem = divmod(stopTime - startTime, 3600)
+minutes, seconds = divmod(rem, 60)
+print("Exécuté en {:0>2}:{:0>2}:{:05.2f}".format(int(hours), int(minutes), seconds))
+