118 lines
3.4 KiB
Python
118 lines
3.4 KiB
Python
import os
|
|
|
|
import psycopg2
|
|
from flask import Flask, redirect, render_template_string, request
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
def get_conn():
|
|
db_url = os.getenv("DATABASE_URL")
|
|
if db_url:
|
|
return psycopg2.connect(db_url)
|
|
return psycopg2.connect(
|
|
host=os.getenv("PGHOST"),
|
|
port=os.getenv("PGPORT", "5432"),
|
|
user=os.getenv("PGUSER"),
|
|
password=os.getenv("PGPASSWORD"),
|
|
dbname=os.getenv("PGDATABASE", "postgres"),
|
|
sslmode=os.getenv("PGSSLMODE", "require"),
|
|
)
|
|
|
|
|
|
def ensure_table():
|
|
with get_conn() as conn, conn.cursor() as cur:
|
|
cur.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS nubes_test_table (
|
|
id SERIAL PRIMARY KEY,
|
|
test_data TEXT,
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
"""
|
|
)
|
|
|
|
|
|
PAGE = """
|
|
<!doctype html>
|
|
<html lang="ru">
|
|
<head>
|
|
<meta charset="utf-8">
|
|
<title>Flask + Postgres Demo</title>
|
|
</head>
|
|
<body>
|
|
<h3>Flask + Postgres Demo</h3>
|
|
{% if error %}<p style="color:red">{{ error }}</p>{% endif %}
|
|
<form method="post">
|
|
<input type="text" name="txt_content" placeholder="Новое сообщение...">
|
|
<button type="submit">Добавить</button>
|
|
</form>
|
|
<table border="1" cellpadding="6" cellspacing="0">
|
|
<tr><th>ID</th><th>Содержимое</th><th>Действия</th></tr>
|
|
{% for row in rows %}
|
|
<tr>
|
|
<td>{{ row[0] }}</td>
|
|
<td>
|
|
<form method="post" action="/update">
|
|
<input type="hidden" name="id" value="{{ row[0] }}">
|
|
<input type="text" name="txt_content" value="{{ row[1] }}">
|
|
<button type="submit">💾</button>
|
|
</form>
|
|
</td>
|
|
<td>
|
|
<form method="post" action="/delete" onsubmit="return confirm('Удалить?')">
|
|
<input type="hidden" name="id" value="{{ row[0] }}">
|
|
<button type="submit">🗑</button>
|
|
</form>
|
|
</td>
|
|
</tr>
|
|
{% endfor %}
|
|
</table>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
|
|
@app.route("/", methods=["GET", "POST"])
|
|
def index():
|
|
ensure_table()
|
|
error = None
|
|
if request.method == "POST":
|
|
content = request.form.get("txt_content") or "Это Flask сделал"
|
|
try:
|
|
with get_conn() as conn, conn.cursor() as cur:
|
|
cur.execute(
|
|
"INSERT INTO nubes_test_table (test_data) VALUES (%s)",
|
|
(content,),
|
|
)
|
|
return redirect("/")
|
|
except Exception as exc:
|
|
error = str(exc)
|
|
with get_conn() as conn, conn.cursor() as cur:
|
|
cur.execute(
|
|
"SELECT id, test_data FROM nubes_test_table ORDER BY id DESC LIMIT 20"
|
|
)
|
|
rows = cur.fetchall()
|
|
return render_template_string(PAGE, rows=rows, error=error)
|
|
|
|
|
|
@app.post("/update")
|
|
def update():
|
|
with get_conn() as conn, conn.cursor() as cur:
|
|
cur.execute(
|
|
"UPDATE nubes_test_table SET test_data=%s WHERE id=%s",
|
|
(request.form.get("txt_content"), request.form.get("id")),
|
|
)
|
|
return redirect("/")
|
|
|
|
|
|
@app.post("/delete")
|
|
def delete():
|
|
with get_conn() as conn, conn.cursor() as cur:
|
|
cur.execute("DELETE FROM nubes_test_table WHERE id=%s", (request.form.get("id"),))
|
|
return redirect("/")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True, host="0.0.0.0", port=5000)
|