class StrategyShareToplevel(Parent): """ Toplevel to display a list of strategy with checkboxes to allow selecting which should be shared. """ def __init__(self, master, client, database, strategy_frame, **kwargs): """ :param master: master widget :param client: network.strategy.client.StrategyClient :param database: results.strategies.StrategyDataBase :param strategy_frame: frames.strategy.StrategyFrame :param kwargs: SnapToplevel keyword arguments """ resizable = kwargs.pop("resizable", False) self._client = client self._database = database self._frame = strategy_frame Parent.__init__(self, master, **kwargs) self.wm_title("GSF Parser: Strategy Sharing") self.wm_resizable(resizable, resizable) # Configure the Treeview self.tree = CheckboxTreeview(master=self, height=16) self.tree.column("#0", width=200) self.tree.heading("#0", text="Strategy") self.tree.config(show=("headings", "tree")) self.scroll_bar = ttk.Scrollbar(self, orient=tk.VERTICAL, command=self.tree.yview) self.tree.config(yscrollcommand=self.scroll_bar.set) self.update_strategy_tree() self.share_button = ttk.Button(self, text="Share Strategies", command=self.share_strategies) self.grid_widgets() def grid_widgets(self): self.tree.grid(row=1, column=1, sticky="nswe", padx=5, pady=5) self.scroll_bar.grid(row=1, column=2, sticky="ns", padx=(0, 5), pady=5) self.share_button.grid(row=2, column=1, columnspan=2, sticky="nswe", padx=5, pady=(0, 5)) def share_strategies(self): for strategy in self.strategies_to_share: self._client.send_strategy(strategy) messagebox.showinfo("Info", "Selected Strategies sent.") def update_strategy_tree(self): self.tree.delete(*self.tree.get_children("")) for strategy in sorted(self._database.keys()): self.tree.insert("", tk.END, iid=strategy, text=strategy) @property def strategies_to_share(self): for strategy in self.tree.get_checked(): yield self._database[strategy]
class LeftPanel: def __init__(self, root, frame): self.root = root self.frame = Frame(frame) # Create the toolbar as a frame self.frame.columnconfigure(0, weight=1) self.frame.columnconfigure(1, weight=1) self.frame.rowconfigure(0, weight=1) self.frame.rowconfigure(1, weight=1) self.frame.grid_rowconfigure(1, minsize=70) self.img1 = Image.open("icons/open-folder.png").convert('RGBA') self.useImg1 = ImageTk.PhotoImage(self.img1) self.path = os.path.dirname(__file__) #self.style = ttk.Style() #self.style.configure('Treeview', rowheight=40) self.tree = CheckboxTreeview(self.frame) #self.tree.pack(expand=YES, fill=BOTH) self.tree.heading("#0", text="Ficheiros") self.dir = self.tree.insert('', 'end', text=self.path, open=True) self.Subs(self.path, self.dir) self.tree.grid(row=0, column=0, columnspan=2, padx=4, pady=4, sticky=ALL) Button(self.frame, image=self.useImg1, width=70, height=70, command=self.browse_button).grid(row=2, column=1, padx=4, pady=4, sticky=E) self.frame.pack(side=TOP, fill=BOTH) def Subs(self, path, parent): for p in os.listdir(path): abspath = os.path.join(path, p) parent_element = self.tree.insert(parent, 'end', text=p, open=True) if os.path.isdir(abspath): self.Subs(abspath, parent_element) def browse_button(self): self.path = tkFileDialog.askdirectory() def set_dir(self): sourcePath = str(self.path) os.chdir(sourcePath)
class TreeviewConstructor(object): logger = CustomAdapter(logging.getLogger(str(__name__)), None) @debug(lvl=logging.NOTSET, prefix='') def __init__(self, master, frame_main, checkwidth=0): self.master = master # Done self.frame = frame_main # Done self.col_obj_dict = {} # Done self.column_lst = [] # Done self.col_disp_lst = [] # Done self.item_obj_dict = {} self.treeview = CheckboxTreeview(self.frame) # Done self.ysb = ttk.Scrollbar(self.frame) # Done self.xsb = ttk.Scrollbar(self.frame) # Done self.checkwidth = checkwidth self.manager = BusyManager(self.frame) self.populate_frame() # Done @debug(lvl=logging.NOTSET, prefix='') def populate_frame(self): self.frame.columnconfigure(0, weight=1) # Done self.frame.rowconfigure(0, weight=1) # Done self.treeview.grid(row=0, column=0, sticky=tk.NSEW) # Done self.ysb.config(orient=tk.VERTICAL, command=self.treeview.yview) # Done self.xsb.config(orient=tk.HORIZONTAL, command=self.treeview.xview) # Done self.treeview['yscroll'] = self.ysb.set # Done self.treeview['xscroll'] = self.xsb.set # Done self.ysb.grid(row=0, column=1, sticky=tk.NS) # Done self.xsb.grid(row=1, column=0, sticky=tk.EW) # Done @debug(lvl=logging.NOTSET, prefix='') def populate_cols(self): col_obj = TreeColumn(order=0, col_id='#0', hdr_txt="", anchor=tk.W, stretch=tk.NO, minwidth=0, width=self.checkwidth, display=False) TreeviewConstructor.logger.log(logging.NOTSET, "Column ID: {col_id} Header Text: {hdr_txt}". format(col_id=col_obj.col_id, hdr_txt=col_obj.hdr_txt)) self.col_obj_dict[col_obj.col_id] = col_obj for col_key, column in sorted(self.col_obj_dict.items(), key=lambda x: x[1].order): self.column_lst.append(column.col_id) if column.display: self.col_disp_lst.append(column.col_id) self.treeview.config(columns=self.column_lst, displaycolumns=self.col_disp_lst) self.treeview.tag_configure('red', foreground='red2') self.treeview.tag_configure('evenrow', background='gray85') self.treeview.tag_configure('oddrow', background='white') for col in self.col_obj_dict.values(): TreeviewConstructor.logger.log( logging.NOTSET, "Column ID: {col_id} Header Text: {hdr_txt} Anchor: {anchor}".format( col_id=col.col_id, hdr_txt=col.hdr_txt, anchor=col.anchor)) if col.hdr_txt in (None, ""): header = col.col_id else: header = col.hdr_txt self.treeview.heading(column=col.col_id, text=header, anchor=col.anchor, command=lambda _col=col.col_id: self.treeview_sort_column(_col, False)) self.treeview.column(col.col_id, minwidth=col.minwidth, width=col.width, stretch=col.stretch) @debug(lvl=logging.DEBUG, prefix='') def populate_items(self): for item in self.item_obj_dict.values(): TreeRow.logger.log(logging.NOTSET, "Item Info: {0}".format(item.iid)) self.treeview.insert(item.parent, item.index, iid=item.iid, values=item.values_list) for tag in item.tags_list: self.treeview.tag_add(item.iid, tag) @debug(lvl=logging.DEBUG, prefix='') def stripe_rows(self): for item in self.item_obj_dict.values(): self.treeview.tag_del(item.iid, 'evenrow') self.treeview.tag_del(item.iid, 'oddrow') row_num = self.treeview.index(item.iid) if row_num % 2 == 0: self.treeview.tag_add(item.iid, 'evenrow') elif row_num % 2 != 0: self.treeview.tag_add(item.iid, 'oddrow') @debug(lvl=logging.DEBUG, prefix='') def treeview_sort_column(self, col, reverse): self.manager.busy() item_list = [(self.treeview.set(k, col), k) for k in self.treeview.get_children('')] item_list.sort(reverse=reverse) # rearrange items in sorted positions for index, (val, k) in enumerate(item_list): self.treeview.move(k, '', index) self.stripe_rows() # reverse sort next time self.treeview.heading(col, command=lambda: self.treeview_sort_column(col, not reverse)) self.manager.not_busy() @debug(lvl=logging.DEBUG, prefix='') def tv_refresh(self): # self.manager.busy() item_list = list(self.treeview.get_children('')) item_list.sort(key=lambda x: int(x)) for item_iid, item_obj in sorted(self.item_obj_dict.items()): TreeRow.logger.log(logging.NOTSET, "Item ID: {0}".format(item_iid)) # l = self.treeview.get_children('') for index, k in enumerate(item_list): TreeRow.logger.log(logging.NOTSET, "k: {0}".format(k)) if str(item_iid) == str(k): self.treeview.delete(k) item_obj.index = index self.treeview.insert(item_obj.parent, item_obj.index, iid=item_obj.iid, values=item_obj.values_list) item_list.remove(k) break self.stripe_rows()
class TreeviewConstructor(ttk.Frame): logger = CustomAdapter(logging.getLogger(str(__name__)), None) @debug(lvl=logging.DEBUG, prefix='') def __init__(self, master, *args, **kwargs): self.col_obj_dict = {} self.column_lst = [] self.col_disp_lst = [] self.item_obj_dict = {} self.master = master # noinspection PyArgumentList super().__init__(self.master, *args, **kwargs) self.columnconfigure(0, weight=1) self.rowconfigure(0, weight=1) self.treeview = CheckboxTreeview(self) TreeviewConstructor.logger.log(logging.DEBUG, "10") self.treeview.grid(row=0, column=0, sticky=tk.NSEW) TreeviewConstructor.logger.log(logging.DEBUG, "11") self.ysb = ttk.Scrollbar(self) TreeviewConstructor.logger.log(logging.DEBUG, "12") self.ysb.grid(row=0, column=1, sticky=tk.NS) TreeviewConstructor.logger.log(logging.DEBUG, "13") self.ysb.config(orient=tk.VERTICAL, command=self.treeview.yview) TreeviewConstructor.logger.log(logging.DEBUG, "14") self.xsb = ttk.Scrollbar(self) TreeviewConstructor.logger.log(logging.DEBUG, "15") self.xsb.grid(row=1, column=0, sticky=tk.EW) TreeviewConstructor.logger.log(logging.DEBUG, "16") self.xsb.config(orient=tk.HORIZONTAL, command=self.treeview.xview) TreeviewConstructor.logger.log(logging.DEBUG, "17") self.treeview['yscroll'] = self.ysb.set TreeviewConstructor.logger.log(logging.DEBUG, "18") self.treeview['xscroll'] = self.xsb.set TreeviewConstructor.logger.log(logging.DEBUG, "19") # self.manager = BusyManager(self.frame) @debug(lvl=logging.DEBUG, prefix='') def populate_cols(self, checkwidth): col_obj = TreeColumn(order=0, col_id='#0', hdr_txt="", anchor=tk.W, stretch=tk.NO, minwidth=0, width=checkwidth, display=False) TreeviewConstructor.logger.log( logging.DEBUG, "Column ID: {col_id} Header Text: {hdr_txt}".format( col_id=col_obj.col_id, hdr_txt=col_obj.hdr_txt)) self.col_obj_dict[col_obj.col_id] = col_obj for col_key, column in sorted(self.col_obj_dict.items(), key=lambda x: x[1].order): self.column_lst.append(column.col_id) if column.display: self.col_disp_lst.append(column.col_id) self.treeview.config(columns=self.column_lst, displaycolumns=self.col_disp_lst) self.treeview.tag_configure('red', foreground='red4', background='IndianRed1') self.treeview.tag_configure('parent_evenrow', background='gray85') self.treeview.tag_configure('child_evenrow', background="honeydew3") self.treeview.tag_configure('parent_oddrow', background='white') self.treeview.tag_configure('child_oddrow', background="honeydew2") for col in self.col_obj_dict.values(): TreeviewConstructor.logger.log( logging.DEBUG, "Column ID: {col_id} Header Text: {hdr_txt} Anchor: {anchor}". format(col_id=col.col_id, hdr_txt=col.hdr_txt, anchor=col.anchor)) if col.hdr_txt in (None, ""): header = col.col_id else: header = col.hdr_txt if col.col_id == '#0': self.treeview.heading(column=col.col_id, text="", anchor=col.anchor, command=lambda _col=col.col_id: self. treeview_sort_column(_col, False)) else: self.treeview.heading(column=col.col_id, text=header, anchor=col.anchor, command=lambda _col=col.col_id: self. treeview_sort_column(_col, False)) self.treeview.column(col.col_id, minwidth=col.minwidth, width=col.width, stretch=col.stretch) @debug(lvl=logging.DEBUG, prefix='') def populate_items(self): for item in self.item_obj_dict.values(): TreeviewConstructor.logger.log(logging.NOTSET, "Item Info: {0}".format(item.iid)) self.treeview.insert(item.parent, item.index, iid=item.iid, values=item.values_list) for tag in item.tags_list: self.treeview.tag_add(item.iid, tag) @debug(lvl=logging.DEBUG, prefix='') def stripe_rows(self): for item in self.item_obj_dict.values(): self.treeview.tag_del(item.iid, 'parent_evenrow') self.treeview.tag_del(item.iid, 'child_evenrow') self.treeview.tag_del(item.iid, 'parent_oddrow') self.treeview.tag_del(item.iid, 'child_oddrow') row_num = self.treeview.index(item.iid) if row_num % 2 == 0: if item.parent == "": self.treeview.tag_add(item.iid, 'child_evenrow') else: self.treeview.tag_add(item.iid, 'parent_evenrow') elif row_num % 2 != 0: if item.parent == "": self.treeview.tag_add(item.iid, 'child_oddrow') else: self.treeview.tag_add(item.iid, 'parent_oddrow') @debug(lvl=logging.DEBUG, prefix='') def treeview_sort_column(self, col, reverse): # self.manager.busy() item_list = [(self.treeview.set(k, col), k) for k in self.treeview.get_children('')] item_list.sort(reverse=reverse) # rearrange items in sorted positions for index, (val, k) in enumerate(item_list): self.treeview.move(k, '', index) self.stripe_rows() # reverse sort next time self.treeview.heading( col, command=lambda: self.treeview_sort_column(col, not reverse)) # self.manager.not_busy() @debug(lvl=logging.DEBUG, prefix='') def tv_refresh(self): # self.manager.busy() item_list = list(self.treeview.get_children('')) item_list.sort(key=lambda x: int(x)) for item_iid, item_obj in sorted(self.item_obj_dict.items()): TreeviewConstructor.logger.log(logging.DEBUG, "Item ID: {0}".format(item_iid)) # l = self.treeview.get_children('') for index, k in enumerate(item_list): TreeviewConstructor.logger.log(logging.DEBUG, "k: {0}".format(k)) if str(item_iid) == str(k): self.treeview.delete(k) item_obj.index = index self.treeview.insert(item_obj.parent, item_obj.index, iid=item_obj.iid, values=item_obj.values_list) item_list.remove(k) break self.stripe_rows() # self.manager.not_busy() @debug(lvl=logging.DEBUG, prefix='') def add_column(self, order=None, col_id=None, hdr_txt="", anchor=tk.W, stretch=tk.YES, minwidth=0, width=50, display=True, desc=None): col_obj = TreeColumn(order=order, col_id=col_id, hdr_txt=hdr_txt, anchor=anchor, stretch=stretch, minwidth=minwidth, width=width, display=display, desc=desc) self.col_obj_dict[col_obj.col_id] = col_obj @debug(lvl=logging.NOTSET, prefix='') def add_item(self, iid, parent="", index=tk.END, values_dict=None): item_obj = TreeRow(treeview_const=self, iid=iid, parent=parent, index=index, values_dict=values_dict) self.item_obj_dict[item_obj.iid] = item_obj @debug(lvl=logging.DEBUG, prefix='') def columns_from_query(self, query, hide_list=None, pref_order=None): if pref_order is None: pref_order = [] if hide_list is None: hide_list = [] col_order = 1 col_order += len(pref_order) TreeviewConstructor.logger.log( logging.DEBUG, "Column Order Starting Value: {0}".format(str(col_order))) for desc in query.column_descriptions: name = desc.get('name').replace("'", "").replace('"', "") TreeviewConstructor.logger.log(logging.DEBUG, "Column Name: {0}".format(name)) disp = True if name in hide_list: disp = False cust_order = col_order if name in pref_order: cust_order = pref_order.index(name) + 1 self.add_column(order=cust_order, col_id=name.replace(" ", "_"), hdr_txt=name, display=disp, desc=desc) if name not in pref_order: col_order += 1 @debug(lvl=logging.DEBUG, prefix='') def rows_from_query(self, query, id_col="line_number", limit=None, parent_col=None): counter = 1 for row in query.all(): temp_dict = {} for col_obj in self.col_obj_dict.values(): # noinspection PyProtectedMember value = row._asdict().get(col_obj.hdr_txt) if isinstance(value, str): value.replace("{", "").replace("}", "") temp_dict[str(col_obj.order)] = value counter += 1 if counter == limit: break parent = "" if parent_col is not None: # noinspection PyProtectedMember if str(row._asdict().get(id_col)) != str( row._asdict().get(parent_col)): # noinspection PyProtectedMember parent = str(row._asdict().get(parent_col)) # noinspection PyProtectedMember self.add_item(iid=str(row._asdict().get(id_col)), values_dict=temp_dict, parent=parent) @debug(lvl=logging.DEBUG, prefix='') def populate_query(self, query, hide_list=None, pref_order=None, id_col="line_number", limit=None, checkwidth=0, parent_col=None): the_query = query self.columns_from_query(the_query, hide_list, pref_order) self.populate_cols(checkwidth=checkwidth) self.rows_from_query(the_query, id_col, limit, parent_col) self.populate_items() self.stripe_rows()
class ScriptManager: """ Open the scripts window manager """ folder_icon = None def __init__(self, parent): """ parent: the tkinter parent view to use for this window construction. """ self.parent = parent self.app = tk.Toplevel(parent) self.app.title("Scripts Manager") self.app.resizable(True, True) self.rvalue = None appFrame = ttk.Frame(self.app) #PANED PART self.paned = tk.PanedWindow(appFrame, height=300) #RIGHT PANE : TAble self.viewframe = ttk.Frame(self.paned) self.file_tree = CheckboxTreeview(self.viewframe) self.file_tree['columns'] = ('name', 'category') self.file_tree.heading('#0', text='Name') self.file_tree.column("#0", stretch=tk.YES, minwidth=300, width=300) self.file_tree.heading('#1', text='Category') self.file_tree.column("#1", stretch=tk.YES, minwidth=300, width=300) self.file_tree.pack(fill=tk.BOTH, expand=True) btn_pane = ttk.Frame(self.viewframe) self.execute_icone = tk.PhotoImage(file = Utils.getIcon("execute.png")) btn_execute = ttk.Button(btn_pane, text="Execute", image=self.execute_icone, command=self.executedSelectedScripts, tooltip="Execute all selected scripts", style="Toolbutton") btn_execute.pack(side=tk.RIGHT, padx=3, pady=5) self.open_folder_icone = tk.PhotoImage(file = Utils.getIcon("folder.png")) btn_openPathForUser = ttk.Button(btn_pane, text="Execute", image=self.open_folder_icone, command=self.openPathForUser, tooltip="Open scripts folder", style="Toolbutton") btn_openPathForUser.pack(side=tk.RIGHT, padx=3, pady=5) btn_pane.pack(fill=tk.X, side=tk.BOTTOM, anchor=tk.E) #LEFT PANE : Treeview self.frameTw = ttk.Frame(self.paned) self.treevw = ttk.Treeview(self.frameTw) self.treevw.pack() scbVSel = ttk.Scrollbar(self.frameTw, orient=tk.VERTICAL, command=self.treevw.yview) self.treevw.configure(yscrollcommand=scbVSel.set) self.treevw.grid(row=0, column=0, sticky=tk.NSEW) scbVSel.grid(row=0, column=1, sticky=tk.NS) self.treevw.grid(row=0, column=0, sticky=tk.NSEW) scbVSel.grid(row=0, column=1, sticky=tk.NS) self.paned.add(self.frameTw) self.paned.add(self.viewframe) self.paned.pack(fill=tk.BOTH, expand=1) self.frameTw.rowconfigure(0, weight=1) # Weight 1 sur un layout grid, sans ça le composant ne changera pas de taille en cas de resize self.frameTw.columnconfigure(0, weight=1) # Weight 1 sur un layout grid, sans ça le composant ne changera pas de taille en cas de resize appFrame.pack(fill=tk.BOTH, ipady=10, ipadx=10, expand=True) self.treevw.bind("<<TreeviewSelect>>", self.onTreeviewSelect) try: self.app.wait_visibility() self.app.focus_force() self.app.lift() except tk.TclError: pass self.refreshUI() def refreshUI(self): for widget in self.treevw.winfo_children(): widget.destroy() script_dir = self.getScriptsDir() if self.__class__.folder_icon is None: self.__class__.folder_icon = ImageTk.PhotoImage(Image.open(Utils.getIcon("folder.png"))) parent = self.treevw.insert("", "end", " ", text="Scripts", image=self.__class__.folder_icon, open=True) self.treevw.focus(parent) self.treevw.selection_set(parent) for root, subFolders, files in os.walk(script_dir): root_name = root.replace(script_dir, "") for folder in subFolders: if folder.startswith("__") or folder.endswith("__"): continue folder_iid = os.path.join(root_name, folder) parent_node = parent if root_name == "" else root_name self.treevw.insert(parent_node, "end", folder_iid, text=folder, image=self.__class__.folder_icon) self.openScriptFolderView() def getScriptsDir(self): return os.path.join(Utils.getMainDir(), "scripts/") def onTreeviewSelect(self, _event=None): selec = self.treevw.selection() if len(selec) == 0: return None item = selec[0] self.openScriptFolderView(str(item)) def openScriptFolderView(self, script_folder=""): full_script_path = os.path.join(self.getScriptsDir(), script_folder.strip()) script_shown = set() self.file_tree.delete(*self.file_tree.get_children()) for root, _, files in os.walk(full_script_path): for file in files: filepath = root + '/' + file if file.endswith(".py"): script_shown.add(filepath) scripts_list = sorted(script_shown) script_dir = self.getScriptsDir() for script in scripts_list: scriptName = os.path.basename(script) category_name = os.path.dirname(script.replace(script_dir, "")) self.file_tree.insert("", "end", script, text=scriptName, values=(category_name)) def executedSelectedScripts(self): for selected in self.file_tree.get_checked(): self.executeScript(selected) def executeScript(self, script_path): script_dir = self.getScriptsDir() category_name = os.path.dirname(script_path.replace(script_dir, "")) script_name = ".".join(os.path.splitext(os.path.basename(script_path))[:-1]) module = os.path.join("pollenisatorgui/scripts/",category_name, script_name).replace("/", '.') imported = importlib.import_module(module) success, res = imported.main(APIClient.getInstance()) if success: tk.messagebox.showinfo("Script finished", f"Script {script_name} finished.\n{res}") else: tk.messagebox.showwarning("Script failed", f"Script {script_name} failed.\n{res}") def openPathForUser(self): selection = self.treevw.selection() if selection: folder = os.path.join(self.getScriptsDir(), selection[0]) else: folder = self.getScriptsDir() Utils.openPathForUser(folder)
class main_Tab: def __init__(self, master, main_path=''): self.main_path = main_path self.master = master self.modules_config = {} #{module_name : } self.modules = {} self.fio = '' self.org = '' #Профили аудита self.profiles = {} self.current_profile_options = {} #Левый фрейм включает данные об аудиторе и о существующих профилях leftFrame = ttk.Frame(master) self.auditor_Frame = ttk.LabelFrame(leftFrame, text='Данные аудитора') self.profiles_Frame = VerticalScrolledFrame(leftFrame, text='Профили аудита') #Правый фрейм включает редактор текущего профиля и кнопку запуска аудита rightFrame = ttk.Frame(master) self.current_Frame = ttk.LabelFrame(rightFrame, text='Текущий профиль') self.btn_Frame = ttk.Frame(rightFrame) #Область данных об аудиторе fio_Label = ttk.Label(self.auditor_Frame, text='ФИО') self.fio_Entry = ttk.Entry(self.auditor_Frame, font=16, width=30) self.fio_Label = ttk.Label(self.auditor_Frame, font=16, width=30) org_Label = ttk.Label(self.auditor_Frame, text='Организация') self.org_Entry = ttk.Entry(self.auditor_Frame, font=16, width=30) self.org_Label = ttk.Label(self.auditor_Frame, font=16, width=30) self.fio_OK_btn = ttk.Button(self.auditor_Frame, text='OK', width=5, command=self.accept_auditor) fio_Label.grid(row=1, column=0, sticky='new', padx=10) self.fio_Entry.grid(row=2, column=0, sticky='nsew', padx=10) org_Label.grid(row=3, column=0, sticky='new', padx=10) self.org_Entry.grid(row=4, column=0, sticky='nsew', padx=10) self.fio_OK_btn.grid(row=5, column=0, sticky='se', padx=10) self.profileView = CheckboxTreeview(self.current_Frame) self.save_audit_btn = ttk.Button(self.btn_Frame, text='Сохранить профиль', command=self.save_btn_click) self.run_audit_btn = ttk.Button(self.btn_Frame, text='Запустить аудит', command=self.run_audit) self.ysb = ttk.Scrollbar(self.current_Frame, orient='vertical', command=self.profileView.yview) self.profileView.configure(yscroll=self.ysb.set) ttk.Style().configure('Vertical.TScrollbar', troughcolor='#f6f4f2', relief=tk.GROOVE) ttk.Style().configure('Treeview', background="#ffffff") ttk.Style().configure('Frame', background="#f6f4f2") #Размещения на фреймах leftFrame.pack(side=tk.LEFT, anchor='nw', fill=tk.Y) rightFrame.pack(side=tk.LEFT, anchor='nw', expand=1, fill=tk.BOTH) self.auditor_Frame.pack(side=tk.TOP, anchor='nw', padx=5, pady=5, fill=tk.X) self.profiles_Frame.pack(side=tk.TOP, anchor='sw', padx=5, pady=10, fill=tk.BOTH, expand=1) self.current_Frame.pack(side=tk.TOP, anchor='nw', padx=5, pady=5, fill=tk.BOTH, expand=1) self.btn_Frame.pack(side=tk.TOP, anchor='nw', fill=tk.X) self.ysb.pack(side=tk.RIGHT, anchor='n', fill=tk.Y) self.profileView.pack(side=tk.TOP, anchor='nw', fill=tk.BOTH, expand=1) self.save_audit_btn.pack(side=tk.LEFT, anchor='sw', padx=5, pady=5) self.run_audit_btn.pack(side=tk.LEFT, anchor='se', padx=5, pady=5, fill=tk.X, expand=1) self.auditor_Frame.grid_rowconfigure(2, minsize=30) self.loadProfiles() self.profileView.bind("<Button-1>", self.check_uncheck_item, True) def run_audit(self): for tab_name, tab in self.modules.items(): tab.run_audit() def sync_vars(self): for tab_name, tab in self.modules.items(): if type(self.current_profile_options[tab_name][1]) != type({}): i = 0 for var in tab.vars: var.set(self.current_profile_options[tab_name][1][i]) i += 1 else: i = len(tab.vars) - 1 for second_tab_name, second_tab in self.current_profile_options[ tab_name][1].items(): for j in range(len(second_tab[1]) - 1, -1, -1): tab.vars[i].set(second_tab[1][j]) i -= 1 def accept_auditor(self): if self.fio_OK_btn.cget('text') == 'OK': self.fio = self.fio_Entry.get() self.org = self.org_Entry.get() self.fio_OK_btn.configure(text='Изменить', width=10) self.fio_Entry.grid_remove() self.org_Entry.grid_remove() self.fio_Label.configure(text=self.fio) self.org_Label.configure(text=self.org) self.fio_Label.grid(row=2, column=0, sticky='nsew', padx=10) self.org_Label.grid(row=4, column=0, sticky='nsew', padx=10) else: self.fio_OK_btn.configure(text='OK', width=5) self.fio_Entry.grid() self.org_Entry.grid() self.fio_Label.grid_remove() self.org_Label.grid_remove() def loadProfiles(self): for child in self.profiles_Frame.interior.winfo_children(): child.destroy() #self.profiles_Frame.pack(side=tk.TOP, anchor='sw', padx=5, pady = 10, fill=tk.BOTH, expand=1) try: with open(self.main_path + '/profiles.json') as file: self.profiles = json.load(file) except: with open(self.main_path + '/profiles.json', 'w') as file: json.dump(self.profiles, file) prof_count = len(self.profiles) self.var = tk.IntVar() for i in range(prof_count): tk.Radiobutton(self.profiles_Frame.interior, variable=self.var, value=i, indicator=0, height=3, text=list(self.profiles.keys())[i], selectcolor="#f19572", activebackground="#f19572", command=self.changeCurrentProfile).pack(side=tk.TOP, anchor='nw', fill=tk.X, padx=5, pady=2) def dumpProfiles(self): with open(self.main_path + '/profiles.json', 'w') as file: json.dump(self.profiles, file) def changeCurrentProfile(self): currentProfileName = list(self.profiles.keys())[self.var.get()] self.current_profile_options = copy.deepcopy( self.profiles[currentProfileName]) self.profileView.heading('#0', text=list( self.profiles.keys())[self.var.get()], anchor='w') self.initTree() self.sync_vars() #self.profiles_Label.configure(text=self.var.get()) def save_btn_click(self): self.saveDialog = tk.Toplevel() save_frame = ttk.Frame(self.saveDialog) self.saveDialog.focus_set() width = self.master.winfo_screenwidth() // 5 + 96 height = self.master.winfo_screenheight() // 5 dw = (self.master.winfo_screenwidth() - width) // 2 dh = (self.master.winfo_screenheight() - height) // 2 self.saveDialog.geometry('{}x{}+{}+{}'.format(width, height, dw, dh)) self.saveDialog.resizable(False, False) self.saveDialog.title('Сохранение профиля') # Настройка виджетов в окне ввода пароля save_Label = ttk.Label(save_frame, text='Названия профиля') self.save_Entry = ttk.Entry(save_frame, font=16) save_Btn = ttk.Button(save_frame, text='Сохранить', width=15) self.save_Entry.bind('<Return>', self.save_click) save_Btn.bind('<Button-1>', self.save_click) save_Label.grid(row=1, column=0) self.save_Entry.grid(row=2, column=0, padx=10, sticky='nsew') save_Btn.grid(row=3, column=0, padx=10, pady=20, sticky='e') save_frame.pack(fill=tk.BOTH) save_frame.grid_rowconfigure(1, minsize=height // 3) save_frame.grid_rowconfigure(2, minsize=30) save_frame.grid_columnconfigure(0, minsize=width) self.save_Entry.focus_set() def save_click(self, event): """Функция-обработчик нажатия кнопки """ profile_name = self.save_Entry.get() self.profiles[profile_name] = self.current_profile_options self.dumpProfiles() self.loadProfiles() self.saveDialog.destroy() self.initTree() def initTree(self): if self.current_profile_options == {}: self.current_profile_options = dict( self.profiles["Профиль по умолчанию"]) for (key, value) in self.current_profile_options.items(): if self.modules_config.__contains__(key): if not self.profileView.exists(key): self.profileView.insert("", "end", key, text=key) self.profileView.change_state(key, value[0]) value = value[1] if type(value) == type("1"): i = 0 for func_key in self.modules_config[key][0].keys(): func_key = func_key.replace('\n', ' ') if not self.profileView.exists(key + func_key + "_" + str(i)): self.profileView.insert(key, "end", key + func_key + "_" + str(i), text=func_key) if value[i] == '0': self.profileView.change_state( key + func_key + "_" + str(i), 'unchecked') if value[i] == '1': self.profileView.change_state( key + func_key + "_" + str(i), 'checked') i += 1 else: j = 0 for (second_key, second_value) in dict(value).items(): if not self.profileView.exists(second_key): self.profileView.insert(key, "end", second_key, text=second_key) self.profileView.change_state(second_key, second_value[0]) second_value = second_value[1] if type(value[second_key][1]) == type("1"): i = 0 for func_key in self.modules_config[key][0][ j].keys(): func_key = func_key.replace('\n', ' ') if not self.profileView.exists(second_key + func_key + "_" + str(i)): self.profileView.insert( second_key, "end", second_key + func_key + "_" + str(i), text=func_key) if value[second_key][1][i] == '0': self.profileView.change_state( second_key + func_key + "_" + str(i), 'unchecked') if value[second_key][1][i] == '1': self.profileView.change_state( second_key + func_key + "_" + str(i), 'checked') i += 1 j += 1 def check_uncheck_item(self, event): x, y, widget = event.x, event.y, event.widget elem = widget.identify("element", x, y) if "image" in elem: # a box was clicked item = self.profileView.identify_row(y) children = self.profileView.get_children(item) parents = [] parent = widget.parent(item) i = 0 while parent != '': parents.append(parent) parent = widget.parent(parent) if parents and children == (): tag = self.profileView.item(parents[-1], "tags") self.current_profile_options[parents[-1]][0] = tag[0] if len(parents) == 2: tag = self.profileView.item(parents[0], "tags") self.current_profile_options[parents[-1]][1][ parents[-2]][0] = tag[0] tag = self.profileView.item(item, "tags") i = int(item.split('_')[1]) varL = self.current_profile_options[parents[-1]][1][ parents[-2]][1][0:i] varR = self.current_profile_options[parents[-1]][1][ parents[-2]][1][i + 1:] if "checked" in tag: self.current_profile_options[parents[-1]][1][ parents[-2]][1] = varL + '1' + varR else: self.current_profile_options[parents[-1]][1][ parents[-2]][1] = varL + '0' + varR else: tag = self.profileView.item(item, "tags") i = int(item.split('_')[1]) varL = self.current_profile_options[parents[-1]][1][0:i] varR = self.current_profile_options[parents[-1]][1][i + 1:] if "checked" in tag: self.current_profile_options[ parents[-1]][1] = varL + '1' + varR else: self.current_profile_options[ parents[-1]][1] = varL + '0' + varR else: tag = self.profileView.item(item, "tags") self.current_profile_options[item][0] = tag[0] profile_1 = '' for child in self.profileView.get_children(item): children = self.profileView.get_children(child) if children != (): tag = self.profileView.item(child, "tags") self.current_profile_options[item][1][child][0] = tag[ 0] profile = '' for s_child in self.profileView.get_children(child): tag = self.profileView.item(s_child, "tags") if 'checked' in tag: profile += '1' else: profile += '0' self.current_profile_options[item][1][child][ 1] = profile else: tag = self.profileView.item(child, "tags") if 'checked' in tag: profile_1 += '1' else: profile_1 += '0' self.current_profile_options[item][1] = profile_1 self.sync_vars()