File size: 880 Bytes
1d64b43 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
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)
|