Function get_table_names

byod/api/app/setup/utils.py:179–212

Get names of tables in the database

(username=get_current_user())

def get_table_names(username=get_current_user()) -> List[str]:
    """
    Get names of tables in the database
    """
    IGNORE_TABLES = ["ai_sql_table_metadata", "ai_sql_type_metadata", "ai_sql_in_context_examples"]
    try:
        with ENGINE.connect() as connection:
            connection = connection.execution_options(
                postgresql_readonly=True
            )
            with connection.begin():
                # sql_text = text(f"""
                #     SELECT tablename
                #     FROM pg_catalog.pg_tables
                #     WHERE tableowner = '{username}';
                # """)
                sql_text = text(f"""
                    SELECT tablename
                    FROM pg_catalog.pg_tables
                    WHERE schemaname = 'public';
                """)
                result = connection.execute(sql_text)
                rows = [list(r) for r in result.all()]

                table_names = []
                for row in rows:
                    if row[0] not in IGNORE_TABLES:
                        table_names.append(row[0])

                return table_names
            
    except Exception as e:
        print(e)
        return None


def generate_type_metadata(type_name):