create_table.py

import psycopg2

def create_tables():
    commands = (
        """
        CREATE TABLE IF NOT EXISTS user_swear_count (
            id SERIAL PRIMARY KEY,
            username VARCHAR(255) NOT NULL,
            swear_count INT NOT NULL,
            detected_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
        )
        """,
    )

    conn = None
    try:
        
        conn = psycopg2.connect(
            host="postgres-server",
            database="slang",
            user="postgres",
            password="#############",
            port = '5432')
        
        cur = conn.cursor()

        for command in commands:
            cur.execute(command)
        cur.close()
        conn.commit()
    
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    
    finally:
        if conn is not None:
            conn.close()



if __name__ == '__main__':
    create_tables()