57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
import streamlit as st
|
|
import os
|
|
import sys
|
|
import importlib.util
|
|
|
|
# Parse command-line arguments.
|
|
if len(sys.argv) > 1:
|
|
folder = os.path.abspath(sys.argv[1])
|
|
else:
|
|
folder = os.path.abspath(os.getcwd())
|
|
|
|
# Get filenames for all files in this path, excluding this script.
|
|
|
|
this_file = os.path.abspath(__file__)
|
|
fnames = []
|
|
|
|
for basename in os.listdir(folder):
|
|
fname = os.path.join(folder, basename)
|
|
|
|
if fname.endswith(".py") and fname != this_file:
|
|
fnames.append(fname)
|
|
|
|
# Make a UI to run different files.
|
|
def format_func(s):
|
|
els = s.split("/")[-1].split(".")[0].split("_")
|
|
return " ".join(el for el in els).capitalize()
|
|
|
|
|
|
fname_to_run = st.sidebar.selectbox(
|
|
"Select an app", fnames, format_func=format_func
|
|
)
|
|
|
|
# Create module from filepath and put in sys.modules, so Streamlit knows
|
|
# to watch it for changes.
|
|
|
|
fake_module_count = 0
|
|
|
|
|
|
def load_module(filepath):
|
|
global fake_module_count
|
|
|
|
modulename = "_dont_care_%s" % fake_module_count
|
|
spec = importlib.util.spec_from_file_location(modulename, filepath)
|
|
module = importlib.util.module_from_spec(spec)
|
|
sys.modules[modulename] = module
|
|
|
|
fake_module_count += 1
|
|
|
|
|
|
# Run the selected file.
|
|
|
|
with open(fname_to_run) as f:
|
|
load_module(fname_to_run)
|
|
filebody = f.read()
|
|
|
|
exec(filebody, {})
|