统一 pg 数据库的连接

This commit is contained in:
JIANG
2025-11-26 10:10:00 +08:00
parent 15d9476858
commit 5ab10886a0
5 changed files with 44 additions and 24 deletions

View File

@@ -29,6 +29,7 @@ import pytz
import influxdb_info
import project_info
import time_api
from postgresql_info import get_pgconn_string
# influxdb数据库连接信息
url = influxdb_info.url
@@ -46,7 +47,7 @@ def query_pg_scada_info_realtime(name: str) -> None:
:return:
"""
# 连接数据库
conn_string = f"dbname={name} host=127.0.0.1"
conn_string = get_pgconn_string(db_name=name)
try:
with psycopg.connect(conn_string) as conn:
with conn.cursor() as cur:
@@ -123,7 +124,7 @@ def query_pg_scada_info_non_realtime(name: str) -> None:
# DingZQ, 2025-03-21
# close_project(name)
# 连接数据库
conn_string = f"dbname={name} host=127.0.0.1"
conn_string = get_pgconn_string(db_name=name)
try:
with psycopg.connect(conn_string) as conn:
with conn.cursor() as cur:
@@ -199,7 +200,7 @@ def query_pg_scada_info(name: str) -> list[dict]:
:return: 包含所有记录的列表,每条记录为一个字典
"""
# 连接数据库
conn_string = f"dbname={name} host=127.0.0.1"
conn_string = get_pgconn_string(db_name=name)
records_list = []
try:
@@ -3360,7 +3361,8 @@ def query_corresponding_query_id_and_element_id(name: str) -> None:
:return:
"""
# 连接数据库
conn_string = f"dbname={name} host=127.0.0.1"
conn_string = f"dbname={db_name} host={db_host} user={db_user} password={db_password}"
conn_string = get_pgconn_string(db_name=name)
try:
with psycopg.connect(conn_string) as conn:
with conn.cursor() as cur: