29 lines
661 B
Python
29 lines
661 B
Python
from flask import Flask
|
|
import os
|
|
import psycopg2
|
|
|
|
app = Flask(__name__)
|
|
|
|
DATABASE_URL = os.getenv("DATABASE_URL", "postgresql://user:password@db:5432/mydatabase")
|
|
|
|
@app.route('/')
|
|
def home():
|
|
return "This is my application!"
|
|
|
|
@app.route('/db-test')
|
|
def db_test():
|
|
try:
|
|
conn = psycopg2.connect(DATABASE_URL)
|
|
cur = conn.cursor()
|
|
cur.execute("SELECT 'Database connection successful!'")
|
|
message = cur.fetchone()[0]
|
|
cur.close()
|
|
conn.close()
|
|
return message
|
|
except Exception as e:
|
|
return f"Database connection failed: {e}"
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=5000)
|
|
|