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()