Function generate_table_metadata

byod/api/app/setup/utils.py:248–289

Generate table metadata

(table_name)

def generate_table_metadata(table_name):
    """
    Generate table metadata
    """
    try:
        with ENGINE.connect() as connection:
            connection = connection.execution_options(
                postgresql_readonly=True
            )
            with connection.begin():
                sql_text = text(f"""
                    SELECT column_name, data_type, udt_name
                    FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = '{table_name}';
                """)
                result = connection.execute(sql_text)

            rows = [list(r) for r in result.all()]

            columns_metadata = []
            for row in rows:
                if row[1] == "USER-DEFINED":
                    column_type = row[2]
                else:
                    column_type = row[1]
                columns_metadata.append({
                    "name": row[0],
                    "type": column_type
                })

            
            # TODO: generate table description
            # TODO: generate column description (FK, PK, etc.)
            table_description = ""
            return {
                "name": table_name,
                "description": table_description,
                "columns": columns_metadata
            }
    
    except Exception as e:
        print(e)
        return None