db / app.py
aigenai's picture
Create app.py
1d64b43 verified
raw
history blame
880 Bytes
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)