Check PostgreSQL Database Connection

check_postgre_database_connection.py

import psycopg2

def check_tables_in_database(host, port, username, password, database):
    try:
        # Attempt to connect to the PostgreSQL database
        connection = psycopg2.connect(
            host=host,
            port=port,
            user=username,
            password=password,
            database=database
        )

        if connection:
            print(f"Connected to the PostgreSQL database: {database}")
            
            # Create a cursor to execute SQL queries
            cursor = connection.cursor()
            
            # Retrieve the list of tables in the database
            cursor.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'")
            
            # Fetch and print the table names
            tables = cursor.fetchall()
            if tables:
                print("Tables in the database:")
                for table in tables:
                    print(table[0])
            
            # Close the cursor and connection
            cursor.close()
            connection.close()
    except psycopg2.Error as e:
        # If there's an error, print an error message
        print(f"Error connecting to the database: {str(e)}")

if __name__ == "__main__":
    # Replace with your PostgreSQL database connection details
    host = ''  # PostgreSQL server host
    port = ''  # PostgreSQL server port
    username = ''  # PostgreSQL username
    password = ''  # PostgreSQL password
    database = ''  # PostgreSQL database name
    
    check_tables_in_database(host, port, username, password, database)