Initial Draft

MigrationUtility.py

@@ -0,0 +1,65 @@
+import streamlit as st
+import pandas as pd
+import re
+import pandasql as psql
+import base64  # For file download
+
+################################
+####### Generic functions ######
+################################
+def process_input():
+    try:
+        if uploaded_files is not None:
+            for file in uploaded_files:
+                df = pd.read_csv(file)
+
+                pattern = r'([^/]+)\.csv$'
+                match = re.search(pattern, file.name)
+                file_name = match.group(1)
+
+                globals()['%s' % file_name] = df
+    except Exception as e:
+        st.error(f"Error processing csv: {str(e)}")
+
+################################
+####### Display of data ########
+################################
+
+# Streamlit Settings
+st.set_page_config(layout='wide')
+col1, col2, col3 = st.columns([1.5, 2, 0.5])
+with col2:
+    st.title("Migration Utility")
+
+# Data Display Layout
+col1, col2, col3 = st.columns([1.25, 2, 0.5])
+with col1:
+    options = ["csv", "xlsx"]
+    file_type = st.selectbox(
+        "Select Input File Type",
+        options, key="file_type"
+    )
+
+# -- Layout for File Upload
+col1, col2, col3 = st.columns([1, 1, 1])
+with col1:
+    uploaded_files = st.file_uploader(
+        "Choose a file",
+        type=file_type,
+        key="uploaded_files",
+        accept_multiple_files=True
+    )
+
+# -- Layout for query and result
+sql_query = st.text_area(label="Sql Query", value="", key="sql_query", height=200)
+if st.button("Run SQL Query"):
+    process_input()
+    try:
+        result_df = psql.sqldf(sql_query, globals())
+        st.write("Query Result")
+        st.dataframe(result_df)
+        csv_data = result_df.to_csv(index=False)
+        b64 = base64.b64encode(csv_data.encode()).decode()
+        st.markdown(f'<a href="data:file/csv;base64,{b64}" download="result.csv">Download Result CSV</a>', unsafe_allow_html=True)
+    except Exception as e:
+        st.error(f"Error executing SQL query: {str(e)}")