api.py

from flask import Flask, request, jsonify
import psycopg2, time

app = Flask(__name__)

def get_connection():
    for _ in range(15):
        try:
            return psycopg2.connect(
                dbname="facts-db", user="postgres", password="postgres", host="db", port="5432"
            )
        except psycopg2.OperationalError:
            time.sleep(3)
    raise Exception("Couldn't connect to the database!")

def create_table():
    connection = get_connection()
    cursor = connection.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS facts (id SERIAL PRIMARY KEY, content TEXT NOT NULL);")
    connection.commit()
    cursor.close()
    connection.close()

@app.route("/")
def home():
    return "Welcome to Facts API!"

@app.route("/facts", methods=["GET"])
def list_facts():
    connection = get_connection()
    cursor = connection.cursor()
    cursor.execute("SELECT * FROM facts ORDER BY id")
    rows = cursor.fetchall()
    cursor.close()
    connection.close()
    
    res = []
    for row in rows:
        res.append({"id": row[0], "content": row[1]})
    return jsonify(res)

@app.route("/facts", methods=["POST"])
def add_fact():
    data = request.get_json()
    content = data.get("content")
    if not content:
        return jsonify("Content missing")
    
    connection = get_connection()
    cursor = connection.cursor()
    cursor.execute("INSERT INTO facts (content) VALUES (%s) RETURNING id", (content,))
    id = cursor.fetchone()[0]
    connection.commit()
    cursor.close()
    connection.close()
    return jsonify({"id": id, "content": content})

@app.route("/facts/<int:id>", methods=["DELETE"])
def delete_fact(id):
    connection = get_connection()
    cursor = connection.cursor()
    cursor.execute("DELETE FROM facts WHERE id = %s RETURNING id", (id,))
    deleted = cursor.fetchone()
    connection.commit()
    cursor.close()
    connection.close()

    if deleted:
        return jsonify(f"Deleted fact with id: {id}")
    else:
        return jsonify("Fact not found")

@app.route("/facts/stats", methods=["GET"])
def stats():
    connection = get_connection()
    cursor = connection.cursor()
    cursor.execute("SELECT COUNT(*), MAX(id), MIN(id) FROM facts")
    count, max_id, min_id = cursor.fetchone()
    cursor.close()
    connection.close()
    return jsonify({"number_facts": count, "highest_id": max_id, "lowest_id": min_id})

if __name__ == "__main__":
    create_table();
    app.run(host="0.0.0.0", port="5000")