136 lines
5.7 KiB
Python
136 lines
5.7 KiB
Python
from flask import Flask, request, jsonify
|
|
import os
|
|
import psycopg2
|
|
import time
|
|
|
|
app = Flask(__name__)
|
|
|
|
# Database Configuration
|
|
db_host = os.environ.get("DB_HOST", "blog-db-service")
|
|
db_name = os.environ.get("POSTGRES_DB", "blogdb")
|
|
db_user = os.environ.get("POSTGRES_USER", "admin")
|
|
db_password = os.environ.get("POSTGRES_PASSWORD", "password")
|
|
|
|
def get_db_connection():
|
|
return psycopg2.connect(host=db_host, database=db_name, user=db_user, password=db_password)
|
|
|
|
def create_table():
|
|
max_retries = 10
|
|
retry_delay = 5 # seconds
|
|
|
|
for attempt in range(max_retries):
|
|
try:
|
|
print(f"Attempt {attempt+1}/{max_retries} to connect to database...")
|
|
conn = get_db_connection()
|
|
cur = conn.cursor()
|
|
cur.execute('''
|
|
CREATE TABLE IF NOT EXISTS articles (
|
|
id SERIAL PRIMARY KEY,
|
|
title TEXT NOT NULL,
|
|
content TEXT NOT NULL
|
|
)
|
|
''')
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
print("Table 'articles' created successfully")
|
|
return True
|
|
except Exception as e:
|
|
print(f"Database connection attempt {attempt+1}/{max_retries} failed: {e}")
|
|
if attempt < max_retries - 1:
|
|
print(f"Retrying in {retry_delay} seconds...")
|
|
time.sleep(retry_delay)
|
|
|
|
print("Failed to create table after multiple attempts")
|
|
return False
|
|
|
|
# Initialize the variable at startup
|
|
table_created = False
|
|
|
|
@app.route("/", methods=["GET", "POST"])
|
|
def index():
|
|
global table_created
|
|
|
|
# If the table is not yet created, try to create it
|
|
if not table_created:
|
|
table_created = create_table()
|
|
|
|
# Check if the table has been created before continuing
|
|
if not table_created:
|
|
return "Database connection failed. Please try again later.", 500
|
|
|
|
try:
|
|
conn = get_db_connection()
|
|
cur = conn.cursor()
|
|
|
|
if request.method == "POST":
|
|
title = request.form.get("title")
|
|
content = request.form.get("content")
|
|
if title and content:
|
|
cur.execute("INSERT INTO articles (title, content) VALUES (%s, %s)", (title, content))
|
|
conn.commit()
|
|
|
|
cur.execute("SELECT * FROM articles ORDER BY id DESC")
|
|
articles = cur.fetchall()
|
|
cur.close()
|
|
conn.close()
|
|
|
|
return f"""
|
|
<html>
|
|
<head>
|
|
<title>My Blog</title>
|
|
<style>
|
|
body {{ font-family: 'Poppins', sans-serif; background-color: #f5f5f5; color: #333; margin: 0; padding: 0; }}
|
|
.container {{ max-width: 1100px; margin: 50px auto; background: #6a0dad; padding: 30px; border-radius: 10px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); display: flex; gap: 20px; }}
|
|
.form-container, .articles-container {{ flex: 1; background: #ffffff; padding: 20px; border-radius: 10px; box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1); }}
|
|
h1{{ text-align: center; color: #6a0dad; font-size: 30px; }}
|
|
form {{ display: flex; flex-direction: column; gap: 10px; align-items: center; }}
|
|
input, textarea {{ width: 90%; padding: 14px; border: 1px solid #ccc; border-radius: 5px; background: #fff; color: #333; font-size: 18px; }}
|
|
button {{ background-color: #6a0dad; color: white; border: none; padding: 12px 25px; cursor: pointer; border-radius: 5px; transition: background 0.3s; font-size: 20px; }}
|
|
button:hover {{ background-color: #5a0dad; }}
|
|
.articles-container {{ display: flex; flex-direction: column; gap: 15px; }}
|
|
.article {{ background: #e0e0e0; padding: 20px; border-radius: 5px; box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1); position: relative; }}
|
|
.article h3 {{ margin: 0; color: #333; font-size: 22px; }}
|
|
.article p {{ margin: 10px 0; color: #555; font-size: 18px; }}
|
|
.delete-form {{ position: absolute; top: 10px; right: 10px; }}
|
|
.delete-button {{ background-color: #ff5555; padding: 8px 12px; font-size: 14px; border-radius: 5px; }}
|
|
.delete-button:hover {{ background-color: #cc4444; }}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<div class="form-container">
|
|
<h1>My Blog</h1>
|
|
<form method="post">
|
|
<input type="text" name="title" placeholder="Title" required>
|
|
<textarea name="content" placeholder="Content" required></textarea>
|
|
<button type="submit">Add Article</button>
|
|
</form>
|
|
</div>
|
|
<div class="articles-container">
|
|
<h1>Articles</h1>
|
|
{''.join(f'<div class="article"><h3>{a[1]}</h3><p>{a[2]}</p><form method="post" action="/delete/{a[0]}" class="delete-form"><button type="submit" class="delete-button">Delete</button></form></div>' for a in articles)}
|
|
</div>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
except Exception as e:
|
|
return f"An error occurred: {str(e)}", 500
|
|
|
|
@app.route("/delete/<int:article_id>", methods=["POST"])
|
|
def delete_article(article_id):
|
|
try:
|
|
conn = get_db_connection()
|
|
cur = conn.cursor()
|
|
cur.execute("DELETE FROM articles WHERE id = %s", (article_id,))
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
return "<script>window.location.href='/'</script>"
|
|
except Exception as e:
|
|
return f"An error occurred while deleting: {str(e)}", 500
|
|
|
|
if __name__ == "__main__":
|
|
create_table()
|
|
app.run(host="0.0.0.0", port=5000) |