OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add extension back

Browse Source
This commit is contained in:
WQY\qiong
2023-03-31 15:18:46 +08:00
parent dcecce165e
commit e57721a29a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
create_template.py
View File

@@ -33,11 +33,13 @@ sql_create = [
"script/sql/create/29.scada_device.sql", "script/sql/create/29.scada_device.sql",
"script/sql/create/30.scada_device_data.sql", "script/sql/create/30.scada_device_data.sql",
"script/sql/create/31.scada_element.sql", "script/sql/create/31.scada_element.sql",
"script/sql/create/32.virtual_district.sql",
"script/sql/create/operation.sql" "script/sql/create/operation.sql"
] ]
sql_drop = [ sql_drop = [
"script/sql/drop/operation.sql", "script/sql/drop/operation.sql",
"script/sql/drop/32.virtual_district.sql",
"script/sql/drop/31.scada_element.sql", "script/sql/drop/31.scada_element.sql",
"script/sql/drop/30.scada_device_data.sql", "script/sql/drop/30.scada_device_data.sql",
"script/sql/drop/29.scada_device.sql", "script/sql/drop/29.scada_device.sql",
@@ -78,6 +80,8 @@ def create_template():
cur.execute("create database project") cur.execute("create database project")
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:
cur.execute('create extension postgis 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") as f:
cur.execute(f.read()) cur.execute(f.read())