shadow_brain_core/check_columns.py

import psycopg2
from db_util import get_db_connection

def get_columns():
    conn = get_db_connection()
    cur = conn.cursor()
    cur.execute("SELECT * FROM taemingames.imperial_team_tasks LIMIT 0")
    colnames = [desc[0] for desc in cur.description]
    print(colnames)
    conn.close()

if __name__ == "__main__":
    get_columns()