diff --git a/create_template.py b/create_template.py index 85ed181..f9a4255 100644 --- a/create_template.py +++ b/create_template.py @@ -40,6 +40,8 @@ sql_create = [ "script/sql/create/36.wda.sql", "script/sql/create/37.history_patterns_flows.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/operation.sql" ] @@ -47,6 +49,8 @@ sql_create = [ sql_drop = [ "script/sql/drop/operation.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/37.history_patterns_flows.sql", "script/sql/drop/36.wda.sql", @@ -97,7 +101,7 @@ def create_template(): cur.execute('create extension postgis cascade') cur.execute('create extension pgrouting cascade') for sql in sql_create: - with open(sql, "r") as f: + with open(sql, "r", encoding="utf-8") as f: cur.execute(f.read()) print(f'executed {sql}') conn.commit() @@ -112,7 +116,7 @@ def delete_template(): with pg.connect(conninfo="dbname=project host=127.0.0.1") as conn: with conn.cursor() as cur: for sql in sql_drop: - with open(sql, "r") as f: + with open(sql, "r", encoding="utf-8") as f: cur.execute(f.read()) print(f'executed {sql}') conn.commit()