123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
import streamlit as st
|
|
import sqlite3
|
|
import pandas as pd
|
|
|
|
def fetch_data_from_db(db_path, table_name):
|
|
conn = sqlite3.connect(db_path)
|
|
query = f"SELECT * FROM {table_name}"
|
|
df = pd.read_sql_query(query, conn)
|
|
conn.close()
|
|
return df
|
|
|
|
def insert_user_into_db(db_path, email, password, domain):
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute('''
|
|
INSERT INTO users (email, password, domain)
|
|
VALUES (?, ?, ?)
|
|
''', (email, password, domain))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def delete_user_from_db(db_path, email):
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute('''
|
|
DELETE FROM users WHERE email = ?
|
|
''', (email,))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def main():
|
|
st.set_page_config(page_title="Archiv Manager")
|
|
st.title("IMAPSync Manager")
|
|
|
|
db_path = "/app/db/imapsync_results.db"
|
|
|
|
# Sidebar for navigation with buttons and unique keys
|
|
st.sidebar.title("Navigation")
|
|
add_user = st.sidebar.button("Add User", key="add_user_button")
|
|
view_users = st.sidebar.button("View Users", key="view_users_button")
|
|
delete_user = st.sidebar.button("Delete User", key="delete_user_button")
|
|
st.sidebar.markdown(
|
|
'<a href="http://180.1.1.164:8000" target="_blank"><button style="background-color: LightGray; color: black; border: none; padding: 8px 16px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer;">View Mails</button></a>',
|
|
unsafe_allow_html=True
|
|
)
|
|
|
|
# Use session state to keep track of the selected page
|
|
if 'page' not in st.session_state:
|
|
st.session_state.page = 'View Users'
|
|
|
|
if add_user:
|
|
st.session_state.page = 'Add User'
|
|
elif view_users:
|
|
st.session_state.page = 'View Users'
|
|
elif delete_user:
|
|
st.session_state.page = 'Delete User'
|
|
|
|
if st.session_state.page == "Add User":
|
|
st.header("Add a New User")
|
|
new_email = st.text_input("Email:")
|
|
new_password = st.text_input("Password:", type="password")
|
|
new_domain = st.text_input("Domain:")
|
|
|
|
if st.button("Add User", key="submit_add_user"):
|
|
if new_email and new_password and new_domain:
|
|
insert_user_into_db(db_path, new_email, new_password, new_domain)
|
|
st.success(f"User {new_email} added successfully!")
|
|
else:
|
|
st.error("Please fill in all fields.")
|
|
|
|
elif st.session_state.page == "View Users":
|
|
st.header("IMAPSync Users and Email Transfer Results")
|
|
|
|
# Fetch users and email transfer data
|
|
users_df = fetch_data_from_db(db_path, "users")
|
|
results_df = fetch_data_from_db(db_path, "sync_results")
|
|
|
|
if users_df.empty:
|
|
st.write("No user data available in the database.")
|
|
else:
|
|
emails = users_df['email'].unique().tolist()
|
|
emails.insert(0, "All")
|
|
selected_email = st.selectbox("Filter by Email:", options=emails)
|
|
|
|
if selected_email == "All":
|
|
filtered_users_df = users_df
|
|
else:
|
|
filtered_users_df = users_df[users_df['email'] == selected_email]
|
|
|
|
st.dataframe(filtered_users_df)
|
|
|
|
# Display email transfer results if available
|
|
if not results_df.empty:
|
|
if selected_email == "All":
|
|
st.write("Email Transfer Results:")
|
|
st.dataframe(results_df)
|
|
else:
|
|
filtered_results_df = results_df[results_df['email'] == selected_email]
|
|
if not filtered_results_df.empty:
|
|
st.write(f"Email Transfer Results for {selected_email}:")
|
|
st.dataframe(filtered_results_df)
|
|
else:
|
|
st.write(f"No transfer data found for {selected_email}.")
|
|
else:
|
|
st.write("No email transfer results available.")
|
|
|
|
elif st.session_state.page == "Delete User":
|
|
st.header("Delete a User")
|
|
|
|
df = fetch_data_from_db(db_path, "users")
|
|
|
|
if df.empty:
|
|
st.write("No data available in the database.")
|
|
else:
|
|
user_to_delete = st.selectbox("Select User to Delete:", df['email'].unique())
|
|
|
|
if st.button("Delete User", key="confirm_delete_user"):
|
|
delete_user_from_db(db_path, user_to_delete)
|
|
st.success(f"User {user_to_delete} deleted successfully!")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|