forked from msei99/pbgui
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pbgui.py
90 lines (79 loc) · 3.5 KB
/
pbgui.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import streamlit as st
import configparser
import os
import inspect
import platform
from pbgui_func import check_password, set_page_config
from Services import Services
from Instance import Instances
from Multi import MultiInstances
from User import Users
def update_dir(key):
choice = st.session_state[key]
if os.path.isdir(os.path.join(st.session_state[key+'curr_dir'], choice)):
st.session_state[key+'curr_dir'] = os.path.normpath(os.path.join(st.session_state[key+'curr_dir'], choice))
files = sorted(os.listdir(st.session_state[key+'curr_dir']))
files.insert(0, '..')
files.insert(0, '.')
st.session_state[key+'files'] = files
def st_file_selector(st_placeholder, path='.', label='Select a file/folder', key = 'selected'):
if key+'curr_dir' not in st.session_state:
base_path = '.' if path is None or path == '' else path
base_path = base_path if os.path.isdir(base_path) else os.path.dirname(base_path)
base_path = '.' if base_path is None or base_path == '' else base_path
files = sorted(os.listdir(base_path))
files.insert(0, '..')
files.insert(0, '.')
st.session_state[key+'files'] = files
st.session_state[key+'curr_dir'] = base_path
else:
base_path = st.session_state[key+'curr_dir']
selected_file = st_placeholder.selectbox(label=label,
options=st.session_state[key+'files'],
key=key,
on_change = lambda: update_dir(key))
selected_path = os.path.normpath(os.path.join(base_path, selected_file))
st_placeholder.write(os.path.abspath(selected_path))
return selected_path
set_page_config()
# Password Check
if not check_password():
st.stop()
st.header("Passivbot GUI")
st.session_state.pbgdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
os.chdir(st.session_state.pbgdir)
# Load ini and initialize session state
pb_config = configparser.ConfigParser()
pb_config.read('pbgui.ini')
if pb_config.has_option("main", "pbdir"):
st.session_state.pbdir = pb_config.get("main", "pbdir")
else:
st.session_state.pbdir = '.'
if pb_config.has_option("main", "pbname"):
st.session_state.pbname = pb_config.get("main", "pbname")
else:
st.session_state.pbname = platform.node()
st.session_state.pbdir = os.path.abspath(st_file_selector(st, path=st.session_state.pbdir, key = 'pbdir_selected', label = 'Choose passivbot directory'))
st.session_state.pbname = st.text_input("Bot Name", value=st.session_state.pbname, max_chars=32)
if not pb_config.has_section("main"):
pb_config.add_section("main")
pb_config.set("main", "pbdir", os.path.abspath(st.session_state.pbdir))
pb_config.set("main", "pbname", st.session_state.pbname)
with open('pbgui.ini', 'w') as pbgui_configfile:
pb_config.write(pbgui_configfile)
# Init Services
if 'services' not in st.session_state:
with st.spinner('Initializing Services...'):
st.session_state.services = Services()
# Init Instances
if 'pbgui_instances' not in st.session_state:
with st.spinner('Initializing Instances...'):
st.session_state.pbgui_instances = Instances()
# Init Multi Instances
if 'multi_instances' not in st.session_state:
with st.spinner('Initializing Multi Instances...'):
st.session_state.multi_instances = MultiInstances()
# Init Users
if 'users' not in st.session_state:
with st.spinner('Initializing Users...'):
st.session_state.users = Users()