121 lines
3.4 KiB
Python
121 lines
3.4 KiB
Python
"""
|
|
app.py — Flask CRUD-интерфейс для PostgreSQL.
|
|
|
|
Переменные окружения (из json_env):
|
|
- PGHOST, PGPORT, PGDATABASE, PGUSER, PGPASSWORD — подключение к БД
|
|
- TABLE_NAME — имя таблицы (по умолчанию crud_items)
|
|
|
|
При старте: CREATE TABLE IF NOT EXISTS (БЕЗ DROP).
|
|
"""
|
|
|
|
import os
|
|
import psycopg2
|
|
import psycopg2.extras
|
|
from flask import Flask, render_template, request, redirect, url_for, g
|
|
|
|
VERSION = "1.0.0"
|
|
TABLE_NAME = os.environ.get("TABLE_NAME", "crud_items")
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
def get_db():
|
|
"""Ленивое подключение к PostgreSQL. Один коннект на запрос."""
|
|
if "db" not in g:
|
|
g.db = psycopg2.connect(
|
|
host=os.environ.get("PGHOST", "127.0.0.1"),
|
|
port=os.environ.get("PGPORT", "5432"),
|
|
dbname=os.environ.get("PGDATABASE", "postgres"),
|
|
user=os.environ.get("PGUSER", "postgres"),
|
|
password=os.environ.get("PGPASSWORD", ""),
|
|
)
|
|
g.db.autocommit = True
|
|
return g.db
|
|
|
|
|
|
def init_db():
|
|
"""CREATE TABLE IF NOT EXISTS — никогда не дропает существующую таблицу."""
|
|
db = get_db()
|
|
cur = db.cursor()
|
|
cur.execute(
|
|
f"""
|
|
CREATE TABLE IF NOT EXISTS {TABLE_NAME} (
|
|
id SERIAL PRIMARY KEY,
|
|
value TEXT NOT NULL DEFAULT ''
|
|
)
|
|
"""
|
|
)
|
|
cur.close()
|
|
|
|
|
|
@app.teardown_appcontext
|
|
def close_db(exc):
|
|
db = g.pop("db", None)
|
|
if db is not None:
|
|
db.close()
|
|
|
|
|
|
@app.route("/", methods=["GET", "POST"])
|
|
def index():
|
|
db = get_db()
|
|
cur = db.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
|
|
|
|
# --- Обработка POST (create / update / delete) — POST-redirect-GET ---
|
|
if request.method == "POST":
|
|
action = request.form.get("action", "")
|
|
|
|
if action == "create":
|
|
cur.execute(
|
|
f"INSERT INTO {TABLE_NAME} (value) VALUES (%s)",
|
|
(request.form.get("value", ""),),
|
|
)
|
|
|
|
elif action == "update":
|
|
cur.execute(
|
|
f"UPDATE {TABLE_NAME} SET value = %s WHERE id = %s",
|
|
(request.form.get("value", ""), int(request.form.get("id", 0))),
|
|
)
|
|
|
|
elif action == "delete":
|
|
cur.execute(
|
|
f"DELETE FROM {TABLE_NAME} WHERE id = %s",
|
|
(int(request.form.get("id", 0)),),
|
|
)
|
|
|
|
cur.close()
|
|
return redirect(url_for("index"))
|
|
|
|
# --- GET: список + форма редактирования ---
|
|
edit_row = None
|
|
edit_id = request.args.get("edit")
|
|
if edit_id:
|
|
cur.execute(
|
|
f"SELECT id, value FROM {TABLE_NAME} WHERE id = %s",
|
|
(int(edit_id),),
|
|
)
|
|
row = cur.fetchone()
|
|
if row:
|
|
edit_row = dict(row)
|
|
|
|
# Список всех записей
|
|
cur.execute(f"SELECT id, value FROM {TABLE_NAME} ORDER BY id")
|
|
rows = cur.fetchall()
|
|
cur.close()
|
|
|
|
return render_template(
|
|
"index.html",
|
|
table_name=TABLE_NAME,
|
|
rows=rows,
|
|
edit_row=edit_row,
|
|
version=VERSION,
|
|
)
|
|
|
|
|
|
# --- Инициализация БД при старте (до первого запроса) ---
|
|
with app.app_context():
|
|
init_db()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 5000)), debug=False)
|