def __init__(self, worker_name, config):
super().__init__()
self.config = config['workers'].get(worker_name)
# Initialize view controller
self.controller = WorkerDetailsController(self, worker_name,
self.config)
self.setupUi(self)
# Add worker's name to the dialog title
self.setWindowTitle("DEXBot - {} details".format(worker_name))
# Get strategy class from the config
strategy_class = getattr(
importlib.import_module(self.config.get('module')), 'Strategy')
# DetailElements that are used to configure worker's detail view
details = strategy_class.configure_details()
# Initialize other data to the dialog
self.controller.initialize_worker_data()
# Add custom tabs
self.add_tabs(details)
# Dialog controls
self.button_box.rejected.connect(self.reject)
class WorkerDetailsView(QtWidgets.QDialog, Ui_details_dialog, Ui_Graph_Tab,
Ui_Table_Tab, Ui_Text_Tab):
def __init__(self, worker_name, config):
super().__init__()
self.config = config['workers'].get(worker_name)
# Initialize view controller
self.controller = WorkerDetailsController(self, worker_name,
self.config)
self.setupUi(self)
# Add worker's name to the dialog title
self.setWindowTitle("DEXBot - {} details".format(worker_name))
# Get strategy class from the config
strategy_class = getattr(
importlib.import_module(self.config.get('module')), 'Strategy')
# DetailElements that are used to configure worker's detail view
details = strategy_class.configure_details()
# Initialize other data to the dialog
self.controller.initialize_worker_data()
# Add custom tabs
self.add_tabs(details)
# Dialog controls
self.button_box.rejected.connect(self.reject)
def add_tabs(self, details):
# Add tabs to the details view
for detail in details:
widget = QWidget(self)
if detail.type == 'graph':
widget = self.add_graph_tab(detail, widget)
elif detail.type == 'table':
widget = self.add_table_tab(detail, widget)
elif detail.type == 'text':
widget = self.add_text_tab(detail, widget)
self.tabs_widget.addTab(widget, detail.name)
def add_graph_tab(self, detail, widget):
tab = Ui_Graph_Tab()
tab.setupUi(widget)
tab.graph_wrap.setTitle(detail.title)
if detail.file:
file = os.path.join(get_user_data_directory(), 'graphs',
detail.file)
tab.table = self.controller.add_graph(tab, file)
return widget
def add_table_tab(self, detail, widget):
tab = Ui_Table_Tab()
tab.setupUi(widget)
tab.table_wrap.setTitle(detail.title)
if detail.file:
file = os.path.join(get_user_data_directory(), 'data', detail.file)
tab.table = self.controller.populate_table_from_csv(tab, file)
return widget
def add_text_tab(self, detail, widget):
tab = Ui_Text_Tab()
tab.setupUi(widget)
tab.text_wrap.setTitle(detail.title)
if detail.file:
file = os.path.join(get_user_data_directory(), 'logs', detail.file)
tab.text = self.controller.populate_text_from_file(tab, file)
return widget