|
from flask import Flask, jsonify |
|
import psycopg2 |
|
import os |
|
|
|
app = Flask(__name__) |
|
|
|
def connect_db(): |
|
try: |
|
connection = psycopg2.connect( |
|
dbname=os.getenv('POSTGRES_DB'), |
|
user=os.getenv('POSTGRES_USER'), |
|
password=os.getenv('POSTGRES_PASSWORD'), |
|
host='localhost' |
|
) |
|
cursor = connection.cursor() |
|
cursor.execute("SELECT version();") |
|
db_version = cursor.fetchone() |
|
|
|
return { |
|
"status": "success", |
|
"db_version": db_version, |
|
"user": os.getenv('POSTGRES_USER'), |
|
"database": os.getenv('POSTGRES_DB') |
|
} |
|
|
|
except Exception as e: |
|
return {"status": "failure", "error": str(e)} |
|
|
|
@app.route('/') |
|
def home(): |
|
db_info = connect_db() |
|
return jsonify(db_info) |
|
|
|
if __name__ == '__main__': |
|
app.run(host='0.0.0.0', port=7860) |
|
|