feat: слои db.py + test_routes (Blueprint /test)

This commit is contained in:
2026-06-13 12:02:46 +04:00
parent 684b4cc9c0
commit dbbc8ea5fe
4 changed files with 89 additions and 18 deletions
+5 -18
View File
@@ -1,37 +1,24 @@
import os
import psycopg2
from dotenv import load_dotenv
from flask import Flask, render_template
from . import db
from .test_routes import test_bp
load_dotenv()
class ContractsApp:
def __init__(self):
self.app = Flask(__name__)
self.db = self._connect_db()
self.add_routes()
def _connect_db(self):
try:
conn = psycopg2.connect(
host=os.getenv("DB_HOST"),
port=os.getenv("DB_PORT", 5432),
dbname=os.getenv("DB_NAME"),
user=os.getenv("DB_USER"),
password=os.getenv("DB_PASS"),
sslmode=os.getenv("DB_SSLMODE", "disable"),
)
return conn
except Exception as e:
print(f"DB connect error: {e}")
return None
def add_routes(self):
self.app.add_url_rule("/", "index", self.index)
self.app.add_url_rule("/health", "health", self.health)
self.app.register_blueprint(test_bp)
def index(self):
db_status = "connected" if self.db else "no DB"
conn = db.connect()
db_status = "connected" if conn else "no DB"
return render_template("index.html", db_status=db_status)
def health(self):