Code Editor for app.py

import streamlit as st
import re

def convert_to_teradata_macro(greenplum_function):
    """
    Converts a Greenplum function to a Teradata macro
    """
    # Remove CREATE OR REPLACE FUNCTION line
    function_name = re.search(r'CREATE\s+(?:OR\s+REPLACE\s+)?FUNCTION\s+(\w+)', greenplum_function, re.IGNORECASE)
    function_name = function_name.group(1) if function_name else "converted_function"
    
    # Extract parameters
    params_match = re.search(r'\((.*?)\)', greenplum_function, re.DOTALL)
    params = params_match.group(1) if params_match else ""
    
    # Convert parameter syntax
    # In Teradata, parameters are defined as IN/OUT without data types in the header
    teradata_params = []
    for param in params.split(','):
        param = param.strip()
        if param:
            # Extract parameter name (assuming format: name type)
            param_name = param.split()[0]
            teradata_params.append(param_name)
    
    # Extract function body
    body_match = re.search(r'AS\s+\$\$\s+(.*?)\$\$', greenplum_function, re.DOTALL | re.IGNORECASE)
    body = body_match.group(1) if body_match else ""
    
    # Replace PostgreSQL/Greenplum specific syntax
    body = body.replace('RETURN', 'RETURN result')
    
    # Create Teradata macro
    teradata_macro = f"""REPLACE MACRO {function_name} ({', '.join(teradata_params)}) AS
(
{body}
);
"""
    return teradata_macro

def convert_to_teradata_stored_proc(greenplum_function):
    """
    Converts a Greenplum function to a Teradata stored procedure
    """
    # Extract function name
    function_name = re.search(r'CREATE\s+(?:OR\s+REPLACE\s+)?FUNCTION\s+(\w+)', greenplum_function, re.IGNORECASE)
    function_name = function_name.group(1) if function_name else "converted_procedure"
    
    # Extract parameters
    params_match = re.search(r'\((.*?)\)', greenplum_function, re.DOTALL)
    params = params_match.group(1) if params_match else ""
    
    # Convert parameter syntax for stored procedure
    # In Teradata, stored procedures use IN/OUT/INOUT parameters with data types
    teradata_params = []
    for param in params.split(','):
        param = param.strip()
        if param:
            parts = param.split()
            if len(parts) >= 2:
                param_name = parts[0]
                param_type = parts[1]
                teradata_params.append(f"IN {param_name} {param_type}")
    
    # Extract function body
    body_match = re.search(r'AS\s+\$\$\s+(.*?)\$\$', greenplum_function, re.DOTALL | re.IGNORECASE)
    body = body_match.group(1) if body_match else ""
    
    # Replace PostgreSQL/Greenplum specific syntax
    body = body.replace('RETURN', 'SET result =')
    
    # Create Teradata stored procedure
    teradata_proc = f"""REPLACE PROCEDURE {function_name} ({', '.join(teradata_params)})
BEGIN
{body}
END;
"""
    return teradata_proc

st.title("Greenplum function to Teradata SQL Converter")

col1, col2 = st.columns(2)

with col1:
    uploaded_file = st.file_uploader("Upload your GP function:", type=["sql", "txt"])

with col2:
    output_type = st.selectbox("Desired output:", ["Macro", "Store Proc"])

# Center the submit button
col_left, col_center, col_right = st.columns([1, 2, 1])
with col_center:
    submit_button = st.button("Submit", use_container_width=True)

# Output section
st.subheader("Output")

if submit_button and uploaded_file is not None:
    greenplum_function = uploaded_file.getvalue().decode("utf-8")
    
    if output_type == "Macro":
        teradata_code = convert_to_teradata_macro(greenplum_function)
    else:  # "Store Proc"
        teradata_code = convert_to_teradata_stored_proc(greenplum_function)
    
    st.code(teradata_code, language="sql")
elif submit_button and uploaded_file is None:
    st.error("Please upload a Greenplum function file first.")