Update create_template for user and scheme

This commit is contained in:
DingZQ
2025-03-24 21:22:41 +08:00
parent 189ad0bf8d
commit 21966d5669

View File

@@ -40,6 +40,8 @@ sql_create = [
"script/sql/create/36.wda.sql", "script/sql/create/36.wda.sql",
"script/sql/create/37.history_patterns_flows.sql", "script/sql/create/37.history_patterns_flows.sql",
"script/sql/create/38.scada_info.sql", "script/sql/create/38.scada_info.sql",
"script/sql/create/39.users.sql",
"script/sql/create/40.scheme_list.sql",
"script/sql/create/extension_data.sql", "script/sql/create/extension_data.sql",
"script/sql/create/operation.sql" "script/sql/create/operation.sql"
] ]
@@ -47,6 +49,8 @@ sql_create = [
sql_drop = [ sql_drop = [
"script/sql/drop/operation.sql", "script/sql/drop/operation.sql",
"script/sql/drop/extension_data.sql", "script/sql/drop/extension_data.sql",
"script/sql/drop/40.scheme_list.sql",
"script/sql/drop/39.users.sql",
"script/sql/drop/38.scada_info.sql", "script/sql/drop/38.scada_info.sql",
"script/sql/drop/37.history_patterns_flows.sql", "script/sql/drop/37.history_patterns_flows.sql",
"script/sql/drop/36.wda.sql", "script/sql/drop/36.wda.sql",
@@ -97,7 +101,7 @@ def create_template():
cur.execute('create extension postgis cascade') cur.execute('create extension postgis cascade')
cur.execute('create extension pgrouting cascade') cur.execute('create extension pgrouting cascade')
for sql in sql_create: for sql in sql_create:
with open(sql, "r") as f: with open(sql, "r", encoding="utf-8") as f:
cur.execute(f.read()) cur.execute(f.read())
print(f'executed {sql}') print(f'executed {sql}')
conn.commit() conn.commit()
@@ -112,7 +116,7 @@ def delete_template():
with pg.connect(conninfo="dbname=project host=127.0.0.1") as conn: with pg.connect(conninfo="dbname=project host=127.0.0.1") as conn:
with conn.cursor() as cur: with conn.cursor() as cur:
for sql in sql_drop: for sql in sql_drop:
with open(sql, "r") as f: with open(sql, "r", encoding="utf-8") as f:
cur.execute(f.read()) cur.execute(f.read())
print(f'executed {sql}') print(f'executed {sql}')
conn.commit() conn.commit()