naeel
cbfba0bc54
По рекомендации Sonnet: - schema.py — все DDL (documents, contracts, supplements, spec_rows, spec_history) - db.py — только транспорт (connect/query), DDL убран - app.py — минимальный: ensure_schema() + register_blueprint() - test_routes.py — подробные комментарии, исправлен импорт psysql
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
"""
|
|
db.py — Транспортный слой к базе данных.
|
|
|
|
Только connect() и query(). Никакой бизнес-логики, никакого DDL.
|
|
DDL → schema.py. Бизнес-запросы → upload.py, extractor.py, api.py.
|
|
|
|
Функции:
|
|
_pg_connect(dbname) — сырое подключение к ЛЮБОЙ базе (используется /test createdb)
|
|
connect() — подключение к ЦЕЛЕВОЙ базе (из DB_NAME)
|
|
query(sql, params) — выполнить запрос → (result, error)
|
|
"""
|
|
|
|
import os
|
|
import psycopg2
|
|
|
|
|
|
def _pg_connect(dbname):
|
|
"""
|
|
Сырое подключение к указанной базе данных.
|
|
Используется:
|
|
- connect() для целевой БД
|
|
- test_routes.py для /test createdb (подключение к 'postgres')
|
|
"""
|
|
return psycopg2.connect(
|
|
host=os.getenv("DB_HOST"),
|
|
port=os.getenv("DB_PORT", 5432),
|
|
dbname=dbname,
|
|
user=os.getenv("DB_USER"),
|
|
password=os.getenv("DB_PASS"),
|
|
sslmode=os.getenv("DB_SSLMODE", "disable"),
|
|
)
|
|
|
|
|
|
def connect():
|
|
"""
|
|
Подключение к ЦЕЛЕВОЙ базе данных (DB_NAME из переменных окружения).
|
|
Возвращает (connection, None) при успехе или (None, error) при ошибке.
|
|
"""
|
|
try:
|
|
return _pg_connect(os.getenv("DB_NAME")), None
|
|
except Exception as e:
|
|
return None, str(e)
|
|
|
|
|
|
def query(sql_text, params=None):
|
|
"""
|
|
Выполнить SQL-запрос к целевой БД.
|
|
Возвращает (result, None) или (None, error).
|
|
|
|
result = {"columns": [...], "rows": [[...], ...]}
|
|
"""
|
|
conn, err = connect()
|
|
if err:
|
|
return None, f"connect: {err}"
|
|
try:
|
|
cur = conn.cursor()
|
|
cur.execute(sql_text, params)
|
|
rows = cur.fetchall()
|
|
cols = [desc[0] for desc in cur.description] if cur.description else []
|
|
cur.close()
|
|
conn.close()
|
|
return {"columns": cols, "rows": [list(r) for r in rows]}, None
|
|
except Exception as e:
|
|
conn.close()
|
|
return None, str(e)
|