def __init__(self, parent, tree: ttk.Treeview) -> None:
Frame.__init__(self, parent)
self.parent = parent
self.values = tree.item(tree.selection())["values"]
self.old = tree.item(tree.selection())["values"]
self.initUI()
class Example(Frame):
def __init__(self, parent):
Frame.__init__(self, parent)
self.tree = Treeview(self)
self.tree["columns"] = ("item1", "item2")
self.tree.heading("item1", text="Column 1")
self.tree.heading("item2", text="Column 2")
self.tree.insert("", 0, text="Item 1", values=("Value 1", "Value 2"))
row2 = self.tree.insert("", 1, "row2", text="Item 2")
self.tree.insert(row2,
"end",
"item1",
text="Item 1",
values=("3", "7"))
self.tree.insert(row2,
"end",
"item2",
text="Item 2",
values=("2", "5"))
self.tree.pack(expand=1, fill="both")
self.delete = Button(self, text="Delete Row", command=self.on_delete)
self.delete.pack(side="bottom")
def on_delete(self):
try:
selected_item = self.tree.selection()[0]
self.tree.delete(selected_item)
except IndexError:
pass
class _List(WidgetBase):
def __init__(self, master, **options):
super().__init__(master)
from tkinter.ttk import Treeview
self.widget = Treeview(self.master, show="tree", **options)
self.widget.bind("<<TreeviewSelect>>", self.callback)
self.value = ()
def set_header(self, column, text):
self.widget.heading("#" + str(column), text=text)
def add_item(self,
parent="",
index=0,
id=None,
label="",
values=None,
**options):
if not id is None:
options.update(iid=id)
if not values is None:
options.update(values=values)
self.widget.insert(parent=parent, index=index, text=label, **options)
def callback(self, event):
self.value = self.widget.selection()
def set_selection(self, items):
self.value = items
self.widget.selection_set(items)
def exist_item(self, id):
return self.widget.exists(id)
class ContactsListView(Frame):
def __init__(self, master, **kwargs):
super().__init__(master, **kwargs)
self.search_text = StringVar()
self.console_text = StringVar()
# Treeview setup
col_ids = ("first_name", "surname", "phone", "email", "last_edited")
self.contacts_list = Treeview(self, columns=col_ids, show='headings')
for col_id, text in zip(
col_ids,
["First name", "Surname", "Phone", "Email", "Last edited"]):
self.contacts_list.column(col_id, stretch=True, width=160)
self.contacts_list.heading(col_id, text=text)
# Search panel
search_panel = Frame(self)
search_entry = Entry(search_panel, textvariable=self.search_text)
self.search_btn = Button(search_panel, text="Search")
search_entry.pack(side=LEFT)
self.search_btn.pack(side=LEFT)
# Controls
self.add_btn = Button(self, text="Add")
self.edit_btn = Button(self, text="Edit")
self.delete_btn = Button(self, text="Delete")
self.console = Label(self, textvariable=self.console_text)
# Layout
search_panel.pack()
self.contacts_list.pack()
self.console.pack()
self.delete_btn.pack(side=RIGHT)
self.edit_btn.pack(side=RIGHT)
self.add_btn.pack(side=RIGHT)
def set(self, contacts):
self.contacts_list.delete(*self.contacts_list.get_children())
for contact_fields in contacts:
rowid, values = str(contact_fields[0]), contact_fields[1:]
self.contacts_list.insert('', 'end', text=rowid, values=values)
self.log()
def get_selected_contacts_ids(self):
focused_rows = self.contacts_list.selection()
return map(lambda focus: self.contacts_list.item(focus)['text'],
focused_rows)
def log(self, message=''):
self.console_text.set(message)
class SelectionFrame(Frame):
def __init__(self, fruit_list, **kw):
# init drop down
super().__init__(**kw)
self.fruit_list = fruit_list
self.type_select = StringVar(self)
header = Label(self)
header.grid(row=0, column=0, columnspan=2, pady=10)
type_options = OptionMenu(
header, self.type_select, *models.TYPES, command=self.set_group)
type_options.config(width=14)
type_options.grid(row=0, column=0)
# init start simulator button
simulator_btn = Button(
header, text="模拟器", command=lambda: sm.Simulator().mainloop())
simulator_btn.grid(row=0, column=1)
# init treeView
self.tree = Treeview(self, columns=["名称", "效果"], show="headings")
self.tree.column("名称", width=50, anchor='center')
self.tree.column("效果", width=150, anchor='center')
self.tree.heading("名称", text="名称")
self.tree.heading("效果", text="效果")
self.tree.grid(row=1, column=0)
self.tree.bind("<Double-1>", self.select_item)
self.select_item_callback = lambda x: x
vbar = Scrollbar(self, orient=VERTICAL, command=self.tree.yview)
self.tree.configure(yscrollcommand=vbar.set)
vbar.grid(row=1, column=1, sticky=NS)
# default value
self.set_group(models.TYPES[0])
def set_group(self, group):
self.type_select.set(group)
x = self.tree.get_children()
for item in x:
self.tree.delete(item)
for x in self.fruit_list:
if x.get_type() == group:
self.tree.insert("", END, value=(x.name, x.description))
def select_item(self, event):
item = self.tree.selection()[0]
name = self.tree.item(item, "value")[0]
self.select_item_callback(
next(filter(lambda x: x.name == name, self.fruit_list)))
def set_select_callback(self, callback):
self.select_item_callback = callback
class ResultForm(tk.Frame):
def __init__(self, master, **kwargs):
super().__init__(master, **kwargs)
self.header_label = tk.Label(self, text="All your saved passwords:", font=("roboto slab", 12, "bold"))
self.header_label.grid(row=0, column=0, pady=(0, 5), sticky=tk.W)
self.scrollbar_y = tk.Scrollbar(self, orient=tk.VERTICAL)
self.tree = Treeview(self, columns=("Place", "Password"), yscrollcommand=self.scrollbar_y.set)
self.scrollbar_y.config(command=self.tree.yview)
self.scrollbar_y.grid(row=1, column=1, sticky="nsew")
self.tree.heading("Place", text="Place")
self.tree.heading("Password", text="Password")
self.tree.column("#0", minwidth=0, width=0)
self.tree.column("#1", width=175)
self.tree.column("#2", width=175)
self.style = Style()
# noinspection SpellCheckingInspection
self.style.configure("Treeview", rowheight=15)
# noinspection SpellCheckingInspection
self.style.configure("Treeview.Heading", font=("roboto slab", 10, "bold"), foreground="#a6a6a6")
self.tree.grid(row=1, column=0)
self.instruction_label = tk.Label(self, text="* First select an item, then choose from below",
font=("roboto slab", 8, "italic"))
self.instruction_label.grid(row=2, column=0, sticky=tk.W)
def insert_to_tree(self, values):
for value in values:
self.tree.insert("", tk.END, value=value)
def empty_tree(self):
if self.tree.get_children():
self.tree.delete(*self.tree.get_children())
def return_selected_item_values(self):
"""Returns the values of the highlighted item inside TreeView"""
highlighted_item = self.tree.selection()
if highlighted_item:
return tuple(self.tree.item(highlighted_item)["values"])
class StructEditor(tk.Frame, Subscriber, Observable):
"""Displays and allow editing of the coordinates and points of one superstructure
Args:
parent (tk.Frame): widget that is the parent of the editor
structure (model.structure.Structure): the ship superstructure that will be edited
"""
def __init__(self, parent, structure, command_stack):
Subscriber.__init__(self, structure)
Observable.__init__(self)
tk.Frame.__init__(self, parent, borderwidth=4, relief="raised")
self._structure = structure
self._command_stack = command_stack
self.bind("<Button-1>", self._on_click)
self.bind("<FocusIn>", self._on_get_focus)
self.bind("<FocusOut>", self._on_lost_focus)
self._tree = Treeview(self,
columns=["#", "X", "Y"],
selectmode="browse")
#kill the icon column
self._tree.column("#0", minwidth=0, width=0)
style = Style()
style.configure("Treeview.Heading", font=(None, 16))
self._tree.column("#", minwidth=20, width=40, anchor=tk.CENTER)
self._tree.column("X", minwidth=20, width=40, anchor=tk.CENTER)
self._tree.column("Y", minwidth=20, width=40, anchor=tk.CENTER)
self._tree.heading("#", text="#")
self._tree.heading("X", text="\u21d5")
self._tree.heading("Y", text="\u21d4")
self._tree.grid(row=0, column=POINTS_TABLE_COL, sticky=tk.N + tk.S)
self._tree.bind("<<TreeviewSelect>>", self._on_point_selected)
self._tree.bind("<FocusIn>", self._on_get_focus)
self._tree.bind("<FocusOut>", self._on_lost_focus)
scroll = Scrollbar(self, command=self._tree.yview)
scroll.grid(row=0, column=SCROLL_COL, sticky=tk.N + tk.S)
scroll.bind("<FocusIn>", self._on_get_focus)
self._tree.configure(yscrollcommand=scroll.set)
self._index_of_sel_point = -1
self._fill_tree()
self._edit_zone = EditZone(self, self._structure, command_stack,
self._on_get_focus)
self._edit_zone.grid(column=EDIT_ZONE_COL, row=0, sticky=tk.N)
def _set_selection(self, new_sel_index):
"""Set the selected point to the new_sel_index
Gives correct focus, update, etc to the editor's widgets
if the index is outside of the self.points, does nothing
"""
if new_sel_index >= 0 and new_sel_index <= len(self.points) - 1:
iid = self._tree.get_children()[new_sel_index]
self._tree.selection_set(iid)
def _on_click(self, *_args):
self._tree.focus_set()
def _on_get_focus(self, *_args):
if self._index_of_sel_point == -1:
self._set_selection(0)
self.configure(relief="sunken")
self._notify("focus", {})
def _on_lost_focus(self, event):
if event.widget not in self.winfo_children():
self.configure(relief="raised")
def _on_point_selected(self, _event):
"""called back when a point is selected in the table/treeview
Updates the editable fields
"""
selected_iid = self._tree.selection()
self._index_of_sel_point = self._tree.index(selected_iid)
self._edit_zone.set_editable_point(
self._tree.item(selected_iid)["values"][0])
self._notify("focus", {})
def _fill_tree(self):
"""fills the treeview with data from the structure
"""
self._tree.delete(*self._tree.get_children())
for point_index, point in enumerate(self._structure.points):
self._tree.insert(
'',
'end',
values=[point_index,
round(point[0]),
round(point[1])])
if point_index == self._index_of_sel_point:
self._set_selection(point_index)
def _on_notification(self, observable, event_type, event_info):
"""Rebuild the treeview on structure update
Depending on the structure state and the operation, change the selcted point
"""
if event_type == "add_point":
self._index_of_sel_point = event_info["index"]
self._fill_tree()
else:
if self._index_of_sel_point >= len(self._structure.points):
self._index_of_sel_point = len(self._structure.points)
self._edit_zone.unset_point()
self._fill_tree()
self._notify("focus", {})
def update_to_coord(self, point):
"""Move the selected point to the position of the given point
Intended to be called from click on the top view
Args:
point (x, y): new position in funnel coordinates
"""
if self._index_of_sel_point != -1 and self._index_of_sel_point <= len(
self.points) - 1:
self._command_stack.do(
model.structure.UpdatePoint(self._structure,
self._index_of_sel_point,
round(point[0]), round(point[1])))
elif self._index_of_sel_point == len(self.points) or not self.points:
self._command_stack.do(
model.structure.AddPoint(self._structure,
self._index_of_sel_point + 1,
round(point[0]), round(point[1])))
if self._index_of_sel_point + 1 >= len(self.points):
self.winfo_toplevel().update()
self._index_of_sel_point = len(self.points)
else:
self._set_selection(self._index_of_sel_point + 1)
self.winfo_toplevel().update()
@property
def points(self):
"""Pipe throught the struct's properties"""
return self._structure.points
@property
def fill(self):
"""Pipe throught the struct's properties"""
return self._structure.fill
@property
def selected_index(self):
"""the index in the struct's point list of the currently selected point
Should be -1 if none selected
"""
return self._index_of_sel_point
class Tree(LibraryListener):
def __init__(self, config, master):
self.views = config['views']
self.library = None
self.__make_widgets(master)
def __make_widgets(self, root):
self.tree = Treeview(root)
def on_library_change(self, library):
self.library = library
self.__make_root_nodes()
def __make_root_nodes(self):
for i in self.tree.get_children():
self.tree.delete(i)
root_nodes = self.query_library({}, self.views[0])
for node in root_nodes:
child = self.tree.insert("", "end", self.__make_node_id(None, self.views[0], node), text=node,
tags=self.views[0])
self.fetch_children(child)
@staticmethod
def __make_node_id(parent_node, node_type, node_value):
node_id = {}
if parent_node:
node_id = copy(parent_node)
node_id[node_type] = node_value
return node_id
def can_be_expanded(self, node_type):
return node_type != self.views[len(self.views) - 1]
def get_child_node_type(self, node_type):
return self.views[self.views.index(node_type) + 1]
def query_library(self, node_id, view):
return self.library.aggregate(node_id, view)
def __has_children(self, node):
return len(self.children(node)) > 0
def fetch_children(self, node):
if not self.__has_children(node):
node_type = self.tree.item(node, "tags")[0]
if self.can_be_expanded(node_type):
children_node_type = self.get_child_node_type(node_type)
node_id = literal_eval(node)
nodes = self.query_library(node_id, children_node_type)
for child_node in nodes:
self.tree.insert(node, "end", self.__make_node_id(node_id, children_node_type, child_node),
text=child_node, tags=children_node_type)
def delete_children(self, parent_node):
children = self.children(parent_node)
for node in children:
self.tree.delete(node)
@property
def selected_node(self):
if len(self.tree.selection()):
return self.tree.selection()[0]
def children(self, node):
return self.tree.get_children(node)
class DialogPluginManager(Toplevel):
def __init__(self, mainWin, modulesWithNewerFileDates):
super(DialogPluginManager, self).__init__(mainWin.parent)
self.ENABLE = _("Enable")
self.DISABLE = _("Disable")
self.parent = mainWin.parent
self.cntlr = mainWin
# copy plugins for temporary display
self.pluginConfig = PluginManager.pluginConfig
self.pluginConfigChanged = False
self.uiClassMethodsChanged = False
self.modelClassesChanged = False
self.customTransformsChanged = False
self.disclosureSystemTypesChanged = False
self.hostSystemFeaturesChanged = False
self.modulesWithNewerFileDates = modulesWithNewerFileDates
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)",
self.parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.title(_("Plug-in Manager"))
frame = Frame(self)
# left button frame
buttonFrame = Frame(frame, width=40)
buttonFrame.columnconfigure(0, weight=1)
addLabel = Label(buttonFrame,
text=_("Find plug-in modules:"),
wraplength=60,
justify="center")
addSelectLocalButton = Button(buttonFrame,
text=_("Select"),
command=self.selectLocally)
ToolTip(
addSelectLocalButton,
text=_(
"Select python module files from the local plugin directory."),
wraplength=240)
addBrowseLocalButton = Button(buttonFrame,
text=_("Browse"),
command=self.browseLocally)
ToolTip(
addBrowseLocalButton,
text=
_("File chooser allows browsing and selecting python module files to add (or reload) plug-ins, from the local file system."
),
wraplength=240)
addWebButton = Button(buttonFrame,
text=_("On Web"),
command=self.findOnWeb)
ToolTip(
addWebButton,
text=
_("Dialog to enter URL full path to load (or reload) plug-ins, from the web or local file system."
),
wraplength=240)
addLabel.grid(row=0, column=0, pady=4)
addSelectLocalButton.grid(row=1, column=0, pady=4)
addBrowseLocalButton.grid(row=2, column=0, pady=4)
addWebButton.grid(row=3, column=0, pady=4)
buttonFrame.grid(row=0,
column=0,
rowspan=3,
sticky=(N, S, W),
padx=3,
pady=3)
# right tree frame (plugins already known to arelle)
modulesFrame = Frame(frame, width=720)
vScrollbar = Scrollbar(modulesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(modulesFrame, orient=HORIZONTAL)
self.modulesView = Treeview(modulesFrame,
xscrollcommand=hScrollbar.set,
yscrollcommand=vScrollbar.set,
height=7)
self.modulesView.grid(row=0, column=0, sticky=(N, S, E, W))
self.modulesView.bind('<<TreeviewSelect>>', self.moduleSelect)
hScrollbar["command"] = self.modulesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E, W))
vScrollbar["command"] = self.modulesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N, S))
modulesFrame.columnconfigure(0, weight=1)
modulesFrame.rowconfigure(0, weight=1)
modulesFrame.grid(row=0,
column=1,
columnspan=4,
sticky=(N, S, E, W),
padx=3,
pady=3)
self.modulesView.focus_set()
self.modulesView.column("#0", width=120, anchor="w")
self.modulesView.heading("#0", text=_("Name"))
self.modulesView["columns"] = ("author", "ver", "status", "date",
"update", "descr", "license")
self.modulesView.column("author", width=100, anchor="w", stretch=False)
self.modulesView.heading("author", text=_("Author"))
self.modulesView.column("ver", width=60, anchor="w", stretch=False)
self.modulesView.heading("ver", text=_("Version"))
self.modulesView.column("status", width=50, anchor="w", stretch=False)
self.modulesView.heading("status", text=_("Status"))
self.modulesView.column("date", width=70, anchor="w", stretch=False)
self.modulesView.heading("date", text=_("File Date"))
self.modulesView.column("update", width=50, anchor="w", stretch=False)
self.modulesView.heading("update", text=_("Update"))
self.modulesView.column("descr", width=200, anchor="w", stretch=False)
self.modulesView.heading("descr", text=_("Description"))
self.modulesView.column("license", width=70, anchor="w", stretch=False)
self.modulesView.heading("license", text=_("License"))
classesFrame = Frame(frame)
vScrollbar = Scrollbar(classesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(classesFrame, orient=HORIZONTAL)
self.classesView = Treeview(classesFrame,
xscrollcommand=hScrollbar.set,
yscrollcommand=vScrollbar.set,
height=5)
self.classesView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.classesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E, W))
vScrollbar["command"] = self.classesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N, S))
classesFrame.columnconfigure(0, weight=1)
classesFrame.rowconfigure(0, weight=1)
classesFrame.grid(row=1,
column=1,
columnspan=4,
sticky=(N, S, E, W),
padx=3,
pady=3)
self.classesView.focus_set()
self.classesView.column("#0", width=200, anchor="w")
self.classesView.heading("#0", text=_("Class"))
self.classesView["columns"] = ("modules", )
self.classesView.column("modules",
width=500,
anchor="w",
stretch=False)
self.classesView.heading("modules", text=_("Modules"))
# bottom frame module info details
moduleInfoFrame = Frame(frame, width=700)
moduleInfoFrame.columnconfigure(1, weight=1)
self.moduleNameLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left",
font=font.Font(family='Helvetica',
size=12,
weight='bold'))
self.moduleNameLabel.grid(row=0, column=0, columnspan=4, sticky=W)
self.moduleAuthorHdr = Label(moduleInfoFrame,
text=_("author:"),
state=DISABLED)
self.moduleAuthorHdr.grid(row=1, column=0, sticky=W)
self.moduleAuthorLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleAuthorLabel.grid(row=1, column=1, columnspan=3, sticky=W)
self.moduleDescrHdr = Label(moduleInfoFrame,
text=_("description:"),
state=DISABLED)
self.moduleDescrHdr.grid(row=2, column=0, sticky=W)
self.moduleDescrLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleDescrLabel.grid(row=2, column=1, columnspan=3, sticky=W)
self.moduleClassesHdr = Label(moduleInfoFrame,
text=_("classes:"),
state=DISABLED)
self.moduleClassesHdr.grid(row=3, column=0, sticky=W)
self.moduleClassesLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleClassesLabel.grid(row=3, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleClassesLabel,
text=_("List of classes that this plug-in handles."),
wraplength=240)
self.moduleVersionHdr = Label(moduleInfoFrame,
text=_("version:"),
state=DISABLED)
self.moduleVersionHdr.grid(row=4, column=0, sticky=W)
self.moduleVersionLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleVersionLabel.grid(row=4, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleVersionLabel,
text=_("Version of plug-in module."),
wraplength=240)
self.moduleUrlHdr = Label(moduleInfoFrame,
text=_("URL:"),
state=DISABLED)
self.moduleUrlHdr.grid(row=5, column=0, sticky=W)
self.moduleUrlLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleUrlLabel.grid(row=5, column=1, columnspan=3, sticky=W)
ToolTip(
self.moduleUrlLabel,
text=_(
"URL of plug-in module (local file path or web loaded file)."),
wraplength=240)
self.moduleDateHdr = Label(moduleInfoFrame,
text=_("date:"),
state=DISABLED)
self.moduleDateHdr.grid(row=6, column=0, sticky=W)
self.moduleDateLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleDateLabel.grid(row=6, column=1, columnspan=3, sticky=W)
ToolTip(
self.moduleDateLabel,
text=
_("Date of currently loaded module file (with parenthetical node when an update is available)."
),
wraplength=240)
self.moduleLicenseHdr = Label(moduleInfoFrame,
text=_("license:"),
state=DISABLED)
self.moduleLicenseHdr.grid(row=7, column=0, sticky=W)
self.moduleLicenseLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleLicenseLabel.grid(row=7, column=1, columnspan=3, sticky=W)
self.moduleImportsHdr = Label(moduleInfoFrame,
text=_("imports:"),
state=DISABLED)
self.moduleImportsHdr.grid(row=8, column=0, sticky=W)
self.moduleImportsLabel = Label(moduleInfoFrame,
wraplength=600,
justify="left")
self.moduleImportsLabel.grid(row=8, column=1, columnspan=3, sticky=W)
self.moduleEnableButton = Button(moduleInfoFrame,
text=self.ENABLE,
state=DISABLED,
command=self.moduleEnable)
ToolTip(self.moduleEnableButton,
text=_("Enable/disable plug in."),
wraplength=240)
self.moduleEnableButton.grid(row=9, column=1, sticky=E)
self.moduleReloadButton = Button(moduleInfoFrame,
text=_("Reload"),
state=DISABLED,
command=self.moduleReload)
ToolTip(self.moduleReloadButton,
text=_("Reload/update plug in."),
wraplength=240)
self.moduleReloadButton.grid(row=9, column=2, sticky=E)
self.moduleRemoveButton = Button(moduleInfoFrame,
text=_("Remove"),
state=DISABLED,
command=self.moduleRemove)
ToolTip(
self.moduleRemoveButton,
text=
_("Remove plug in from plug in table (does not erase the plug in's file)."
),
wraplength=240)
self.moduleRemoveButton.grid(row=9, column=3, sticky=E)
moduleInfoFrame.grid(row=2,
column=0,
columnspan=5,
sticky=(N, S, E, W),
padx=3,
pady=3)
moduleInfoFrame.config(borderwidth=4, relief="groove")
okButton = Button(frame, text=_("Close"), command=self.ok)
ToolTip(okButton,
text=_("Accept and changes (if any) and close dialog."),
wraplength=240)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
ToolTip(cancelButton,
text=_("Cancel changes (if any) and close dialog."),
wraplength=240)
okButton.grid(row=3, column=3, sticky=(S, E), pady=3)
cancelButton.grid(row=3, column=4, sticky=(S, E), pady=3, padx=3)
enableDisableFrame = Frame(frame)
enableDisableFrame.grid(row=3, column=1, sticky=(S, W), pady=3)
enableAllButton = Button(enableDisableFrame,
text=_("Enable All"),
command=self.enableAll)
ToolTip(enableAllButton,
text=_("Enable all plug ins."),
wraplength=240)
disableAllButton = Button(enableDisableFrame,
text=_("Disable All"),
command=self.disableAll)
ToolTip(disableAllButton,
text=_("Disable all plug ins."),
wraplength=240)
enableAllButton.grid(row=1, column=1)
disableAllButton.grid(row=1, column=2)
self.loadTreeViews()
self.geometry("+{0}+{1}".format(dialogX + 50, dialogY + 100))
frame.grid(row=0, column=0, sticky=(N, S, E, W))
frame.columnconfigure(0, weight=0)
frame.columnconfigure(1, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeViews(self):
self.selectedModule = None
# clear previous treeview entries
for previousNode in self.modulesView.get_children(""):
self.modulesView.delete(previousNode)
def loadSubtree(parentNode, moduleItems):
for moduleItem in sorted(moduleItems, key=lambda item: item[0]):
moduleInfo = moduleItem[1]
if parentNode or not moduleInfo.get("isImported"):
nodeName = moduleItem[0]
if parentNode:
nodeName = parentNode + GROUPSEP + nodeName
name = moduleInfo.get("name", nodeName)
node = self.modulesView.insert(parentNode,
"end",
nodeName,
text=name)
self.modulesView.set(node, "author",
moduleInfo.get("author"))
self.modulesView.set(node, "ver",
moduleInfo.get("version"))
self.modulesView.set(node, "status",
moduleInfo.get("status"))
self.modulesView.set(node, "date",
moduleInfo.get("fileDate"))
if name in self.modulesWithNewerFileDates:
self.modulesView.set(node, "update", _("available"))
self.modulesView.set(node, "descr",
moduleInfo.get("description"))
self.modulesView.set(node, "license",
moduleInfo.get("license"))
if moduleInfo.get("imports"):
loadSubtree(
node,
[(importModuleInfo["name"], importModuleInfo)
for importModuleInfo in moduleInfo["imports"]])
loadSubtree("", self.pluginConfig.get("modules", {}).items())
# clear previous treeview entries
for previousNode in self.classesView.get_children(""):
self.classesView.delete(previousNode)
for i, classItem in enumerate(
sorted(self.pluginConfig.get("classes", {}).items())):
className, moduleList = classItem
node = self.classesView.insert("",
"end",
className,
text=className)
self.classesView.set(node, "modules", ', '.join(moduleList))
self.moduleSelect() # clear out prior selection
def ok(self, event=None):
# check for orphaned classes (for which there is no longer a corresponding module)
_moduleNames = self.pluginConfig.get("modules", {}).keys()
_orphanedClassNames = set()
for className, moduleList in self.pluginConfig.get("classes",
{}).items():
for _moduleName in moduleList.copy():
if _moduleName not in _moduleNames: # it's orphaned
moduleList.remove(_moduleName)
self.pluginConfigChanged = True
if not moduleList: # now orphaned
_orphanedClassNames.add(className)
self.pluginConfigChanged = True
for _orphanedClassName in _orphanedClassNames:
del self.pluginConfig["classes"][_orphanedClassName]
if self.pluginConfigChanged:
PluginManager.pluginConfig = self.pluginConfig
PluginManager.pluginConfigChanged = True
PluginManager.reset() # force reloading of modules
if self.uiClassMethodsChanged or self.modelClassesChanged or self.customTransformsChanged or self.disclosureSystemTypesChanged or self.hostSystemFeaturesChanged: # may require reloading UI
affectedItems = ""
if self.uiClassMethodsChanged:
affectedItems += _("menus of the user interface")
if self.modelClassesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("model objects of the processor")
if self.customTransformsChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("custom transforms")
if self.disclosureSystemTypesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("disclosure system types")
if self.hostSystemFeaturesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("host system features")
if messagebox.askyesno(
_("User interface plug-in change"),
_("A change in plug-in class methods may have affected {0}. "
"Please restart Arelle to due to these changes. \n\n"
"Should Arelle restart itself now "
"(if there are any unsaved changes they would be lost!)?"
).format(affectedItems),
parent=self):
self.cntlr.uiThreadQueue.put((self.cntlr.quit, [None, True]))
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def moduleSelect(self, *args):
node = (self.modulesView.selection() or (None, ))[0]
if node:
node = node.rpartition(GROUPSEP)[
2] # drop leading path names for module name
moduleInfo = self.pluginConfig.get("modules", {}).get(node)
if moduleInfo:
self.selectedModule = node
name = moduleInfo["name"]
self.moduleNameLabel.config(text=name)
self.moduleAuthorHdr.config(state=ACTIVE)
self.moduleAuthorLabel.config(text=moduleInfo.get("author"))
self.moduleDescrHdr.config(state=ACTIVE)
self.moduleDescrLabel.config(text=moduleInfo.get("description"))
self.moduleClassesHdr.config(state=ACTIVE)
self.moduleClassesLabel.config(
text=', '.join(moduleInfo["classMethods"]))
self.moduleVersionHdr.config(state=ACTIVE)
self.moduleVersionLabel.config(text=moduleInfo.get("version"))
self.moduleUrlHdr.config(state=ACTIVE)
self.moduleUrlLabel.config(text=moduleInfo["moduleURL"])
self.moduleDateHdr.config(state=ACTIVE)
self.moduleDateLabel.config(
text=moduleInfo["fileDate"] + " " +
(_("(an update is available)") if name in
self.modulesWithNewerFileDates else ""))
self.moduleLicenseHdr.config(state=ACTIVE)
self.moduleLicenseLabel.config(text=moduleInfo.get("license"))
if moduleInfo.get("imports"):
self.moduleImportsHdr.config(state=ACTIVE)
_text = ", ".join(mi["name"]
for mi in moduleInfo["imports"][:3])
if len(moduleInfo["imports"]) >= 3:
_text += ", ..."
self.moduleImportsLabel.config(text=_text)
_buttonState = DISABLED if moduleInfo.get("isImported") else ACTIVE
self.moduleEnableButton.config(state=_buttonState,
text={
"enabled": self.DISABLE,
"disabled": self.ENABLE
}[moduleInfo["status"]])
self.moduleReloadButton.config(state=_buttonState)
self.moduleRemoveButton.config(state=_buttonState)
else:
self.selectedModule = None
self.moduleNameLabel.config(text="")
self.moduleAuthorHdr.config(state=DISABLED)
self.moduleAuthorLabel.config(text="")
self.moduleDescrHdr.config(state=DISABLED)
self.moduleDescrLabel.config(text="")
self.moduleClassesHdr.config(state=DISABLED)
self.moduleClassesLabel.config(text="")
self.moduleVersionHdr.config(state=DISABLED)
self.moduleVersionLabel.config(text="")
self.moduleUrlHdr.config(state=DISABLED)
self.moduleUrlLabel.config(text="")
self.moduleDateHdr.config(state=DISABLED)
self.moduleDateLabel.config(text="")
self.moduleLicenseHdr.config(state=DISABLED)
self.moduleLicenseLabel.config(text="")
self.moduleImportsHdr.config(state=DISABLED)
self.moduleImportsLabel.config(text="")
self.moduleEnableButton.config(state=DISABLED, text=self.ENABLE)
self.moduleReloadButton.config(state=DISABLED)
self.moduleRemoveButton.config(state=DISABLED)
def selectLocally(self):
choices = [] # list of tuple of (file name, description)
def sortOrder(key):
return {
"EdgarRenderer": "1",
"validate": "2",
"xbrlDB": "3"
}.get(key, "4") + key.lower()
def selectChoices(dir, indent=""):
dirHasEntries = False
for f in sorted(os.listdir(dir), key=sortOrder):
if f not in (".", "..", "__pycache__", "__init__.py"):
fPath = os.path.join(dir, f)
fPkgInit = os.path.join(fPath, "__init__.py")
dirInsertPoint = len(choices)
moduleInfo = None
if ((os.path.isdir(fPath) and os.path.exists(fPkgInit)) or
((os.path.isfile(fPath) and f.endswith(".py")))):
moduleInfo = PluginManager.moduleModuleInfo(fPath)
if moduleInfo:
choices.append((
indent + f,
"name: {}\ndescription: {}\nversion: {}\nlicense: {}"
.format(moduleInfo["name"],
moduleInfo.get("description"),
moduleInfo.get("version"),
moduleInfo.get("license")), fPath,
moduleInfo["name"], moduleInfo.get("version"),
moduleInfo.get("description"),
moduleInfo.get("license")))
dirHasEntries = True
if os.path.isdir(fPath) and f not in ("DQC_US_Rules", ):
if selectChoices(fPath, indent=indent +
" ") and not moduleInfo:
choices.insert(dirInsertPoint,
(indent + f, None, None, None, None,
None, None))
return dirHasEntries
selectChoices(self.cntlr.pluginDir)
selectedPath = DialogOpenArchive.selectPlugin(self, choices)
if selectedPath:
moduleInfo = PluginManager.moduleModuleInfo(
selectedPath[len(self.cntlr.pluginDir) + 1:])
self.loadFoundModuleInfo(moduleInfo, selectedPath)
def browseLocally(self):
initialdir = self.cntlr.pluginDir # default plugin directory
if not self.cntlr.isMac: # can't navigate within app easily, always start in default directory
initialdir = self.cntlr.config.setdefault("pluginOpenDir",
initialdir)
filename = self.cntlr.uiFileDialog(
"open",
parent=self,
title=_("Choose plug-in module file"),
initialdir=initialdir,
filetypes=[(_("Python files"), "*.py")],
defaultextension=".py")
if filename:
# check if a package is selected (any file in a directory containing an __init__.py
#if (os.path.basename(filename) == "__init__.py" and os.path.isdir(os.path.dirname(filename)) and
# os.path.isfile(filename)):
# filename = os.path.dirname(filename) # refer to the package instead
self.cntlr.config["pluginOpenDir"] = os.path.dirname(filename)
moduleInfo = PluginManager.moduleModuleInfo(filename)
self.loadFoundModuleInfo(moduleInfo, filename)
def findOnWeb(self):
url = DialogURL.askURL(self)
if url: # url is the in-cache or local file
moduleInfo = PluginManager.moduleModuleInfo(url)
self.cntlr.showStatus("") # clear web loading status
self.loadFoundModuleInfo(moduleInfo, url)
def loadFoundModuleInfo(self, moduleInfo, url):
if moduleInfo and moduleInfo.get("name"):
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
else:
messagebox.showwarning(
_("Module is not itself a plug-in or in a directory with package __init__.py plug-in. "
),
_("File does not itself contain a python program with an appropriate __pluginInfo__ declaration: \n\n{0}"
).format(url),
parent=self)
def checkIfImported(self, moduleInfo):
if moduleInfo.get("isImported"):
messagebox.showwarning(
_("Plug-in is imported by a parent plug-in. "),
_("Plug-in has a parent, please request operation on the parent: \n\n{0}"
).format(moduleInfo.get("name")),
parent=self)
return True
return False
def checkClassMethodsChanged(self, moduleInfo):
for classMethod in moduleInfo["classMethods"]:
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
elif classMethod == "ModelObjectFactory.ElementSubstitutionClasses":
self.modelClassesChanged = True # model object factor classes changed
elif classMethod == "ModelManager.LoadCustomTransforms":
self.customTransformsChanged = True
elif classMethod == "DisclosureSystem.Types":
self.disclosureSystemTypesChanged = True # disclosure system types changed
elif classMethod.startswith("Proxy."):
self.hostSystemFeaturesChanged = True # system features (e.g., proxy) changed
def removePluginConfigModuleInfo(self, name):
moduleInfo = self.pluginConfig["modules"].get(name)
if moduleInfo:
if self.checkIfImported(moduleInfo):
return
def _removePluginConfigModuleInfo(moduleInfo):
_name = moduleInfo.get("name")
if _name:
self.checkClassMethodsChanged(moduleInfo)
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].get(
classMethod)
if classMethods and _name in classMethods:
classMethods.remove(_name)
if not classMethods: # list has become unused
del self.pluginConfig["classes"][
classMethod] # remove class
for importModuleInfo in moduleInfo.get(
"imports", EMPTYLIST):
_removePluginConfigModuleInfo(importModuleInfo)
self.pluginConfig["modules"].pop(_name, None)
_removePluginConfigModuleInfo(moduleInfo)
if not self.pluginConfig["modules"] and self.pluginConfig[
"classes"]:
self.pluginConfig["classes"].clear() # clean orphan classes
self.pluginConfigChanged = True
def addPluginConfigModuleInfo(self, moduleInfo):
if self.checkIfImported(moduleInfo):
return
name = moduleInfo.get("name")
self.removePluginConfigModuleInfo(
name) # remove any prior entry for this module
def _addPlugin(moduleInfo):
_name = moduleInfo.get("name")
if _name:
self.modulesWithNewerFileDates.discard(
_name) # no longer has an update available
self.pluginConfig["modules"][_name] = moduleInfo
# add classes
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].setdefault(
classMethod, [])
if name not in classMethods:
classMethods.append(_name)
self.checkClassMethodsChanged(moduleInfo)
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_addPlugin(importModuleInfo)
_addPlugin(moduleInfo)
self.pluginConfigChanged = True
def moduleEnable(self):
if self.selectedModule in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][self.selectedModule]
if self.checkIfImported(moduleInfo):
return
def _moduleEnable(moduleInfo):
if self.moduleEnableButton['text'] == self.ENABLE:
moduleInfo["status"] = "enabled"
elif self.moduleEnableButton['text'] == self.DISABLE:
moduleInfo["status"] = "disabled"
self.checkClassMethodsChanged(moduleInfo)
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_moduleEnable(
importModuleInfo) # set status on nested moduleInfo
if importModuleInfo['name'] in self.pluginConfig[
"modules"]: # set status on top level moduleInfo
_moduleEnable(self.pluginConfig["modules"][
importModuleInfo['name']])
_moduleEnable(moduleInfo)
if self.moduleEnableButton['text'] == self.ENABLE:
self.moduleEnableButton['text'] = self.DISABLE
elif self.moduleEnableButton['text'] == self.DISABLE:
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
def moduleReload(self):
if self.selectedModule in self.pluginConfig["modules"]:
url = self.pluginConfig["modules"][self.selectedModule].get(
"moduleURL")
if url:
moduleInfo = PluginManager.moduleModuleInfo(url, reload=True)
if moduleInfo:
if self.checkIfImported(moduleInfo):
return
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
self.cntlr.showStatus(_("{0} reloaded").format(
moduleInfo["name"]),
clearAfter=5000)
else:
messagebox.showwarning(
_("Module error"),
_("File or module cannot be reloaded: \n\n{0}").format(
url),
parent=self)
def moduleRemove(self):
if self.selectedModule in self.pluginConfig["modules"]:
self.removePluginConfigModuleInfo(self.selectedModule)
self.pluginConfigChanged = True
self.loadTreeViews()
def enableAll(self):
self.enableDisableAll(True)
def disableAll(self):
self.enableDisableAll(False)
def enableDisableAll(self, doEnable):
for module in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][module]
if not moduleInfo.get("isImported"):
def _enableDisableAll(moduleInfo):
if doEnable:
moduleInfo["status"] = "enabled"
else:
moduleInfo["status"] = "disabled"
for importModuleInfo in moduleInfo.get(
"imports", EMPTYLIST):
_enableDisableAll(importModuleInfo)
_enableDisableAll(moduleInfo)
if doEnable:
self.moduleEnableButton['text'] = self.DISABLE
else:
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
class BlackWhite:
def __init__(self):
self.win = Tk()
self.win.title("White/Black List")
self.win.geometry('{}x{}'.format(800, 450))
self.canvas = Canvas(self.win, bg='#36393F')
self.frame = Frame(self.canvas, bg='#36393F', width=600, height=340)
self.userSelected = StringVar()
self.whitebox = Treeview(self.frame,
columns=1,
show="headings",
height="5")
self.blackbox = Treeview(self.frame,
columns=1,
show="headings",
height="5")
self.userList = Combobox(self.frame,
width=20,
textvariable=self.userSelected,
state="readonly")
self.user = db.getName()
def main(self):
self.canvas.pack(expand=TRUE, fill=BOTH)
self.frame.pack(expand=TRUE)
Button(self.frame,
text="Add",
font='Arial 10 bold',
bg='#36393F',
fg="#f7cc35",
command=self.addWhite).grid(row=0, column=0)
Button(self.frame,
text="Remove",
font='Arial 10 bold',
bg='#36393F',
fg="#f7cc35",
command=self.removeWhite).grid(row=1, column=0, padx=10)
self.whitebox.grid(row=0, column=1, rowspan=2)
self.whitebox.heading(1, text="WhiteBox")
self.whitebox.column(1, width=100)
self.blackbox.grid(row=0, column=2, rowspan=2)
self.blackbox.heading(1, text="BlackBox")
self.blackbox.column(1, width=100)
Button(self.frame,
text="Add",
font='Arial 10 bold',
bg='#36393F',
fg="#f7cc35",
command=self.addBlack).grid(row=0, column=3)
Button(self.frame,
text="Remove",
font='Arial 10 bold',
bg='#36393F',
fg="#f7cc35",
command=self.removeBlack).grid(row=1, column=3, padx=10)
db.cursor.execute(
"SELECT whitelisted FROM white_list WHERE whitelister = '%s'" %
self.user)
for row in db.cursor.fetchall():
self.whitebox.insert('', END, values=row)
db.cursor.execute(
"SELECT blacklisted FROM black_list WHERE blacklister = '%s'" %
self.user)
for row in db.cursor.fetchall():
self.blackbox.insert('', END, values=row)
self.userList.grid(row=2, column=0, columnspan=4, pady=10)
self.userList['values'] = inputUsers()
self.userList.current(0)
self.win.mainloop()
def addWhite(self):
# add selected user from combobox to whitebox
self.whitebox.insert('', END, values=self.userSelected.get())
# add to white_list table
db.cursor.execute("INSERT INTO white_list VALUES(%s, %s)",
(self.user, self.userSelected.get()))
def removeWhite(self):
for selected_item in self.whitebox.selection():
whitelisted = self.whitebox.item(selected_item, 'values')[0]
self.whitebox.delete(selected_item)
db.cursor.execute(
"DELETE FROM white_list WHERE whitelister = %s AND whitelisted = %s",
(self.user, whitelisted))
def addBlack(self):
self.blackbox.insert('', END, values=self.userSelected.get())
db.cursor.execute("INSERT INTO black_list VALUES(%s, %s)",
(self.user, self.userSelected.get()))
def removeBlack(self):
for selected_item in self.blackbox.selection():
blacklisted = self.blackbox.item(selected_item, 'values')[0]
self.blackbox.delete(selected_item)
db.cursor.execute(
"DELETE FROM black_list WHERE blacklister = %s AND blacklisted = %s",
(self.user, blacklisted))
class DebitList(Frame, MoneySubject, MoneyObserver):
def __init__(self, parent):
super().__init__(parent)
MoneySubject.__init__(self)
MoneyObserver.__init__(self)
self.initUI()
def initUI(self):
self.records = []
self.debitList = Treeview(self, height=15)
columns = ("Amount", "Desc")
self.debitList["columns"] = columns
self.debitList.column("#0", width=70, minwidth=70, stretch=NO)
self.debitList.column("Amount", width=70, stretch=NO)
self.debitList.column("Desc", width=250, stretch=NO)
self.debitList.heading("#0", text="Date", anchor=W)
for field in columns:
self.debitList.heading(field, text=field, anchor=W)
self.debitList.pack(side=LEFT)
self.pack()
self.debitList.bind("<Delete>", self.keydown)
def keydown(self, e):
item = self.debitList.selection()
parent = self.debitList.parent(item)
if item != () and parent != '':
values = self.debitList.item(item)['values']
amount = float(values[0][1:])
self.moneyNotify({"debit": -amount})
self.debitList.delete(item)
self.records.remove({
"debit": amount,
"desc": str(values[1]),
"date": parent
})
def reset(self):
self.records = []
for child in self.debitList.get_children():
self.debitList.delete(child)
def export(self):
return self.records
def restore(self, debits, startDate):
self.reset()
self.setStartDate(startDate)
dateTable = {}
for d in range(7):
day = startDate + timedelta(days=d)
dateStr = day.strftime("%d/%m")
dateTable[dateStr] = day
for debit in debits:
debit["date"] = dateTable[debit["date"]]
self.moneyUpdate(debit)
self.moneyNotify(debit)
def moneyUpdate(self, monies):
if "debit" in monies:
dateStr = monies["date"].strftime("%d/%m")
toSave = monies
toSave["date"] = dateStr
self.records.append(toSave)
values = (f"${monies['debit']}", monies["desc"])
self.debitList.insert(dateStr, "end", text="", values=values)
def setStartDate(self, startDate):
for d in range(7):
day = startDate + timedelta(days=d)
dateStr = day.strftime("%d/%m")
self.debitList.insert("",
"end",
id=dateStr,
text=dateStr,
values=("", ""))
class GUI:
def __init__(self):
self.root = Tk()
self.right_frame = Frame(self.root)
self.left_frame = Frame(self.root)
self.image = None
self.scrollbarx = Scrollbar(self.left_frame, orient=HORIZONTAL)
self.scrollbary = Scrollbar(self.left_frame, orient=VERTICAL)
self.tree = None
self.files = {}
self.tree_index = None
self.selected_item = None
self.test_data_creator = TestDataCreator()
self.char_images = []
self.index = 0
self.panel = None
self.buttons = {}
def create_root(self):
self.root.title('Data Creator')
w = 720
h = 450
sw = self.root.winfo_screenwidth()
sh = self.root.winfo_screenheight()
x = (sw - w) / 2
y = (sh - h) / 2
self.root.geometry('%dx%d+%d+%d' % (w, h, x, y))
self.root.resizable(False, False)
self.add_left_frame()
self.add_buttons()
self.left_frame.pack(side=LEFT)
self.right_frame.pack(side=RIGHT)
def add_buttons(self):
self.panel = Label(self.right_frame, image=self.image)
self.panel.pack(side=TOP)
digit_frame = Frame(self.right_frame)
for i in [str(i) for i in range(10)]:
button_digit = Button(digit_frame, text=i, width=3)
button_digit.bind('<Button-1>', self.button_click)
button_digit.pack(side=LEFT, padx=2.)
digit_frame.pack(pady=5)
alpha_frame = Frame(self.right_frame)
for i in ['A', 'B', 'E', 'K', 'M', 'H', 'O', 'P', 'C', 'T', 'Y', 'X']:
button_alpha = Button(alpha_frame, text=i, width=3)
button_alpha.bind('<Button-1>', self.button_click)
button_alpha.pack(side=LEFT, padx=2.)
alpha_frame.pack(pady=5)
button_nothing = Button(self.right_frame, text='-', width=10)
button_nothing.bind('<Button-1>', self.button_click)
button_nothing.pack()
def create_tree(self):
if self.tree:
self.tree.destroy()
style = Style(self.left_frame)
style.configure('Calendar.Treeview', rowheight=50)
self.tree = Treeview(self.left_frame,
columns='#1',
height=400,
selectmode="extended",
yscrollcommand=self.scrollbary.set,
xscrollcommand=self.scrollbarx.set,
style='Calendar.Treeview')
self.scrollbary.config(command=self.tree.yview)
self.scrollbary.pack(side=RIGHT, fill=Y)
self.scrollbarx.config(command=self.tree.xview)
self.scrollbarx.pack(side=BOTTOM, fill=X)
self.tree.heading('#0', text='image', anchor=CENTER)
self.tree.heading('#1', text='file_name')
self.tree.column('#0', width=80, anchor=CENTER)
if self.files:
for key in self.files:
self.tree.insert("", 'end', values=key, image=self.files[key])
self.tree.pack(side=BOTTOM)
def add_left_frame(self):
load_button = Button(self.left_frame,
text="load files",
command=self.load_files)
load_button.pack(side=TOP, anchor=NW, pady=5)
start_button = Button(self.left_frame,
text='Start recognition',
command=self.run)
start_button.pack(side=TOP, anchor=NW)
self.create_tree()
def load_files(self):
file_names = fd.askopenfilenames(
filetypes=[('Image file',
'*.png'), ('Image file',
'*.jpg'), ('Image file', '*.jpeg')])
for file in file_names:
self.files[file] = ImageTk.PhotoImage(
Image.open(file).resize((60, 40)))
self.create_tree()
def button_click(self, event):
image_name = event.widget.cget('text')
if image_name != '-':
self.test_data_creator.multiply_image(self.char_images[self.index],
image_name)
self.index += 1
if self.index >= len(self.char_images):
del self.files[self.selected_item]
mb.showinfo("Recognition", "Characters are over!")
self.image = None
self.create_tree()
else:
self.image = ImageTk.PhotoImage(
Image.fromarray(self.char_images[self.index].image).resize(
(120, 200)))
self.panel.configure(image=self.image)
def run(self):
index = self.tree.selection()
if not index:
mb.showinfo("Choice image", "Not selected item")
return
index = index[0]
self.selected_item = ' '.join(self.tree.item(index)['values'])
self.char_images = self.test_data_creator.start(self.selected_item)
self.index = 0
if not self.char_images:
mb.showerror("Recognition", "Car numbers not recognized")
del self.files[self.selected_item]
self.create_tree()
self.image = None
else:
self.image = ImageTk.PhotoImage(
Image.fromarray(self.char_images[self.index].image).resize(
(120, 200)))
self.panel.configure(image=self.image)
class FormChildAED:
def __init__(self, frm_parent, title, connection):
self.connection = connection
self.directive = Message()
self.title = title
self.decide = True
self.id_selected = 0
self.frm_child_list = LabelFrame(frm_parent)
self.frm_child_crud = LabelFrame(frm_parent)
self.frm_child_crud.config(fg=TEXT_COLOR, font=SUBTITLE_FONT)
self.initialize_components()
def initialize_components(self):
"""
Method that initialize the visual components for each form associated with the local administration
"""
# Resources for the Forms
self.new_icon = PhotoImage(file=r"./Resources/create.png")
self.modify_icon = PhotoImage(file=r"./Resources/modify.png")
self.remove_icon = PhotoImage(file=r"./Resources/delete.png")
self.save_icon = PhotoImage(file=r"./Resources/save.png")
self.cancel_icon = PhotoImage(file=r"./Resources/cancel.png")
# Components for List Form
lbl_sep1 = Label(self.frm_child_list)
lbl_sep1.grid(row=0, column=0, padx=10, pady=25)
self.trv_available = Treeview(self.frm_child_list,
height=15,
columns=('N', 'Name', 'Surname',
'E-mail'))
self.trv_available.heading('#0', text='ID', anchor=CENTER)
self.trv_available.heading('#1', text='N', anchor=CENTER)
self.trv_available.heading('#2', text='Name', anchor=CENTER)
self.trv_available.heading('#3', text='Surname', anchor=CENTER)
self.trv_available.heading('#4', text='E-mail', anchor=CENTER)
self.trv_available.column('#0', width=0, minwidth=50, stretch=NO)
self.trv_available.column('#1', width=20, minwidth=20, stretch=NO)
self.trv_available.column('#2', width=200, minwidth=200, stretch=NO)
self.trv_available.column('#3', width=200, minwidth=200, stretch=NO)
self.trv_available.column('#4', width=400, minwidth=400, stretch=NO)
self.trv_available.grid(row=0, column=1, sticky=W, pady=25)
vsb_trv_av = Scrollbar(self.frm_child_list,
orient="vertical",
command=self.trv_available.yview)
vsb_trv_av.grid(row=0, column=2, pady=25, sticky=NS)
self.trv_available.configure(yscrollcommand=vsb_trv_av.set)
frm_aux4 = Frame(self.frm_child_list)
btn_new = Button(frm_aux4, image=self.new_icon, command=self.click_new)
btn_new.grid(row=0, column=0, pady=5, padx=5, sticky=E)
btn_new_ttp = CreateToolTip(btn_new, 'New ' + self.title.lower())
btn_edit = Button(frm_aux4,
image=self.modify_icon,
command=self.click_update)
btn_edit.grid(row=1, column=0, pady=5, padx=5, sticky=E)
btn_edit_ttp = CreateToolTip(btn_edit, 'Edit ' + self.title.lower())
btn_delete = Button(frm_aux4,
image=self.remove_icon,
command=self.click_delete)
btn_delete.grid(row=2, column=0, pady=5, padx=5, sticky=E)
btn_delete_ttp = CreateToolTip(btn_delete,
'Delete ' + self.title.lower())
frm_aux4.grid(row=0, column=3, pady=25, padx=25, sticky=NW)
# Components for CRUD FRM
lbl_name = Label(self.frm_child_crud, text='Name*')
lbl_name.config(fg=TEXT_COLOR, font=LABEL_FONT)
lbl_name.grid(row=0, column=0, pady=10, padx=20, sticky=W)
lbl_surname = Label(self.frm_child_crud, text='Surname*')
lbl_surname.config(fg=TEXT_COLOR, font=LABEL_FONT)
lbl_surname.grid(row=1, column=0, pady=10, padx=20, sticky=W)
lbl_email = Label(self.frm_child_crud, text='E-mail*')
lbl_email.config(fg=TEXT_COLOR, font=LABEL_FONT)
lbl_email.grid(row=2, column=0, pady=10, padx=20, sticky=W)
self.lbl_old_passwd = Label(self.frm_child_crud, text='Old password*')
self.lbl_old_passwd.config(fg=TEXT_COLOR, font=LABEL_FONT)
self.lbl_passwd = Label(self.frm_child_crud, text='New password*')
self.lbl_passwd.config(fg=TEXT_COLOR, font=LABEL_FONT)
self.lbl_passwd_conf = Label(self.frm_child_crud,
text='Confirm new password*')
self.lbl_passwd_conf.config(fg=TEXT_COLOR, font=LABEL_FONT)
self.txt_name = Entry(self.frm_child_crud)
self.txt_name.grid(row=0, column=1, pady=10, padx=20, sticky=W)
self.txt_surname = Entry(self.frm_child_crud)
self.txt_surname.grid(row=1, column=1, pady=10, padx=20, sticky=W)
self.txt_email = Entry(self.frm_child_crud)
self.txt_email.grid(row=2, column=1, pady=10, padx=20, sticky=W)
self.txt_old_passwd = Entry(self.frm_child_crud, show="*")
self.txt_passwd = Entry(self.frm_child_crud, show="*")
self.txt_passwd_conf = Entry(self.frm_child_crud, show="*")
sep_aux2 = Separator(self.frm_child_crud, orient=VERTICAL)
sep_aux2.grid(row=0, column=2, sticky=NS, rowspan=6)
frm_aux = Frame(self.frm_child_crud)
btn_save = Button(frm_aux,
image=self.save_icon,
command=self.click_save)
btn_save.grid(row=0, column=0, padx=5, pady=5, sticky=E)
btn_save_ttp = CreateToolTip(btn_save, 'Save ' + self.title.lower())
btn_cancel = Button(frm_aux,
image=self.cancel_icon,
command=self.click_cancel)
btn_cancel.grid(row=1, column=0, padx=5, pady=5, sticky=E)
btn_cancel_ttp = CreateToolTip(btn_cancel, 'Cancel')
frm_aux.grid(row=0, column=3, pady=10, padx=25, sticky=N, rowspan=6)
def retrieve_list(self):
"""
Method that retrieve users information from the server and displays them in the TreeView from
the List Form
"""
# Remove existing elements in the list
for item in self.trv_available.get_children():
self.trv_available.delete(item)
# Retrieve information from the server
if self.title == 'Experimenter':
self.directive = Message(action=17)
elif self.title == 'Designer':
self.directive = Message(action=22)
elif self.title == 'Administrator':
self.directive = Message(action=12)
else:
raise Exception('Error en recuperacion: tipo de usuario')
self.connection = self.directive.send_directive(self.connection)
# Adding elements in the list
for index, item in enumerate(self.connection.message.information):
elements = item.split('¥')
self.trv_available.insert('',
'end',
text=elements[0],
values=(index + 1, elements[1],
elements[2], elements[3]))
# Mark first element of the treeview if exist
if len(self.trv_available.get_children()) != 0:
self.trv_available.selection_set(
self.trv_available.get_children()[0])
def show_frm(self):
"""
Show the List form when the User administration is called
"""
self.retrieve_list()
self.frm_child_list.grid(row=1,
column=0,
columnspan=9,
rowspan=8,
pady=10,
padx=10)
def hide_frm(self):
"""
Hide the User administration Forms
"""
self.clear_fields()
self.frm_child_list.grid_forget()
self.frm_child_crud.grid_forget()
def click_new(self):
"""
Initialize CRUD Form for creating a new user.
"""
self.user = Designer()
self.frm_child_list.grid_forget()
self.txt_name.focus_set()
self.frm_child_crud['text'] = 'New ' + self.title.lower()
self.lbl_passwd.grid(row=3, column=0, pady=10, padx=20, sticky=W)
self.lbl_passwd_conf.grid(row=4, column=0, pady=10, padx=20, sticky=W)
self.txt_passwd.grid(row=3, column=1, pady=10, padx=20, sticky=W)
self.txt_passwd_conf.grid(row=4, column=1, pady=10, padx=20, sticky=W)
self.frm_child_crud.grid(row=1,
column=0,
columnspan=9,
rowspan=8,
pady=10,
padx=10)
def click_update(self):
"""
Initialize CRUD Form for updating a user. It loads information of selected User into visual components
"""
if len(self.trv_available.selection()) == 1:
id_selected = int(
self.trv_available.item(
self.trv_available.selection())['text'])
if self.title == 'Experimenter':
self.directive = Message(action=20, information=[id_selected])
elif self.title == 'Designer':
self.directive = Message(action=25, information=[id_selected])
else:
self.directive = Message(action=15, information=[id_selected])
self.connection = self.directive.send_directive(self.connection)
if self.connection.message.action == 5: # An error ocurred while trying to update the item
messagebox.showerror(
parent=self.frm_child_list,
title='Can not update the item',
message=self.connection.message.information[0])
else:
self.user = Designer(
id=id_selected,
name=self.connection.message.information[0],
surname=self.connection.message.information[1],
user=self.connection.message.information[2],
password=self.connection.message.information[3])
self.txt_name.insert(0, self.user.name)
self.txt_surname.insert(0, self.user.surname)
self.txt_email.insert(0, self.user.user)
self.frm_child_list.grid_forget()
self.txt_name.focus_set()
self.frm_child_crud['text'] = 'Update ' + self.title.lower()
self.lbl_old_passwd.grid(row=3,
column=0,
pady=10,
padx=20,
sticky=W)
self.lbl_passwd.grid(row=4,
column=0,
pady=10,
padx=20,
sticky=W)
self.lbl_passwd_conf.grid(row=5,
column=0,
pady=10,
padx=20,
sticky=W)
self.txt_old_passwd.grid(row=3,
column=1,
pady=10,
padx=20,
sticky=W)
self.txt_passwd.grid(row=4,
column=1,
pady=10,
padx=20,
sticky=W)
self.txt_passwd_conf.grid(row=5,
column=1,
pady=10,
padx=20,
sticky=W)
self.frm_child_crud.grid(row=1,
column=0,
columnspan=9,
rowspan=8,
pady=10,
padx=10)
else:
messagebox.showwarning(parent=self.frm_child_list,
title='No selection',
message='You must select one item')
def click_delete(self):
"""
Method that removes a selected user from the initial list (changes are updated in DB)
"""
if len(self.trv_available.selection()) == 1:
decision = messagebox.askyesno(
parent=self.frm_child_list,
title='Confirmation',
message='Are you sure you want to delete the item?')
if decision:
id_selected = int(
self.trv_available.item(
self.trv_available.selection())['text'])
if self.title == 'Experimenter':
self.directive = Message(action=19,
information=[id_selected])
elif self.title == 'Designer':
self.directive = Message(action=24,
information=[id_selected])
else:
self.directive = Message(action=14,
information=[id_selected])
self.connection = self.directive.send_directive(
self.connection)
if self.connection.message.action == 5: # An error ocurred while deleting the item
messagebox.showerror(
parent=self.frm_child_list,
title='Can not delete the item',
message=self.connection.message.information[0])
else:
self.retrieve_list()
else:
messagebox.showwarning(parent=self.frm_child_list,
title='No selection',
message='You must select one item')
def click_save(self):
"""
Saves information of the user inserted into the visual components and sends to the server
"""
if self.validate_fields():
self.user.name = self.txt_name.get()
self.user.surname = self.txt_surname.get()
self.user.user = self.txt_email.get()
self.user.password = self.txt_passwd.get()
if self.user.id == 0: # Creating an user
if self.title == 'Experimenter':
self.directive = Message(
action=16,
information=[
self.user.name, self.user.surname, self.user.user,
hashlib.sha1(
self.user.password.encode()).hexdigest()
])
elif self.title == 'Designer':
self.directive = Message(
action=21,
information=[
self.user.name, self.user.surname, self.user.user,
hashlib.sha1(
self.user.password.encode()).hexdigest()
])
else:
self.directive = Message(
action=11,
information=[
self.user.name, self.user.surname, self.user.user,
hashlib.sha1(
self.user.password.encode()).hexdigest()
])
else: # Updating an user
if self.title == 'Experimenter':
self.directive = Message(
action=18,
information=[
self.user.id, self.user.name, self.user.surname,
self.user.user,
hashlib.sha1(
self.user.password.encode()).hexdigest()
])
elif self.title == 'Designer':
self.directive = Message(
action=23,
information=[
self.user.id, self.user.name, self.user.surname,
self.user.user,
hashlib.sha1(
self.user.password.encode()).hexdigest()
])
else:
self.directive = Message(action=13,
information=[
self.user.id, self.user.name,
self.user.surname,
self.user.user,
self.user.password
])
self.connection = self.directive.send_directive(self.connection)
if self.connection.message.action == 5:
messagebox.showwarning(parent=self.frm_child_crud,
title='Repeated e-mail',
message=self.connection.message.comment)
else:
self.clear_fields()
self.frm_child_crud.grid_forget()
self.show_frm()
def click_cancel(self):
"""
Function activated when 'Cancel' button is pressed in frm_child_crud
"""
decision = True
if self.txt_name.get() != self.user.name or \
self.txt_surname.get() != self.user.surname or \
self.txt_email.get() != self.user.user or len(self.txt_passwd.get()) != 0 or \
len(self.txt_passwd_conf.get()) != 0:
if self.user.id != 0 and len(
self.txt_passwd_conf.get()) != 0 or self.user.id == 0:
decision = messagebox.askyesno(
parent=self.frm_child_crud,
title='Cancel',
message='Are you sure you want to cancel?')
if decision:
self.clear_fields()
self.frm_child_crud.grid_forget()
self.show_frm()
def validate_fields(self):
if len(self.txt_name.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must insert a name for the {}'.format(
self.title.lower()))
return False
if len(self.txt_surname.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must insert a surname for the {}'.format(
self.title.lower()))
return False
if len(self.txt_email.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must insert an e-mail for the {}'.format(
self.title.lower()))
return False
# If updating an user
if self.user.id != 0 and len(self.txt_old_passwd.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must insert the old password for the {}'.format(
self.title.lower()))
return False
if len(self.txt_passwd.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must insert a new password for the {}'.format(
self.title.lower()))
return False
if len(self.txt_passwd_conf.get()) == 0:
messagebox.showwarning(
parent=self.frm_child_crud,
title='Missing information',
message='You must confirm the new password for the {}'.format(
self.title.lower()))
return False
if self.txt_passwd.get() != self.txt_passwd_conf.get():
messagebox.showwarning(
parent=self.frm_child_crud,
title='Password field',
message=
'The new password you provided does not match the confirmation'
)
return False
# If updating an user
if self.user.id != 0 and self.user.password != hashlib.sha1(
self.txt_old_passwd.get().encode()).hexdigest():
messagebox.showwarning(parent=self.frm_child_crud,
title='Old password field',
message='The old password is incorrect')
return False
return True
def clear_fields(self):
self.txt_name.delete(0, END)
self.txt_surname.delete(0, END)
self.txt_email.delete(0, END)
self.txt_old_passwd.delete(0, END)
self.txt_passwd.delete(0, END)
self.txt_passwd_conf.delete(0, END)
self.lbl_old_passwd.grid_forget()
self.lbl_passwd.grid_forget()
self.lbl_passwd_conf.grid_forget()
self.txt_old_passwd.grid_forget()
self.txt_passwd.grid_forget()
self.txt_passwd_conf.grid_forget()
class DialogPluginManager(Toplevel):
def __init__(self, mainWin, modulesWithNewerFileDates):
super(DialogPluginManager, self).__init__(mainWin.parent)
self.ENABLE = _("Enable")
self.DISABLE = _("Disable")
self.parent = mainWin.parent
self.cntlr = mainWin
# copy plugins for temporary display
self.pluginConfig = PluginManager.pluginConfig
self.pluginConfigChanged = False
self.uiClassMethodsChanged = False
self.modulesWithNewerFileDates = modulesWithNewerFileDates
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", self.parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.title(_("Plug-in Manager"))
frame = Frame(self)
# left button frame
buttonFrame = Frame(frame, width=40)
buttonFrame.columnconfigure(0, weight=1)
addLabel = Label(buttonFrame, text=_("Find plug-in modules:"), wraplength=60, justify="center")
addLocalButton = Button(buttonFrame, text=_("Locally"), command=self.findLocally)
ToolTip(addLocalButton, text=_("File chooser allows selecting python module files to add (or reload) plug-ins, from the local file system."), wraplength=240)
addWebButton = Button(buttonFrame, text=_("On Web"), command=self.findOnWeb)
ToolTip(addWebButton, text=_("Dialog to enter URL full path to load (or reload) plug-ins, from the web or local file system."), wraplength=240)
addLabel.grid(row=0, column=0, pady=4)
addLocalButton.grid(row=1, column=0, pady=4)
addWebButton.grid(row=2, column=0, pady=4)
buttonFrame.grid(row=0, column=0, rowspan=2, sticky=(N, S, W), padx=3, pady=3)
# right tree frame (plugins already known to arelle)
modulesFrame = Frame(frame, width=700)
vScrollbar = Scrollbar(modulesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(modulesFrame, orient=HORIZONTAL)
self.modulesView = Treeview(modulesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=7)
self.modulesView.grid(row=0, column=0, sticky=(N, S, E, W))
self.modulesView.bind('<<TreeviewSelect>>', self.moduleSelect)
hScrollbar["command"] = self.modulesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.modulesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
modulesFrame.columnconfigure(0, weight=1)
modulesFrame.rowconfigure(0, weight=1)
modulesFrame.grid(row=0, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.modulesView.focus_set()
self.modulesView.column("#0", width=120, anchor="w")
self.modulesView.heading("#0", text=_("Name"))
self.modulesView["columns"] = ("author", "ver", "status", "date", "update", "descr", "license")
self.modulesView.column("author", width=100, anchor="w", stretch=False)
self.modulesView.heading("author", text=_("Author"))
self.modulesView.column("ver", width=50, anchor="w", stretch=False)
self.modulesView.heading("ver", text=_("Version"))
self.modulesView.column("status", width=50, anchor="w", stretch=False)
self.modulesView.heading("status", text=_("Status"))
self.modulesView.column("date", width=70, anchor="w", stretch=False)
self.modulesView.heading("date", text=_("File Date"))
self.modulesView.column("update", width=50, anchor="w", stretch=False)
self.modulesView.heading("update", text=_("Update"))
self.modulesView.column("descr", width=200, anchor="w", stretch=False)
self.modulesView.heading("descr", text=_("Description"))
self.modulesView.column("license", width=70, anchor="w", stretch=False)
self.modulesView.heading("license", text=_("License"))
classesFrame = Frame(frame)
vScrollbar = Scrollbar(classesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(classesFrame, orient=HORIZONTAL)
self.classesView = Treeview(classesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=5)
self.classesView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.classesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.classesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
classesFrame.columnconfigure(0, weight=1)
classesFrame.rowconfigure(0, weight=1)
classesFrame.grid(row=1, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.classesView.focus_set()
self.classesView.column("#0", width=200, anchor="w")
self.classesView.heading("#0", text=_("Class"))
self.classesView["columns"] = ("modules")
self.classesView.column("modules", width=500, anchor="w", stretch=False)
self.classesView.heading("modules", text=_("Modules"))
# bottom frame module info details
moduleInfoFrame = Frame(frame, width=700)
moduleInfoFrame.columnconfigure(1, weight=1)
self.moduleNameLabel = Label(moduleInfoFrame, wraplength=600, justify="left",
font=font.Font(family='Helvetica', size=12, weight='bold'))
self.moduleNameLabel.grid(row=0, column=0, columnspan=4, sticky=W)
self.moduleAuthorHdr = Label(moduleInfoFrame, text=_("author:"), state=DISABLED)
self.moduleAuthorHdr.grid(row=1, column=0, sticky=W)
self.moduleAuthorLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleAuthorLabel.grid(row=1, column=1, columnspan=3, sticky=W)
self.moduleDescrHdr = Label(moduleInfoFrame, text=_("description:"), state=DISABLED)
self.moduleDescrHdr.grid(row=2, column=0, sticky=W)
self.moduleDescrLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDescrLabel.grid(row=2, column=1, columnspan=3, sticky=W)
self.moduleClassesHdr = Label(moduleInfoFrame, text=_("classes:"), state=DISABLED)
self.moduleClassesHdr.grid(row=3, column=0, sticky=W)
self.moduleClassesLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleClassesLabel.grid(row=3, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleClassesLabel, text=_("List of classes that this plug-in handles."), wraplength=240)
self.moduleUrlHdr = Label(moduleInfoFrame, text=_("URL:"), state=DISABLED)
self.moduleUrlHdr.grid(row=4, column=0, sticky=W)
self.moduleUrlLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleUrlLabel.grid(row=4, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleUrlLabel, text=_("URL of plug-in module (local file path or web loaded file)."), wraplength=240)
self.moduleDateHdr = Label(moduleInfoFrame, text=_("date:"), state=DISABLED)
self.moduleDateHdr.grid(row=5, column=0, sticky=W)
self.moduleDateLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDateLabel.grid(row=5, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleDateLabel, text=_("Date of currently loaded module file (with parenthetical node when an update is available)."), wraplength=240)
self.moduleLicenseHdr = Label(moduleInfoFrame, text=_("license:"), state=DISABLED)
self.moduleLicenseHdr.grid(row=6, column=0, sticky=W)
self.moduleLicenseLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleLicenseLabel.grid(row=6, column=1, columnspan=3, sticky=W)
self.moduleEnableButton = Button(moduleInfoFrame, text=self.ENABLE, state=DISABLED, command=self.moduleEnable)
ToolTip(self.moduleEnableButton, text=_("Enable/disable plug in."), wraplength=240)
self.moduleEnableButton.grid(row=7, column=1, sticky=E)
self.moduleReloadButton = Button(moduleInfoFrame, text=_("Reload"), state=DISABLED, command=self.moduleReload)
ToolTip(self.moduleReloadButton, text=_("Reload/update plug in."), wraplength=240)
self.moduleReloadButton.grid(row=7, column=2, sticky=E)
self.moduleRemoveButton = Button(moduleInfoFrame, text=_("Remove"), state=DISABLED, command=self.moduleRemove)
ToolTip(self.moduleRemoveButton, text=_("Remove plug in from plug in table (does not erase the plug in's file)."), wraplength=240)
self.moduleRemoveButton.grid(row=7, column=3, sticky=E)
moduleInfoFrame.grid(row=2, column=0, columnspan=5, sticky=(N, S, E, W), padx=3, pady=3)
moduleInfoFrame.config(borderwidth=4, relief="groove")
okButton = Button(frame, text=_("Close"), command=self.ok)
ToolTip(okButton, text=_("Accept and changes (if any) and close dialog."), wraplength=240)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
ToolTip(cancelButton, text=_("Cancel changes (if any) and close dialog."), wraplength=240)
okButton.grid(row=3, column=3, sticky=(S,E), pady=3)
cancelButton.grid(row=3, column=4, sticky=(S,E), pady=3, padx=3)
self.loadTreeViews()
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=1)
frame.columnconfigure(1, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeViews(self):
self.selectedModule = None
# clear previous treeview entries
for previousNode in self.modulesView.get_children(""):
self.modulesView.delete(previousNode)
for i, moduleItem in enumerate(sorted(self.pluginConfig.get("modules", {}).items())):
moduleInfo = moduleItem[1]
name = moduleInfo.get("name", moduleItem[0])
node = self.modulesView.insert("", "end", name, text=name)
self.modulesView.set(node, "author", moduleInfo.get("author"))
self.modulesView.set(node, "ver", moduleInfo.get("version"))
self.modulesView.set(node, "status", moduleInfo.get("status"))
self.modulesView.set(node, "date", moduleInfo.get("fileDate"))
if name in self.modulesWithNewerFileDates:
self.modulesView.set(node, "update", _("available"))
self.modulesView.set(node, "descr", moduleInfo.get("description"))
self.modulesView.set(node, "license", moduleInfo.get("license"))
# clear previous treeview entries
for previousNode in self.classesView.get_children(""):
self.classesView.delete(previousNode)
for i, classItem in enumerate(sorted(self.pluginConfig.get("classes", {}).items())):
className, moduleList = classItem
node = self.classesView.insert("", "end", className, text=className)
self.classesView.set(node, "modules", ', '.join(moduleList))
self.moduleSelect() # clear out prior selection
def ok(self, event=None):
if self.pluginConfigChanged:
PluginManager.pluginConfig = self.pluginConfig
PluginManager.pluginConfigChanged = True
PluginManager.reset() # force reloading of modules
if self.uiClassMethodsChanged: # may require reloading UI
if messagebox.askyesno(_("User interface plug-in change"),
_("A change in plug-in class methods may have affected the menus "
"of the user interface. It may be necessary to restart Arelle to "
"access the menu entries or the changes to their plug-in methods. \n\n"
"Should Arelle restart with changed user interface language, "
"(if there are any unsaved changes they would be lost!)?"),
parent=self):
self.cntlr.uiThreadQueue.put((self.cntlr.quit, [None, True]))
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def moduleSelect(self, *args):
node = (self.modulesView.selection() or (None,))[0]
moduleInfo = self.pluginConfig.get("modules", {}).get(node)
if moduleInfo:
self.selectedModule = node
name = moduleInfo["name"]
self.moduleNameLabel.config(text=name)
self.moduleAuthorHdr.config(state=ACTIVE)
self.moduleAuthorLabel.config(text=moduleInfo["author"])
self.moduleDescrHdr.config(state=ACTIVE)
self.moduleDescrLabel.config(text=moduleInfo["description"])
self.moduleClassesHdr.config(state=ACTIVE)
self.moduleClassesLabel.config(text=', '.join(moduleInfo["classMethods"]))
self.moduleUrlHdr.config(state=ACTIVE)
self.moduleUrlLabel.config(text=moduleInfo["moduleURL"])
self.moduleDateHdr.config(state=ACTIVE)
self.moduleDateLabel.config(text=moduleInfo["fileDate"] + " " +
(_("(an update is available)") if name in self.modulesWithNewerFileDates else ""))
self.moduleLicenseHdr.config(state=ACTIVE)
self.moduleLicenseLabel.config(text=moduleInfo["license"])
self.moduleEnableButton.config(state=ACTIVE,
text={"enabled":self.DISABLE,
"disabled":self.ENABLE}[moduleInfo["status"]])
self.moduleReloadButton.config(state=ACTIVE)
self.moduleRemoveButton.config(state=ACTIVE)
else:
self.selectedModule = None
self.moduleNameLabel.config(text="")
self.moduleAuthorHdr.config(state=DISABLED)
self.moduleAuthorLabel.config(text="")
self.moduleDescrHdr.config(state=DISABLED)
self.moduleDescrLabel.config(text="")
self.moduleClassesHdr.config(state=DISABLED)
self.moduleClassesLabel.config(text="")
self.moduleUrlHdr.config(state=DISABLED)
self.moduleUrlLabel.config(text="")
self.moduleDateHdr.config(state=DISABLED)
self.moduleDateLabel.config(text="")
self.moduleLicenseHdr.config(state=DISABLED)
self.moduleLicenseLabel.config(text="")
self.moduleEnableButton.config(state=DISABLED, text=self.ENABLE)
self.moduleReloadButton.config(state=DISABLED)
self.moduleRemoveButton.config(state=DISABLED)
def findLocally(self):
filename = self.cntlr.uiFileDialog("open",
owner=self,
title=_("Choose plug-in module file"),
initialdir=self.cntlr.config.setdefault("pluginOpenDir","."),
filetypes=[(_("Python files"), "*.py")],
defaultextension=".py")
if filename:
self.cntlr.config["pluginOpenDir"] = os.path.dirname(filename)
moduleInfo = PluginManager.moduleModuleInfo(filename)
self.loadFoundModuleInfo(moduleInfo, filename)
def findOnWeb(self):
url = DialogURL.askURL(self)
if url: # url is the in-cache or local file
moduleInfo = PluginManager.moduleModuleInfo(url)
self.cntlr.showStatus("") # clear web loading status
self.loadFoundModuleInfo(moduleInfo, url)
def loadFoundModuleInfo(self, moduleInfo, url):
if moduleInfo and moduleInfo.get("name"):
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
else:
messagebox.showwarning(_("Module is not a plug-in"),
_("File does not contain a python program with an appropriate __pluginInfo__ declaration: \n\n{0}")
.format(url),
parent=self)
def removePluginConfigModuleInfo(self, name):
moduleInfo = self.pluginConfig["modules"].get(name)
if moduleInfo:
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].get(classMethod)
if classMethods and name in classMethods:
classMethods.remove(name)
if not classMethods: # list has become unused
del self.pluginConfig["classes"][classMethod] # remove class
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
del self.pluginConfig["modules"][name]
self.pluginConfigChanged = True
def addPluginConfigModuleInfo(self, moduleInfo):
name = moduleInfo["name"]
self.removePluginConfigModuleInfo(name) # remove any prior entry for this module
self.modulesWithNewerFileDates.discard(name) # no longer has an update available
self.pluginConfig["modules"][name] = moduleInfo
# add classes
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].setdefault(classMethod, [])
if name not in classMethods:
classMethods.append(name)
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
self.pluginConfigChanged = True
def moduleEnable(self):
if self.selectedModule in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][self.selectedModule]
if self.moduleEnableButton['text'] == self.ENABLE:
moduleInfo["status"] = "enabled"
self.moduleEnableButton['text'] = self.DISABLE
elif self.moduleEnableButton['text'] == self.DISABLE:
moduleInfo["status"] = "disabled"
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
def moduleReload(self):
if self.selectedModule in self.pluginConfig["modules"]:
url = self.pluginConfig["modules"][self.selectedModule].get("moduleURL")
if url:
moduleInfo = PluginManager.moduleModuleInfo(url, reload=True)
if moduleInfo:
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
self.cntlr.showStatus(_("{0} reloaded").format(moduleInfo.get("name")), clearAfter=5000)
else:
messagebox.showwarning(_("Module error"),
_("File or module cannot be reloaded: \n\n{0}")
.format(url),
parent=self)
def moduleRemove(self):
if self.selectedModule in self.pluginConfig["modules"]:
self.removePluginConfigModuleInfo(self.selectedModule)
self.pluginConfigChanged = True
self.loadTreeViews()
class Window:
def fillTree(self,path, parent, list):
for file in os.listdir(path):
abspath = os.path.join(path,file)
color = ""
treelist = None
for mini in list:
if abspath in mini:
color = 'red'
treelist = mini
else:
for lk in mini:
if abspath in lk:
color = 'purple'
child = None
if color == 'red':
child = self.tree.insert(parent,'end',text=file,open=False,tags=(abspath,'red',str(treelist)),)
elif color == 'purple':
child = self.tree.insert(parent,'end',text=file,open=False,tags=(abspath,'purple'))
else:
child = self.tree.insert(parent,'end',text=file,open=False,tags=(abspath,'white'))
if(os.path.isdir(abspath)):
self.tree.insert(child,'end',text='',open=False)
def __init__(self,list,dirlist):
self.root = Tk()
self.root.wm_title("Duplicate_Files")
self.min = None
self.list = list
self.root.geometry('600x600+0+0')
self.tree = Treeview(self.root ,height=15)
self.tree.pack(expand='yes',fill='both')
self.tree.heading('#0',text="files")
self.tree.tag_configure('red',foreground='red')
self.tree.tag_configure('purple',foreground='#cc00ff')
self.tree.bind("<Double-1>",self.onDoubleClick)
self.tree.bind("<<TreeviewOpen>>",self.onOpen)
self.tree.bind("<<TreeviewClose>>",self.onClose)
for path in dirlist:
branch = self.tree.insert('','end',text=path,open=True,tags=(path,'white'))
self.fillTree(path,branch,list)
self.root.mainloop()
def onDoubleClick(self,event):
item = self.tree.selection()[0]
print ("clicked" + str(self.tree.item(item,'tags')[0]))
if str(self.tree.item(item,'tags')[1]) == "red":
list_of_files = ast.literal_eval(str(self.tree.item(item,'tags')[2]))
if self.min != None:
if self.min.mini.winfo_exists():
self.min.mini.destroy()
self.min = MiniWindow(self.root,list_of_files)
def onOpen(self,event):
item = self.tree.selection()[0]
if self.tree.parent(item) != '':
if len(self.tree.get_children(item))>0:
self.tree.delete(self.tree.get_children(item))
abspath = str(self.tree.item(item,'tags')[0])
if(os.path.isdir(abspath)):
self.fillTree(abspath, item,self.list)
def onClose(self,event):
item = self.tree.selection()[0]
if self.tree.parent(item) != '':
if len(self.tree.get_children(item))>0:
self.tree.delete(self.tree.get_children(item))
class NameView(object):
"""Shows a treeview of unique names."""
def __init__(self, master, names):
self.widget = Frame(master)
self._tree = Treeview(self.widget, columns='name')
self._tree.grid(row=0, column=0, sticky=(N, S, W, E))
self._tree.view = self
self.widget.columnconfigure(0, weight=1)
self.widget.rowconfigure(0, weight=1)
self._tree.column('name', width=50)
self._tree['show'] = 'tree'
actions = {
'edit': lambda e: self.edit(),
'search': lambda e: self.search(),
'focus_next': lambda e: self.focus_next(),
'focus_prev': lambda e: self.focus_prev(),
'select':
lambda e: self._tree.selection_toggle(self._tree.focus()),
'clear_selection': lambda e: self._tree.selection_set([])
}
kb.make_bindings(kb.tagview, actions, self._tree.bind)
self._iids = dict()
self._names = dict()
logger.debug('Names: %s', names)
self.widget.focus_set = self._tree.focus_set
for name in sorted(names):
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
self._scroll = Scrollbar(self.widget, command=self._tree.yview)
self._tree['yscrollcommand'] = self._scroll.set
self._scroll.grid(row=0, column=1, sticky=(N, S))
self.widget.columnconfigure(1, weight=0)
def selection(self):
logger.debug('Selection: %s', self._tree.selection())
return [self._names[iid] for iid in self._tree.selection()]
def edit(self):
self._tree.event_generate('<<NameViewEdit>>')
def search(self):
if len(self._tree.selection()) == 0:
self._tree.selection_add(self._tree.focus())
self._tree.event_generate('<<NameViewSearch>>')
def append(self, names):
logger.debug('Append names: %s', names)
for name in names:
if name not in self._names.values():
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
def delete(self, name):
self._tree.delete(self._iids[name])
del self._names[self._iids[name]]
del self._iids[name]
def _focus(self, iid):
self._tree.focus(iid)
self._tree.see(iid)
def focus_next(self):
cur_iid = self._tree.focus()
next_iid = self._tree.next(cur_iid)
if next_iid == '':
iids = self._tree.get_children()
next_iid = iids[0]
self._focus(next_iid)
def focus_prev(self):
cur_iid = self._tree.focus()
prev_iid = self._tree.prev(cur_iid)
if prev_iid == '':
iids = self._tree.get_children()
prev_iid = iids[-1]
self._focus(prev_iid)
def jump_to(self, name):
try:
iid = self._iids[name]
self._focus(iid)
except KeyError:
pass
def get_names(self):
return tuple(self._names.values())
def set(self, names):
self._tree.delete(*self._iids.values())
self._iids.clear()
self._names.clear()
for name in sorted(names):
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
class Expenses(Frame):
# Creates the first option menus in the expense window
def createOptionButtons(self):
self.master.title("Expenses")
# Creates the add item to inventory button
addItem = Button(root, text="Add item to inventory", command=lambda: self.sequence(self.addItem))#addItem(master))) # This button will send to the user to the add item page
addItem.place(x = 130, y = 100)
# Creates the view items in inventory button
inventoryButton = Button(root, text="View items in inventory", command=lambda: self.sequence(self.viewInveroty)) # This button will send the user to the view inventory page
inventoryButton.place(x = 130, y = 150)
# Create the total cost button
totalCost = Button(root, text="Total Cost", command=lambda: self.sequence(self.viewTotalCost))
totalCost.place(x = 130, y = 200)
# Creates the back button
backButton = Button(root, text="Back", command=returnHome) # This button will return the user to the main page. Still working on it.
backButton.place(x = 50, y = 350)
# Creates the add item to inventory button and entries
def addItem(self):
self.master.title("Add new item") # Changes the title of the page to Add New Item
# Creates a label called nameOfItems and an entry called itemName
nameOfItem = Label(root, text="Item Name: ")
nameOfItem.place(x = 110, y = 100)
self.itemName = Entry(root) # This will allow the user to enter the name of the item that they will be adding
self.itemName.place(x = 190, y = 100)
# Creates the label called itemTypeLabel and a drop down menu called itemTypeChoice
itemTypeLabel = Label(root, text = "Item's type: ")
itemTypeLabel.place(x = 110, y = 160)
self.itemTypeChoice = StringVar(root) # This makes itemTypeChoice a permanent String
self.itemTypeChoice.set("Tree") # Tree is set to the default string of itemTypeChoice
typeChoices = OptionMenu(root, self.itemTypeChoice, "Tree", "Animal", "Machine") # Drop down menu is created and options Tree, Animal, and Machine are added to the menu
typeChoices.place(x = 190, y = 160)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.createOptionButtons))
backButton.place(x = 50, y = 350)
# Next button
nextButton = Button(root, text = "Next", command=self.saveNameAndType) #This button will send the user to the add inventory page
nextButton.place(x = 350, y = 350)
# Function that creates a new item object and assigns it a name and the type
def saveNameAndType(self):
name = self.itemName.get()
self.item = Inventory(name)
itemType = self.itemTypeChoice.get()
self.item.itemType = itemType
self.sequence(self.addToInventory)
# Creates the add to inventory options
def addToInventory(self):
self.master.title("Add %s to %s inventory" % (self.item.name, self.item.itemType))
# This assigns the variables month, day, and year to be value holder for integer values
# They are also set to be values of the class expenses (by using self) so that they can
# be used in the function updateDay and SaveDate
self.month = IntVar(self)
self.day = IntVar(self)
self.year = IntVar(self)
# This trace function is used to keep track of when the selected months and years change. This is
# done to adjust the days of the month according to the month or the year
self.month.trace('w', self.updateDay)
self.year.trace('w', self.updateDay)
numMonths = self.nums(1, 12) # Runs the nums function that creates a list from 1 to 12
numYears = self.nums(2015, 2030) # Runs the nums function that creates a list from 2015 to 2030
# This creates the drop down menu and assigns the options is the menu. The day menu is left empty and
# is assigned in the updateDay function
self.optionmenu_month = OptionMenu(root, self.month, *numMonths)
self.optionmenu_day = OptionMenu(root, self.day, '')
self.optionmenu_year = OptionMenu(root, self.year, *numYears)
# Sets the default value of the month and year options to 1 and 2015 respectively
self.month.set(numMonths[0])
self.year.set(numYears[0])
self.optionmenu_month.place(x = 100, y = 120)
self.optionmenu_day.place(x = 150, y = 120)
self.optionmenu_year.place(x = 200, y = 120)
datePurchased = Label(root, text = "Date Purchased")
datePurchased.place(x = 150, y = 95)
quantityPurchasedLabel = Label(root, text="Amount purchased:")
quantityPurchasedLabel.place(x = 50, y = 180)
self.quantityPurchasedEntry = Entry(root, bd=5) # Creates input box for user to insert the amount of items purchased
self.quantityPurchasedEntry.place(x = 180, y = 180)
pricePaidLabe = Label(root, text="Price paid for all: ")
pricePaidLabe.place(x = 50, y = 210)
self.pricePaidEntry = Entry(root, bd=5) # Creates input box for user to insert the price paid for the item
self.pricePaidEntry.place(x = 180, y = 210)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.addItem))
backButton.place(x = 50, y = 350)
nextButton = Button(root, text = "Next", command=self.saveQuanAndPrice)
nextButton.place(x = 350, y = 350)
# This function will update the days of the month according to the selected month and year
def updateDay(self, *args):
# The .get() will obtain the selected month and year values from the drop down menu above
month = self.month.get()
year = self.year.get()
# Creates a loop which chooses the last day of the month according to the month or the year
if month == 1 or month == 3 or month == 5 or month == 7 or month == 8 or month == 10 or month == 12:
lastDay = 31
elif month == 4 or month == 6 or month == 9 or month == 11:
lastDay = 30
# This elif loop uses the leap year formula at account for leap years
elif month == 2:
if (year % 4) == 0:
if (year % 100) == 0:
if (year % 400) == 0:
lastDay = 29
else:
lastDay = 28
else:
lastDay = 29
else:
lastDay = 28
numDays = self.nums(1,lastDay)
# Assigns menu to the day drop down menu and deletes all of the options in the menu
menu = self.optionmenu_day['menu']
menu.delete(0, 'end')
# Loop for generating the new day menu
for day in numDays:
menu.add_command(label=day, command=lambda d = day: self.day.set(d))
self.day.set(1)
# Function that creates the range of numbers for the drop down menu
def nums(self, numStart, numEnd):
num = range(numStart, numEnd + 1)
return num
# Function that assigns the price and quantity to an item
def saveQuanAndPrice(self):
self.item.price = self.pricePaidEntry.get()
self.item.quantity = self.quantityPurchasedEntry.get()
self.saveDate()
self.sequence(self.confirmation)
# Function that assigns the purchase date to an item
def saveDate(self):
self.item.purchaseMonth = self.month.get()
self.item.purchaseDay = self.day.get()
self.item.purchaseYear = self.year.get()
self.item.purchaseDate = ("%s/%s/%s" % (self.item.purchaseMonth, self.item.purchaseDay, self.item.purchaseYear))
# Function that displays the user inputted information
def confirmation(self):
self.master.title("Confirm %s information" % self.item.name)
name = Label(root, text="Name of item: ")
name.place(x = 100, y = 50)
itemName = Label(root, text=self.item.name)
itemName.place(x = 100, y = 65)
type = Label(root, text="%s type: " % self.item.name)
type.place(x = 100, y = 90)
itemType = Label(root, text=self.item.itemType)
itemType.place(x = 100, y = 105)
quantity = Label(root, text="How many %s were bought?" % self.item.name)
quantity.place(x = 100, y = 130)
itemQuantity = Label(root, text=self.item.quantity)
itemQuantity.place(x = 100, y = 145)
price = Label(root, text="How much did the %s %s cost?" % (self.item.quantity, self.item.name))
price.place(x = 100, y = 170)
itemPrice = Label(root, text=self.item.price)
itemPrice.place(x = 100, y = 185)
date = Label(root, text="When were %s bought?" % self.item.name)
date.place(x = 100, y = 210)
itemDate = Label(root, text=self.item.purchaseDate)
itemDate.place(x = 100, y = 225)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.addToInventory))
backButton.place(x = 50, y = 350)
startOverButton = Button(root, text = "Start Over", command=lambda: self.sequence(self.createOptionButtons))
startOverButton.place(x = 200, y = 350)
confirmButton = Button(root, text = "Confirm", command=lambda: self.sequence(self.addToDatabase))
confirmButton.place(x = 320, y = 350)
# Adds the item to the database
def addToDatabase(self):
self.inventoryDB.insertInvetory(self.item)
return self.successful()
# Displays a success message when the item is added
def successful(self):
self.master.title("%s was added successfully!" % self.item.name)
succMessage = Message(root, text = "%s was successfully added to the %s list!" % (self.item.name, self.item.itemType))
succMessage.place(x = 150, y = 150)
startOverButton = Button(root, text = "Start Over", command=lambda: self.sequence(self.createOptionButtons))#self.saveNameAndType(itemName)))#(self.saveNameAndType(itemName))) # (itemName)))# lambda: self.sequence(self.test)) #This button will send the user to the add inventory page
startOverButton.place(x = 150, y = 350)
# Used to view the inventory
def viewInveroty(self):
# Creates the label called chooseTypeLabel and a drop down menu called chooseItemType
chooseTypeLabel = Label(root, text = "Item's type: ")
chooseTypeLabel.place(x = 110, y = 160)
self.chooseItemType = StringVar(root) # The drop down menu is created and assigned to chooseItemType
self.chooseItemType.set("Tree") # Tree is set to the default option in the drop down menu
typeChoices = OptionMenu(root, self.chooseItemType, "Tree", "Animal", "Machine", "All") # Options Tree, Animal, Machine, and ALL are added to the drop down menu
typeChoices.place(x = 190, y = 160)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.createOptionButtons)) # This button will return the user to the expenses option page
backButton.place(x = 50, y = 350)
nextButton = Button(root, text = "Next", command=lambda: self.sequence(self.displayGeneralInventory))#self.saveNameAndType(itemName)))#(self.saveNameAndType(itemName))) # (itemName)))# lambda: self.sequence(self.test)) #This button will send the user to the add inventory page
nextButton.place(x = 350, y = 350)
# Used to create the inventory table
def displayGeneralInventory(self):
# This creates a table using the function Treeview
self.tree = Treeview(height="20", columns=("Name", "Current Quantity"))
self.tree.pack()
self.tree.heading('#1', text = "Name", anchor = CENTER)
self.tree.heading('#2', text = "Current Quantity", anchor = CENTER)
self.tree.column('#1', minwidth=0, width = 100)
self.tree.column('#2', minwidth=0, width = 100)
self.tree.column('#0', minwidth=0, width = 0)
itemType = self.chooseItemType.get()
if(itemType == "All"):
self.obtainData("Tree")
self.obtainData("Animal")
self.obtainData("Machine")
else:
self.obtainData(itemType)
# Adds database data to the inventory table
def obtainData(self, type):
for row in (self.inventoryDB.getOverviewInventory(type)):
name = row[0]
totalQuantity = row[1]
# Inserts data into the table. Each entry is tagged with the name and the type
# This is done in order to make identifying the entries easier for when detailed
# tables are requested
self.tree.insert("", "end", values = (name,totalQuantity), tag= [name, type])
# Creates a bak function that is used in the displayGeneralInventory functions
self.backFunction = self.displayGeneralInventory
# Binds a double click function to the Treeview table. If an entry is double clicked,
# the function displayGeneralInventory is ran
self.tree.bind("<Double-1>", self.displayDetailedInventory)
backButton = Button(root, text="Back", command=lambda: self.sequence(self.viewInveroty)) # This button will return the user to the main page. Still working on it.
backButton.place(x = 50, y = 350)
# Creates table when an entry is double clicked
def displayDetailedInventory(self, event):
# The selected item's tag are extracted and assigned to name and type
itemSelected = self.tree.selection()
name = self.tree.item(itemSelected,"tag")[0]
type = self.tree.item(itemSelected, "tag")[1]
for child in root.winfo_children():
child.destroy()
self.createDisplayTable()
self.obtainDetailedData(name, type)
# Adds detailed database data to the inventory table
def obtainDetailedData(self,name, type):
for row in (self.inventoryDB.getDetailedInventory(type, name)):
name = row[0]
purchaseDate = row[1]
Quantity = row[3]
Price = row[4]
self.tree.insert("", "end", values = (name,purchaseDate,Quantity, Price))
backButton = Button(root, text="Back", command=lambda: self.sequence(self.backFunction))
backButton.place(x = 50, y = 350)
# Creates the view total cost by month and year buttons
def viewTotalCost(self):
viewMonth = Button(root, text="View by month", command=lambda: self.sequence(self.viewByMonth))
viewMonth.place(x = 120, y = 100)
viewYear = Button(root, text="View by year", command=lambda: self.sequence(self.viewByYear))
viewYear.place(x = 120, y = 150)
backButton = Button(root, text="Back", command=lambda: self.sequence(self.createOptionButtons))#displayGeneralInventory)) # This button will return the user to the main page. Still working on it.
backButton.place(x = 50, y = 350)
# Creates the options for the user to select a month and year
def viewByMonth(self):
monthLabel = Label(root, text="Month")
yearLabel = Label(root, text="Year")
self.month = IntVar(self)
self.year = IntVar(self)
numMonths = self.nums(1, 12)
numYears = self.nums(2015, 2030)
self.optionmenu_month = OptionMenu(root, self.month, *numMonths)
self.optionmenu_year = OptionMenu(root, self.year, *numYears)
self.month.set(numMonths[0])
self.year.set(numYears[0])
self.optionmenu_month.place(x = 100, y = 100)
self.optionmenu_year.place(x = 150, y = 100)
monthLabel.place(x = 100, y = 140)
yearLabel.place(x = 150, y = 140)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.viewTotalCost)) # This button will return the user to the expenses option page
backButton.place(x = 50, y = 350)
nextButton = Button(root, text = "Next", command= self.viewTotalCostMonth)#self.viewTotalCostMonth)#self.saveNameAndType(itemName)))#(self.saveNameAndType(itemName))) # (itemName)))# lambda: self.sequence(self.test)) #This button will send the user to the add inventory page
nextButton.place(x = 350, y = 350)
# Creates database table and inserts the respective values by month and year
def viewTotalCostMonth(self):
self.createDisplayTable()
self.totalPrice = 0
month = self.month.get()
year = self.year.get()
self.lengthMonth = len(str(month))
self.searchDate = str(month) + "/" + str(year)
InventoryDB = getDatabaseConnection()
database = InventoryDB.cursor()
self.insertData("DetailedTreeInventory", "Tree", database, "Month")
self.insertData("DetailedAnimalInventory", "Animal", database, "Month")
self.insertData("DetailedMachineInventory", "Machine", database, "Month")
InventoryDB.close()
totalPriceLabel = Label(root, text=("Total price for " + calendar.month_name[month] + " in " + str(year) + " is: " + str(self.totalPrice)))
totalPriceLabel.place(x = 100, y = 350)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.viewByMonth)) # This button will return the user to the expenses option page
backButton.place(x = 50, y = 350)
# Creates the option for the user to select the year
def viewByYear(self):
yearLabel = Label(root, text="Year")
self.year = IntVar(self)
numYears = self.nums(2015, 2030)
self.optionmenu_year = OptionMenu(root, self.year, *numYears)
self.year.set(numYears[0])
self.optionmenu_year.place(x = 100, y = 100)
yearLabel.place(x = 100, y = 140)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.viewTotalCost)) # This button will return the user to the expenses option page
backButton.place(x = 50, y = 350)
nextButton = Button(root, text = "Next", command= self.viewTotalCostYear)#self.viewTotalCostMonth)#self.saveNameAndType(itemName)))#(self.saveNameAndType(itemName))) # (itemName)))# lambda: self.sequence(self.test)) #This button will send the user to the add inventory page
nextButton.place(x = 350, y = 350)
# Creates database table and inserts the respective values by year
def viewTotalCostYear(self):
self.createDisplayTable()
self.totalPrice = 0
year = self.year.get()
InventoryDB = getDatabaseConnection()
database = InventoryDB.cursor()
self.insertData("DetailedTreeInventory", "Tree", database, "Year")
self.insertData("DetailedAnimalInventory", "Animal", database, "Year")
self.insertData("DetailedMachineInventory", "Machine", database, "Year")
totalPriceLabel = Label(root, text="Total price for " + str(year) + " is: " + str(self.totalPrice))
totalPriceLabel.place(x = 100, y = 350)
backButton = Button(root, text = "Back", command=lambda: self.sequence(self.viewByYear)) # This button will return the user to the expenses option page
backButton.place(x = 50, y = 350)
# Inserts the detailed values into the detailed table
def insertData(self, table, type, database, yearOrMonth):
if yearOrMonth == "Year":
for row in database.execute("SELECT * FROM %s" % table ):
itemdate = row[1]
if ( str(self.year.get()) == itemdate[-4:]):
name = row[0]
purchaseDate = row[1]
Quantity = row[3]
Price = row[4]
self.tree.insert("", "end", values = (name,purchaseDate,Quantity, Price),tag = [name, type] )
self.totalPrice = self.totalPrice + Price
self.backFunction = self.viewTotalCostYear
else:
for row in database.execute("SELECT * FROM %s" % table ):
itemdate = row[1]
if (self.searchDate == (itemdate[0:(self.lengthMonth + 1)] + itemdate[-4:])):
name = row[0]
purchaseDate = row[1]
Quantity = row[3]
Price = row[4]
self.tree.insert("", "end", values = (name,purchaseDate,Quantity, Price), tag = [name, type])
self.totalPrice = self.totalPrice + Price
self.backFunction = self.viewTotalCostMonth
# If entry is double clicked, the table will acknoledge the click and display the detailed table
self.tree.bind("<Double-1>", self.displayDetailedInventory)
def createDisplayTable(self):
for child in root.winfo_children():
child.destroy()
self.tree = Treeview(height="15", columns=("Name", "Purchase Date", "Quantity", "Price"))#, "Description"))
self.tree.pack()
self.tree.heading('#1', text = "Name", anchor = CENTER)
self.tree.heading('#2', text = "Purchase Date", anchor = CENTER)
self.tree.heading('#3', text = "Quantity", anchor = CENTER)
self.tree.heading('#4', text = "Price", anchor = CENTER)
self.tree.column('#1', minwidth=0, width = 95)
self.tree.column('#2', minwidth=0, width = 95)
self.tree.column('#3', minwidth=0, width = 95)
self.tree.column('#4', minwidth=0, width = 95)
self.tree.column('#0', minwidth=0, width = 0)
# This is a helper function that will delete the current widgets of the frame
def sequence(self, run):
for child in root.winfo_children():
child.destroy()
run()
def __init__(self, master=None):
Frame.__init__(self, master)
self.place();
self.inventoryDB = InventoryDatabase()
# self.inventoryDB.createTable()
self.createOptionButtons()
class MyApp( Tk ):
def __init__(self):
Tk.__init__( self )
self.title( 'TaobaoAndTMall-Crawler' )
self.geometry( '850x650' )
self.CrawlTime = 20
self.DelayTime = 1
self.ThreadNum = 24
self.SpeedLimited = 200
self.downloadSpeed = 0
self.startFlag = 0
self.setUI()
self.pic_num = 1
def refresh_pb(self):
self.ProgressBar['value'] += (10 / (self.CrawlTime) * 1.0)
if self.ProgressBar['value'] >= 99:
print ("Done")
self.startFlag = 0
self.downloadingSpeed.config(text=u"下载速度：" + '0kb / s')
else:
self.after(100,self.refresh_pb)
def refresh_downloadSpeed(self):
L = self.SpeedLimited - 100
if L < 0:
L = 0
R = self.SpeedLimited
self.downloadSpeed = random.randint(L,R)
if self.startFlag == 1:
self.downloadingSpeed.config(text=u"下载速度：" + str( self.downloadSpeed ) + 'kb / s')
self.after(1000,self.refresh_downloadSpeed)
def refresh_photo(self):
try:
# os.listdir(os.getcwd()+"/2/taobao_spider/picture")
path = os.getcwd()
# print('OK')
name = str(random.choice(os.listdir(path+"/2/taobao_spider/picture")))
# print(name)
# print('./2/taobao_spider/picture/'+name)
# self.img2 = ImageTk.PhotoImage(Image.open( "./2/taobao_spider/picture/" + str(random.choice(os.listdir("./2/taobao_spider/piuture")) ).resize((320, 320), Image.ANTIALIAS))
img = Image.open('./2/taobao_spider/picture/'+name).resize((320, 320), Image.ANTIALIAS)
self.photo = ImageTk.PhotoImage(img)
# self.img2 = ImageTk.PhotoImage(Image.open(r'C:\Users\17990\Desktop\界面\返回.jpg')).resize((320, 320),Image.ANTIALIAS)
# print('helol')
self.photoLabel.configure(image = self.photo)
self.photoLabel.image = self.photo
self.after(5000,self.refresh_photo)
except:
print ("pic ::::::: " + str(self.pic_num))
print ("Error!!!!")
def refresh_treeview(self):
items = self.tree.get_children()
for item in items:
self.tree.delete(item)
# 读取csv
if os.path.exists('/Users/luodian/PycharmProjects/csdn-crawler/2/taobao_spider/info.csv'):
with open('/Users/luodian/PycharmProjects/csdn-crawler/2/taobao_spider/info.csv','r') as csvfile:
lines = []
reader = csv.reader(csvfile)
i = 0
# 转存至数组
for line in reader:
# 不输出第一行
if i > 0:
lines.append(line)
i = i + 1
i = 0
try:
for line in lines:
if len(line) != 7 or line[0] == 'address':
continue
self.tree.insert('', i, values=(line[6], line[3],
line[5], line[4],
line[0], line[1][:20]))
i = i + 1
except:
print ("**********")
self.after(100,self.refresh_treeview)
def returnPrimary(self):
parentFilePath = os.path.abspath(os.path.join(os.getcwd(), os.pardir))
os.system('cd ' + parentFilePath +'/csdn-crawler && chmod 777 ./return.sh && ./return.sh')
os._exit(0)
def robo3t(self):
os.system('robo3t.exe.lnk')
def start_crawl(self):
# path = os.getcwd()
# os.system( 'cd ' + path + ' && chmod 777 ./3/startScrapy.sh && ./3/startScrapy.sh' )
# os.system('cd /Users/luodian/PycharmProjects/csdn-crawler/2/taobao_spider && chmod 777 ./conthread.sh && ./conthread.sh')
self.startFlag = 1
self.after(3300,self.refresh_pb)
self.after(3300,self.refresh_downloadSpeed)
self.after(5000,self.refresh_treeview)
self.after(8000,self.refresh_photo)
self.pic_num = 1
def stop_crawl(self):
self.ProgressBar['value'] = 0
def open_finder(self):
path = os.getcwd()
operation = 'cd ' + path
os.system( operation + ' && chmod 777 ./openFinder.sh && ./openFinder.sh' )
def setUI(self):
self.photo = ImageTk.PhotoImage(Image.open('淘宝&天猫.jpg').resize((320, 320), Image.ANTIALIAS))
self.photoLabel = Label(self, image=self.photo, width=320, height=320)
# start_button = Button( self, text=u"开始", width=4, height=6, command=self.start_crawl )
# stop_button = Button( self, text=u"停止", width=4, height=6, command=self.stop_crawl )
# setting_button = Button( self, text=u"设置", width=4, height=6, command=self.setup_config )
# directory_button = Button( self, text=u"目录", width=4, height=6, command=self.open_finder )
start_button = Button(self, text=u"开始", width=7, height=2, command=self.start_crawl,state=ACTIVE)
stop_button = Button(self, text=u"暂停", width=7, height=2, command=self.stop_crawl,state=ACTIVE)
setting_button = Button(self, text=u"设置", width=7, height=2, command=self.setup_config,state=ACTIVE)
directory_button = Button(self, text=u"查询", width=7, height=2,command=self.robo3t,state=ACTIVE)
self.downloadingSpeed = Label( self, text=u"下载速度：" + str( self.downloadSpeed ) + 'kb / s')
# 需要用设置修改
self.threadingNum = Label( self, text=u"线程数：" + str( self.ThreadNum ) )
self.delayTime = Label( self, text=u"下载延时：" + str( self.DelayTime ) )
self.crawlTimeLabel = Label( self, text=u"爬取时间：" + str( self.CrawlTime ) )
self.photoLabel.place(x=20, y=20)
start_button.place(x=420, y=65)
setting_button.place(x=540, y=65)
stop_button.place(x=420, y=180)
directory_button.place(x=540, y=180)
self.downloadingSpeed.place(x=680, y=65)
self.threadingNum.place(x=680, y=120)
self.delayTime.place(x=680, y=175)
self.crawlTimeLabel.place(x=680, y=230)
# self.photoLabel.grid( row=0, rowspan=4, pady=20, padx=20 )
# start_button.grid( row=0, column=1, padx=60 )
# setting_button.grid( row=2, column=1, padx=60 )
# stop_button.grid( row=1, column=1, padx=60 )
# directory_button.grid( row=3, column=1, padx=60 )
#
# self.downloadingSpeed.grid( row=0, column=2, padx=60, sticky=W )
# self.threadingNum.grid( row=1, column=2, padx=60, sticky=W )
# self.delayTime.grid( row=2, column=2, padx=60, sticky=W )
# self.crawlTimeLabel.grid( row=3, column=2, padx=60, sticky=W )
returnImg = Image.open('返回.jpg').resize((30, 30), Image.ANTIALIAS)
self.returnPhoto = ImageTk.PhotoImage(returnImg)
return_button = Button(self, width=30, height=30, command=self.returnPrimary, image=self.returnPhoto,
text=u'返回', state=ACTIVE)
return_button.place(x=2, y=2)
bottomFrame = Frame( self, width=80 )
# bottomFrame.grid( row=5, column=0, columnspan=3, padx=24 )
bottomFrame.place(x=35, y=390)
scrollBar = Scrollbar( bottomFrame )
scrollBar.pack( side=RIGHT, fill=Y )
# 进度条位置
self.ProgressBar = Progressbar( self, orient="horizontal", length=320, mode='determinate' )
# self.ProgressBar.grid( row=4, column=2, columnspan=2, sticky=W + E)
self.ProgressBar.place(x=400, y=300)
self.tree = Treeview( bottomFrame, column=('c1', 'c2', 'c3', 'c4', 'c5', 'c6'), show="headings",
yscrollcommand=scrollBar.set )
self.tree.column( 'c1', width=220, anchor='center' )
self.tree.column( 'c2', width=80, anchor='center' )
self.tree.column( 'c3', width=80, anchor='center' )
self.tree.column( 'c4', width=80, anchor='center' )
self.tree.column( 'c5', width=120, anchor='center' )
self.tree.column( 'c6', width=120, anchor='center' )
# 设置每列表头标题文本
self.tree.heading( 'c1', text='商品标题' )
self.tree.heading( 'c2', text='商品链接' )
self.tree.heading( 'c3', text='店铺' )
self.tree.heading( 'c4', text='价格' )
self.tree.heading( 'c5', text='地址' )
self.tree.heading( 'c6', text='评论' )
self.tree.pack( side=LEFT, fill="both" )
scrollBar.config( command=self.tree.yview )
items = self.tree.get_children()
for item in items:
self.tree.delete(item)
self.tree.bind("<Double-1>",self.onDBClick)
def onDBClick(self,event):
print("selected items:")
for item in self.tree.selection():
item_text = self.tree.item(item,"text")
print(item_text)
def setup_config(self):
pw = PopupDialog( self )
self.wait_window( pw )
return
class ElementListWidget(Frame):
def __init__(self, parent, label, columns, showError):
Frame.__init__(self, parent)
self.showError = showError
self.columnconfigure(0, weight = 1)
self.rowconfigure(1, weight = 1)
# Название таблицы
self.titleLabel = Label(self, text = label)
self.titleLabel.grid(column = 0, row = 0, sticky = W + E)
# Таблица значений
columns = ("Метка", "№") + columns
self.tree = Treeview(self, columns = columns, displaycolumns = columns,
selectmode = "browse")
self.tree.grid(column = 0, row = 1, sticky = W + N + E + S)
# Настраиваем внешний вид таблицы (первые колонки)
self.tree.column("#0", width = 0, stretch = 0) # Прячем колонку с иконкой
self.tree.column( columns[0], anchor = W, width = 150)
self.tree.heading(columns[0], anchor = W, text = columns[0])
self.tree.column( columns[1], anchor = E, width = 80)
self.tree.heading(columns[1], anchor = E, text = columns[1])
self.tree.bind("<<TreeviewSelect>>", self.onSelectionChanged)
# Панель с кнопками
self.buttonPanel = Frame(self)
self.buttonPanel.grid(column = 0, row = 2, sticky = W + E)
self.buttonPanel.columnconfigure(0, weight = 1)
self.buttonPanel.columnconfigure(3, minsize = emptySpaceSize, weight = 0)
self.buttonPanel.columnconfigure(6, minsize = emptySpaceSize, weight = 0)
self.buttonPanel.columnconfigure(9, weight = 1)
# Кнопки добавления/удаления элемента
self.buttonAdd = Button(self.buttonPanel, text = "+", width = 3,
command = self.onButtonAddClicked)
self.buttonAdd.grid(column = 1, row = 0)
self.buttonRemove = Button(self.buttonPanel, text = "-", width = 3, state = DISABLED,
command = self.onButtonRemoveClicked)
self.buttonRemove.grid(column = 2, row = 0)
# Кнопки перемещения элемента
self.buttonUp = Button(self.buttonPanel, text = "↑", width = 3, state = DISABLED,
command = self.onButtonUpClicked)
self.buttonUp.grid(column = 4, row = 0)
self.buttonDown = Button(self.buttonPanel, text = "↓", width = 3, state = DISABLED,
command = self.onButtonDownClicked)
self.buttonDown.grid(column = 5, row = 0)
# Кнопки применить/отменить (для выбранного элемента)
self.buttonCancel = Button(self.buttonPanel, text = "✗", width = 3,
command = self.updateSelectedFrame)
self.buttonCancel.grid(column = 7, row = 0)
self.buttonApply = Button(self.buttonPanel, text = "✓", width = 3,
command = self.onButtonApplyClicked)
self.buttonApply.grid(column = 8, row = 0)
# Редактирование выделенного элемента
self.i = StringVar()
self.label = (StringVar(), StringVar())
self.selectedFrame = Frame(self)
self.selectedFrame.grid(column = 0, row = 3, sticky = W + E)
# Номер
Label(self.selectedFrame, text = "№:") \
.grid(column = 0, row = 0)
Label(self.selectedFrame, textvariable = self.i, width = 3, justify = RIGHT) \
.grid(column = 1, row = 0)
# Пустое пространство
self.selectedFrame.columnconfigure(2, minsize = emptySpaceSize, weight = 0)
# Метка
Entry(self.selectedFrame, textvariable = self.label[0]) \
.grid(column = 3, row = 0, sticky = W + E)
Entry(self.selectedFrame, textvariable = self.label[1], bg = defaultValueBG) \
.grid(column = 4, row = 0, sticky = W + E)
# Виджет для элементов классов-потомков
self.detailFrame = Frame(self.selectedFrame)
self.detailFrame.grid(column = 3, row = 1, columnspan = 2, sticky = W + N + E + S)
self.selectedFrame.columnconfigure(3, weight = 1)
self.selectedFrame.columnconfigure(4, weight = 1)
self.selectedFrame.rowconfigure(1, weight = 1)
def onButtonUpClicked(self):
item = self.selectedItem()
if item is None: return
prev = self.tree.prev(item)
if prev != "":
parent, index = self.tree.parent(item), self.tree.index(item)
self.tree.move(item, parent, index - 1)
# Корректируем номера элементов
self.tree.set(item, "№", index - 1)
self.tree.set(prev, "№", index)
self.updateSelectedFrame(item)
def onButtonDownClicked(self):
item = self.selectedItem()
if item is None: return
next = self.tree.next(item)
if next != "":
parent, index = self.tree.parent(item), self.tree.index(item)
self.tree.move(item, parent, index + 1)
# Корректируем номера элементов
self.tree.set(item, "№", index + 1)
self.tree.set(next, "№", index)
self.updateSelectedFrame(item)
def onButtonAddClicked(self):
pass
def onButtonRemoveClicked(self):
item = self.selectedItem()
if item is None: return
next = self.tree.next(item)
self.tree.delete(item)
while next != "":
i = int(self.tree.set(next, "№"))
self.tree.set(next, "№", i - 1)
next = self.tree.next(next)
self.onSelectionChanged()
def onButtonApplyClicked(self, item = None):
if item is None: item = self.selectedItem()
if item is None: return None
label = self.label[0].get()
self.tree.set(item, "Метка", label)
return item
def onSelectionChanged(self, event = None):
item = self.selectedItem()
# Обновляем состояние кнопок
state = DISABLED if item is None else NORMAL
for x in (self.buttonRemove, self.buttonUp, self.buttonDown):
x["state"] = state
self.updateSelectedFrame(item)
def selectedItem(self):
selection = self.tree.selection()
return None if type(selection) == type("") else selection[0]
def clear(self):
for item in self.tree.get_children():
self.tree.delete(item)
def updateSelectedFrame(self, item = None, values = None):
if item is None: item = self.selectedItem()
values = None
if item is None:
i = ""
label = ""
else:
if values is None: values = self.tree.set(item)
i = values["№"]
label = values["Метка"]
self.i.set(i)
self.label[0].set(label)
return (item, values)
def addElement(self, values):
self.tree.insert(parent = "", index = END, values = values)
def setDefaultElement(self, label):
self.label[1].set(label)
def elementsCount(self):
return len(self.tree.get_children())
def elements(self, transform):
return [ transform(item) for item in self.tree.get_children() ]
class DialogOpenArchive(Toplevel):
def __init__(self, parent, openType, filesource, filenames, title, colHeader, showAltViewButton=False):
if isinstance(parent, Cntlr):
cntlr = parent
parent = parent.parent # parent is cntlrWinMain
else: # parent is a Toplevel dialog
cntlr = parent.cntlr
super(DialogOpenArchive, self).__init__(parent)
self.parent = parent
self.showAltViewButton = showAltViewButton
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.accepted = False
self.transient(self.parent)
frame = Frame(self)
treeFrame = Frame(frame, width=500)
vScrollbar = Scrollbar(treeFrame, orient=VERTICAL)
hScrollbar = Scrollbar(treeFrame, orient=HORIZONTAL)
self.treeView = Treeview(treeFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set)
self.treeView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.treeView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.treeView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
treeFrame.columnconfigure(0, weight=1)
treeFrame.rowconfigure(0, weight=1)
treeFrame.grid(row=0, column=0, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.treeView.focus_set()
if openType not in (PLUGIN, PACKAGE):
cntlr.showStatus(_("loading archive {0}").format(filesource.url))
self.filesource = filesource
self.filenames = filenames
self.selection = filesource.selection
self.hasToolTip = False
selectedNode = None
if openType == ENTRY_POINTS:
try:
metadataFiles = filesource.taxonomyPackageMetadataFiles
''' take first for now
if len(metadataFiles) != 1:
raise IOError(_("Taxonomy package contained more than one metadata file: {0}.")
.format(', '.join(metadataFiles)))
'''
metadataFile = metadataFiles[0]
metadata = filesource.url + os.sep + metadataFile
self.metadataFilePrefix = os.sep.join(os.path.split(metadataFile)[:-1])
if self.metadataFilePrefix:
self.metadataFilePrefix += "/" # zip contents have /, never \ file seps
self.taxonomyPkgMetaInf = '{}/META-INF/'.format(
os.path.splitext(os.path.basename(filesource.url))[0])
self.taxonomyPackage = parsePackage(cntlr, filesource, metadata,
os.sep.join(os.path.split(metadata)[:-1]) + os.sep)
if self.taxonomyPackage["entryPoints"]:
# may have instance documents too
self.packageContainedInstances = []
packageContentTypeCounts = {}
for suffix in (".xhtml", ".htm", ".html"):
for potentialInstance in filesource.dir:
if potentialInstance.endswith(".xhtml"):
_type = "Inline Instance"
self.packageContainedInstances.append([potentialInstance, _type])
packageContentTypeCounts[potentialInstance] = packageContentTypeCounts.get(potentialInstance, 0) + 1
if self.packageContainedInstances:
break
if self.packageContainedInstances: # add sequences to any duplicated entry types
for _type, count in packageContentTypeCounts.items():
if count > 1:
_dupNo = 0
for i in range(len(self.packageContainedInstances)):
if self.packageContainedInstances[i][0] == _type:
_dupNo += 1
self.packageContainedInstances[i][0] = "{} {}".format(_type, _dupNo)
else:
# may be a catalog file with no entry oint names
openType = ARCHIVE # no entry points to show, just archive
self.showAltViewButton = False
except Exception as e:
self.close()
err = _("Failed to parse metadata; the underlying error was: {0}").format(e)
messagebox.showerror(_("Malformed taxonomy package"), err)
cntlr.addToLog(err)
return
if openType not in (PLUGIN, PACKAGE):
cntlr.showStatus(None)
if openType in (DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
y = 3
else:
y = 1
okButton = Button(frame, text=_("OK"), command=self.ok)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
okButton.grid(row=y, column=2, sticky=(S,E,W), pady=3)
cancelButton.grid(row=y, column=3, sticky=(S,E,W), pady=3, padx=3)
if self.showAltViewButton:
self.altViewButton = Button(frame, command=self.showAltView)
self.altViewButton.grid(row=y, column=0, sticky=(S,W), pady=3, padx=3)
self.loadTreeView(openType, colHeader, title)
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.toolTipText = StringVar()
if self.hasToolTip:
self.treeView.bind("<Motion>", self.motion, '+')
self.treeView.bind("<Leave>", self.leave, '+')
self.toolTipText = StringVar()
self.toolTip = ToolTip(self.treeView,
textvariable=self.toolTipText,
wraplength=640,
follow_mouse=True,
state="disabled")
self.toolTipRowId = None
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeView(self, openType, title, colHeader):
self.title(title)
self.openType = openType
selectedNode = None
# clear previous treeview entries
for previousNode in self.treeView.get_children(""):
self.treeView.delete(previousNode)
# set up treeView widget and tabbed pane
if openType in (ARCHIVE, DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
if openType in (PLUGIN, PACKAGE): width = 770
else: width = 500
self.treeView.column("#0", width=width, anchor="w")
self.treeView.heading("#0", text=colHeader)
self.isRss = getattr(self.filesource, "isRss", False)
if self.isRss:
self.treeView.column("#0", width=350, anchor="w")
self.treeView["columns"] = ("descr", "date", "instDoc")
self.treeView.column("descr", width=50, anchor="center", stretch=False)
self.treeView.heading("descr", text="Form")
self.treeView.column("date", width=170, anchor="w", stretch=False)
self.treeView.heading("date", text="Pub Date")
self.treeView.column("instDoc", width=200, anchor="w", stretch=False)
self.treeView.heading("instDoc", text="Instance Document")
elif openType == PLUGIN:
self.treeView.column("#0", width=150, anchor="w")
self.treeView["columns"] = ("name", "vers", "descr", "license")
self.treeView.column("name", width=150, anchor="w", stretch=False)
self.treeView.heading("name", text="Name")
self.treeView.column("vers", width=60, anchor="w", stretch=False)
self.treeView.heading("vers", text="Version")
self.treeView.column("descr", width=300, anchor="w", stretch=False)
self.treeView.heading("descr", text="Description")
self.treeView.column("license", width=60, anchor="w", stretch=False)
self.treeView.heading("license", text="License")
elif openType == PACKAGE:
self.treeView.column("#0", width=200, anchor="w")
self.treeView["columns"] = ("vers", "descr", "license")
self.treeView.column("vers", width=100, anchor="w", stretch=False)
self.treeView.heading("vers", text="Version")
self.treeView.column("descr", width=400, anchor="w", stretch=False)
self.treeView.heading("descr", text="Description")
self.treeView.column("license", width=70, anchor="w", stretch=False)
self.treeView.heading("license", text="License")
else:
self.treeView["columns"] = tuple()
loadedPaths = []
for i, filename in enumerate(self.filenames):
if isinstance(filename,tuple):
if self.isRss:
form, date, instDoc = filename[2:5]
elif openType == PLUGIN:
name, vers, descr, license = filename[3:7]
elif openType == PACKAGE:
vers, descr, license = filename[3:6]
filename = filename[0] # ignore tooltip
self.hasToolTip = True
if filename.endswith("/"):
filename = filename[:-1]
path = filename.split("/")
if not self.isRss and len(path) > 1 and path[:-1] in loadedPaths:
parent = "file{0}".format(loadedPaths.index(path[:-1]))
else:
parent = ""
node = self.treeView.insert(parent, "end", "file{0}".format(i), text=path[-1])
if self.isRss:
self.treeView.set(node, "descr", form)
self.treeView.set(node, "date", date)
self.treeView.set(node, "instDoc", os.path.basename(instDoc))
elif openType == PLUGIN:
self.treeView.set(node, "name", name)
self.treeView.set(node, "vers", vers)
self.treeView.set(node, "descr", descr)
self.treeView.set(node, "license", license)
elif openType == PACKAGE:
self.treeView.set(node, "vers", vers)
self.treeView.set(node, "descr", descr)
self.treeView.set(node, "license", license)
if self.selection == filename:
selectedNode = node
loadedPaths.append(path)
elif openType == ENTRY_POINTS:
self.treeView.column("#0", width=200, anchor="w")
self.treeView.heading("#0", text="Name")
self.treeView["columns"] = ("url",)
self.treeView.column("url", width=300, anchor="w")
self.treeView.heading("url", text="URL")
for fileType, fileUrl in getattr(self, "packageContainedInstances", ()):
self.treeView.insert("", "end", fileUrl,
values=fileType,
text=fileUrl or urls[0][2])
for name, urls in sorted(self.taxonomyPackage["entryPoints"].items(), key=lambda i:i[0][2]):
self.treeView.insert("", "end", name,
values="\n".join(url[1] for url in urls),
text=name or urls[0][2])
self.hasToolTip = True
else: # unknown openType
return None
if selectedNode:
self.treeView.see(selectedNode)
self.treeView.selection_set(selectedNode)
if self.showAltViewButton:
self.altViewButton.config(text=_("Show Files") if openType == ENTRY_POINTS else _("Show Entries"))
def ok(self, event=None):
selection = self.treeView.selection()
if len(selection) > 0:
if hasattr(self, "taxonomyPackage"):
# load file source remappings
self.filesource.mappedPaths = self.taxonomyPackage["remappings"]
filename = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
filename = self.filenames[int(selection[0][4:])]
if isinstance(filename,tuple):
if self.isRss:
filename = filename[4]
else:
filename = filename[0]
elif self.openType == ENTRY_POINTS:
epName = selection[0]
#index 0 is the remapped Url, as opposed to the canonical one used for display
# Greg Acsone reports [0] does not work for Corep 1.6 pkgs, need [1], old style packages
filenames = []
for _url, _type in self.packageContainedInstances: # check if selection was an inline instance
if _type == epName:
filenames.append(_url)
if not filenames: # else if it's a named taxonomy entry point
for url in self.taxonomyPackage["entryPoints"][epName]:
filename = url[0]
if not filename.endswith("/"):
# check if it's an absolute URL rather than a path into the archive
if not isHttpUrl(filename) and self.metadataFilePrefix != self.taxonomyPkgMetaInf:
# assume it's a path inside the archive:
filename = self.metadataFilePrefix + filename
filenames.append(filename)
if filenames:
self.filesource.select(filenames)
self.accepted = True
self.close()
return
elif self.openType in (PLUGIN, PACKAGE):
filename = self.filenames[int(selection[0][4:])][2]
if filename is not None and not filename.endswith("/"):
if hasattr(self, "taxonomyPackage"):
# attempt to unmap the filename to original file
# will be mapped again in loading, but this allows schemaLocation to be unmapped
for prefix, remapping in self.taxonomyPackage["remappings"].items():
if isHttpUrl(remapping):
remapStart = remapping
else:
remapStart = self.metadataFilePrefix + remapping
if filename.startswith(remapStart):
# set unmmapped file
filename = prefix + filename[len(remapStart):]
break
if self.openType in (PLUGIN, PACKAGE):
self.filesource.selection = filename
else:
self.filesource.select(filename)
self.accepted = True
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def showAltView(self, event=None):
if self.openType == ENTRY_POINTS:
self.loadTreeView(ARCHIVE, _("Select Entry Point"), _("File"))
else:
self.loadTreeView(ENTRY_POINTS, _("Select Archive File"), _("File"))
def leave(self, *args):
self.toolTipRowId = None
def motion(self, *args):
tvRowId = self.treeView.identify_row(args[0].y)
if tvRowId != self.toolTipRowId:
text = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
self.toolTipRowId = tvRowId
if tvRowId and len(tvRowId) > 4:
try:
text = self.filenames[ int(tvRowId[4:]) ]
if isinstance(text, tuple):
text = (text[1] or "").replace("\\n","\n")
except (KeyError, ValueError):
pass
elif self.openType == ENTRY_POINTS:
try:
text = "{0}\n{1}".format(tvRowId,
"\n".join(url[1] for url in self.taxonomyPackage["entryPoints"][tvRowId]))
except KeyError:
pass
self.setToolTip(text)
def setToolTip(self, text):
self.toolTip._hide()
if text:
self.toolTipText.set(text)
self.toolTip.configure(state="normal")
self.toolTip._schedule()
else:
self.toolTipText.set("")
self.toolTip.configure(state="disabled")
class TestcaseSelector:
def start(self):
# Create TK window
self.root = Tk()
# Create Style object for TK window
self.root.style = Style()
# Set default frame size to 800x640
self.root.geometry('800x640')
# Set Treeview row height to 40 so there's no overlap
self.root.style.configure('Treeview',rowheight=40)
# Set title and window size
self.root.wm_title("Select Testcases to Run")
# Create a frame for the treeview
self.testcase_frame = Frame(self.root)
# Create scrollbar for treeview
scrollbar = Scrollbar(self.root)
scrollbar.pack(side=RIGHT,fill=Y)
# Create Treeview
self.treeView = Treeview(self.testcase_frame)
self.treeView.pack(expand=1,fill=BOTH)
# Attach scrollbar to Treeview
self.treeView.config(yscrollcommand=scrollbar.set)
scrollbar.config(command=self.treeView.yview)
# Get testcase name dictionary to use for filling out treeview
testcase_dictionary = get_testcase_name_dictonary()
self.testcase_data = {}
self.testcase_run_data = {}
for key in testcase_dictionary.keys():
subsection = testcase_dictionary[key]
self.testcase_data[key] = subsection
s_p = self.treeView.insert('', END, text=key)
for test in subsection:
testcase_name = test._testMethodName
testcase_name = testcase_name
self.treeView.insert(s_p, END, text=testcase_name)
self.testcase_run_data[testcase_name] = test
self.webData = self.testcase_run_data
# Create buttons for cancel and run tests
run_button = Button(self.testcase_frame, text="Run", fg="green",command=self._save_selection,width=25,height=5)
run_button.pack(side=LEFT,expand=1,fill=BOTH)
quit_button = Button(self.testcase_frame, text="Cancel", fg="red", command=self.treeView.quit,width=25,height=5)
quit_button.pack(side=RIGHT,expand=1,fill=BOTH)
# Pack the rest of the frame and tell it to scale on both x and y axis
self.testcase_frame.pack(expand=1,fill=BOTH)
def get_tests_from_selected_names(self,names):
ret_tests = {}
for name in names:
ret_tests[name] = self.webData[name]
return ret_tests
def _save_selection(self):
selected_tests = self.treeView.selection()
output=[]
for selection in selected_tests:
item_text = self.treeView.item(selection,'text')
if 'test_' in item_text:
if item_text not in output:
output.append(item_text)
else:
pass
elif 'Tests' in item_text:
for test in self.testcase_data[item_text]:
output.append(test._testMethodName)
# output = output + self.testSectionData[item_text]
self.testcases = self.get_tests_from_selected_names(output)
self.root.quit()
def get_testcases(self):
self.start()
self.root.mainloop()
self.root.destroy()
# Try/Except to fail gracefully
try:
return self.testcases
except:
exit(0)
class FensterTrans(GuiTemplate):
def __init__(self, master):
super().__init__(master)
self.grid()
tv_column_width_id = 50
tv_column_width = 200
tv_column_width_min = 200
self.__pd_PLQuelle = Punkt_Dic()
self.__pd_PLZiel = Punkt_Dic()
self.__pd_PLTrans = Punkt_Dic()
self.__dict_para = {}
self.str_filePathSource = StringVar()
self.str_filePathDest = StringVar()
self.int_radioTrans = IntVar(value=0)
self.dbl_ParaMY = DoubleVar()
self.dbl_ParaMX = DoubleVar()
self.dbl_ParaRotY = DoubleVar()
self.dbl_ParaRotX = DoubleVar()
self.dbl_ParaTransY = DoubleVar()
self.dbl_ParaTransX = DoubleVar()
lf_transType = LabelFrame(self, text="Ebene Transformationen")
lf_transType.grid(row=0, column=0, padx=3, pady=3, sticky="w")
lf_Source = LabelFrame(self, text="Quellsystem")
lf_Source.grid(row=1,
column=0,
padx=3,
pady=3,
sticky="w",
columnspan=2)
lf_Dest = LabelFrame(
self,
text=
"Zielsystem (Markierung: Nichtbeachtung des Passpunktes bei der Transformation)"
)
lf_Dest.grid(row=1, column=3, padx=3, pady=3, sticky="w", columnspan=2)
lf_Parameter = LabelFrame(self, text="Parameter")
lf_Parameter.grid(row=2,
column=2,
padx=3,
pady=3,
sticky="",
columnspan=3)
lf_Trans = LabelFrame(self, text="Zielsystem / transformiert")
lf_Trans.grid(row=4, column=0, padx=3, pady=3, sticky="", columnspan=4)
Radiobutton(lf_transType,text="Helmerttransformation", variable=self.int_radioTrans, value=0)\
.grid(row=0, column=0, padx=3, pady=3)
Radiobutton(lf_transType,text="Affintransformation", variable=self.int_radioTrans, value=1)\
.grid(row=0, column=1, padx=3, pady=3)
Button(self,text="Lade Punkte",command=self.BtnPressedLoadPoints)\
.grid(row=0, column=1, padx=3, pady=3, sticky="w")
Button(self,text="Berechnen",command=self.BtnPressedCalc)\
.grid(row=2, column=0, padx=3, pady=3, columnspan=1,sticky="e")
Button(self,text="Zeige Grafik",command=self.BtnPressedPlot)\
.grid(row=2, column=1, padx=3, pady=3, columnspan=1,sticky="w")
self.tv_punktListSource = Treeview(lf_Source, selectmode="none")
self.tv_punktListSource.grid(row=0, column=0, padx=3, pady=3)
self.tv_punktListSource["columns"] = ("y", "x")
self.tv_punktListSource.column("#0",
width=tv_column_width_id,
minwidth=tv_column_width_id)
self.tv_punktListSource.column("y",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListSource.column("x",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListSource.heading("#0", text="id")
self.tv_punktListSource.heading("y", text="y")
self.tv_punktListSource.heading("x", text="x")
self.scr_punktListSourceScroll = Scrollbar(
lf_Source,
orient="vertical",
command=self.tv_punktListSource.yview)
self.scr_punktListSourceScroll.grid(row=0, column=1, sticky="nse")
self.tv_punktListSource.configure(
yscrollcommand=self.scr_punktListSourceScroll.set)
self.tv_punktListDest = Treeview(lf_Dest)
self.tv_punktListDest.grid(row=0, column=0, padx=3, pady=3)
self.tv_punktListDest["columns"] = ("Y", "X")
self.tv_punktListDest.column("#0",
width=tv_column_width_id,
minwidth=tv_column_width_id)
self.tv_punktListDest.column("Y",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListDest.column("X",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListDest.heading("#0", text="id")
self.tv_punktListDest.heading("Y", text="Y")
self.tv_punktListDest.heading("X", text="X")
self.scr_punktListDestScroll = Scrollbar(
lf_Dest, orient="vertical", command=self.tv_punktListDest.yview)
self.scr_punktListDestScroll.grid(row=0, column=1, sticky="nse")
self.tv_punktListDest.configure(
yscrollcommand=self.scr_punktListDestScroll.set)
Label(lf_Parameter, text="Maßstab Y").grid(row=0,
column=0,
padx=3,
pady=3,
sticky="w")
Label(lf_Parameter, text="Maßstab X").grid(row=0,
column=3,
padx=3,
pady=3,
sticky="e")
Label(lf_Parameter, text="Rotation Y / gon").grid(row=1,
column=0,
padx=3,
pady=3,
sticky="w")
Label(lf_Parameter, text="Rotation X / gon").grid(row=1,
column=3,
padx=3,
pady=3,
sticky="e")
Label(lf_Parameter, text="Translation Y").grid(row=2,
column=0,
padx=3,
pady=3,
sticky="w")
Label(lf_Parameter, text="Translation X").grid(row=2,
column=3,
padx=3,
pady=3,
sticky="e")
Entry(lf_Parameter, textvariable=self.dbl_ParaMY,
state="readonly").grid(row=0, column=1, padx=3, pady=3)
Entry(lf_Parameter, textvariable=self.dbl_ParaMX,
state="readonly").grid(row=0, column=2, padx=3, pady=3)
Entry(lf_Parameter, textvariable=self.dbl_ParaRotY,
state="readonly").grid(row=1, column=1, padx=3, pady=3)
Entry(lf_Parameter, textvariable=self.dbl_ParaRotX,
state="readonly").grid(row=1, column=2, padx=3, pady=3)
Entry(lf_Parameter, textvariable=self.dbl_ParaTransY,
state="readonly").grid(row=2, column=1, padx=3, pady=3)
Entry(lf_Parameter, textvariable=self.dbl_ParaTransX,
state="readonly").grid(row=2, column=2, padx=3, pady=3)
self.tv_punktListTrans = Treeview(lf_Trans, selectmode="none")
self.tv_punktListTrans.grid(row=0, column=0, padx=3, pady=3)
self.tv_punktListTrans["columns"] = ("Y", "X", "Rk Y", "Rk X")
self.tv_punktListTrans.column("#0",
width=tv_column_width_id,
minwidth=tv_column_width_id)
self.tv_punktListTrans.column("Y",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListTrans.column("X",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListTrans.column("Rk Y",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListTrans.column("Rk X",
width=tv_column_width,
minwidth=tv_column_width_min)
self.tv_punktListTrans.heading("#0", text="id")
self.tv_punktListTrans.heading("Y", text="Y")
self.tv_punktListTrans.heading("X", text="X")
self.tv_punktListTrans.heading("Rk Y", text="Restklaffe Y")
self.tv_punktListTrans.heading("Rk X", text="Restklaffe X")
self.scr_punktListTransScroll = Scrollbar(
lf_Trans, orient="vertical", command=self.tv_punktListTrans.yview)
self.scr_punktListTransScroll.grid(row=0, column=1, sticky="nse")
self.tv_punktListTrans.configure(
yscrollcommand=self.scr_punktListTransScroll.set)
def loadPoints(self, text, sepDec, sepVal, systemtype):
if systemtype == 0:
self.__pd_PLQuelle.clear()
self.__pd_PLQuelle.einlesenListe(text, sepDec, sepVal)
self.showPoints(systemtype, self.__pd_PLQuelle)
else:
self.__pd_PLZiel.clear()
self.__pd_PLZiel.einlesenListe(text, sepDec, sepVal)
self.showPoints(systemtype, self.__pd_PLZiel)
def showPoints(self, systemtype, dP):
if systemtype == 0:
self.fillTree(self.tv_punktListSource, dP)
elif systemtype == 1:
self.fillTree(self.tv_punktListDest, dP)
else:
self.fillTree(self.tv_punktListTrans, dP)
def showParam(self, parameter):
self.dbl_ParaMY.set(self.runde(parameter["m_Y"]))
self.dbl_ParaRotY.set(
self.runde(Winkel(parameter["rot_Y"], "rad").get_w("gon")))
self.dbl_ParaTransY.set(self.runde(parameter["Y0"]))
self.dbl_ParaTransX.set(self.runde(parameter["X0"]))
if "m_X" in parameter:
self.dbl_ParaMX.set(self.runde(parameter["m_X"]))
self.dbl_ParaRotX.set(
self.runde(Winkel(parameter["rot_X"], "rad").get_w("gon")))
else:
self.dbl_ParaMX.set(self.runde(parameter["m_Y"]))
self.dbl_ParaRotX.set(
self.runde(Winkel(parameter["rot_Y"], "rad").get_w("gon")))
def fillTree(self, treename, dP):
for row in treename.get_children():
treename.delete(row)
dP = dP.get_dic()
keys = list(dP.keys())
for i in range(len(dP)):
pId = keys[i]
y = self.runde(dP[pId]["coord"].get_y())
x = self.runde(dP[pId]["coord"].get_x())
if "w" in dP[pId]:
wy = self.runde(dP[pId]["w"].get_y())
wx = self.runde(dP[pId]["w"].get_x())
treename.insert("", i + 1, text=pId, values=(y, x, wy, wx))
else:
treename.insert("", i + 1, text=pId, values=(y, x))
def BtnPressedLoadPoints(self):
top = Toplevel()
top.title("Punkte laden")
self.F_loadPList = Fenster_loadPList(top, self)
def BtnPressedCalc(self):
l_p1exclude = []
for i in self.tv_punktListDest.selection():
l_p1exclude.append(self.tv_punktListDest.item(i, "text"))
if self.int_radioTrans.get() == 0:
self.__pd_PLTrans, self.__dict_para = HelmertTrans(
self.__pd_PLQuelle, self.__pd_PLZiel,
l_p1exclude).get_result()
else:
self.__pd_PLTrans, self.__dict_para = AffinTrans(
self.__pd_PLQuelle, self.__pd_PLZiel,
l_p1exclude).get_result()
self.showPoints(2, self.__pd_PLTrans)
self.showParam(self.__dict_para)
def load_json(self, s):
pDic = json.loads(s)
for k, v in pDic.items():
if k == "source":
self.__pd_PLQuelle.from_json(v)
elif k == "dest":
self.__pd_PLZiel.from_json(v)
elif k == "trans":
self.__pd_PLTrans.from_json(v)
elif k == "parameter":
self.__dict_para = v
self.showPoints(0, self.__pd_PLQuelle)
self.showPoints(1, self.__pd_PLZiel)
self.showPoints(2, self.__pd_PLTrans)
self.showParam(self.__dict_para)
def save_json(self):
return json.dumps(
{
"source": self.__pd_PLQuelle.get_json(),
"dest": self.__pd_PLZiel.get_json(),
"trans": self.__pd_PLTrans.get_json(),
"parameter": self.__dict_para
},
default=lambda obj: obj.get_json(),
sort_keys=True,
indent=4)
def BtnPressedPlot(self):
fig = plt.figure()
ax = fig.add_subplot()
ax.set_aspect("equal")
pl_plot = []
for k in self.__pd_PLTrans.get_dic().values():
pl_plot.append(k)
l_y = []
l_x = []
l_id = []
for p in pl_plot:
l_y.append(p["coord"].get_y())
l_x.append(p["coord"].get_x())
l_id.append(p["coord"].get_id())
for i in range(len(l_y)):
ax.text(l_y[i] + 10, l_x[i] + 10, l_id[i])
plt.plot(l_y, l_x, '.', color='black')
ax.relim()
ax.autoscale_view()
ax.set_xlabel("Y")
ax.set_ylabel("X")
ax.grid(True)
plt.show()
class StoreView(BaseView):
########
# Initializes and places all GUI elements.
def _create_widgets(self):
# left frame
self._leftFrame = Frame(self)
self._imgLabel = Label(self._leftFrame)
self._locLabel = Label(self._leftFrame,
text="Location:",
font=BaseView.NORMAL_FONT)
self._locPreview = SimpleLocationView(self._leftFrame)
# right frame
self._rightFrame = Frame(self)
self._nameLabel = Label(self._rightFrame,
compound=LEFT,
text="Name",
font=BaseView.LARGE_FONT)
self._descLabel = Label(self._rightFrame,
text="Description:",
font=BaseView.NORMAL_FONT)
self._descFrame = Frame(self._rightFrame)
self._descText = Text(self._descFrame,
width=50,
height=4,
state=DISABLED,
wrap=WORD,
font=BaseView.NORMAL_FONT)
self._descScroll = Scrollbar(self._descFrame,
command=self._descText.yview)
self._descText.config(yscrollcommand=self._descScroll.set)
self._notesLabel = Label(self._rightFrame,
text="Notes:",
font=BaseView.NORMAL_FONT)
self._notesFrame = Frame(self._rightFrame)
self._notesText = Text(self._notesFrame,
width=50,
height=4,
state=DISABLED,
wrap=WORD,
font=BaseView.NORMAL_FONT)
self._notesScroll = Scrollbar(self._notesFrame,
command=self._notesText.yview)
self._notesText.config(yscrollcommand=self._notesScroll.set)
# bottom
self._sep1 = Separator(self, orient="horizontal")
self._invLabel = Label(self,
text="Inventory:",
font=BaseView.NORMAL_FONT)
self._invFrame = Frame(self)
self._inventory = Treeview(
self._invFrame,
columns=["type", "price", "qty", "stockDays"],
selectmode="browse",
height=15)
self._inventory.heading("#0", text="Item", anchor=N + W)
self._inventory.column("#0", width=300, anchor=N + W, stretch=True)
self._inventory.heading("type", text="Type", anchor=N + W)
self._inventory.column("type", width=100, anchor=N + W)
self._inventory.heading("price", text="Price", anchor=N + W)
self._inventory.column("price", width=60, anchor=N + W)
self._inventory.heading("qty", text="Qty", anchor=N + W)
self._inventory.column("qty", width=40, anchor=N + W)
self._inventory.heading("stockDays", text="Stock Days", anchor=N + W)
self._inventory.column("stockDays",
width=200,
anchor=N + W,
stretch=True)
self._invScroll = Scrollbar(self._invFrame,
command=self._inventory.yview)
self._inventory.config(yscrollcommand=self._invScroll.set)
# placement: scrollbars
self._descText.grid(row=0, column=0, sticky=N + W + E + S)
self._descScroll.grid(row=0, column=1, sticky=N + S)
self._notesText.grid(row=0, column=0, sticky=N + W + E + S)
self._notesScroll.grid(row=0, column=1, sticky=N + S)
self._inventory.grid(row=0, column=0, sticky=N + W + E + S)
self._invScroll.grid(row=0, column=1, sticky=N + S)
# placement: left frame
self._imgLabel.grid(row=0, column=0, sticky=N + W + E + S)
self._locLabel.grid(row=1, column=0, sticky=N + W)
self._locPreview.grid(row=2, column=0, sticky=W + E)
# placement: right frame
self._nameLabel.grid(row=0, column=0, sticky=W)
self._descLabel.grid(row=1, column=0, sticky=W)
self._descFrame.grid(row=2, column=0, sticky=W)
self._notesLabel.grid(row=3, column=0, sticky=W)
self._notesFrame.grid(row=4, column=0, sticky=W)
# bottom
self._sep1.grid(row=1, column=0, columnspan=2, sticky=W + E)
self._invLabel.grid(row=2, column=0, columnspan=2, sticky=N + W)
self._invFrame.grid(row=3, column=0, columnspan=2, sticky=W + E)
self._leftFrame.grid(row=0, column=0, sticky=N + W)
self._rightFrame.grid(row=0, column=1, sticky=N + W)
########
# Add callbacks for all GUI element events and Tkinter variables.
def _bind_widgets(self):
self._locPreview.bind("<Double-Button-1>", self._open_loc)
self._inventory.bind("<Double-Button-1>",
self._inventory_on_double_click)
########
# Populates all GUI elements with new data.
def populate(self, data):
self._data = data
if data == None: # null check
self.set_defaults()
return
for k, v in data.items():
if k == "name":
# non-null
self._nameLabel.config(text=v)
elif k == "img":
if v == None: # null check
v = BaseView.DEFAULT_IMG
utility.update_img(self._imgLabel, v, maxSize=300)
elif k == "location":
# non-null
self._locPreview.populate(v)
elif k == "description":
if v == None: # null check
v = BaseView.EMPTY_STR
utility.update_text(self._descText, v)
elif k == "notes":
if v == None: # null check
v = BaseView.EMPTY_STR
utility.update_text(self._notesText, v)
elif k == "sells":
self._update_inventory()
########
# Resets GUI elements to default values.
def set_defaults(self):
utility.update_img(self._imgLabel, BaseView.DEFAULT_IMG, maxSize=300)
utility.update_text(self._descText, BaseView.EMPTY_STR)
utility.update_text(self._notesText, BaseView.EMPTY_STR)
########
# Populates inventory with correct values
def _update_inventory(self):
allItems = self._inventory.get_children()
for item in allItems:
self._inventory.delete(item)
self._imgs = []
if self._data == None:
return
for entry in self._data["sells"]:
img = entry["item"]["img"]
if img == None:
img = BaseView.DEFAULT_IMG
img = utility.get_img(img, maxSize=20)
name = entry["item"]["name"]
fields = [
BaseView.TYPE_MAP[entry["item"]["type"]], entry["price"],
entry["qty"], entry["stockDays"]
]
for i in range(len(fields)):
if fields[i] == None:
fields[i] = BaseView.EMPTY_STR
self._imgs.append(img)
self._inventory.insert("",
END,
image=img,
text=name,
values=fields)
########
# Opens location view through refBook.
def _open_loc(self, *args, **kwargs):
if self._refBook == None or self._data == None:
return
self._refBook.show_location(self._data["location"]["name"])
########
# Callback for double clicking in inventory treeview.
def _inventory_on_double_click(self, *args, **kwargs):
if self._refBook == None or self._data == None or len(
self._inventory.selection()) == 0:
return
self._refBook.show_item(self._data["sells"][self._inventory.index(
self._inventory.selection())]["item"]["name"])
class DialogOpenArchive(Toplevel):
def __init__(self, mainWin, openType, filesource, filenames, title, colHeader, showAltViewButton=False):
parent = mainWin.parent
super(DialogOpenArchive, self).__init__(parent)
self.parent = parent
self.showAltViewButton = showAltViewButton
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.accepted = False
self.transient(self.parent)
frame = Frame(self)
treeFrame = Frame(frame, width=500)
vScrollbar = Scrollbar(treeFrame, orient=VERTICAL)
hScrollbar = Scrollbar(treeFrame, orient=HORIZONTAL)
self.treeView = Treeview(treeFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set)
self.treeView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.treeView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.treeView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
treeFrame.columnconfigure(0, weight=1)
treeFrame.rowconfigure(0, weight=1)
treeFrame.grid(row=0, column=0, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.treeView.focus_set()
mainWin.showStatus(_("loading archive {0}").format(filesource.url))
self.filesource = filesource
self.filenames = filenames
self.selection = filesource.selection
self.hasToolTip = False
selectedNode = None
if openType == ENTRY_POINTS:
try:
metadataFiles = filesource.taxonomyPackageMetadataFiles
''' take first for now
if len(metadataFiles) != 1:
raise IOError(_("Taxonomy package contained more than one metadata file: {0}.")
.format(', '.join(metadataFiles)))
'''
metadataFile = metadataFiles[0]
metadata = filesource.url + os.sep + metadataFile
self.metadataFilePrefix = os.sep.join(os.path.split(metadataFile)[:-1])
if self.metadataFilePrefix:
self.metadataFilePrefix += "/" # zip contents have /, never \ file seps
self.taxonomyPkgMetaInf = '{}/META-INF/'.format(
os.path.splitext(os.path.basename(filesource.url))[0])
self.taxonomyPackage = parsePackage(mainWin, filesource, metadata,
os.sep.join(os.path.split(metadata)[:-1]) + os.sep)
# may be a catalog file with no entry oint names
if not self.taxonomyPackage["nameToUrls"]:
openType = ARCHIVE # no entry points to show, just archive
self.showAltViewButton = False
except Exception as e:
self.close()
err = _("Failed to parse metadata; the underlying error was: {0}").format(e)
messagebox.showerror(_("Malformed taxonomy package"), err)
mainWin.addToLog(err)
return
mainWin.showStatus(None)
if openType == DISCLOSURE_SYSTEM:
y = 3
else:
y = 1
okButton = Button(frame, text=_("OK"), command=self.ok)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
okButton.grid(row=y, column=2, sticky=(S,E,W), pady=3)
cancelButton.grid(row=y, column=3, sticky=(S,E,W), pady=3, padx=3)
if self.showAltViewButton:
self.altViewButton = Button(frame, command=self.showAltView)
self.altViewButton.grid(row=y, column=0, sticky=(S,W), pady=3, padx=3)
self.loadTreeView(openType, colHeader, title)
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.toolTipText = StringVar()
if self.hasToolTip:
self.treeView.bind("<Motion>", self.motion, '+')
self.treeView.bind("<Leave>", self.leave, '+')
self.toolTipText = StringVar()
self.toolTip = ToolTip(self.treeView,
textvariable=self.toolTipText,
wraplength=640,
follow_mouse=True,
state="disabled")
self.toolTipRowId = None
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeView(self, openType, title, colHeader):
self.title(title)
self.openType = openType
selectedNode = None
# clear previous treeview entries
for previousNode in self.treeView.get_children(""):
self.treeView.delete(previousNode)
# set up treeView widget and tabbed pane
if openType in (ARCHIVE, DISCLOSURE_SYSTEM):
self.treeView.column("#0", width=500, anchor="w")
self.treeView.heading("#0", text=colHeader)
try:
self.isRss = self.filesource.isRss
if self.isRss:
self.treeView.column("#0", width=350, anchor="w")
self.treeView["columns"] = ("descr", "date", "instDoc")
self.treeView.column("descr", width=50, anchor="center", stretch=False)
self.treeView.heading("descr", text="Form")
self.treeView.column("date", width=170, anchor="w", stretch=False)
self.treeView.heading("date", text="Pub Date")
self.treeView.column("instDoc", width=200, anchor="w", stretch=False)
self.treeView.heading("instDoc", text="Instance Document")
except AttributeError:
self.isRss = False
self.treeView["columns"] = tuple()
loadedPaths = []
for i, filename in enumerate(self.filenames):
if isinstance(filename,tuple):
if self.isRss:
form, date, instDoc = filename[2:5]
filename = filename[0] # ignore tooltip
self.hasToolTip = True
if filename.endswith("/"):
filename = filename[:-1]
path = filename.split("/")
if not self.isRss and len(path) > 1 and path[:-1] in loadedPaths:
parent = "file{0}".format(loadedPaths.index(path[:-1]))
else:
parent = ""
node = self.treeView.insert(parent, "end", "file{0}".format(i), text=path[-1])
if self.isRss:
self.treeView.set(node, "descr", form)
self.treeView.set(node, "date", date)
self.treeView.set(node, "instDoc", os.path.basename(instDoc))
if self.selection == filename:
selectedNode = node
loadedPaths.append(path)
elif openType == ENTRY_POINTS:
self.treeView.column("#0", width=150, anchor="w")
self.treeView.heading("#0", text="Name")
self.treeView["columns"] = ("url",)
self.treeView.column("url", width=350, anchor="w")
self.treeView.heading("url", text="URL")
for name, urls in self.taxonomyPackage["nameToUrls"].items():
displayUrl = urls[1] # display the canonical URL
self.treeView.insert("", "end", name, values=[displayUrl], text=name)
self.hasToolTip = True
else: # unknown openType
return None
if selectedNode:
self.treeView.see(selectedNode)
self.treeView.selection_set(selectedNode)
if self.showAltViewButton:
self.altViewButton.config(text=_("Show Files") if openType == ENTRY_POINTS else _("Show Entries"))
def ok(self, event=None):
selection = self.treeView.selection()
if len(selection) > 0:
if hasattr(self, "taxonomyPackage"):
# load file source remappings
self.filesource.mappedPaths = self.taxonomyPackage["remappings"]
filename = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
filename = self.filenames[int(selection[0][4:])]
if isinstance(filename,tuple):
if self.isRss:
filename = filename[4]
else:
filename = filename[0]
elif self.openType == ENTRY_POINTS:
epName = selection[0]
#index 0 is the remapped Url, as opposed to the canonical one used for display
filename = self.taxonomyPackage["nameToUrls"][epName][0]
if not filename.endswith("/"):
# check if it's an absolute URL rather than a path into the archive
if not isHttpUrl(filename) and self.metadataFilePrefix != self.taxonomyPkgMetaInf:
# assume it's a path inside the archive:
filename = self.metadataFilePrefix + filename
if filename is not None and not filename.endswith("/"):
if hasattr(self, "taxonomyPackage"):
# attempt to unmap the filename to original file
# will be mapped again in loading, but this allows schemaLocation to be unmapped
for prefix, remapping in self.taxonomyPackage["remappings"].items():
if isHttpUrl(remapping):
remapStart = remapping
else:
remapStart = self.metadataFilePrefix + remapping
if filename.startswith(remapStart):
# set unmmapped file
filename = prefix + filename[len(remapStart):]
break
self.filesource.select(filename)
self.accepted = True
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def showAltView(self, event=None):
if self.openType == ENTRY_POINTS:
self.loadTreeView(ARCHIVE, _("Select Entry Point"), _("File"))
else:
self.loadTreeView(ENTRY_POINTS, _("Select Archive File"), _("File"))
def leave(self, *args):
self.toolTipRowId = None
def motion(self, *args):
tvRowId = self.treeView.identify_row(args[0].y)
if tvRowId != self.toolTipRowId:
text = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
self.toolTipRowId = tvRowId
if tvRowId and len(tvRowId) > 4:
try:
text = self.filenames[ int(tvRowId[4:]) ]
if isinstance(text, tuple):
text = text[1].replace("\\n","\n")
except (KeyError, ValueError):
pass
elif self.openType == ENTRY_POINTS:
try:
epUrl = self.taxonomyPackage["nameToUrls"][tvRowId][1]
text = "{0}\n{1}".format(tvRowId, epUrl)
except KeyError:
pass
self.setToolTip(text)
def setToolTip(self, text):
self.toolTip._hide()
if text:
self.toolTipText.set(text)
self.toolTip.configure(state="normal")
self.toolTip._schedule()
else:
self.toolTipText.set("")
self.toolTip.configure(state="disabled")
class App(ttk.Frame):
def __init__(self, parent=None, *args, **kwargs):
self._running = threading.Event()
self._stop = threading.Event()
self._stop.clear()
self.init_UI(parent)
self.init()
def init_UI(self, parent=None):
ttk.Frame.__init__(self, parent)
self.parent = parent
'''
### Frame setting
'''
self.frame_l = Frame(self.parent)
self.frame_l.grid(row=0, column=0, padx=1, pady=1, sticky='NSWE')
self.frame_r = Frame(self.parent)
self.frame_r.grid(row=0, column=1, padx=1, pady=1, sticky='NSWE')
Grid.rowconfigure(self.frame_r, 0, weight=1)
Grid.columnconfigure(self.frame_r, 0, weight=1)
'''
### setting_panel
'''
self.setting_panel = ttk.LabelFrame(self.frame_l, text=u"设置")
self.setting_panel.grid(column=0, row=0, padx=5, pady=0)
self.add_btn = Button(
self.setting_panel, text=u"添加",
command=self.add_btn_click, bg='#2E8B57')
self.add_btn.grid(row=0, column=2, padx=1, pady=5, sticky='NWSE')
self.new_url = StringVar()
self.entry_new_url = Entry(
self.setting_panel, textvariable=self.new_url)
self.entry_new_url.grid(row=0, column=0, pady=5, columnspan=2, sticky='NWSE')
Label(self.setting_panel, text=u'刷新频率(秒)：').grid(
row=1, column=0, padx=0, pady=5, sticky='NWSE')
self.set_btn = Button(
self.setting_panel, text=u"设置",
command=self.set_btn_click, bg='#2E8B57')
self.set_btn.grid(row=1, column=2, pady=5, sticky='NWSE')
self.refresh_delay = StringVar()
self.entry_refresh_delay = Entry(
self.setting_panel, textvariable=self.refresh_delay)
self.entry_refresh_delay.grid(row=1, column=1, pady=5, sticky='NWSE')
self.refresh_delay.set('60')
'''
### control_panel
'''
self.control_panel = ttk.LabelFrame(self.frame_l, text=u"操作")
self.control_panel.grid(column=0, row=1, padx=5, pady=0, sticky='NWSE')
Grid.columnconfigure(self.control_panel, 0, weight=1)
self.displayed_msg = StringVar()
self.displayed_msg.set(u'正在初始化...')
self.label = Label(self.control_panel, textvariable=self.displayed_msg, bg='#AAAAAA')
self.label.grid(row=0, column=0, padx=1, pady=5, rowspan=2, sticky='NWSE')
self.start_btn = Button(
self.control_panel, text=u"开始",
command=self.start_btn_click, bg='#2E8B57')
self.start_btn.grid(row=0, column=1, padx=1, pady=5, sticky='NWSE')
self.stop_btn = Button(
self.control_panel, text=u"停止",
command=self.stop_btn_click, bg='#2E8B57')
self.stop_btn.grid(row=1, column=1, padx=1, pady=5, sticky='NWSE')
'''
### Treeview
'''
self.tree = Treeview(
self.frame_r,
columns=['c1', 'c2', 'c3'],
displaycolumns=['c1', 'c2'],
selectmode='browse',
)
# 设置每列宽度和对齐方式
self.tree.column('#0', anchor='center', width=60)
self.tree.column('c1', anchor='w')
self.tree.column('c2', anchor='center', width=80)
self.tree.column('c3', anchor='center')
# 设置每列表头标题文本
self.tree.heading('c1', text=u'视频名称')
self.tree.heading('c2', text=u'发布时间')
self.tree.heading('c3', text=u'url')
self.tree.grid(row=0, column=0, sticky='NSWE')
# ----vertical scrollbar------------
self.vbar = ttk.Scrollbar(
self.frame_r, orient=VERTICAL, command=self.tree.yview)
self.tree.configure(yscrollcommand=self.vbar.set)
self.vbar.grid(row=0, column=1, sticky='NS')
'''
# 定义并绑定Treeview组件的鼠标单击事件
'''
menu = Menu(self.frame_r, tearoff=0)
menu.add_command(label=u"删除选中行", command=self.delete)
def treeviewDoubleClick(event):
'''
如果是根结点，打开个人主页
如果是子节点，打开视频页面
'''
webbrowser.open(
self.tree.item(self.tree.selection()[0], "values")[2])
if self.tree.tag_has('unread', self.tree.selection()[0]):
self.tree.item(
self.tree.selection()[0],
image=image_old,
tag=['root', 'read'])
def treeviewOpen(event):
'''
如果是根结点，打开个人主页
如果是子节点，打开视频页面
'''
if self.tree.tag_has('unread', self.tree.selection()[0]):
self.tree.item(
self.tree.selection()[0],
image=image_old,
tag=['root', 'read']
)
def treeviewPopupmenu(event):
menu.post(event.x_root, event.y_root)
self.tree.bind('<Double-1>', treeviewDoubleClick)
self.tree.bind('<<TreeviewOpen>>', treeviewOpen)
self.tree.bind('<Button-3>', treeviewPopupmenu)
def init(self):
if not os.path.exists(DATA_PATH):
os.makedirs(DATA_PATH)
if not os.path.exists(CONF_PATH):
os.makedirs(CONF_PATH)
self.data_list = []
files = os.listdir(DATA_PATH)
for file in files:
with open('%s%s' %(DATA_PATH, file), 'r') as f:
data = json.load(f)
self.new_watcher(json.dumps(data))
if os.path.exists(CONF_FILE):
with open(CONF_FILE, 'r') as f:
try:
self.refresh_delay.set(json.load(f)['wait_time'])
except:
pass
self.p = threading.Thread(target=self.run)
self.delay = int(self.refresh_delay.get())
self.status_audit()
def new_watcher(self, data):
logging.info('new_watcher')
logging.info(data)
info = json.loads(data)
try:
self.tree.insert('', 'end',
info['u_id'],
text=info['u_name'],
image=image_old,
values=['', '', info['url']],
tag=['root', 'read'])
self.data_list.append(info)
except Exception, ex:
logging.error(ex)
class Configurator(tk.Tk):
"""
The main Tk window representing the main app.
Attributes
----------
treeview : :py:class:`~tkinter.Treeview`
The treeview widget.
treeview_popup_target_id : `int`
The pop target id relating to the id of the selected element.
treeview_popup : :py:class:`~tkinter.Widget`
The treeview popup widget.
cfg_file_name : `str`
The file name of the current configuration.
element_dict : `dict`
The dictionary of elements. Keys are the element ids.
root_element : :py:class:`~enrich2.base.storemanager.StoreManager`
An instance inheriting from storemanager that acts as a root object.
force_recalculate :py:class:`tkinter.BooleanVar`
The tkinter boolean variable for this option.
component_outliers :py:class:`tkinter.BooleanVar`
The tkinter boolean variable for this option.
tsv_requested : :py:class:`tkinter.BooleanVar`
The tkinter boolean variable for this option.
treeview_buttons : `list`
The ``new``, ``edit`` and ``delete`` buttons.
go_button : :py:class`~tkinter.ttk.Button`
The button that begins the analysis
scorer_widget : :py:class:`~enrich2.gui.options_frame.ScorerScriptsDropDown`
The ScorerScriptsDropDown instance associated with this app.
scorer : :py:class:`~enrich2.plugins.scoring.BaseScorerPlugin`
The scorer class loaded from a plugin
scorer_attrs : `dict`
The scoring attributes for the plugin.
scorer_path : `str`
The path to the currently selected scoring plugin.
analysis_thread : :py:class:`~threading.Thread`
The thread object that runs the computation method to prevent
GUI blocking.
Methods
-------
create_main_frame
create_menubar
create_treeview_context_menu
create_new_element
menu_open
menu_save
menu_saveas
menu_selectall
refresh_treeview
treeview_context_menu
set_treeview_properties
populate_tree
go_button_press
new_button_press
edit_button_press
delete_button_press
delete_element
apply_seqlib_fastq
get_element
get_focused_element
get_selected_elements
get_selected_scorer_class
get_selected_scorer_attrs
get_selected_scorer_path
run_analysis
set_gui_state
configure_analysis
refresh_plugins
show_plugin_source_window
See Also
--------
:py:class:`~tkinter.Tk`
"""
def __init__(self):
tk.Tk.__init__(self)
self.title("Enrich 2")
# Main app variables
self.cfg_file_name = tk.StringVar()
self.element_dict = dict()
self.root_element = None
self.analysis_thread = None
self.plugin_source_window = None
self.queue = queue.Queue()
# Treeview variables
self.treeview = None
self.treeview_popup_target_id = None
self.treeview_popup = None
# analysis options
self.force_recalculate = tk.BooleanVar()
self.component_outliers = tk.BooleanVar()
self.tsv_requested = tk.BooleanVar()
# allow resizing
self.rowconfigure(0, weight=1)
self.columnconfigure(0, weight=1)
# create UI elements
self.treeview_buttons = []
self.go_button = None
self.scorer_widget = None
self.scorer = None
self.scorer_attrs = None
self.scorer_path = None
self.create_main_frame()
self.create_menubar()
self.create_treeview_context_menu()
self.after(10, self.poll_logging_queue)
self.plugin_source_window = SourceWindow(master=self)
self.plugin_source_window.hide()
self.refresh_plugins()
# ---------------------------------------------------------------------- #
# Creation Methods
# ---------------------------------------------------------------------- #
def create_treeview_context_menu(self):
"""
This creates the tree-like view rendering the experiment heirachy.
"""
self.treeview_popup = tk.Menu(self, tearoff=0)
self.treeview_popup.add_command(label="Apply FASTQ...",
command=self.apply_seqlib_fastq)
def create_main_frame(self):
"""
Large function creating all the basic elements of the main app frame.
Creates the treeview and associated buttons, the scoring plugin frame
and the go button.
"""
# Frame for the Treeview and New/Edit/Delete buttons
main = Frame(self, padding=(3, 3, 12, 12))
main.rowconfigure(0, weight=1)
main.columnconfigure(0, weight=1)
main.columnconfigure(1, weight=0)
main.grid(row=0, column=0, sticky="nsew")
# ------------------------------------------------------- #
# Frame for the Treeview and its scrollbars
tree_frame = Frame(main, padding=(3, 3, 12, 12))
tree_frame.rowconfigure(0, weight=1)
tree_frame.rowconfigure(1, weight=0)
tree_frame.columnconfigure(0, weight=1)
tree_frame.columnconfigure(1, weight=0)
tree_frame.grid(row=0, column=0, sticky="nsew")
# ------------------------------------------------------- #
# Treeview with column headings
self.treeview = Treeview(tree_frame)
self.treeview["columns"] = ("class", "barcodes", "variants")
self.treeview.column("class", width=120)
self.treeview.heading("class", text="Type")
self.treeview.column("barcodes",
width=25,
stretch=tk.NO,
anchor=tk.CENTER)
self.treeview.heading("barcodes", text="BC")
self.treeview.column("variants",
width=25,
stretch=tk.NO,
anchor=tk.CENTER)
self.treeview.heading("variants", text="V")
self.treeview.grid(row=0, column=0, sticky="nsew")
# Treeview context menu bindings
self.treeview.bind("<Button-2>", self.treeview_context_menu)
# Treeview scrollbars
tree_ysb = tk.Scrollbar(tree_frame,
orient="vertical",
command=self.treeview.yview)
tree_xsb = tk.Scrollbar(tree_frame,
orient="horizontal",
command=self.treeview.xview)
tree_ysb.grid(row=0, column=1, sticky="nsw")
tree_xsb.grid(row=1, column=0, sticky="ewn")
self.treeview.config(yscroll=tree_ysb.set, xscroll=tree_xsb.set)
# ------------------------------------------------------- #
# Frame for New/Edit/Delete buttons
button_frame = Frame(main, padding=(3, 3, 12, 12))
button_frame.grid(row=1, column=0)
new_button = Button(button_frame,
text="New...",
command=self.new_button_press)
new_button.grid(row=0, column=0)
edit_button = Button(button_frame,
text="Edit...",
command=self.edit_button_press)
edit_button.grid(row=0, column=1)
delete_button = Button(button_frame,
text="Delete",
command=self.delete_button_press)
delete_button.grid(row=0, column=2)
self.treeview_buttons = [new_button, delete_button, edit_button]
# ------------------------------------------------------- #
# Frame for Plugin and Analysis Options
right_frame = Frame(main, padding=(3, 3, 12, 12))
right_frame.rowconfigure(0, weight=1)
right_frame.rowconfigure(1, weight=0)
right_frame.columnconfigure(0, weight=1)
right_frame.columnconfigure(1, weight=0)
right_frame.grid(row=0, column=1, sticky="new")
# ------------------------------------------------------- #
# LabelFrame for plugin and options
scoring_plugin = ScorerScriptsDropDown(right_frame,
text="Scoring Options",
padding=(3, 3, 12, 12))
scoring_plugin.grid(row=0, column=0, sticky="new")
self.scorer_widget = scoring_plugin
# ------------------------------------------------------- #
# LabelFrame for Analysis Options
row = 0
options_frame = LabelFrame(right_frame,
text="Analysis Options",
padding=(3, 3, 12, 12))
options_frame.grid(row=1, column=0, sticky="new", pady=4)
# force recalculate
force_recalculate = Checkbutton(options_frame,
text="Force Recalculation",
variable=self.force_recalculate)
force_recalculate.grid(column=0, row=row, sticky="w")
row += 1
# component outliers
component_outliers = Checkbutton(
options_frame,
text="Component Outlier Statistics",
variable=self.component_outliers,
)
component_outliers.grid(column=0, row=row, sticky="w")
row += 1
# write tsv
tsv_requested = Checkbutton(options_frame,
text="Write TSV Files",
variable=self.tsv_requested)
tsv_requested.grid(column=0, row=row, sticky="w")
tsv_requested.invoke()
row += 1
# ------------------------------------------------------- #
# Run Analysis button frame
go_button_frame = Frame(main, padding=(3, 3, 12, 12))
go_button_frame.grid(row=1, column=1)
go_button = Button(go_button_frame,
text="Run Analysis",
command=self.go_button_press)
go_button.grid(column=0, row=0)
self.go_button = go_button
def create_new_element(self):
"""
Create and return a new element based on the current selection.
This element is not added to the treeview.
"""
element = None
parent_element = self.get_focused_element()
if isinstance(parent_element, Experiment):
element = Condition()
element.parent = parent_element
elif isinstance(parent_element, Condition):
element = Selection()
element.parent = parent_element
elif isinstance(parent_element, Selection):
element = CreateSeqLibDialog(self).element_type()
element.parent = parent_element
elif isinstance(parent_element, SeqLib):
# special case: creates a copy of the selected SeqLib as a sibling
element = type(parent_element)()
element.configure(parent_element.serialize())
element.parent = parent_element.parent
# clear out the seqlib-specific values
element.name = None
element.timepoint = None
element.counts_file = None
element.reads = None
else:
raise ValueError("Unrecognized parent object "
"type '{}'".format(type(parent_element)))
return element
def create_menubar(self):
"""
Creates the menubar for the main app, with associated drop down menus.
"""
# make platform-specific keybinds
if platform.system() == "Darwin":
accel_string = "Command+"
accel_bind = "Command-"
else:
accel_string = "Ctrl+"
accel_bind = "Control-"
# create the menubar
menubar = tk.Menu(self)
# file menu
filemenu = tk.Menu(menubar, tearoff=0)
filemenu.add_command(
label="Open...",
accelerator="{}O".format(accel_string),
command=self.menu_open,
)
filemenu.add_command(label="Save",
accelerator="{}S".format(accel_string),
command=self.menu_save)
filemenu.add_command(
label="Save As...",
accelerator="{}Shift+S".format(accel_string),
command=self.menu_saveas,
)
menubar.add_cascade(label="File", menu=filemenu)
# edit menu
filemenu = tk.Menu(menubar, tearoff=0)
filemenu.add_command(
label="Select All",
accelerator="{}A".format(accel_string),
command=self.menu_selectall,
)
menubar.add_cascade(label="Edit", menu=filemenu)
# tools menu
filemenu = tk.Menu(menubar, tearoff=0)
filemenu.add_command(
label="Show Log",
accelerator="{}L".format(accel_string),
command=show_log_window,
)
filemenu.add_command(
label="Plugin Sources",
accelerator="{}P".format(accel_string),
command=self.show_plugin_source_window,
)
filemenu.add_command(
label="Refresh Plugins",
accelerator="{}R".format(accel_string),
command=self.refresh_plugins,
)
menubar.add_cascade(label="Tools", menu=filemenu)
# add the menubar
self.config(menu=menubar)
# add file menu keybinds
self.bind("<{}o>".format(accel_bind), lambda event: self.menu_open())
self.bind("<{}s>".format(accel_bind), lambda event: self.menu_save())
self.bind("<{}Shift-s>".format(accel_bind),
lambda event: self.menu_saveas())
# add edit menu keybinds
self.bind("<{}a>".format(accel_bind),
lambda event: self.menu_selectall())
# add show log menu keybinds
# add edit menu keybinds
self.bind("<{}l>".format(accel_bind), lambda event: show_log_window())
self.bind("<{}p>".format(accel_bind),
lambda event: self.show_plugin_source_window())
self.bind("<{}r>".format(accel_bind),
lambda event: self.refresh_plugins())
# ---------------------------------------------------------------------- #
# Treeview Methods
# ---------------------------------------------------------------------- #
def treeview_context_menu(self, click):
"""
Sets the currently selected treeview object id in the variable
``treeview_popup_target_id``.
Parameters
----------
click :
tkinter click event
"""
target = self.treeview.identify_row(click.y)
if target != "":
self.treeview_popup_target_id = target
self.treeview_popup.post(click.x_root, click.y_root)
self.treeview_popup_target_id = None
def apply_seqlib_fastq(self):
"""
Applies settings to the seqlib object by running the configuration
method.
"""
SeqLibApplyDialog(self, self, self.treeview_popup_target_id)
def new_button_press(self):
"""
Spawns a dialog box depending on the currently selected treeview item
to create a new element.
"""
if self.treeview.focus() == "" and self.root_element is not None:
tkinter.messagebox.showwarning(None, "No parent element selected.")
else:
if self.treeview.focus() == "" and self.root_element is None:
element = CreateRootDialog(self).element
if isinstance(element, SeqLib):
EditDialog(self, self, element)
self.root_element = element
else:
element = self.create_new_element()
EditDialog(self, self, element)
# refresh the treeview and re-assign treeview id's
self.refresh_treeview()
# select the newly added element if it was successfully added
if element.treeview_id in list(self.element_dict.keys()):
self.treeview.focus(element.treeview_id)
self.treeview.selection_set(element.treeview_id)
else:
if element.parent is not None:
self.treeview.focus(element.parent.treeview_id)
self.treeview.selection_set(element.parent.treeview_id)
del element
def edit_button_press(self):
"""
Spawns a dialog box depending on the currently selected treeview item
to edit the selected element.
"""
if self.treeview.focus() == "":
tkinter.messagebox.showwarning(None, "No element selected.")
else:
EditDialog(self, self, self.get_focused_element())
def delete_button_press(self):
"""
Deletes the selected treeview element and it's children.
"""
if self.treeview.focus() == "":
tkinter.messagebox.showwarning(None, "No element selected.")
else:
DeleteDialog(self, self)
def delete_element(self, tree_id):
"""
Delete element with Treeview id *tree_id* from the tree, from the
element dictionary, and from the associated data structure. Recursively
deletes all children of *tree_id*.
The tree should be refreshed using :py:meth:`refresh_tree` after
each deletion. This is the responsibility of the caller.
Parameters
----------
tree_id : `int`
The id of the currently selected treeview element.
"""
if tree_id in self.element_dict:
# recursively delete children
if self.element_dict[tree_id].children is not None:
for child in self.element_dict[tree_id].children:
self.delete_element(child.treeview_id)
# check if deleting the root element
if self.root_element.treeview_id == tree_id:
# clear the root element
print("None {}".format(tree_id))
self.root_element = None
else:
try:
# remove the element from its parent's list of children
self.element_dict[tree_id].parent.remove_child_id(tree_id)
except AttributeError:
raise AttributeError(
"Non-root element lacks proper parent")
# delete the element from the dictionary
del self.element_dict[tree_id]
def refresh_treeview(self):
"""
Clears the Treeview and repopulates it with the
current contents of the tree.
"""
# clear the entries in the Treeview
for x in self.treeview.get_children():
self.treeview.delete(x)
# clear the id-element dictionary
# elements may be given new id's after repopulation
self.element_dict.clear()
# repopulate
if self.root_element is not None:
self.populate_tree(self.root_element)
def set_treeview_properties(self, element):
"""
Set the information text for the Treeview *element*.
Parameters
----------
element : :py:class:`~enrich2.base.storemanager.StoreManager`
The storemanager object to configure.
"""
# set class property
self.treeview.set(element.treeview_id, "class",
element.treeview_class_name)
# add the check marks for barcodes/variants
if "variants" in element.labels:
self.treeview.set(element.treeview_id, "variants", u"\u2713")
else:
self.treeview.set(element.treeview_id, "variants", "")
if "barcodes" in element.labels:
self.treeview.set(element.treeview_id, "barcodes", u"\u2713")
else:
self.treeview.set(element.treeview_id, "barcodes", "")
self.treeview.set(element.treeview_id, "class",
element.treeview_class_name)
def populate_tree(self, element, parent_id=""):
"""
Recursively populate the Treeview.
Also populates the *id_cfgstrings*.
Parameters
----------
element : :py:class:`~enrich2.base.storemanager.StoreManager`
The storemanager object to configure.
parent_id : `int`
``treeview_id`` of element's parent.
"""
# insert into the Treeview
element.treeview_id = self.treeview.insert(parent_id,
"end",
text=element.name,
open=True)
# add id-element pair to dictionary
self.element_dict[element.treeview_id] = element
# set information fields
self.set_treeview_properties(element)
# populate for children
if element.children is not None:
for child in element.children:
self.populate_tree(child, parent_id=element.treeview_id)
# ---------------------------------------------------------------------- #
# Getter Methods
# ---------------------------------------------------------------------- #
def get_selected_scorer_class(self):
"""
Returns the currently selected scoring class object.
"""
return self.scorer
def get_selected_scorer_attrs(self):
"""
Returns the currently selected scoring class attribute `dict`.
"""
return self.scorer_attrs
def get_selected_scorer_path(self):
"""
Returns the currently selected scoring path.
"""
return self.scorer_path
def get_element(self, treeview_id):
"""
Returns the element with *treeview_id*.
Parameters
----------
treeview_id : `int`
``treeview_id`` attribute of element to get.
Returns
-------
:py:class:`~enrich2.base.storemanager.StoreManager`
The instance with matching ``treeview_id``
"""
return self.element_dict[treeview_id]
def get_focused_element(self):
"""
Gets the focused element in the treeview.
Returns
-------
:py:class:`~enrich2.base.storemanager.StoreManager`
Returns the element that is currently being focused in the
Treeview. ``None`` if nothing is focused.
"""
if self.treeview.focus() != "":
return self.get_element(self.treeview.focus())
else:
return None
def get_selected_elements(self):
"""
Returns a list of currently selected elements in the treeview.
Returns
-------
`list`
Returns a list of elements that are currently selected in the
Treeview. If no elements are selected, it returns an empty list.
"""
return [self.get_element(x) for x in self.treeview.selection()]
# ---------------------------------------------------------------------- #
# Menubar Methods
# ---------------------------------------------------------------------- #
def menu_open(self):
"""
Spawns an `askopenfilename` dialog to open a configuration file.
"""
message_title = "Open Configuration"
fname = tkinter.filedialog.askopenfilename()
if len(fname) > 0: # file was selected
try:
with open(fname, "rU") as handle:
cfg = json.load(handle)
except ValueError:
tkinter.messagebox.showerror(message_title,
"Failed to parse config file.")
except IOError:
tkinter.messagebox.showerror(message_title,
"Could not read config file.")
else:
if is_experiment(cfg):
obj = Experiment()
elif is_selection(cfg):
obj = Selection()
elif is_seqlib(cfg):
sltype = seqlib_type(cfg)
obj = globals()[sltype]()
else:
tkinter.messagebox.showerror(
message_title, "Unrecognized config format.")
return
obj.output_dir_override = False
try:
if isinstance(obj, Experiment) or isinstance(
obj, Selection):
obj.configure(cfg, init_from_gui=True)
else:
obj.configure(cfg)
# Try load the scorer into the GUI
scorer_path = cfg.get(SCORER, {}).get(SCORER_PATH, "")
scorer_attrs = cfg.get(SCORER, {}).get(SCORER_OPTIONS, {})
if scorer_path:
self.scorer_widget.load_from_cfg_file(
scorer_path, scorer_attrs)
else:
log_message(
logging_callback=logging.warning,
msg="No plugin could be loaded from configuration.",
extra={"oname": self.__class__.__name__},
)
except Exception as e:
tkinter.messagebox.showerror(
message_title,
"Failed to load config file:\n\n{}".format(e))
else:
self.root_element = obj
self.cfg_file_name.set(fname)
self.refresh_treeview()
def menu_save(self):
"""
Asks the user where to save the current configuration.
"""
if len(self.cfg_file_name.get()) == 0:
self.menu_saveas()
elif self.root_element is None:
tkinter.messagebox.showwarning("Save Configuration",
"Cannot save empty configuration.")
else:
save = askyesno("Save Configuration",
"Overwrite existing configuration?")
if not save:
return
try:
with open(self.cfg_file_name.get(), "w") as handle:
cfg = self.root_element.serialize()
# Get the currently selected scorer
if not isinstance(self.root_element,
SeqLib) and not isinstance(
self.root_element, Condition):
(
_,
attrs,
scorer_path,
) = self.scorer_widget.get_scorer_class_attrs_path()
cfg[SCORER] = {
SCORER_PATH: scorer_path,
SCORER_OPTIONS: attrs
}
write_json(cfg, handle)
except IOError:
tkinter.messagebox.showerror("Save Configuration",
"Failed to save config file.")
else:
tkinter.messagebox.showinfo(
"Save Configuration",
"Saved file at location:\n\n{}".format(
self.cfg_file_name.get()),
)
def menu_saveas(self):
"""
Asks the user where to save the current configuration.
"""
if self.root_element is None:
tkinter.messagebox.showwarning("Save Configuration",
"Cannot save empty configuration.")
else:
fname = tkinter.filedialog.asksaveasfilename()
if len(fname) > 0: # file was selected
try:
with open(fname, "w") as handle:
cfg = self.root_element.serialize()
# Get the currently selected scorer
if not isinstance(self.root_element,
SeqLib) and not isinstance(
self.root_element, Condition):
(
_,
attrs,
scorer_path,
) = self.scorer_widget.get_scorer_class_attrs_path(
)
cfg[SCORER] = {
SCORER_PATH: scorer_path,
SCORER_OPTIONS: attrs,
}
write_json(cfg, handle)
except IOError:
tkinter.messagebox.showerror(
"Save Configuration", "Failed to save config file.")
else:
self.cfg_file_name.set(fname)
tkinter.messagebox.showinfo(
"Save Configuration",
"Saved file at location:\n\n{}".format(
self.cfg_file_name.get()),
)
def menu_selectall(self):
"""
Add all elements in the Treeview to the selection.
"""
for k in self.element_dict.keys():
self.treeview.selection_add(k)
def show_plugin_source_window(self):
"""
Show the pop-up window to modify plugin sources
"""
if not self.plugin_source_window:
self.plugin_source_window = SourceWindow(master=self)
else:
self.plugin_source_window.toggle_show()
# ---------------------------------------------------------------------- #
# Run Analysis Methods
# ---------------------------------------------------------------------- #
def go_button_press(self):
"""
Starts the analysis if all elements have been properly configured.
This will run the analysis in a new thread and block out GUI editing
to prevent the analysis breaking.
"""
(
self.scorer,
self.scorer_attrs,
self.scorer_path,
) = self.scorer_widget.get_scorer_class_attrs_path()
if self.scorer is None or self.scorer_attrs is None:
tkinter.messagebox.showwarning("Incomplete Configuration",
"No scoring plugin selected.")
elif self.root_element is None:
tkinter.messagebox.showwarning(
"Incomplete Configuration",
"No experimental design specified.")
else:
plugin, *_ = self.scorer_widget.get_selected_plugin()
if plugin.md5_has_changed():
proceed = askokcancel(
"Selected plugin has been modified.",
"The selected plugin has been modified on disk. Do you "
"want to proceed with the current version? To see changes "
"click 'Cancel' and refresh plugins before proceeding.",
)
if not proceed:
return
if askyesno(
"Save Configuration?",
"Would you like to save the confiugration "
"file before proceeding?",
):
self.menu_save()
run = askyesno(
"Begin Analysis?",
"Click Yes when you are ready to start.\n\nThis could "
"take some time so grab a cup of tea, or a beer if that's "
"your thing, and enjoy the show.",
)
if run:
self.configure_analysis()
self.set_gui_state(tk.DISABLED)
thread = threading.Thread(target=self.run_analysis)
thread.setDaemon(True)
self.analysis_thread = thread
self.analysis_thread.start()
self.after(100, self.poll_analysis_thread)
def poll_logging_queue(self):
"""
Polls the logging queue for messages to log.
"""
try:
log = get_logging_queue(init=True).get(0)
log[CALLBACK](log[MESSAGE], **log[KWARGS])
self.after(10, self.poll_logging_queue)
except queue.Empty:
self.after(10, self.poll_logging_queue)
def poll_analysis_thread(self):
"""
Polls the thread to check it's state. When it is finished, all stores
are closed.
"""
try:
analysis_result = self.queue.get(0)
self.handle_analysis_result(analysis_result)
except queue.Empty:
self.after(100, self.poll_analysis_thread)
def handle_analysis_result(self, success):
"""
Shows the appropriate messagebox and logs exceptions upon analysis
completing.
Parameters
----------
success : `bool`
Exception object if an error occured during analysis, otherwise
None to indicate successful computation.
"""
log_message(
logging_callback=logging.info,
msg="Closing stores...",
extra={"oname": self.root_element.name},
)
self.root_element.store_close(children=True)
log_message(
logging_callback=logging.info,
msg="Stores closed.",
extra={"oname": self.root_element.name},
)
if success:
showinfo("Analysis completed.",
"Analysis has completed successfully!")
log_message(
logging_callback=logging.info,
msg="Completed successfully!",
extra={"oname": self.root_element.name},
)
else:
showwarning(
"Error during analysis.",
"An error occurred during the analysis. See log for details",
)
log_message(
logging_callback=logging.info,
msg="Completed, but with errors!",
extra={"oname": self.root_element.name},
)
self.set_gui_state(tk.NORMAL)
def run_analysis(self):
"""
Runs the storemanager compute method.
"""
try:
self.root_element.validate()
self.root_element.store_open(children=True)
self.root_element.calculate()
if self.root_element.tsv_requested:
self.root_element.write_tsv()
self.queue.put(True, block=False)
except Exception as exception:
log_message(
logging_callback=logging.exception,
msg=exception,
extra={"oname": self.root_element.name},
)
self.queue.put(False, block=False)
finally:
return
def configure_analysis(self):
"""
Configures the attributes of the root_element by querying the GUI
options.
"""
try:
self.root_element.force_recalculate = self.force_recalculate.get()
self.root_element.component_outliers = self.component_outliers.get(
)
self.root_element.tsv_requested = self.tsv_requested.get()
scorer_class = self.get_selected_scorer_class()
scorer_class_attrs = self.get_selected_scorer_attrs()
scorer_path = self.get_selected_scorer_path()
self.root_element.scorer_class = scorer_class
self.root_element.scorer_class_attrs = scorer_class_attrs
self.root_element.scorer_path = scorer_path
except Exception as e:
log_message(
logging_callback=logging.info,
msg="An error occurred when trying to configure the "
"root element.",
extra={"oname": self.root_element.name},
)
log_message(
logging_callback=logging.exception,
msg=e,
extra={"oname": self.root_element.name},
)
# ---------------------------------------------------------------------- #
# GUI Modifications
# ---------------------------------------------------------------------- #
def set_gui_state(self, state):
"""
Sets the state of the `go_button`, `treeview` and `treeview_buttons`.
Parameters
----------
state : `str`
State to set, usually ``'normal'`` or ``'disabled'``
"""
for btn in self.treeview_buttons:
btn.config(state=state)
self.go_button.config(state=state)
if state == "normal":
self.treeview.bind("<Button-2>", self.treeview_context_menu)
else:
self.treeview.bind("<Button-2>", lambda event: event)
def refresh_plugins(self):
"""
Refresh the plugins by re-checking the sources file.
"""
if self.plugin_source_window:
sources = self.plugin_source_window.sources
self.scorer_widget.refresh_sources(sources)
class Search(Module):
def __init__(self, app):
self.app = app
self.window = None
self.context_size = 3
self.results = []
self.regex = StringVar(self.app)
# if anything ever changes the contents of any intervals
# it should call SearchModule.loadIntervals()
self.intervals = []
self.loadIntervals()
def handleClose(self, event=None):
self.window.destroy()
self.window = None
def createWindow(self):
self.window = Toplevel(self.app)
self.window.title('Search')
self.window.protocol("WM_DELETE_WINDOW", self.handleClose)
self.input = Entry(self.window, textvariable=self.regex)
self.input.grid(row=0, column=0)
self.input.bind('<Return>', self.search)
self.input.bind('<Escape>', lambda ev: self.window.focus())
self.searchButton = Button(self.window,
text='Search',
command=self.search,
takefocus=0)
self.searchButton.grid(row=0, column=1)
self.resultCount = Label(self.window, text='0 results')
self.resultCount.grid(row=0, column=2)
cols = ('File', 'Tier', 'Time', 'Text')
self.scroll = Scrollbar(self.window, orient='vertical')
self.resultList = Treeview(self.window,
columns=cols,
show="headings",
yscrollcommand=self.scroll.set,
selectmode='browse')
self.scroll.config(command=self.resultList.yview)
for col in cols:
self.resultList.heading(col, text=col)
self.resultList.grid(row=2, column=0, columnspan=3, sticky='news')
self.resultList.bind('<Double-1>', self.onClick)
Grid.rowconfigure(self.window, 2, weight=1)
Grid.columnconfigure(self.window, 0, weight=1)
self.scroll.grid(row=2, column=3, sticky='ns')
def openSearch(self):
if self.window == None:
self.createWindow()
self.window.lift()
self.input.focus()
def loadIntervals(self):
filecount = len(self.app.Data.getTopLevel('files'))
self.intervals = []
for f in range(filecount):
filename = self.app.Data.getFileLevel('name', f)
tg = self.app.Data.checkFileLevel('.TextGrid',
f,
shoulderror=False)
if tg:
grid = self.app.TextGrid.fromFile(tg)
for tier in grid:
if TextGrid.isIntervalTier(tier):
for el in tier:
if el.mark:
self.intervals.append(
(el, tier.name, filename))
def search(self, event=None):
if self.regex.get() == '':
self.results = []
else:
pat = re.compile(self.regex.get(),
re.IGNORECASE | re.MULTILINE | re.DOTALL)
self.results = []
for i in self.intervals:
s = pat.search(i[0].mark)
if s:
disp = i[0].mark
a = max(0, s.start() - self.context_size)
b = min(s.end() + self.context_size, len(disp))
self.results.append(i +
(('...' if a > 0 else '') + disp[a:b] +
('...' if b < len(disp) else ''), ))
self.resultCount.configure(text='%s results' % len(self.results))
for kid in self.resultList.get_children():
self.resultList.delete(kid)
for row, res in enumerate(self.results):
ls = (res[2], res[1], '%s-%s' % (res[0].minTime, res[0].maxTime),
res[3])
self.resultList.insert('', 'end', iid=str(row), values=ls)
def onClick(self, event=None):
self.jumpTo(int(self.resultList.selection()[0]))
def jumpTo(self, index):
self.app.filesJumpTo(self.results[index][2])
self.app.TextGrid.selectedTier.set(self.results[index][1])
self.app.TextGrid.start = self.results[index][0].minTime
self.app.TextGrid.end = self.results[index][0].maxTime
for i, f in enumerate(self.app.TextGrid.frameTier):
if f.time >= self.results[index][0].minTime:
self.app.frameSV.set(str(i))
self.app.framesJumpTo()
break
self.app.TextGrid.fillCanvases()
def reset(self, *args, **kwargs):
raise NotImplementedError('cannot call SearchModule::reset')
def update(self, *args, **kwargs):
raise NotImplementedError('cannot call SearchModule::update')
def grid(self, *args, **kwargs):
raise NotImplementedError('cannot call SearchModule::grid')
def grid_remove(self, *args, **kwargs):
raise NotImplementedError('cannot call SearchModule::grid_remove')
class RssPage(tk.Frame): # 继承Frame类
def __init__(self, parent, controller):
tk.Frame.__init__(self, parent)
self.controller = controller
self.rss_list = []
self.rss_path = './conf/rss_config.pickle'
if not os.path.isfile(self.rss_path):
self.create_config()
self.var_rss_link = StringVar()
self.var_rss_site = StringVar()
self.var_rss_name = StringVar()
self.label_link = Label(self, text='链接：', anchor=W)
self.label_name = Label(self, text='别名：', anchor=W)
self.label_site = Label(self, text='站点：', anchor=W)
self.entry_rss_link = tk.Entry(self,
textvariable=self.var_rss_link,
width=58,
borderwidth=2,
font=('Helvetica', '12'))
self.entry_rss_name = tk.Entry(self,
textvariable=self.var_rss_name,
width=58,
borderwidth=2,
font=('Helvetica', '12'))
self.entry_rss_site = tk.Entry(self,
textvariable=self.var_rss_site,
width=58,
borderwidth=2,
font=('Helvetica', '12'))
self.button_add = tk.Button(self,
text='添加',
width=6,
command=self.add_to_list)
self.frame_m = Frame(self)
self.scrollBar = Scrollbar(self.frame_m)
self.tree = Treeview(self.frame_m,
columns=('c1', 'c2', 'c3'),
show="headings",
yscrollcommand=self.scrollBar.set)
self.button_delete = tk.Button(self,
text='删除',
command=self.delete_rss)
self.button_save = tk.Button(self, text='保存', command=self.save_config)
self.create_page()
self.load_config()
def load_config(self):
try:
with open(self.rss_path, "rb") as rss_file:
rss_list = pickle.load(rss_file)
for item in rss_list:
self.tree.insert('',
'end',
values=[item[0], item[1], item[2]])
except FileNotFoundError:
pass
def create_config(self):
with open(self.rss_path, 'wb') as rss_file:
self.rss_list = []
pickle.dump(self.rss_list, rss_file)
def save_config(self):
rss_all = self.tree.get_children()
self.rss_list.clear()
for item in rss_all:
value_rss_link = self.tree.item(item, 'values')[0]
value_rss_name = self.tree.item(item, 'values')[1]
value_rss_site = self.tree.item(item, 'values')[2]
rss_tuple = (value_rss_link, value_rss_name, value_rss_site)
self.rss_list.append(rss_tuple)
with open(self.rss_path, 'wb') as rss_file:
pickle.dump(self.rss_list, rss_file)
def create_page(self):
self.label_link.place(x=32, y=30, width=80, height=30)
self.label_name.place(x=32, y=70, width=80, height=30)
self.label_site.place(x=330, y=70, width=80, height=30)
self.entry_rss_link.place(x=80, y=30, width=610, height=30)
self.entry_rss_name.place(x=80, y=70, width=200, height=30)
self.entry_rss_site.place(x=380, y=70, width=200, height=30)
self.button_add.place(x=610, y=70, width=80, height=30)
self.frame_m.place(x=32, y=120, width=680, height=420)
# 在Frame容器中创建滚动条
self.scrollBar.pack(side=RIGHT, fill=Y)
# 在Frame容器中使用Treeview组件实现表格功能
# Treeview组件，三列，显示表头，带垂直滚动条
# 设置每列宽度和对齐方式
self.tree.column('c1', width=400, anchor='center')
self.tree.column('c2', width=140, anchor='center')
self.tree.column('c3', width=120, anchor='center')
# 设置每列表头标题文本
self.tree.heading(
'c1',
text='链接',
command=lambda: self.treeview_sort_column(self.tree, 'c1', False))
self.tree.heading(
'c2',
text='别名',
command=lambda: self.treeview_sort_column(self.tree, 'c2', False))
self.tree.heading(
'c3',
text='站点名称',
command=lambda: self.treeview_sort_column(self.tree, 'c3', False))
# 左对齐，纵向填充
self.tree.pack(side=LEFT, fill=Y)
self.tree.bind('<Double-1>', self.treeviewclick)
# Treeview组件与垂直滚动条结合
self.scrollBar.config(command=self.tree.yview)
# 删除按钮
self.button_delete.place(x=160, y=555, width=120, height=30)
# 刷新按钮
self.button_save.place(x=460, y=555, width=120, height=30)
def treeview_sort_column(self, tv, col, reverse): # Treeview、列名、排列方式
ll = [(tv.set(k, col), k) for k in tv.get_children('')]
ll.sort(reverse=reverse) # 排序方式
for index, (val, k) in enumerate(ll): # 根据排序后索引移动
tv.move(k, '', index)
tv.heading(col,
command=lambda: self.treeview_sort_column(
tv, col, not reverse)) # 重写标题，使之成为再点倒序的标题
def add_to_list(self):
value_rss_site = self.var_rss_site.get()
value_rss_name = self.var_rss_name.get()
value_rss_link = self.var_rss_link.get()
values = [value_rss_link, value_rss_name, value_rss_site]
self.tree.insert('', 'end', values=values)
self.var_rss_link.set('')
self.var_rss_name.set('')
self.var_rss_site.set('')
def delete_rss(self):
if not self.tree.selection():
tk.messagebox.showerror('抱歉', '你还没有选择，不能删除')
return
for item in self.tree.selection():
self.tree.delete(item)
def treeviewclick(self, event):
selected_item = self.tree.selection()[0]
link = self.tree.item(selected_item, 'values')[0]
webbrowser.open(link)
class IncomeEntry(Frame, MoneySubject):
def __init__(self, parent):
super().__init__(parent)
MoneySubject.__init__(self)
self.parent = parent
self.initUI()
def initUI(self):
self.entries = {}
self.records = []
titleFrame = Frame(self, borderwidth=1)
titleFrame.pack(fill=BOTH, expand=True)
lbl = Label(titleFrame, text="Incomes:", font=("Ubuntu", 20))
lbl.pack(side=LEFT)
entryFrame = Frame(self, borderwidth=1)
for data in [("Hours", 3), ("Rate", 5), ("Desc", 15)]:
dataFrame = Frame(entryFrame, borderwidth=2)
lbl = Label(dataFrame, text=data[0], font=("Ubuntu", 10))
lbl.pack()
entry = Entry(dataFrame, width=data[1])
entry.pack()
self.entries[data[0]] = entry
dataFrame.pack(side=LEFT)
okButton = Button(entryFrame, text="Submit", command=self.submitIncome)
okButton.pack(side=LEFT)
entryFrame.pack(fill=BOTH, expand=True)
self.incomeList = Treeview(self, height=3)
columns = ("Hours", "Rate", "Desc", "Total")
self.incomeList["columns"] = columns
self.incomeList.column("#0", width=0, minwidth=0, stretch=NO)
for field in columns:
self.incomeList.column(field, width=100, stretch=NO)
self.incomeList.heading(field, text=field, anchor=W)
self.incomeList.pack(side=LEFT)
self.pack(fill=BOTH, expand=False)
self.incomeList.bind("<Delete>", self.keydown)
def keydown(self, e):
item = self.incomeList.selection()
if item != ():
values = self.incomeList.item(item)['values']
hours = float(values[0])
rate = float(values[1])
desc = str(values[2])
amt = hours * rate
self.moneyNotify({"income": -amt})
self.incomeList.delete(item)
self.records.remove({"hours": hours, "rate": rate, "desc": desc})
def _emitIncome(self, hours, rate, desc):
values = (hours, rate, desc, f"${hours*rate:.2f}")
self.incomeList.insert("", "end", text="", values=values)
self.records.append({"hours": hours, "rate": rate, "desc": desc})
self.moneyNotify({"income": hours * rate})
def submitIncome(self):
# consume input in entry boxes and store in tree
hours = float(self.entries["Hours"].get())
rate = float(self.entries["Rate"].get())
desc = self.entries["Desc"].get()
self._emitIncome(hours, rate, desc)
for entry in self.entries.values():
entry.delete(0, 'end')
def clear(self):
for child in self.incomeList.get_children():
self.incomeList.delete(child)
self.records = []
def export(self):
return self.records
def restore(self, incomes, startDate):
self.clear()
for income in incomes:
self._emitIncome(income["hours"], income["rate"], income["desc"])
class CustomerApp:
"""Main customer window."""
def __init__(self, master):
"""Initializes main customer window."""
self.master = master
self.master.geometry(CUSTOMER_WINDOW_SIZE)
self.master.configure(bg=my_config.BACKGROUND)
self.master.title(my_config.APP_NAME)
# main frames
self.frame = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame2 = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame3 = tk.Frame(self.master, bg=my_config.BACKGROUND)
# it contains error messages, for example not all entry are filled.
self.error_label = tk.Label()
self.product_tree = None
self.my_orders_tree = None
self.location_entry = None
self.quantity_entry = None
self.id_product_entry = None
def initialize_main_buttons(self):
"""Initializes main buttons.
Used in other functions repeatedly, that's why it's not in __init__"""
if self.frame:
self.frame.destroy()
if self.function_frame:
self.function_frame.destroy()
if self.function_frame2:
self.function_frame2.destroy()
if self.function_frame3:
self.function_frame3.destroy()
self.frame = tk.Frame(self.master, bg=my_config.BACKGROUND)
search_button = tk.Button(self.frame, text='List of products',
bg=my_config.FOREGROUND, command=self.list_products, width=16)
search_button.grid(row=0, column=0, pady=(10, 3))
edit_button = tk.Button(self.frame, text='Edit account', bg=my_config.FOREGROUND,
command=self.account_edit, width=16)
edit_button.grid(row=1, column=0, pady=(0, 3))
orders_button = tk.Button(self.frame, text='My Orders', bg=my_config.FOREGROUND,
command=self.my_orders, width=16)
orders_button.grid(row=2, column=0, pady=(0, 3))
logoff_button = tk.Button(self.frame, text='Logoff', bg=my_config.FOREGROUND,
command=self.log_off, width=16)
logoff_button.grid(row=3, column=0, pady=(0, 3))
self.frame.pack()
def list_products(self):
"""Lists all of the customer products under menu."""
self.initialize_main_buttons()
# frame for listbox
self.function_frame = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame.pack()
self.function_frame2 = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame2.pack()
list_label = tk.Label(self.function_frame, text='list of products',
width=100, bg=my_config.BACKGROUND)
list_label.grid(row=0, column=0, pady=(10, 0))
# creating treeview for customers
self.product_tree = Treeview(self.function_frame, columns=PRODUCT_COLUMNS,
show='headings', height=10)
self.product_tree.grid(row=1, column=0, padx=8)
for column_name, width in zip(PRODUCT_COLUMNS, PRODUCT_COLUMNS_SIZE):
self.product_tree.column(column_name, width=width, anchor=tk.CENTER)
self.product_tree.heading(column_name, text=column_name)
scrollbar = tk.Scrollbar(self.function_frame, orient=tk.VERTICAL)
scrollbar.configure(command=self.product_tree.set)
self.product_tree.configure(yscrollcommand=scrollbar)
self.product_tree.bind('<ButtonRelease-1>', self.product_selection)
# adding records from DB to Listbox
records = db.return_products()
for record in records:
self.product_tree.insert('', tk.END, values=[record[0], record[1], record[2], record[3]])
# crating labels
id_product_label = tk.Label(self.function_frame2, text='Product ID:', bg=my_config.BACKGROUND)
id_product_label.grid(row=0, column=0, sticky=tk.E)
quantity_label = tk.Label(self.function_frame2, text='Quantity:', bg=my_config.BACKGROUND)
quantity_label.grid(row=1, column=0, sticky=tk.E)
location_label = tk.Label(self.function_frame2, text='Order location:', bg=my_config.BACKGROUND)
location_label.grid(row=2, column=0, sticky=tk.E)
# creating entry boxes
self.id_product_entry = tk.Entry(self.function_frame2, width=30, bg=my_config.FOREGROUND)
self.id_product_entry.grid(row=0, column=1)
self.quantity_entry = tk.Entry(self.function_frame2, width=30, bg=my_config.FOREGROUND)
self.quantity_entry.grid(row=1, column=1)
self.location_entry = tk.Entry(self.function_frame2, width=30, bg=my_config.FOREGROUND)
self.location_entry.grid(row=2, column=1)
# buttons
place_order_button = tk.Button(self.function_frame2, text='Place order',
bg=my_config.FOREGROUND, command=self.place_order, width=16)
place_order_button.grid(row=4, column=0)
details_button = tk.Button(self.function_frame2, text='details',
bg=my_config.FOREGROUND, command=self.product_details, width=16)
details_button.grid(row=4, column=1, )
def place_order(self):
"""Place new order, if all required entries are filled."""
if self.error_label:
self.error_label.destroy()
# checking if all required entries are filled properly
if not self.id_product_entry.get():
self.error_message("'id product' missing")
elif not my_config.is_integer(self.quantity_entry.get()) or int(self.quantity_entry.get()) < 1:
self.error_message("'quantity' Must be an positive integer")
elif not self.location_entry.get():
self.error_message("'location' missing")
# checking if customer and product exists
elif not db.is_customer_id_exist(my_config.MY_ID) or not db.is_product_id_exists(
self.id_product_entry.get()):
self.error_message("product or customer id not Exists")
# function itself check if there is enough products, and count total price (quantity*price)
elif db.add_order(my_config.MY_ID, self.id_product_entry.get(), self.quantity_entry.get(),
self.location_entry.get()):
messagebox.showinfo("Mendiona bytes", 'successfully added.')
self.list_products()
else:
self.error_message("not enough products in stock.")
def product_details(self):
"""show details of selected product."""
if self.error_label:
self.error_label.destroy()
if self.function_frame3:
self.function_frame3.destroy()
if not self.id_product_entry.get():
self.error_message("select product.")
elif db.is_product_id_exists(self.id_product_entry.get()):
self.function_frame3 = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame3.pack(side=tk.TOP)
# creating Message instead of Label (description might be long)
description = db.return_product(self.id_product_entry.get())[4]
self.error_label = tk.Message(self.function_frame3, text="Description: {}".format(description),
bg=my_config.BACKGROUND, width=300)
self.error_label.grid(row=5, column=0)
else:
self.error_message("Product not exist.")
def product_selection(self, event):
"""Adds id of selected product to designated entry."""
try:
if self.product_tree.selection():
record = self.product_tree.set(self.product_tree.selection())
self.id_product_entry.delete(0, tk.END)
self.id_product_entry.insert(tk.END, record[PRODUCT_COLUMNS[0]])
except KeyError:
pass
def order_selection(self, event):
"""Shows details of selected order."""
if self.my_orders_tree.selection():
record = self.my_orders_tree.set(self.my_orders_tree.selection())
record = db.return_order(record[PRODUCT_COLUMNS[0]])
if self.function_frame2:
self.function_frame2.destroy()
self.function_frame2 = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame2.pack(side=tk.TOP)
# creating Message instead of Label (might be long)
order_info = ("quantity: \t{}\ntotal_price: \t{}\npayment_status: \t{}\n"
"send_status: \t{}\noder_date: \t{}\nlocation: \t{}\n"
).format(record[3], record[4], record[5], record[6], record[7], record[8])
self.error_label = tk.Message(self.function_frame2, text=order_info,
bg=my_config.BACKGROUND, width=300)
self.error_label.grid(row=0, column=0)
def account_edit(self):
"""Runs new window for editing account."""
if self.frame:
self.frame.destroy()
if self.function_frame:
self.function_frame.destroy()
if self.function_frame2:
self.function_frame2.destroy()
if self.function_frame3:
self.function_frame3.destroy()
AccountEdit(self.master)
def my_orders(self):
"""Creates menu with list of user orders."""
self.initialize_main_buttons()
self.function_frame = tk.Frame(self.master, bg=my_config.BACKGROUND)
self.function_frame.pack()
# creating listbox for customers
list_label = tk.Label(self.function_frame, text='my orders:', width=100, bg=my_config.BACKGROUND)
list_label.grid(row=0, column=0, pady=(10, 0))
# creating treeview for customers
self.my_orders_tree = Treeview(self.function_frame, columns=MY_ORDERS_COLUMNS,
show='headings', height=10)
self.my_orders_tree.grid(row=1, column=0)
for column_name, width in zip(MY_ORDERS_COLUMNS, MY_ORDERS_COLUMNS_SIZE):
self.my_orders_tree.column(column_name, width=width, anchor=tk.CENTER)
self.my_orders_tree.heading(column_name, text=column_name)
scrollbar = tk.Scrollbar(self.function_frame, orient=tk.VERTICAL)
scrollbar.configure(command=self.my_orders_tree.set)
self.my_orders_tree.configure(yscrollcommand=scrollbar)
self.my_orders_tree.bind('<ButtonRelease-1>', self.order_selection)
# adding records from DB to treeview
records = db.orders_product_info(my_config.MY_ID)
for record in records:
self.my_orders_tree.insert('', tk.END, values=[record[0], record[1], record[2], record[3]])
def error_message(self, name):
"""Shows passed message in designated place
Used to clear code and make it more readable as it is
called multiple times."""
# deleting missing label from last add_order call if it exists
if self.error_label:
self.error_label.destroy()
self.error_label = tk.Label(self.function_frame2, text=name, bg=my_config.BACKGROUND,
fg=my_config.ERROR_FOREGROUND)
self.error_label.grid(row=3, column=1)
def log_off(self):
"""Returns User to logging window."""
if self.frame:
self.frame.destroy()
if self.function_frame:
self.function_frame.destroy()
if self.function_frame2:
self.function_frame2.destroy()
if self.function_frame3:
self.function_frame3.destroy()
application = login_window.LoginWindow(self.master)
application.initialize_login_window()
class AutoCorrectConfig(Frame):
"""Configuration window for autocorrect."""
def __init__(self, master, app, **kwargs):
Frame.__init__(self, master, padding=4, **kwargs)
self.rowconfigure(2, weight=1)
self.columnconfigure(0, weight=1)
self.columnconfigure(1, weight=1)
self.tree = Treeview(self,
columns=('replace', 'by'),
show='',
selectmode='browse')
scroll_x = AutoScrollbar(self,
orient='horizontal',
command=self.tree.xview)
scroll_y = AutoScrollbar(self,
orient='vertical',
command=self.tree.yview)
self.tree.configure(xscrollcommand=scroll_x.set,
yscrollcommand=scroll_y.set)
self.reset()
self.replace = StringVar(self)
self.by = StringVar(self)
add_trace(self.replace, 'write', self._trace_replace)
add_trace(self.by, 'write', self._trace_by)
b_frame = Frame(self)
self.b_add = Button(b_frame, text=_('New'), command=self.add)
self.b_rem = Button(b_frame, text=_('Delete'), command=self.remove)
self.b_add.state(('disabled', ))
self.b_rem.state(('disabled', ))
self.b_add.pack(pady=4, fill='x')
self.b_rem.pack(pady=4, fill='x')
Button(b_frame, text=_('Reset'), command=self.reset).pack(pady=8,
fill='x')
Label(self, text=_('Replace')).grid(row=0,
column=0,
sticky='w',
pady=4)
Label(self, text=_('By')).grid(row=0, column=1, sticky='w', pady=4)
Entry(self, textvariable=self.replace).grid(row=1,
column=0,
sticky='ew',
pady=4,
padx=(0, 4))
Entry(self, textvariable=self.by).grid(row=1,
column=1,
sticky='ew',
pady=4)
self.tree.grid(row=2, columnspan=2, sticky='ewsn', pady=(4, 0))
scroll_x.grid(row=3, columnspan=2, sticky='ew', pady=(0, 4))
scroll_y.grid(row=2, column=2, sticky='ns', pady=(4, 0))
b_frame.grid(row=1, rowspan=2, padx=(4, 0), sticky='nw', column=3)
self.tree.bind('<<TreeviewSelect>>', self._on_treeview_select)
def _trace_by(self, *args):
key = self.replace.get().strip()
val = self.by.get().strip()
self.by.set(val)
if key in self.tree.get_children(''):
if val != self.tree.set(key, 'by'):
self.b_add.state(('!disabled', ))
else:
self.b_add.state(('disabled', ))
else:
self.b_add.state(('!disabled', ))
if not val:
self.b_add.state(('disabled', ))
def _trace_replace(self, *args):
key = self.replace.get().strip()
val = self.by.get().strip()
self.replace.set(key)
if not key:
self.b_add.state(('disabled', ))
self.b_rem.state(('disabled', ))
else:
self.b_add.state(('!disabled', ))
sel = self.tree.selection()
if key in self.tree.get_children(''):
if key not in sel:
self.tree.selection_set(key)
self.b_add.configure(text=_('Replace'))
self.b_rem.state(('!disabled', ))
if val != self.tree.set(key, 'by'):
self.b_add.state(('!disabled', ))
else:
self.b_add.state(('disabled', ))
else:
self.b_rem.state(('disabled', ))
self.b_add.configure(text=_('New'))
if sel:
self.tree.selection_remove(*sel)
if not val:
self.b_add.state(('disabled', ))
def _on_treeview_select(self, event):
sel = self.tree.selection()
if sel:
key, val = self.tree.item(sel[0], 'values')
self.replace.set(key)
self.by.set(val)
def reset(self):
self.tree.delete(*self.tree.get_children(''))
keys = list(AUTOCORRECT.keys())
keys.sort()
for key in keys:
self.tree.insert('', 'end', key, values=(key, AUTOCORRECT[key]))
def add(self):
key = self.replace.get().strip()
val = self.by.get().strip()
if key in self.tree.get_children(''):
self.tree.item(key, values=(key, val))
elif key and val:
self.tree.insert('', 'end', key, values=(key, val))
def remove(self):
key = self.replace.get()
if key in self.tree.get_children(''):
self.tree.delete(key)
def ok(self):
keys = self.tree.get_children('')
AUTOCORRECT.clear()
for key in keys:
AUTOCORRECT[key] = self.tree.set(key, 'by')
class DialogPluginManager(Toplevel):
def __init__(self, mainWin, modulesWithNewerFileDates):
super(DialogPluginManager, self).__init__(mainWin.parent)
self.ENABLE = _("Enable")
self.DISABLE = _("Disable")
self.parent = mainWin.parent
self.cntlr = mainWin
# copy plugins for temporary display
self.pluginConfig = PluginManager.pluginConfig
self.pluginConfigChanged = False
self.uiClassMethodsChanged = False
self.modelClassesChanged = False
self.disclosureSystemTypesChanged = False
self.modulesWithNewerFileDates = modulesWithNewerFileDates
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", self.parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.title(_("Plug-in Manager"))
frame = Frame(self)
# left button frame
buttonFrame = Frame(frame, width=40)
buttonFrame.columnconfigure(0, weight=1)
addLabel = Label(buttonFrame, text=_("Find plug-in modules:"), wraplength=60, justify="center")
addLocalButton = Button(buttonFrame, text=_("Locally"), command=self.findLocally)
ToolTip(addLocalButton, text=_("File chooser allows selecting python module files to add (or reload) plug-ins, from the local file system."), wraplength=240)
addWebButton = Button(buttonFrame, text=_("On Web"), command=self.findOnWeb)
ToolTip(addWebButton, text=_("Dialog to enter URL full path to load (or reload) plug-ins, from the web or local file system."), wraplength=240)
addLabel.grid(row=0, column=0, pady=4)
addLocalButton.grid(row=1, column=0, pady=4)
addWebButton.grid(row=2, column=0, pady=4)
buttonFrame.grid(row=0, column=0, rowspan=2, sticky=(N, S, W), padx=3, pady=3)
# right tree frame (plugins already known to arelle)
modulesFrame = Frame(frame, width=700)
vScrollbar = Scrollbar(modulesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(modulesFrame, orient=HORIZONTAL)
self.modulesView = Treeview(modulesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=7)
self.modulesView.grid(row=0, column=0, sticky=(N, S, E, W))
self.modulesView.bind('<<TreeviewSelect>>', self.moduleSelect)
hScrollbar["command"] = self.modulesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.modulesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
modulesFrame.columnconfigure(0, weight=1)
modulesFrame.rowconfigure(0, weight=1)
modulesFrame.grid(row=0, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.modulesView.focus_set()
self.modulesView.column("#0", width=120, anchor="w")
self.modulesView.heading("#0", text=_("Name"))
self.modulesView["columns"] = ("author", "ver", "status", "date", "update", "descr", "license")
self.modulesView.column("author", width=100, anchor="w", stretch=False)
self.modulesView.heading("author", text=_("Author"))
self.modulesView.column("ver", width=50, anchor="w", stretch=False)
self.modulesView.heading("ver", text=_("Version"))
self.modulesView.column("status", width=50, anchor="w", stretch=False)
self.modulesView.heading("status", text=_("Status"))
self.modulesView.column("date", width=70, anchor="w", stretch=False)
self.modulesView.heading("date", text=_("File Date"))
self.modulesView.column("update", width=50, anchor="w", stretch=False)
self.modulesView.heading("update", text=_("Update"))
self.modulesView.column("descr", width=200, anchor="w", stretch=False)
self.modulesView.heading("descr", text=_("Description"))
self.modulesView.column("license", width=70, anchor="w", stretch=False)
self.modulesView.heading("license", text=_("License"))
classesFrame = Frame(frame)
vScrollbar = Scrollbar(classesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(classesFrame, orient=HORIZONTAL)
self.classesView = Treeview(classesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=5)
self.classesView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.classesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.classesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
classesFrame.columnconfigure(0, weight=1)
classesFrame.rowconfigure(0, weight=1)
classesFrame.grid(row=1, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.classesView.focus_set()
self.classesView.column("#0", width=200, anchor="w")
self.classesView.heading("#0", text=_("Class"))
self.classesView["columns"] = ("modules",)
self.classesView.column("modules", width=500, anchor="w", stretch=False)
self.classesView.heading("modules", text=_("Modules"))
# bottom frame module info details
moduleInfoFrame = Frame(frame, width=700)
moduleInfoFrame.columnconfigure(1, weight=1)
self.moduleNameLabel = Label(moduleInfoFrame, wraplength=600, justify="left",
font=font.Font(family='Helvetica', size=12, weight='bold'))
self.moduleNameLabel.grid(row=0, column=0, columnspan=4, sticky=W)
self.moduleAuthorHdr = Label(moduleInfoFrame, text=_("author:"), state=DISABLED)
self.moduleAuthorHdr.grid(row=1, column=0, sticky=W)
self.moduleAuthorLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleAuthorLabel.grid(row=1, column=1, columnspan=3, sticky=W)
self.moduleDescrHdr = Label(moduleInfoFrame, text=_("description:"), state=DISABLED)
self.moduleDescrHdr.grid(row=2, column=0, sticky=W)
self.moduleDescrLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDescrLabel.grid(row=2, column=1, columnspan=3, sticky=W)
self.moduleClassesHdr = Label(moduleInfoFrame, text=_("classes:"), state=DISABLED)
self.moduleClassesHdr.grid(row=3, column=0, sticky=W)
self.moduleClassesLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleClassesLabel.grid(row=3, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleClassesLabel, text=_("List of classes that this plug-in handles."), wraplength=240)
self.moduleUrlHdr = Label(moduleInfoFrame, text=_("URL:"), state=DISABLED)
self.moduleUrlHdr.grid(row=4, column=0, sticky=W)
self.moduleUrlLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleUrlLabel.grid(row=4, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleUrlLabel, text=_("URL of plug-in module (local file path or web loaded file)."), wraplength=240)
self.moduleDateHdr = Label(moduleInfoFrame, text=_("date:"), state=DISABLED)
self.moduleDateHdr.grid(row=5, column=0, sticky=W)
self.moduleDateLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDateLabel.grid(row=5, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleDateLabel, text=_("Date of currently loaded module file (with parenthetical node when an update is available)."), wraplength=240)
self.moduleLicenseHdr = Label(moduleInfoFrame, text=_("license:"), state=DISABLED)
self.moduleLicenseHdr.grid(row=6, column=0, sticky=W)
self.moduleLicenseLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleLicenseLabel.grid(row=6, column=1, columnspan=3, sticky=W)
self.moduleEnableButton = Button(moduleInfoFrame, text=self.ENABLE, state=DISABLED, command=self.moduleEnable)
ToolTip(self.moduleEnableButton, text=_("Enable/disable plug in."), wraplength=240)
self.moduleEnableButton.grid(row=7, column=1, sticky=E)
self.moduleReloadButton = Button(moduleInfoFrame, text=_("Reload"), state=DISABLED, command=self.moduleReload)
ToolTip(self.moduleReloadButton, text=_("Reload/update plug in."), wraplength=240)
self.moduleReloadButton.grid(row=7, column=2, sticky=E)
self.moduleRemoveButton = Button(moduleInfoFrame, text=_("Remove"), state=DISABLED, command=self.moduleRemove)
ToolTip(self.moduleRemoveButton, text=_("Remove plug in from plug in table (does not erase the plug in's file)."), wraplength=240)
self.moduleRemoveButton.grid(row=7, column=3, sticky=E)
moduleInfoFrame.grid(row=2, column=0, columnspan=5, sticky=(N, S, E, W), padx=3, pady=3)
moduleInfoFrame.config(borderwidth=4, relief="groove")
okButton = Button(frame, text=_("Close"), command=self.ok)
ToolTip(okButton, text=_("Accept and changes (if any) and close dialog."), wraplength=240)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
ToolTip(cancelButton, text=_("Cancel changes (if any) and close dialog."), wraplength=240)
okButton.grid(row=3, column=3, sticky=(S,E), pady=3)
cancelButton.grid(row=3, column=4, sticky=(S,E), pady=3, padx=3)
enableDisableFrame = Frame(frame)
enableDisableFrame.grid(row=3, column=1, sticky=(S,W), pady=3)
enableAllButton = Button(enableDisableFrame, text=_("Enable All"), command=self.enableAll)
ToolTip(enableAllButton, text=_("Enable all plug ins."), wraplength=240)
disableAllButton = Button(enableDisableFrame, text=_("Disable All"), command=self.disableAll)
ToolTip(disableAllButton, text=_("Disable all plug ins."), wraplength=240)
enableAllButton.grid(row=1, column=1)
disableAllButton.grid(row=1, column=2)
self.loadTreeViews()
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=0)
frame.columnconfigure(1, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeViews(self):
self.selectedModule = None
# clear previous treeview entries
for previousNode in self.modulesView.get_children(""):
self.modulesView.delete(previousNode)
for i, moduleItem in enumerate(sorted(self.pluginConfig.get("modules", {}).items())):
moduleInfo = moduleItem[1]
name = moduleInfo.get("name", moduleItem[0])
node = self.modulesView.insert("", "end", name, text=name)
self.modulesView.set(node, "author", moduleInfo.get("author"))
self.modulesView.set(node, "ver", moduleInfo.get("version"))
self.modulesView.set(node, "status", moduleInfo.get("status"))
self.modulesView.set(node, "date", moduleInfo.get("fileDate"))
if name in self.modulesWithNewerFileDates:
self.modulesView.set(node, "update", _("available"))
self.modulesView.set(node, "descr", moduleInfo.get("description"))
self.modulesView.set(node, "license", moduleInfo.get("license"))
# clear previous treeview entries
for previousNode in self.classesView.get_children(""):
self.classesView.delete(previousNode)
for i, classItem in enumerate(sorted(self.pluginConfig.get("classes", {}).items())):
className, moduleList = classItem
node = self.classesView.insert("", "end", className, text=className)
self.classesView.set(node, "modules", ', '.join(moduleList))
self.moduleSelect() # clear out prior selection
def ok(self, event=None):
if self.pluginConfigChanged:
PluginManager.pluginConfig = self.pluginConfig
PluginManager.pluginConfigChanged = True
PluginManager.reset() # force reloading of modules
if self.uiClassMethodsChanged or self.modelClassesChanged or self.disclosureSystemTypesChanged: # may require reloading UI
affectedItems = ""
if self.uiClassMethodsChanged:
affectedItems += _("menus of the user interface")
if self.modelClassesChanged:
if self.uiClassMethodsChanged:
affectedItems += _(" and ")
affectedItems += _("model objects of the processor")
if (self.uiClassMethodsChanged or self.modelClassesChanged):
affectedItems += _(" and ")
if self.disclosureSystemTypesChanged:
if (self.uiClassMethodsChanged or self.modelClassesChanged):
affectedItems += _(" and ")
affectedItems += _("disclosure system types")
if messagebox.askyesno(_("User interface plug-in change"),
_("A change in plug-in class methods may have affected {0}. "
"Please restart Arelle to due to these changes. \n\n"
"Should Arelle restart itself now "
"(if there are any unsaved changes they would be lost!)?"
).format(affectedItems),
parent=self):
self.cntlr.uiThreadQueue.put((self.cntlr.quit, [None, True]))
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def moduleSelect(self, *args):
node = (self.modulesView.selection() or (None,))[0]
moduleInfo = self.pluginConfig.get("modules", {}).get(node)
if moduleInfo:
self.selectedModule = node
name = moduleInfo["name"]
self.moduleNameLabel.config(text=name)
self.moduleAuthorHdr.config(state=ACTIVE)
self.moduleAuthorLabel.config(text=moduleInfo["author"])
self.moduleDescrHdr.config(state=ACTIVE)
self.moduleDescrLabel.config(text=moduleInfo["description"])
self.moduleClassesHdr.config(state=ACTIVE)
self.moduleClassesLabel.config(text=', '.join(moduleInfo["classMethods"]))
self.moduleUrlHdr.config(state=ACTIVE)
self.moduleUrlLabel.config(text=moduleInfo["moduleURL"])
self.moduleDateHdr.config(state=ACTIVE)
self.moduleDateLabel.config(text=moduleInfo["fileDate"] + " " +
(_("(an update is available)") if name in self.modulesWithNewerFileDates else ""))
self.moduleLicenseHdr.config(state=ACTIVE)
self.moduleLicenseLabel.config(text=moduleInfo["license"])
self.moduleEnableButton.config(state=ACTIVE,
text={"enabled":self.DISABLE,
"disabled":self.ENABLE}[moduleInfo["status"]])
self.moduleReloadButton.config(state=ACTIVE)
self.moduleRemoveButton.config(state=ACTIVE)
else:
self.selectedModule = None
self.moduleNameLabel.config(text="")
self.moduleAuthorHdr.config(state=DISABLED)
self.moduleAuthorLabel.config(text="")
self.moduleDescrHdr.config(state=DISABLED)
self.moduleDescrLabel.config(text="")
self.moduleClassesHdr.config(state=DISABLED)
self.moduleClassesLabel.config(text="")
self.moduleUrlHdr.config(state=DISABLED)
self.moduleUrlLabel.config(text="")
self.moduleDateHdr.config(state=DISABLED)
self.moduleDateLabel.config(text="")
self.moduleLicenseHdr.config(state=DISABLED)
self.moduleLicenseLabel.config(text="")
self.moduleEnableButton.config(state=DISABLED, text=self.ENABLE)
self.moduleReloadButton.config(state=DISABLED)
self.moduleRemoveButton.config(state=DISABLED)
def findLocally(self):
initialdir = self.cntlr.pluginDir # default plugin directory
if not self.cntlr.isMac: # can't navigate within app easily, always start in default directory
initialdir = self.cntlr.config.setdefault("pluginOpenDir", initialdir)
filename = self.cntlr.uiFileDialog("open",
parent=self,
title=_("Choose plug-in module file"),
initialdir=initialdir,
filetypes=[(_("Python files"), "*.py")],
defaultextension=".py")
if filename:
# check if a package is selected (any file in a directory containing an __init__.py
if (os.path.isdir(os.path.dirname(filename)) and
os.path.isfile(os.path.join(os.path.dirname(filename), "__init__.py"))):
filename = os.path.dirname(filename) # refer to the package instead
self.cntlr.config["pluginOpenDir"] = os.path.dirname(filename)
moduleInfo = PluginManager.moduleModuleInfo(filename)
self.loadFoundModuleInfo(moduleInfo, filename)
def findOnWeb(self):
url = DialogURL.askURL(self)
if url: # url is the in-cache or local file
moduleInfo = PluginManager.moduleModuleInfo(url)
self.cntlr.showStatus("") # clear web loading status
self.loadFoundModuleInfo(moduleInfo, url)
def loadFoundModuleInfo(self, moduleInfo, url):
if moduleInfo and moduleInfo.get("name"):
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
else:
messagebox.showwarning(_("Module is not itself a plug-in or in a directory with package __init__.py plug-in. "),
_("File does not itself contain a python program with an appropriate __pluginInfo__ declaration: \n\n{0}")
.format(url),
parent=self)
def removePluginConfigModuleInfo(self, name):
moduleInfo = self.pluginConfig["modules"].get(name)
if moduleInfo:
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].get(classMethod)
if classMethods and name in classMethods:
classMethods.remove(name)
if not classMethods: # list has become unused
del self.pluginConfig["classes"][classMethod] # remove class
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
elif classMethod == "ModelObjectFactory.ElementSubstitutionClasses":
self.modelClassesChanged = True # model object factor classes changed
elif classMethod == "DisclosureSystem.Types":
self.disclosureSystemTypesChanged = True # disclosure system types changed
del self.pluginConfig["modules"][name]
self.pluginConfigChanged = True
def addPluginConfigModuleInfo(self, moduleInfo):
name = moduleInfo["name"]
self.removePluginConfigModuleInfo(name) # remove any prior entry for this module
self.modulesWithNewerFileDates.discard(name) # no longer has an update available
self.pluginConfig["modules"][name] = moduleInfo
# add classes
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].setdefault(classMethod, [])
if name not in classMethods:
classMethods.append(name)
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
elif classMethod == "ModelObjectFactory.ElementSubstitutionClasses":
self.modelClassesChanged = True # model object factor classes changed
elif classMethod == "DisclosureSystem.Types":
self.disclosureSystemTypesChanged = True # disclosure system types changed
self.pluginConfigChanged = True
def moduleEnable(self):
if self.selectedModule in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][self.selectedModule]
if self.moduleEnableButton['text'] == self.ENABLE:
moduleInfo["status"] = "enabled"
self.moduleEnableButton['text'] = self.DISABLE
elif self.moduleEnableButton['text'] == self.DISABLE:
moduleInfo["status"] = "disabled"
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
def moduleReload(self):
if self.selectedModule in self.pluginConfig["modules"]:
url = self.pluginConfig["modules"][self.selectedModule].get("moduleURL")
if url:
moduleInfo = PluginManager.moduleModuleInfo(url, reload=True)
if moduleInfo:
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
self.cntlr.showStatus(_("{0} reloaded").format(moduleInfo.get("name")), clearAfter=5000)
else:
messagebox.showwarning(_("Module error"),
_("File or module cannot be reloaded: \n\n{0}")
.format(url),
parent=self)
def moduleRemove(self):
if self.selectedModule in self.pluginConfig["modules"]:
self.removePluginConfigModuleInfo(self.selectedModule)
self.pluginConfigChanged = True
self.loadTreeViews()
def enableAll(self):
self.enableDisableAll(True)
def disableAll(self):
self.enableDisableAll(False)
def enableDisableAll(self, doEnable):
for module in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][module]
if doEnable:
moduleInfo["status"] = "enabled"
self.moduleEnableButton['text'] = self.DISABLE
else:
moduleInfo["status"] = "disabled"
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
class SuperUser:
def __init__(self):
self.win = Tk()
self.win.title("The Hive")
self.win.geometry('{}x{}'.format(1000, 450))
self.canvas = Canvas(self.win, bg='#454b54')
self.acceptButton = Button(self.canvas, text="Accept", font='Arial 15 bold', bg='#454b54',
fg="#f7cc35", command=self.accept)
self.rejectButton = Button(self.canvas, text="Reject", font='Arial 15 bold', bg='#454b54',
fg="#f7cc35", command=self.reject)
self.list = Treeview(self.canvas, columns=(1, 2, 3, 4, 5, 6, 7, 8), show="headings", height="15")
def main(self):
self.canvas.pack(expand=TRUE, fill=BOTH)
self.list.pack()
self.list.heading(1, text="ID")
self.list.column(1, width=20)
self.list.heading(2, text="Name")
self.list.column(2, width=100)
self.list.heading(3, text="Email")
self.list.column(3, width=150)
self.list.heading(4, text="Reference")
self.list.column(4, width=100)
self.list.heading(5, text="Interest")
self.list.column(5, width=100)
self.list.heading(6, text="Credential")
self.list.column(6, width=100)
self.list.heading(7, text="Rejected #")
self.list.column(7, width=120)
self.list.heading(8, text="Appeal")
self.list.column(8, width=300)
db.cursor.execute('SELECT * FROM pending_users')
for row in db.cursor.fetchall():
self.list.insert('', END, values=row)
self.acceptButton.pack(expand=TRUE, side=LEFT)
self.rejectButton.pack(expand=TRUE, side=LEFT)
self.win.mainloop()
def accept(self):
password = ''.join(random.choice(string.ascii_lowercase) for i in range(6))
for selected_item in self.list.selection():
a, b, c, d, e, f, g, h = self.list.item(selected_item, 'values')
email = c
username = generate_username(b)
self.list.delete(selected_item)
db.cursor.execute('INSERT INTO users (email, username, password, reputation_score, user_type, login_time) VALUES (%s, %s, %s, 10, "OU", "FIRST")',
(email, username, password))
db.cursor.execute("DELETE FROM pending_users WHERE email = %s", (email,))
subject = "Application Accepted!"
content = '''\
Congratz! Please change your password once you log in with the following credentials. \n
Username: {username} \n
Password: {password} \
'''.format(username=username, password=password)
send_email(subject, content, email)
def reject(self):
for selected_item in self.list.selection():
email = self.list.item(selected_item, 'values')[2]
self.list.delete(selected_item)
db.cursor.execute("SELECT rejected FROM pending_users WHERE email = %s", (email,))
rejNum = db.cursor.fetchone()[0]
if rejNum == 0:
rejNum += 1
db.cursor.execute("UPDATE pending_users SET rejected = %s WHERE email = %s", (rejNum,email))
subject = "Application Denied"
content = '''\
Sorry, but your application has been denied. \n
You have one chance to appeal and the SU will make a final decision to \n
reverse the rejection. If you receive another rejection, then you \n
will be put in blacklist forever. \
'''
send_email(subject, content, email)
elif rejNum == 1:
db.cursor.execute("DELETE FROM pending_users WHERE email = %s", (email,))
db.cursor.execute("INSERT INTO black_list VALUES (%s, %s)", (db.getName(), email))
class NameView(object):
"""Shows a treeview of unique names."""
def __init__(self, master, names):
self.widget = Frame(master)
self._tree = Treeview(self.widget, columns='name')
self._tree.grid(row=0,column=0, sticky=(N,S,W,E))
self._tree.view = self
self.widget.columnconfigure(0, weight=1)
self.widget.rowconfigure(0,weight=1)
self._tree.column('name', width=50)
self._tree['show'] = 'tree'
actions = {'edit': lambda e: self.edit(),
'search': lambda e: self.search(),
'focus_next': lambda e: self.focus_next(),
'focus_prev': lambda e: self.focus_prev(),
'select': lambda e: self._tree.selection_toggle(self._tree.focus()),
'clear_selection': lambda e: self._tree.selection_set([])
}
kb.make_bindings(kb.tagview, actions, self._tree.bind)
self._iids = dict()
self._names = dict()
logger.debug('Names: %s', names)
self.widget.focus_set = self._tree.focus_set
for name in sorted(names):
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
self._scroll = Scrollbar(self.widget, command=self._tree.yview)
self._tree['yscrollcommand'] = self._scroll.set
self._scroll.grid(row=0, column=1, sticky=(N, S))
self.widget.columnconfigure(1, weight=0)
def selection(self):
logger.debug('Selection: %s', self._tree.selection())
return [self._names[iid] for iid in self._tree.selection()]
def edit(self):
self._tree.event_generate('<<NameViewEdit>>')
def search(self):
if len(self._tree.selection()) == 0:
self._tree.selection_add(self._tree.focus())
self._tree.event_generate('<<NameViewSearch>>')
def append(self, names):
logger.debug('Append names: %s', names)
for name in names:
if name not in self._names.values():
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
def delete(self, name):
self._tree.delete(self._iids[name])
del self._names[self._iids[name]]
del self._iids[name]
def _focus(self, iid):
self._tree.focus(iid)
self._tree.see(iid)
def focus_next(self):
cur_iid = self._tree.focus()
next_iid = self._tree.next(cur_iid)
if next_iid == '':
iids = self._tree.get_children()
next_iid = iids[0]
self._focus(next_iid)
def focus_prev(self):
cur_iid = self._tree.focus()
prev_iid = self._tree.prev(cur_iid)
if prev_iid == '':
iids = self._tree.get_children()
prev_iid = iids[-1]
self._focus(prev_iid)
def jump_to(self, name):
try:
iid = self._iids[name]
self._focus(iid)
except KeyError:
pass
def get_names(self):
return tuple(self._names.values())
def set(self, names):
self._tree.delete(*self._iids.values())
self._iids.clear()
self._names.clear()
for name in sorted(names):
iid = self._tree.insert('', 'end', text=name)
self._names[iid] = name
self._iids[name] = iid
class DialogPackageManager(Toplevel):
def __init__(self, mainWin, packageNamesWithNewerFileDates):
super(DialogPackageManager, self).__init__(mainWin.parent)
self.ENABLE = _("Enable")
self.DISABLE = _("Disable")
self.parent = mainWin.parent
self.cntlr = mainWin
# copy plugins for temporary display
self.packagesConfig = PackageManager.packagesConfig
self.packagesConfigChanged = False
self.packageNamesWithNewerFileDates = packageNamesWithNewerFileDates
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", self.parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.title(_("Taxonomy Packages Manager"))
frame = Frame(self)
# left button frame
buttonFrame = Frame(frame, width=40)
buttonFrame.columnconfigure(0, weight=1)
addLabel = Label(buttonFrame, text=_("Find taxonomy packages:"), wraplength=64, justify="center")
addLocalButton = Button(buttonFrame, text=_("Locally"), command=self.findLocally)
ToolTip(addLocalButton, text=_("File chooser allows selecting taxonomy packages to add (or reload), from the local file system. "
"Select either a taxonomy package zip file, or a taxonomy manifest (.taxonomyPackage.xml) within an unzipped taxonomy package. "), wraplength=240)
addWebButton = Button(buttonFrame, text=_("On Web"), command=self.findOnWeb)
ToolTip(addWebButton, text=_("Dialog to enter URL full path to load (or reload) package, from the web or local file system. "
"URL may be either a taxonomy package zip file, or a taxonomy manifest (.taxonomyPackage.xml) within an unzipped taxonomy package. "), wraplength=240)
manifestNameButton = Button(buttonFrame, text=_("Manifest"), command=self.manifestName)
ToolTip(manifestNameButton, text=_("Provide non-standard archive manifest file name pattern (e.g., *taxonomyPackage.xml). "
"Uses unix file name pattern matching. "
"Multiple manifest files are supported in archive (such as oasis catalogs). "
"(Replaces search for either .taxonomyPackage.xml or catalog.xml). "), wraplength=240)
self.manifestNamePattern = ""
addLabel.grid(row=0, column=0, pady=4)
addLocalButton.grid(row=1, column=0, pady=4)
addWebButton.grid(row=2, column=0, pady=4)
manifestNameButton.grid(row=3, column=0, pady=4)
buttonFrame.grid(row=0, column=0, rowspan=3, sticky=(N, S, W), padx=3, pady=3)
# right tree frame (packages already known to arelle)
packagesFrame = Frame(frame, width=700)
vScrollbar = Scrollbar(packagesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(packagesFrame, orient=HORIZONTAL)
self.packagesView = Treeview(packagesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=7)
self.packagesView.grid(row=0, column=0, sticky=(N, S, E, W))
self.packagesView.bind('<<TreeviewSelect>>', self.packageSelect)
hScrollbar["command"] = self.packagesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.packagesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
packagesFrame.columnconfigure(0, weight=1)
packagesFrame.rowconfigure(0, weight=1)
packagesFrame.grid(row=0, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.packagesView.focus_set()
self.packagesView.column("#0", width=120, anchor="w")
self.packagesView.heading("#0", text=_("Name"))
self.packagesView["columns"] = ("ver", "status", "date", "update", "descr")
self.packagesView.column("ver", width=150, anchor="w", stretch=False)
self.packagesView.heading("ver", text=_("Version"))
self.packagesView.column("status", width=50, anchor="w", stretch=False)
self.packagesView.heading("status", text=_("Status"))
self.packagesView.column("date", width=170, anchor="w", stretch=False)
self.packagesView.heading("date", text=_("File Date"))
self.packagesView.column("update", width=50, anchor="w", stretch=False)
self.packagesView.heading("update", text=_("Update"))
self.packagesView.column("descr", width=200, anchor="w", stretch=False)
self.packagesView.heading("descr", text=_("Description"))
remappingsFrame = Frame(frame)
vScrollbar = Scrollbar(remappingsFrame, orient=VERTICAL)
hScrollbar = Scrollbar(remappingsFrame, orient=HORIZONTAL)
self.remappingsView = Treeview(remappingsFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=5)
self.remappingsView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.remappingsView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.remappingsView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
remappingsFrame.columnconfigure(0, weight=1)
remappingsFrame.rowconfigure(0, weight=1)
remappingsFrame.grid(row=1, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.remappingsView.focus_set()
self.remappingsView.column("#0", width=200, anchor="w")
self.remappingsView.heading("#0", text=_("Prefix"))
self.remappingsView["columns"] = ("remapping")
self.remappingsView.column("remapping", width=500, anchor="w", stretch=False)
self.remappingsView.heading("remapping", text=_("Remapping"))
# bottom frame package info details
packageInfoFrame = Frame(frame, width=700)
packageInfoFrame.columnconfigure(1, weight=1)
self.packageNameLabel = Label(packageInfoFrame, wraplength=600, justify="left",
font=font.Font(family='Helvetica', size=12, weight='bold'))
self.packageNameLabel.grid(row=0, column=0, columnspan=6, sticky=W)
self.packageVersionHdr = Label(packageInfoFrame, text=_("version:"), state=DISABLED)
self.packageVersionHdr.grid(row=1, column=0, sticky=W)
self.packageVersionLabel = Label(packageInfoFrame, wraplength=600, justify="left")
self.packageVersionLabel.grid(row=1, column=1, columnspan=5, sticky=W)
self.packageDescrHdr = Label(packageInfoFrame, text=_("description:"), state=DISABLED)
self.packageDescrHdr.grid(row=2, column=0, sticky=W)
self.packageDescrLabel = Label(packageInfoFrame, wraplength=600, justify="left")
self.packageDescrLabel.grid(row=2, column=1, columnspan=5, sticky=W)
self.packagePrefixesHdr = Label(packageInfoFrame, text=_("prefixes:"), state=DISABLED)
self.packagePrefixesHdr.grid(row=3, column=0, sticky=W)
self.packagePrefixesLabel = Label(packageInfoFrame, wraplength=600, justify="left")
self.packagePrefixesLabel.grid(row=3, column=1, columnspan=5, sticky=W)
ToolTip(self.packagePrefixesLabel, text=_("List of prefixes that this package remaps."), wraplength=240)
self.packageUrlHdr = Label(packageInfoFrame, text=_("URL:"), state=DISABLED)
self.packageUrlHdr.grid(row=4, column=0, sticky=W)
self.packageUrlLabel = Label(packageInfoFrame, wraplength=600, justify="left")
self.packageUrlLabel.grid(row=4, column=1, columnspan=5, sticky=W)
ToolTip(self.packageUrlLabel, text=_("URL of taxonomy package (local file path or web loaded file)."), wraplength=240)
self.packageDateHdr = Label(packageInfoFrame, text=_("date:"), state=DISABLED)
self.packageDateHdr.grid(row=5, column=0, sticky=W)
self.packageDateLabel = Label(packageInfoFrame, wraplength=600, justify="left")
self.packageDateLabel.grid(row=5, column=1, columnspan=5, sticky=W)
ToolTip(self.packageDateLabel, text=_("Date of currently loaded package file (with parenthetical node when an update is available)."), wraplength=240)
self.packageEnableButton = Button(packageInfoFrame, text=self.ENABLE, state=DISABLED, command=self.packageEnable)
ToolTip(self.packageEnableButton, text=_("Enable/disable package."), wraplength=240)
self.packageEnableButton.grid(row=6, column=1, sticky=E)
self.packageMoveUpButton = Button(packageInfoFrame, text=_("Move Up"), state=DISABLED, command=self.packageMoveUp)
ToolTip(self.packageMoveUpButton, text=_("Move package up (above other remappings)."), wraplength=240)
self.packageMoveUpButton.grid(row=6, column=2, sticky=E)
self.packageMoveDownButton = Button(packageInfoFrame, text=_("Move Down"), state=DISABLED, command=self.packageMoveDown)
ToolTip(self.packageMoveDownButton, text=_("Move package down (below other remappings)."), wraplength=240)
self.packageMoveDownButton.grid(row=6, column=3, sticky=E)
self.packageReloadButton = Button(packageInfoFrame, text=_("Reload"), state=DISABLED, command=self.packageReload)
ToolTip(self.packageReloadButton, text=_("Reload/update package."), wraplength=240)
self.packageReloadButton.grid(row=6, column=4, sticky=E)
self.packageRemoveButton = Button(packageInfoFrame, text=_("Remove"), state=DISABLED, command=self.packageRemove)
ToolTip(self.packageRemoveButton, text=_("Remove package from packages table (does not erase the package file)."), wraplength=240)
self.packageRemoveButton.grid(row=6, column=5, sticky=E)
packageInfoFrame.grid(row=2, column=0, columnspan=5, sticky=(N, S, E, W), padx=3, pady=3)
packageInfoFrame.config(borderwidth=4, relief="groove")
okButton = Button(frame, text=_("Close"), command=self.ok)
ToolTip(okButton, text=_("Accept and changes (if any) and close dialog."), wraplength=240)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
ToolTip(cancelButton, text=_("Cancel changes (if any) and close dialog."), wraplength=240)
okButton.grid(row=3, column=3, sticky=(S,E), pady=3)
cancelButton.grid(row=3, column=4, sticky=(S,E), pady=3, padx=3)
self.loadTreeViews()
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=0)
frame.columnconfigure(1, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeViews(self):
self.selectedModule = None
# clear previous treeview entries
for previousNode in self.packagesView.get_children(""):
self.packagesView.delete(previousNode)
for i, packageInfo in enumerate(self.packagesConfig.get("packages", [])):
name = packageInfo.get("name", "package{}".format(i))
node = self.packagesView.insert("", "end", "_{}".format(i), text=name)
self.packagesView.set(node, "ver", packageInfo.get("version"))
self.packagesView.set(node, "status", packageInfo.get("status"))
self.packagesView.set(node, "date", packageInfo.get("fileDate"))
if name in self.packageNamesWithNewerFileDates:
self.packagesView.set(node, "update", _("available"))
self.packagesView.set(node, "descr", packageInfo.get("description"))
# clear previous treeview entries
for previousNode in self.remappingsView.get_children(""):
self.remappingsView.delete(previousNode)
for i, remappingItem in enumerate(sorted(self.packagesConfig.get("remappings", {}).items())):
prefix, remapping = remappingItem
node = self.remappingsView.insert("", "end", prefix, text=prefix)
self.remappingsView.set(node, "remapping", remapping)
self.packageSelect() # clear out prior selection
def ok(self, event=None):
if self.packagesConfigChanged:
PackageManager.packagesConfig = self.packagesConfig
PackageManager.packagesConfigChanged = True
self.cntlr.onPackageEnablementChanged()
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def packageSelect(self, *args):
node = (self.packagesView.selection() or (None,))[0]
try:
nodeIndex = int(node[1:])
except (ValueError, TypeError):
nodeIndex = -1
if 0 <= nodeIndex < len(self.packagesConfig["packages"]):
packageInfo = self.packagesConfig["packages"][nodeIndex]
self.selectedPackageIndex = nodeIndex
name = packageInfo["name"]
self.packageNameLabel.config(text=name)
self.packageVersionHdr.config(state=ACTIVE)
self.packageVersionLabel.config(text=packageInfo["version"])
self.packageDescrHdr.config(state=ACTIVE)
self.packageDescrLabel.config(text=packageInfo["description"])
self.packagePrefixesHdr.config(state=ACTIVE)
self.packagePrefixesLabel.config(text=', '.join(packageInfo["remappings"].keys()))
self.packageUrlHdr.config(state=ACTIVE)
self.packageUrlLabel.config(text=packageInfo["URL"])
self.packageDateHdr.config(state=ACTIVE)
self.packageDateLabel.config(text=packageInfo["fileDate"] + " " +
(_("(an update is available)") if name in self.packageNamesWithNewerFileDates else ""))
self.packageEnableButton.config(state=ACTIVE,
text={"enabled":self.DISABLE,
"disabled":self.ENABLE}[packageInfo["status"]])
self.packageMoveUpButton.config(state=ACTIVE if 0 < nodeIndex else DISABLED)
self.packageMoveDownButton.config(state=ACTIVE if nodeIndex < (len(self.packagesConfig["packages"]) - 1) else DISABLED)
self.packageReloadButton.config(state=ACTIVE)
self.packageRemoveButton.config(state=ACTIVE)
else:
self.selectedPackageIndex = -1
self.packageNameLabel.config(text="")
self.packageVersionHdr.config(state=DISABLED)
self.packageVersionLabel.config(text="")
self.packageDescrHdr.config(state=DISABLED)
self.packageDescrLabel.config(text="")
self.packagePrefixesHdr.config(state=DISABLED)
self.packagePrefixesLabel.config(text="")
self.packageUrlHdr.config(state=DISABLED)
self.packageUrlLabel.config(text="")
self.packageDateHdr.config(state=DISABLED)
self.packageDateLabel.config(text="")
self.packageEnableButton.config(state=DISABLED, text=self.ENABLE)
self.packageMoveUpButton.config(state=DISABLED)
self.packageMoveDownButton.config(state=DISABLED)
self.packageReloadButton.config(state=DISABLED)
self.packageRemoveButton.config(state=DISABLED)
def findLocally(self):
initialdir = self.cntlr.pluginDir # default plugin directory
if not self.cntlr.isMac: # can't navigate within app easily, always start in default directory
initialdir = self.cntlr.config.setdefault("packageOpenDir", initialdir)
filename = self.cntlr.uiFileDialog("open",
parent=self,
title=_("Choose taxonomy package file"),
initialdir=initialdir,
filetypes=[(_("Taxonomy package files (*.zip)"), "*.zip"),
(_("Manifest (*.taxonomyPackage.xml)"), "*.taxonomyPackage.xml"),
(_("Oasis Catalog (*catalog.xml)"), "*catalog.xml")],
defaultextension=".zip")
if filename:
# check if a package is selected (any file in a directory containing an __init__.py
self.cntlr.config["packageOpenDir"] = os.path.dirname(filename)
packageInfo = PackageManager.packageInfo(filename, packageManifestName=self.manifestNamePattern)
self.loadFoundPackageInfo(packageInfo, filename)
def findOnWeb(self):
url = DialogURL.askURL(self)
if url: # url is the in-cache or local file
packageInfo = PackageManager.packageInfo(url, packageManifestName=self.manifestNamePattern)
self.cntlr.showStatus("") # clear web loading status
self.loadFoundPackageInfo(packageInfo, url)
def manifestName(self):
self.manifestNamePattern = simpledialog.askstring(_("Archive manifest file name pattern"),
_("Provide non-standard archive manifest file name pattern (e.g., *taxonomyPackage.xml). \n"
"Uses unix file name pattern matching. \n"
"Multiple manifest files are supported in archive (such as oasis catalogs). \n"
"(If blank, search for either .taxonomyPackage.xml or catalog.xml). "),
initialvalue=self.manifestNamePattern,
parent=self)
def loadFoundPackageInfo(self, packageInfo, url):
if packageInfo and packageInfo.get("name"):
self.addPackageInfo(packageInfo)
self.loadTreeViews()
else:
messagebox.showwarning(_("Package is not itself a taxonomy package. "),
_("File does not itself contain a manifest file: \n\n{0}\n\n "
"If opening an archive file, the manifest file search pattern currently is \"\", please press \"Manifest\" to change manifest file name pattern, e.g.,, \"*.taxonomyPackage.xml\", if needed. ")
.format(url),
parent=self)
def removePackageInfo(self, name, version):
# find package entry
packagesList = self.packagesConfig["packages"]
j = -1
for i, packageInfo in enumerate(packagesList):
if packageInfo['name'] == name and packageInfo['version'] == version:
j = i
break
if 0 <= j < len(packagesList):
del self.packagesConfig["packages"][i]
self.packagesConfigChanged = True
def addPackageInfo(self, packageInfo):
name = packageInfo["name"]
version = packageInfo["version"]
self.removePackageInfo(name, version) # remove any prior entry for this package
self.packageNamesWithNewerFileDates.discard(name) # no longer has an update available
self.packagesConfig["packages"].append(packageInfo)
PackageManager.rebuildRemappings()
self.packagesConfigChanged = True
def packageEnable(self):
if 0 <= self.selectedPackageIndex < len(self.packagesConfig["packages"]):
packageInfo = self.packagesConfig["packages"][self.selectedPackageIndex]
if self.packageEnableButton['text'] == self.ENABLE:
packageInfo["status"] = "enabled"
self.packageEnableButton['text'] = self.DISABLE
elif self.packageEnableButton['text'] == self.DISABLE:
packageInfo["status"] = "disabled"
self.packageEnableButton['text'] = self.ENABLE
self.packagesConfigChanged = True
PackageManager.rebuildRemappings()
self.loadTreeViews()
def packageMoveUp(self):
if 1 <= self.selectedPackageIndex < len(self.packagesConfig["packages"]):
packages = self.packagesConfig["packages"]
packageInfo = packages[self.selectedPackageIndex]
del packages[self.selectedPackageIndex]
packages.insert(self.selectedPackageIndex -1, packageInfo)
self.packagesConfigChanged = True
PackageManager.rebuildRemappings()
self.loadTreeViews()
def packageMoveDown(self):
if 0 <= self.selectedPackageIndex < len(self.packagesConfig["packages"]) - 1:
packages = self.packagesConfig["packages"]
packageInfo = packages[self.selectedPackageIndex]
del packages[self.selectedPackageIndex]
packages.insert(self.selectedPackageIndex + 1, packageInfo)
self.packagesConfigChanged = True
PackageManager.rebuildRemappings()
self.loadTreeViews()
def packageReload(self):
if 0 <= self.selectedPackageIndex < len(self.packagesConfig["packages"]):
packageInfo = self.packagesConfig["packages"][self.selectedPackageIndex]
url = packageInfo.get("URL")
if url:
packageInfo = PackageManager.packageInfo(url, reload=True, packageManifestName=packageInfo.get("manifestName"))
if packageInfo:
self.addPackageInfo(packageInfo)
PackageManager.rebuildRemappings()
self.loadTreeViews()
self.cntlr.showStatus(_("{0} reloaded").format(packageInfo.get("name")), clearAfter=5000)
else:
messagebox.showwarning(_("Package error"),
_("File or package cannot be reloaded: \n\n{0}")
.format(url),
parent=self)
def packageRemove(self):
if 0 <= self.selectedPackageIndex < len(self.packagesConfig["packages"]):
packageInfo = self.packagesConfig["packages"][self.selectedPackageIndex]
self.removePackageInfo(packageInfo["name"], packageInfo["version"])
self.packagesConfigChanged = True
PackageManager.rebuildRemappings()
self.loadTreeViews()
class Main_Page:
"""Displaying available books and user's borrowed books"""
def __init__(self, master, logged_userid):
self.user_app = master
self.user_app.title('Library')
self.user_app.geometry('850x350')
self.users_list = Treeview(self.user_app, columns=BORROWINGS_COLUMNS, show='headings', height=10)
self.books_list = Treeview(self.user_app, columns=LIBRARY_COLUMNS, show='headings', height=10)
self.logged_userid = logged_userid
self.borrowed_books_counter = 0
self.dbd = db.DatesDB('library.db')
self.udb = db.UsersDB('library.db')
self.bdb = db.BooksDB('library.db')
self.search_text = tk.StringVar()
self.search_entry = None
def logout(self):
"""Log out"""
if messagebox.askyesno("Logging out", "Do you want to log out?"):
self.user_app.destroy()
self.user_app = tk.Tk()
login_window = Login(self.user_app)
login_window.welcome_screen()
self.user_app.mainloop()
def populate_list(self):
"""Displays books available to borrow"""
self.books_list.delete(*self.books_list.get_children())
for row in self.bdb.fetch(1):
self.books_list.insert('', tk.END, values=row[0:6])
def user_list(self):
"""Daysplays books borrowed by user"""
self.users_list.delete(*self.users_list.get_children())
for row in self.dbd.fetch_users(self.logged_userid):
self.borrowed_books_counter += 1
self.users_list.insert('', tk.END, values=row[0:4])
def borrow_book(self):
"""Borrowing selected book"""
if self.books_list.selection():
if self.borrowed_books_counter < 3:
selected_book = self.books_list.set(self.books_list.selection())
id = selected_book.get('Id')
self.dbd.insert(self.logged_userid, id)
self.bdb.status_update(id, 0)
self.user_list()
self.populate_list()
self.borrowed_books_counter += 1
else:
messagebox.showerror('Limit', 'You can have up to 3 books borrowed!')
return
def return_book(self):
"""Returning seleted book"""
if self.users_list.selection():
selected_book = self.users_list.set(self.users_list.selection())
id = selected_book.get('Id')
self.bdb.status_update(id, 1)
self.dbd.remove(id)
self.populate_list()
self.user_list()
self.borrowed_books_counter -= 1
def menu_screen(self):
"""Prepares labels, listboxes, buttons"""
# Available book treeview
self.books_list.grid(row=1, column=0, padx=8)
for column_name, width in zip(LIBRARY_COLUMNS, LIBRARY_COLUMNS_SIZE):
self.books_list.column(column_name, width=width, anchor=tk.CENTER)
self.books_list.heading(column_name, text=column_name)
scrollbar = tk.Scrollbar(self.user_app, orient=tk.VERTICAL)
scrollbar.configure(command=self.books_list.set)
self.books_list.configure(yscrollcommand=scrollbar)
# User's borrowed books treeview
self.users_list.grid(row=1, column=1, padx=8)
for column_name, width in zip(BORROWINGS_COLUMNS, BORROWINGS_COLUMNS_SIZE):
self.users_list.column(column_name, width=width, anchor=tk.CENTER)
self.users_list.heading(column_name, text=column_name)
scrollbar = tk.Scrollbar(self.user_app, orient=tk.VERTICAL)
scrollbar.configure(command=self.users_list.set)
self.users_list.configure(yscrollcommand=scrollbar)
borrow_btn = tk.Button(self.user_app, text='Borrow book', width=12, command=self.borrow_book)
borrow_btn.grid(row=9, column=0, pady=10)
return_btn = tk.Button(self.user_app, text='Return book', width=12, command=self.return_book)
return_btn.grid(row=9, column=1, pady=10)
logout_btn = tk.Button(self.user_app, text='Logout', width=12, command=self.logout)
logout_btn.grid(row=0, column=1, pady=10)
self.populate_list()
self.user_list()
class DialogOpenArchive(Toplevel):
def __init__(self,
parent,
openType,
filesource,
filenames,
title,
colHeader,
showAltViewButton=False):
if isinstance(parent, Cntlr):
cntlr = parent
parent = parent.parent # parent is cntlrWinMain
else: # parent is a Toplevel dialog
cntlr = parent.cntlr
super(DialogOpenArchive, self).__init__(parent)
self.parent = parent
self.showAltViewButton = showAltViewButton
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)",
parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.accepted = False
self.transient(self.parent)
frame = Frame(self)
treeFrame = Frame(frame, width=500)
vScrollbar = Scrollbar(treeFrame, orient=VERTICAL)
hScrollbar = Scrollbar(treeFrame, orient=HORIZONTAL)
self.treeView = Treeview(treeFrame,
xscrollcommand=hScrollbar.set,
yscrollcommand=vScrollbar.set)
self.treeView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.treeView.xview
hScrollbar.grid(row=1, column=0, sticky=(E, W))
vScrollbar["command"] = self.treeView.yview
vScrollbar.grid(row=0, column=1, sticky=(N, S))
treeFrame.columnconfigure(0, weight=1)
treeFrame.rowconfigure(0, weight=1)
treeFrame.grid(row=0,
column=0,
columnspan=4,
sticky=(N, S, E, W),
padx=3,
pady=3)
self.treeView.focus_set()
if openType not in (PLUGIN, PACKAGE):
cntlr.showStatus(_("loading archive {0}").format(filesource.url))
self.filesource = filesource
self.filenames = filenames
self.selection = filesource.selection
self.hasToolTip = False
selectedNode = None
if openType == ENTRY_POINTS:
try:
metadataFiles = filesource.taxonomyPackageMetadataFiles
''' take first for now
if len(metadataFiles) != 1:
raise IOError(_("Taxonomy package contained more than one metadata file: {0}.")
.format(', '.join(metadataFiles)))
'''
metadataFile = metadataFiles[0]
metadata = filesource.url + os.sep + metadataFile
self.metadataFilePrefix = os.sep.join(
os.path.split(metadataFile)[:-1])
if self.metadataFilePrefix:
self.metadataFilePrefix += "/" # zip contents have /, never \ file seps
self.taxonomyPkgMetaInf = '{}/META-INF/'.format(
os.path.splitext(os.path.basename(filesource.url))[0])
self.taxonomyPackage = parsePackage(
cntlr, filesource, metadata,
os.sep.join(os.path.split(metadata)[:-1]) + os.sep)
if self.taxonomyPackage["entryPoints"]:
# may have instance documents too
self.packageContainedInstances = []
packageContentTypeCounts = {}
for suffix in (".xhtml", ".htm", ".html"):
for potentialInstance in filesource.dir:
if potentialInstance.endswith(".xhtml"):
_type = "Inline Instance"
self.packageContainedInstances.append(
[potentialInstance, _type])
packageContentTypeCounts[
potentialInstance] = packageContentTypeCounts.get(
potentialInstance, 0) + 1
if self.packageContainedInstances:
break
if self.packageContainedInstances: # add sequences to any duplicated entry types
for _type, count in packageContentTypeCounts.items():
if count > 1:
_dupNo = 0
for i in range(
len(self.packageContainedInstances)):
if self.packageContainedInstances[i][
0] == _type:
_dupNo += 1
self.packageContainedInstances[i][
0] = "{} {}".format(_type, _dupNo)
else:
# may be a catalog file with no entry oint names
openType = ARCHIVE # no entry points to show, just archive
self.showAltViewButton = False
except Exception as e:
self.close()
err = _(
"Failed to parse metadata; the underlying error was: {0}"
).format(e)
messagebox.showerror(_("Malformed taxonomy package"), err)
cntlr.addToLog(err)
return
if openType not in (PLUGIN, PACKAGE):
cntlr.showStatus(None)
if openType in (DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
y = 3
else:
y = 1
okButton = Button(frame, text=_("OK"), command=self.ok)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
okButton.grid(row=y, column=2, sticky=(S, E, W), pady=3)
cancelButton.grid(row=y, column=3, sticky=(S, E, W), pady=3, padx=3)
if self.showAltViewButton:
self.altViewButton = Button(frame, command=self.showAltView)
self.altViewButton.grid(row=y,
column=0,
sticky=(S, W),
pady=3,
padx=3)
self.loadTreeView(openType, colHeader, title)
self.geometry("+{0}+{1}".format(dialogX + 50, dialogY + 100))
frame.grid(row=0, column=0, sticky=(N, S, E, W))
frame.columnconfigure(0, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.toolTipText = StringVar()
if self.hasToolTip:
self.treeView.bind("<Motion>", self.motion, '+')
self.treeView.bind("<Leave>", self.leave, '+')
self.toolTipText = StringVar()
self.toolTip = ToolTip(self.treeView,
textvariable=self.toolTipText,
wraplength=640,
follow_mouse=True,
state="disabled")
self.toolTipRowId = None
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeView(self, openType, title, colHeader):
self.title(title)
self.openType = openType
selectedNode = None
# clear previous treeview entries
for previousNode in self.treeView.get_children(""):
self.treeView.delete(previousNode)
# set up treeView widget and tabbed pane
if openType in (ARCHIVE, DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
if openType in (PLUGIN, PACKAGE): width = 770
else: width = 500
self.treeView.column("#0", width=width, anchor="w")
self.treeView.heading("#0", text=colHeader)
self.isRss = getattr(self.filesource, "isRss", False)
if self.isRss:
self.treeView.column("#0", width=350, anchor="w")
self.treeView["columns"] = ("descr", "date", "instDoc")
self.treeView.column("descr",
width=50,
anchor="center",
stretch=False)
self.treeView.heading("descr", text="Form")
self.treeView.column("date",
width=170,
anchor="w",
stretch=False)
self.treeView.heading("date", text="Pub Date")
self.treeView.column("instDoc",
width=200,
anchor="w",
stretch=False)
self.treeView.heading("instDoc", text="Instance Document")
elif openType == PLUGIN:
self.treeView.column("#0", width=150, anchor="w")
self.treeView["columns"] = ("name", "vers", "descr", "license")
self.treeView.column("name",
width=150,
anchor="w",
stretch=False)
self.treeView.heading("name", text="Name")
self.treeView.column("vers",
width=60,
anchor="w",
stretch=False)
self.treeView.heading("vers", text="Version")
self.treeView.column("descr",
width=300,
anchor="w",
stretch=False)
self.treeView.heading("descr", text="Description")
self.treeView.column("license",
width=60,
anchor="w",
stretch=False)
self.treeView.heading("license", text="License")
elif openType == PACKAGE:
self.treeView.column("#0", width=200, anchor="w")
self.treeView["columns"] = ("vers", "descr", "license")
self.treeView.column("vers",
width=100,
anchor="w",
stretch=False)
self.treeView.heading("vers", text="Version")
self.treeView.column("descr",
width=400,
anchor="w",
stretch=False)
self.treeView.heading("descr", text="Description")
self.treeView.column("license",
width=70,
anchor="w",
stretch=False)
self.treeView.heading("license", text="License")
else:
self.treeView["columns"] = tuple()
loadedPaths = []
for i, filename in enumerate(self.filenames):
if isinstance(filename, tuple):
if self.isRss:
form, date, instDoc = filename[2:5]
elif openType == PLUGIN:
name, vers, descr, license = filename[3:7]
elif openType == PACKAGE:
vers, descr, license = filename[3:6]
filename = filename[0] # ignore tooltip
self.hasToolTip = True
if filename.endswith("/"):
filename = filename[:-1]
path = filename.split("/")
if not self.isRss and len(
path) > 1 and path[:-1] in loadedPaths:
parent = "file{0}".format(loadedPaths.index(path[:-1]))
else:
parent = ""
node = self.treeView.insert(parent,
"end",
"file{0}".format(i),
text=path[-1])
if self.isRss:
self.treeView.set(node, "descr", form)
self.treeView.set(node, "date", date)
self.treeView.set(node, "instDoc",
os.path.basename(instDoc))
elif openType == PLUGIN:
self.treeView.set(node, "name", name)
self.treeView.set(node, "vers", vers)
self.treeView.set(node, "descr", descr)
self.treeView.set(node, "license", license)
elif openType == PACKAGE:
self.treeView.set(node, "vers", vers)
self.treeView.set(node, "descr", descr)
self.treeView.set(node, "license", license)
if self.selection == filename:
selectedNode = node
loadedPaths.append(path)
elif openType == ENTRY_POINTS:
self.treeView.column("#0", width=200, anchor="w")
self.treeView.heading("#0", text="Name")
self.treeView["columns"] = ("url", )
self.treeView.column("url", width=300, anchor="w")
self.treeView.heading("url", text="URL")
for fileType, fileUrl in getattr(self, "packageContainedInstances",
()):
self.treeView.insert("",
"end",
fileUrl,
values=fileType,
text=fileUrl or urls[0][2])
for name, urls in sorted(
self.taxonomyPackage["entryPoints"].items(),
key=lambda i: i[0][2]):
self.treeView.insert("",
"end",
name,
values="\n".join(url[1] for url in urls),
text=name or urls[0][2])
self.hasToolTip = True
else: # unknown openType
return None
if selectedNode:
self.treeView.see(selectedNode)
self.treeView.selection_set(selectedNode)
if self.showAltViewButton:
self.altViewButton.config(text=_("Show Files") if openType ==
ENTRY_POINTS else _("Show Entries"))
def ok(self, event=None):
selection = self.treeView.selection()
if len(selection) > 0:
if hasattr(self, "taxonomyPackage"):
# load file source remappings
self.filesource.mappedPaths = self.taxonomyPackage[
"remappings"]
filename = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
filename = self.filenames[int(selection[0][4:])]
if isinstance(filename, tuple):
if self.isRss:
filename = filename[4]
else:
filename = filename[0]
elif self.openType == ENTRY_POINTS:
epName = selection[0]
#index 0 is the remapped Url, as opposed to the canonical one used for display
# Greg Acsone reports [0] does not work for Corep 1.6 pkgs, need [1], old style packages
filenames = []
for _url, _type in self.packageContainedInstances: # check if selection was an inline instance
if _type == epName:
filenames.append(_url)
if not filenames: # else if it's a named taxonomy entry point
for url in self.taxonomyPackage["entryPoints"][epName]:
filename = url[0]
if not filename.endswith("/"):
# check if it's an absolute URL rather than a path into the archive
if not isHttpUrl(
filename
) and self.metadataFilePrefix != self.taxonomyPkgMetaInf:
# assume it's a path inside the archive:
filename = self.metadataFilePrefix + filename
filenames.append(filename)
if filenames:
self.filesource.select(filenames)
self.accepted = True
self.close()
return
elif self.openType in (PLUGIN, PACKAGE):
filename = self.filenames[int(selection[0][4:])][2]
if filename is not None and not filename.endswith("/"):
if hasattr(self, "taxonomyPackage"):
# attempt to unmap the filename to original file
# will be mapped again in loading, but this allows schemaLocation to be unmapped
for prefix, remapping in self.taxonomyPackage[
"remappings"].items():
if isHttpUrl(remapping):
remapStart = remapping
else:
remapStart = self.metadataFilePrefix + remapping
if filename.startswith(remapStart):
# set unmmapped file
filename = prefix + filename[len(remapStart):]
break
if self.openType in (PLUGIN, PACKAGE):
self.filesource.selection = filename
else:
self.filesource.select(filename)
self.accepted = True
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def showAltView(self, event=None):
if self.openType == ENTRY_POINTS:
self.loadTreeView(ARCHIVE, _("Select Entry Point"), _("File"))
else:
self.loadTreeView(ENTRY_POINTS, _("Select Archive File"),
_("File"))
def leave(self, *args):
self.toolTipRowId = None
def motion(self, *args):
tvRowId = self.treeView.identify_row(args[0].y)
if tvRowId != self.toolTipRowId:
text = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM, PLUGIN, PACKAGE):
self.toolTipRowId = tvRowId
if tvRowId and len(tvRowId) > 4:
try:
text = self.filenames[int(tvRowId[4:])]
if isinstance(text, tuple):
text = (text[1] or "").replace("\\n", "\n")
except (KeyError, ValueError):
pass
elif self.openType == ENTRY_POINTS:
try:
text = "{0}\n{1}".format(
tvRowId, "\n".join(
url[1] for url in
self.taxonomyPackage["entryPoints"][tvRowId]))
except KeyError:
pass
self.setToolTip(text)
def setToolTip(self, text):
self.toolTip._hide()
if text:
self.toolTipText.set(text)
self.toolTip.configure(state="normal")
self.toolTip._schedule()
else:
self.toolTipText.set("")
self.toolTip.configure(state="disabled")
class DialogOpenArchive(Toplevel):
def __init__(self, mainWin, openType, filesource, filenames, title, colHeader, showAltViewButton=False):
parent = mainWin.parent
super(DialogOpenArchive, self).__init__(parent)
self.parent = parent
self.showAltViewButton = showAltViewButton
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.accepted = False
self.transient(self.parent)
frame = Frame(self)
treeFrame = Frame(frame, width=500)
vScrollbar = Scrollbar(treeFrame, orient=VERTICAL)
hScrollbar = Scrollbar(treeFrame, orient=HORIZONTAL)
self.treeView = Treeview(treeFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set)
self.treeView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.treeView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.treeView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
treeFrame.columnconfigure(0, weight=1)
treeFrame.rowconfigure(0, weight=1)
treeFrame.grid(row=0, column=0, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.treeView.focus_set()
mainWin.showStatus(_("loading archive {0}").format(filesource.url))
self.filesource = filesource
self.filenames = filenames
self.selection = filesource.selection
self.hasToolTip = False
selectedNode = None
if openType == ENTRY_POINTS:
try:
metadataFiles = filesource.taxonomyPackageMetadataFiles
if len(metadataFiles) > 1:
raise IOError(_("Taxonomy package contained more than one metadata file: {0}.")
.format(', '.join(metadataFiles)))
metadataFile = metadataFiles[0]
metadata = filesource.file(filesource.url + os.sep + metadataFile)[0]
self.metadataFilePrefix = os.sep.join(os.path.split(metadataFile)[:-1])
if self.metadataFilePrefix:
self.metadataFilePrefix += os.sep
self.nameToUrls, self.remappings = parseTxmyPkg(mainWin, metadata)
except Exception as e:
self.close()
err = _("Failed to parse metadata; the underlying error was: {0}").format(e)
messagebox.showerror(_("Malformed taxonomy package"), err)
mainWin.addToLog(err)
return
mainWin.showStatus(None)
if openType == DISCLOSURE_SYSTEM:
y = 3
else:
y = 1
okButton = Button(frame, text=_("OK"), command=self.ok)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
okButton.grid(row=y, column=2, sticky=(S,E,W), pady=3)
cancelButton.grid(row=y, column=3, sticky=(S,E,W), pady=3, padx=3)
if showAltViewButton:
self.altViewButton = Button(frame, command=self.showAltView)
self.altViewButton.grid(row=y, column=0, sticky=(S,W), pady=3, padx=3)
self.loadTreeView(openType, colHeader, title)
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.toolTipText = StringVar()
if self.hasToolTip:
self.treeView.bind("<Motion>", self.motion, '+')
self.treeView.bind("<Leave>", self.leave, '+')
self.toolTipText = StringVar()
self.toolTip = ToolTip(self.treeView,
textvariable=self.toolTipText,
wraplength=640,
follow_mouse=True,
state="disabled")
self.toolTipRowId = None
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeView(self, openType, title, colHeader):
self.title(title)
self.openType = openType
selectedNode = None
# clear previous treeview entries
for previousNode in self.treeView.get_children(""):
self.treeView.delete(previousNode)
# set up treeView widget and tabbed pane
if openType in (ARCHIVE, DISCLOSURE_SYSTEM):
self.treeView.column("#0", width=500, anchor="w")
self.treeView.heading("#0", text=colHeader)
try:
self.isRss = self.filesource.isRss
if self.isRss:
self.treeView.column("#0", width=350, anchor="w")
self.treeView["columns"] = ("descr", "date", "instDoc")
self.treeView.column("descr", width=50, anchor="center", stretch=False)
self.treeView.heading("descr", text="Form")
self.treeView.column("date", width=170, anchor="w", stretch=False)
self.treeView.heading("date", text="Pub Date")
self.treeView.column("instDoc", width=200, anchor="w", stretch=False)
self.treeView.heading("instDoc", text="Instance Document")
except AttributeError:
self.isRss = False
self.treeView["columns"] = tuple()
loadedPaths = []
for i, filename in enumerate(self.filenames):
if isinstance(filename,tuple):
if self.isRss:
form, date, instDoc = filename[2:5]
filename = filename[0] # ignore tooltip
self.hasToolTip = True
if filename.endswith("/"):
filename = filename[:-1]
path = filename.split("/")
if not self.isRss and len(path) > 1 and path[:-1] in loadedPaths:
parent = "file{0}".format(loadedPaths.index(path[:-1]))
else:
parent = ""
node = self.treeView.insert(parent, "end", "file{0}".format(i), text=path[-1])
if self.isRss:
self.treeView.set(node, "descr", form)
self.treeView.set(node, "date", date)
self.treeView.set(node, "instDoc", os.path.basename(instDoc))
if self.selection == filename:
selectedNode = node
loadedPaths.append(path)
elif openType == ENTRY_POINTS:
self.treeView.column("#0", width=150, anchor="w")
self.treeView.heading("#0", text="Name")
self.treeView["columns"] = ("url",)
self.treeView.column("url", width=350, anchor="w")
self.treeView.heading("url", text="URL")
for name, urls in self.nameToUrls.items():
displayUrl = urls[1] # display the canonical URL
self.treeView.insert("", "end", name, values=[displayUrl], text=name)
self.hasToolTip = True
else: # unknown openType
return None
if selectedNode:
self.treeView.see(selectedNode)
self.treeView.selection_set(selectedNode)
if self.showAltViewButton:
self.altViewButton.config(text=_("Show Files") if openType == ENTRY_POINTS else _("Show Entries"))
def ok(self, event=None):
selection = self.treeView.selection()
if len(selection) > 0:
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
filename = self.filenames[int(selection[0][4:])]
if isinstance(filename,tuple):
if self.isRss:
filename = filename[4]
else:
filename = filename[0]
if not filename.endswith("/"):
self.filesource.select(filename)
self.accepted = True
self.close()
elif self.openType == ENTRY_POINTS:
epName = selection[0]
#index 0 is the remapped Url, as opposed to the canonical one used for display
urlOrFile = self.nameToUrls[epName][0]
# load file source remappings
self.filesource.mappedPaths = \
dict((prefix,
remapping if isHttpUrl(remapping)
else (self.filesource.baseurl + os.sep + self.metadataFilePrefix +remapping.replace("/", os.sep)))
for prefix, remapping in self.remappings.items())
if not urlOrFile.endswith("/"):
# check if it's an absolute URL rather than a path into the archive
if isHttpUrl(urlOrFile):
self.filesource.select(urlOrFile) # absolute path selection
else:
# assume it's a path inside the archive:
self.filesource.select(self.metadataFilePrefix + urlOrFile)
self.accepted = True
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def showAltView(self, event=None):
if self.openType == ENTRY_POINTS:
self.loadTreeView(ARCHIVE, _("Select Entry Point"), _("File"))
else:
self.loadTreeView(ENTRY_POINTS, _("Select Archive File"), _("File"))
def leave(self, *args):
self.toolTipRowId = None
def motion(self, *args):
tvRowId = self.treeView.identify_row(args[0].y)
if tvRowId != self.toolTipRowId:
text = None
if self.openType in (ARCHIVE, DISCLOSURE_SYSTEM):
self.toolTipRowId = tvRowId
if tvRowId and len(tvRowId) > 4:
try:
text = self.filenames[ int(tvRowId[4:]) ]
if isinstance(text, tuple):
text = text[1].replace("\\n","\n")
except (KeyError, ValueError):
pass
elif self.openType == ENTRY_POINTS:
try:
epUrl = self.nameToUrls[tvRowId][1]
text = "{0}\n{1}".format(tvRowId, epUrl)
except KeyError:
pass
self.setToolTip(text)
def setToolTip(self, text):
self.toolTip._hide()
if text:
self.toolTipText.set(text)
self.toolTip.configure(state="normal")
self.toolTip._schedule()
else:
self.toolTipText.set("")
self.toolTip.configure(state="disabled")
class Cerberus:
def __init__(self, master, root):
self.exportToCSV = False
self.versionApp, self.key, self.salt = self.initApp()
self.key = cerberusCryptography.getMasterKey()
self.cipher_suite = Fernet(self.key)
self.master = master
self.master.title('Cerberus')
self.windowWidth = 1060
self.windowHeight = 450
self.screenWidth = self.master.winfo_screenwidth()
self.screenHeight = self.master.winfo_screenheight()
self.positionRight = int(self.screenWidth / 2 - self.windowWidth / 2)
self.positionDown = int(self.screenHeight / 3 - self.windowHeight / 2)
self.master.geometry("{}x{}+{}+{}".format(self.windowWidth,
self.windowHeight,
self.positionRight,
self.positionDown))
self.img = PhotoImage(data=icons.getAppIcon())
self.master.wm_iconphoto(True, self.img)
self.master.resizable(0, 0)
self.menubar = Menu(master)
filemenu = Menu(self.menubar, tearoff=0)
self.menubar.add_cascade(label="Cerberus", menu=filemenu)
self.addIcon = PhotoImage(data=icons.getAddIcon())
filemenu.add_command(label="Εισαγωγή Υπηρεσίας",
image=self.addIcon,
compound='left',
command=self.getAddNewServiceForm)
self.editIcon = PhotoImage(data=icons.getEditIcon())
filemenu.add_command(label="Επεξεργασία Υπηρεσίας",
image=self.editIcon,
compound='left',
command=self.getEditServiceForm)
self.deleteIcon = PhotoImage(data=icons.getDeleteIcon())
filemenu.add_command(label="Διαγραφή Υπηρεσίας",
image=self.deleteIcon,
compound='left',
command=self.deleteService)
filemenu.add_separator()
self.excelIcon = PhotoImage(data=icons.getExcelIcon())
filemenu.add_command(label="Εξαγωγή σε Excel",
image=self.excelIcon,
compound='left',
command=self.checkPasswordToExportToCSV)
filemenu.add_separator()
self.exitIcon = PhotoImage(data=icons.getExitIcon())
filemenu.add_command(label="Έξοδος",
image=self.exitIcon,
compound='left',
command=self.exitApp)
settingsMenu = Menu(self.menubar, tearoff=0)
self.menubar.add_cascade(label="Ρυθμίσεις", menu=settingsMenu)
self.settingsIcon = PhotoImage(data=icons.getSettingsIcon())
settingsMenu.add_command(label="Επεξεργασία Στοιχείων",
image=self.settingsIcon,
compound='left')
#command=self.getSettingsForm)
aboutMenu = Menu(self.menubar, tearoff=0)
self.menubar.add_cascade(label="Βοήθεια", menu=aboutMenu)
self.infoIcon = PhotoImage(data=icons.getInfoIcon())
aboutMenu.add_command(label="Περί",
image=self.infoIcon,
compound='left',
command=self.getAboutAppForm)
self.master.config(menu=self.menubar)
self.copyIcon = PhotoImage(data=icons.getCopyIcon())
self.popup = Menu(root, tearoff=0)
self.popup.add_command(label=" Αντιγραφή Email",
image=self.copyIcon,
compound='left',
command=self.copyEmail)
self.popup.add_command(label=" Αντιγραφή Username",
image=self.copyIcon,
compound='left',
command=self.copyUsername)
self.popup.add_command(label=" Αντιγραφή Κωδικού",
image=self.copyIcon,
compound='left',
command=self.copyPasswd)
self.popup.add_command(label=" Αντιγραφή ID",
image=self.copyIcon,
compound='left',
command=self.copyID)
self.popup.add_separator()
self.popup.add_command(label=" Επεξεργασία Υπηρεσίας",
image=self.editIcon,
compound='left',
command=self.getEditServiceForm)
self.popup.add_command(label=" Διαγραφή Υπηρεσίας",
image=self.deleteIcon,
compound='left',
command=self.deleteService)
self.popup.add_separator()
self.popup.add_command(label=" Έξοδος",
image=self.exitIcon,
compound='left',
command=self.exitApp)
self.frame = Frame(self.master,
background="white",
borderwidth=1,
relief="sunken",
highlightthickness=1)
self.frame.pack(side="top", fill="x", padx=4, pady=4)
self.search = StringVar()
self.searchEntry = Entry(self.frame,
textvariable=self.search,
borderwidth=0,
highlightthickness=0,
background="white")
self.searchEntry.insert(0, 'Αναζήτηση Υπηρεσίας')
self.searchEntry['fg'] = 'grey'
self.search.trace(
"w",
lambda name, index, mode, sv=self.search: self.searchService())
self.searchEntry.image = PhotoImage(data=icons.getSearchIcon())
imageLabel = Label(self.frame, image=self.searchEntry.image)
imageLabel.pack(side="left")
imageLabel['bg'] = 'white'
self.searchEntry.pack(side="left", fill="both", expand=True)
# Fix BUG with Treeview colors in Python3.7
def fixed_map(option):
return [
elm for elm in style.map('Treeview', query_opt=option)
if elm[:2] != ('!disabled', '!selected')
]
style = ttk.Style(root)
style.map('Treeview',
foreground=fixed_map('foreground'),
background=fixed_map('background'))
# Fix BUG with Treeview colors in Python3.7
self.table = Treeview(self.master)
self.table['show'] = 'headings'
self.table['columns'] = ('Services', 'email', 'username', 'passwd',
'id', 'category', 'url', 'ID')
self.table["displaycolumns"] = ('Services', 'email', 'username',
'passwd', 'id', 'category', 'url')
for col in self.table['columns']:
self.table.heading(
col, command=lambda c=col: self.sortby(self.table, c, 0))
self.table.heading('Services', text='Services')
self.table.column('Services', anchor='center', width=200)
self.table.heading('email', text='Email')
self.table.column('email', anchor='center', width=200)
self.table.heading('username', text='Username')
self.table.column('username', anchor='center', width=100)
self.table.heading('passwd', text='Password')
self.table.column('passwd', anchor='center', width=100)
self.table.heading('url', text='URL')
self.table.column('url', anchor='center', width=120)
self.table.heading('id', text='ID')
self.table.column('id', anchor='center', width=100)
self.table.heading('category', text='Category')
self.table.column('category', anchor='center', width=100)
self.table.heading('ID', text='ID')
self.table.column('ID', anchor='center', width=200)
self.table.tag_configure('oddrow', background='#e6eef2')
self.table.tag_configure('evenrow', background='#b3cfdd')
self.table.tag_configure('focus', background='#c6b6b4')
self.last_focus = None
self.last_focus_tag = None
self.table.focus()
self.table.pack(fill=BOTH, expand=1)
self.table.bind("<<TreeviewSelect>>", self.onTableSelect)
self.table.bind("<ButtonRelease-1>", self.openURLService)
self.table.bind("<Motion>", self.changePointerOnHover)
self.table.bind("<Button-3>", self.popupMenu)
self.searchEntry.bind("<FocusIn>", self.foc_in)
self.searchEntry.bind("<FocusOut>", self.foc_out)
self.popup.bind("<FocusOut>", self.popupFocusOut)
self.master.protocol("WM_DELETE_WINDOW", self.exitApp)
self.loadTable(self)
self.master.bind("<Escape>", self.exitApp)
def popupFocusOut(self, event=None):
self.popup.unpost()
def foc_in(self, *args):
if self.search.get() == 'Αναζήτηση Υπηρεσίας':
self.searchEntry.delete('0', 'end')
self.searchEntry['fg'] = 'black'
def foc_out(self, *args):
if not self.search.get():
self.searchEntry.insert(0, 'Αναζήτηση Υπηρεσίας')
self.searchEntry['fg'] = 'grey'
self.loadTable(self)
def changePointerOnHover(self, event):
_iid = self.table.identify_row(event.y)
if _iid != self.last_focus:
if self.last_focus:
self.table.item(self.last_focus, tags=[self.last_focus_tag])
self.last_focus_tag = self.table.item(_iid, "tag")
self.table.item(_iid, tags=['focus'])
self.last_focus = _iid
curItem = self.table.item(self.table.identify('item', event.x,
event.y))
if curItem['values'] != '':
col = self.table.identify_column(event.x)
url = curItem['values'][int(col[-1]) - 1]
if col[-1] == "7" and url != '---':
self.master.config(cursor="hand2")
else:
self.master.config(cursor="")
def openURLService(self, event):
curItem = self.table.item(self.table.focus())
col = self.table.identify_column(event.x)
region = self.table.identify("region", event.x, event.y)
if col[-1] == "7" and region != 'heading':
url = curItem['values'][int(col[-1]) - 1]
if url != '---':
webbrowser.open_new_tab('http://' + str(url))
def onTableSelect(self, event):
for item in self.table.selection():
item_text = self.table.item(item, "values")
print(item_text[0])
def getSelectedService(self, event):
for item in self.table.selection():
selectedRow = self.table.item(item, "value")
return selectedRow
def initApp(self):
print("Initialize Cerberus App")
try:
conn = sqlite3.connect('cerberus.db')
except sqlite3.Error as e:
print(e)
cur = conn.cursor()
cur.execute(
"SELECT version, masterToken, salt FROM cerberusParameters")
row = cur.fetchone()
cur.close()
return row
def copyEmail(self):
for item in self.table.selection():
item_text = self.table.item(item, "values")
self.master.clipboard_clear()
root.clipboard_append(item_text[1])
def copyUsername(self):
for item in self.table.selection():
item_text = self.table.item(item, "values")
self.master.clipboard_clear()
root.clipboard_append(item_text[2])
def copyPasswd(self):
for item in self.table.selection():
item_text = self.table.item(item, "values")
self.master.clipboard_clear()
root.clipboard_append(item_text[3])
def copyID(self):
for item in self.table.selection():
item_text = self.table.item(item, "values")
self.master.clipboard_clear()
root.clipboard_append(item_text[4])
def searchService(self):
try:
conn = sqlite3.connect('cerberus.db')
except sqlite3.Error as e:
print(e)
cur = conn.cursor()
if self.search.get() == 'Αναζήτηση Υπηρεσίας':
pass
elif self.search.get():
cur.execute(
"SELECT id, name, email, username, password, value, category, url FROM service WHERE name LIKE '%"
+ self.search.get() + "%' or name LIKE '%" +
self.search.get().upper() +
"%'") # ('%'+self.search.get()+'%',),'Α')
elif not self.search.get():
cur.execute(
"SELECT id, name, email, username, password, value, category, url FROM service "
)
rows = cur.fetchall()
cur.close()
for k in self.table.get_children():
self.table.delete(k)
i = 1
for row in rows:
if (i % 2) == 0:
tag = "oddrow"
else:
tag = "evenrow"
self.table.insert(
'',
'end',
values=(
row[1],
self.cipher_suite.decrypt(row[2]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[3]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[4]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[5]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[6]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[7]).decode("utf-8").split(),
row[0]),
tags=tag)
i = i + 1
@staticmethod
def exitApp(event=None):
root.destroy()
@staticmethod
def getAboutAppForm():
import aboutApp
aboutApp.aboutApp()
def getAddNewServiceForm(self):
self.master.withdraw()
import addNewServiceForm
addNewServiceForm.addNewServiceForm(self)
def getEditServiceForm(self):
service = self.getSelectedService(self)
if service is None:
messagebox.showerror(
"Μήνυμα Σφάλματος",
"Παρακαλώ επιλέξτε την Υπηρεσία που θέλετε να Επεξεργαστείτε.")
else:
self.master.withdraw()
import editServiceForm
editServiceForm.editServiceForm(self, service)
def getSettingsForm(self):
import settingsForm
settingsForm.settingsForm()
def sortby(self, tree, col, descending):
data = [(tree.set(child, col), child)
for child in tree.get_children('')]
data.sort(reverse=descending)
for ix, item in enumerate(data):
if (ix % 2) == 0:
tag = "evenrow"
else:
tag = "oddrow"
tree.move(item[1], '', ix)
tree.item(item[1], tags=tag)
# switch the heading so that it will sort in the opposite direction
tree.heading(
col,
command=lambda x=col: self.sortby(tree, col, int(not descending)))
@staticmethod
def loadTable(self):
try:
conn = sqlite3.connect('cerberus.db')
except sqlite3.Error as e:
print(e)
cur = conn.cursor()
cur.execute(
"SELECT id, name, email, username, password, value, category, url value FROM service"
)
rows = cur.fetchall()
for row in self.table.get_children():
self.table.delete(row)
i = 1
for row in rows:
if (i % 2) == 0:
tag = "oddrow"
else:
tag = "evenrow"
self.table.insert(
'',
'end',
values=(
row[1],
self.cipher_suite.decrypt(row[2]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[3]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[4]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[5]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[6]).decode("utf-8").split(),
self.cipher_suite.decrypt(row[7]).decode("utf-8").split(),
row[0]),
tags=tag)
i = i + 1
conn.close()
self.last_focus = None
self.table.selection()
def deleteService(self):
service = self.getSelectedService(self)
if service is None:
messagebox.showerror(
"Μήνυμα Σφάλματος",
"Παρακαλώ επιλέξτε την Υπηρεσία που θέλετε να Διαγράξετε.")
else:
msgBox = messagebox.askquestion(
'Διαγραφή: {}'.format(service[0]),
'Είστε σίγουρος ότι θέλετε να διαγράψετε την Υπηρεσία: '
'{}'
' ?'.format(service[0]),
icon='warning')
if msgBox == 'yes':
try:
conn = sqlite3.connect('cerberus.db')
except sqlite3.Error as e:
print(e)
sql = 'DELETE FROM service WHERE id=?'
cur = conn.cursor()
cur.execute(sql, (service[-1], ))
conn.commit()
conn.close()
self.loadTable(self)
def popupMenu(self, event):
serviceId = self.table.identify_row(event.y)
if serviceId:
self.table.selection_set(serviceId)
try:
self.popup.tk_popup(event.x_root, event.y_root)
finally:
self.popup.grab_release()
def checkPasswordToExportToCSV(self):
print("Check Password..")
import logInForm
self.master.withdraw()
logInForm.logIn(self)
@staticmethod
def exportToCSV():
print("Export Services to CSV...")
try:
conn = sqlite3.connect('cerberus.db')
except sqlite3.Error as e:
print(e)
key = cerberusCryptography.getMasterKey()
cipher_suite = Fernet(key)
cur = conn.cursor()
cur.execute(
"SELECT category, name, email, username, password, value, url value FROM service"
)
rows = cur.fetchall()
csvData = [[
'Κατηγορία',
'Υπηρεσία',
'Email',
'Όνομα Χρήστη',
'Κωδικός',
'ID',
'URL',
]]
for row in rows:
csvData = csvData + [[
cipher_suite.decrypt(row[0]).decode("utf-8").split(),
cipher_suite.decrypt(row[1]).decode("utf-8").split(),
cipher_suite.decrypt(row[2]).decode("utf-8").split(),
cipher_suite.decrypt(row[3]).decode("utf-8").split(),
cipher_suite.decrypt(row[4]).decode("utf-8").split(),
cipher_suite.decrypt(row[5]).decode("utf-8").split(),
cipher_suite.decrypt(row[6]).decode("utf-8").split(),
]]
try:
homeFolder = str(Path.home())
filePath = filedialog.asksaveasfile(
initialdir=homeFolder,
initialfile='cerberus.csv',
title="Επιλογή Αρχείου",
filetypes=(("csv files", "*.csv"), ("all files", "*.*")))
if filePath:
try:
with open(filePath.name, 'w') as csvFile:
csvFile = csv.writer(csvFile, delimiter='\t')
csvFile.writerows(csvData)
messagebox.showinfo(
"Μήνυμα Επιτυχίας",
"Το αρχείο αποθηκέυτηκε με Επιτυχία στην τοποθεσία {}."
.format(filePath.name))
except Exception as e:
messagebox.showerror(
"Μήνυμα Σφάλματος",
"Δεν ήταν δυνατή η Εξαγωγή του αρχείου.")
except Exception as e:
print(e)
messagebox.showerror("Μήνυμα Σφάλματος",
"Δεν ήταν δυνατή η Εξαγωγή του αρχείου.")
class DialogPluginManager(Toplevel):
def __init__(self, mainWin, modulesWithNewerFileDates):
super(DialogPluginManager, self).__init__(mainWin.parent)
self.ENABLE = _("Enable")
self.DISABLE = _("Disable")
self.parent = mainWin.parent
self.cntlr = mainWin
# copy plugins for temporary display
self.pluginConfig = PluginManager.pluginConfig
self.pluginConfigChanged = False
self.uiClassMethodsChanged = False
self.modelClassesChanged = False
self.disclosureSystemTypesChanged = False
self.hostSystemFeaturesChanged = False
self.modulesWithNewerFileDates = modulesWithNewerFileDates
parentGeometry = re.match("(\d+)x(\d+)[+]?([-]?\d+)[+]?([-]?\d+)", self.parent.geometry())
dialogX = int(parentGeometry.group(3))
dialogY = int(parentGeometry.group(4))
self.title(_("Plug-in Manager"))
frame = Frame(self)
# left button frame
buttonFrame = Frame(frame, width=40)
buttonFrame.columnconfigure(0, weight=1)
addLabel = Label(buttonFrame, text=_("Find plug-in modules:"), wraplength=60, justify="center")
addLocalButton = Button(buttonFrame, text=_("Locally"), command=self.findLocally)
ToolTip(addLocalButton, text=_("File chooser allows selecting python module files to add (or reload) plug-ins, from the local file system."), wraplength=240)
addWebButton = Button(buttonFrame, text=_("On Web"), command=self.findOnWeb)
ToolTip(addWebButton, text=_("Dialog to enter URL full path to load (or reload) plug-ins, from the web or local file system."), wraplength=240)
addLabel.grid(row=0, column=0, pady=4)
addLocalButton.grid(row=1, column=0, pady=4)
addWebButton.grid(row=2, column=0, pady=4)
buttonFrame.grid(row=0, column=0, rowspan=2, sticky=(N, S, W), padx=3, pady=3)
# right tree frame (plugins already known to arelle)
modulesFrame = Frame(frame, width=700)
vScrollbar = Scrollbar(modulesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(modulesFrame, orient=HORIZONTAL)
self.modulesView = Treeview(modulesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=7)
self.modulesView.grid(row=0, column=0, sticky=(N, S, E, W))
self.modulesView.bind('<<TreeviewSelect>>', self.moduleSelect)
hScrollbar["command"] = self.modulesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.modulesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
modulesFrame.columnconfigure(0, weight=1)
modulesFrame.rowconfigure(0, weight=1)
modulesFrame.grid(row=0, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.modulesView.focus_set()
self.modulesView.column("#0", width=120, anchor="w")
self.modulesView.heading("#0", text=_("Name"))
self.modulesView["columns"] = ("author", "ver", "status", "date", "update", "descr", "license")
self.modulesView.column("author", width=100, anchor="w", stretch=False)
self.modulesView.heading("author", text=_("Author"))
self.modulesView.column("ver", width=50, anchor="w", stretch=False)
self.modulesView.heading("ver", text=_("Version"))
self.modulesView.column("status", width=50, anchor="w", stretch=False)
self.modulesView.heading("status", text=_("Status"))
self.modulesView.column("date", width=70, anchor="w", stretch=False)
self.modulesView.heading("date", text=_("File Date"))
self.modulesView.column("update", width=50, anchor="w", stretch=False)
self.modulesView.heading("update", text=_("Update"))
self.modulesView.column("descr", width=200, anchor="w", stretch=False)
self.modulesView.heading("descr", text=_("Description"))
self.modulesView.column("license", width=70, anchor="w", stretch=False)
self.modulesView.heading("license", text=_("License"))
classesFrame = Frame(frame)
vScrollbar = Scrollbar(classesFrame, orient=VERTICAL)
hScrollbar = Scrollbar(classesFrame, orient=HORIZONTAL)
self.classesView = Treeview(classesFrame, xscrollcommand=hScrollbar.set, yscrollcommand=vScrollbar.set, height=5)
self.classesView.grid(row=0, column=0, sticky=(N, S, E, W))
hScrollbar["command"] = self.classesView.xview
hScrollbar.grid(row=1, column=0, sticky=(E,W))
vScrollbar["command"] = self.classesView.yview
vScrollbar.grid(row=0, column=1, sticky=(N,S))
classesFrame.columnconfigure(0, weight=1)
classesFrame.rowconfigure(0, weight=1)
classesFrame.grid(row=1, column=1, columnspan=4, sticky=(N, S, E, W), padx=3, pady=3)
self.classesView.focus_set()
self.classesView.column("#0", width=200, anchor="w")
self.classesView.heading("#0", text=_("Class"))
self.classesView["columns"] = ("modules",)
self.classesView.column("modules", width=500, anchor="w", stretch=False)
self.classesView.heading("modules", text=_("Modules"))
# bottom frame module info details
moduleInfoFrame = Frame(frame, width=700)
moduleInfoFrame.columnconfigure(1, weight=1)
self.moduleNameLabel = Label(moduleInfoFrame, wraplength=600, justify="left",
font=font.Font(family='Helvetica', size=12, weight='bold'))
self.moduleNameLabel.grid(row=0, column=0, columnspan=4, sticky=W)
self.moduleAuthorHdr = Label(moduleInfoFrame, text=_("author:"), state=DISABLED)
self.moduleAuthorHdr.grid(row=1, column=0, sticky=W)
self.moduleAuthorLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleAuthorLabel.grid(row=1, column=1, columnspan=3, sticky=W)
self.moduleDescrHdr = Label(moduleInfoFrame, text=_("description:"), state=DISABLED)
self.moduleDescrHdr.grid(row=2, column=0, sticky=W)
self.moduleDescrLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDescrLabel.grid(row=2, column=1, columnspan=3, sticky=W)
self.moduleClassesHdr = Label(moduleInfoFrame, text=_("classes:"), state=DISABLED)
self.moduleClassesHdr.grid(row=3, column=0, sticky=W)
self.moduleClassesLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleClassesLabel.grid(row=3, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleClassesLabel, text=_("List of classes that this plug-in handles."), wraplength=240)
self.moduleUrlHdr = Label(moduleInfoFrame, text=_("URL:"), state=DISABLED)
self.moduleUrlHdr.grid(row=4, column=0, sticky=W)
self.moduleUrlLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleUrlLabel.grid(row=4, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleUrlLabel, text=_("URL of plug-in module (local file path or web loaded file)."), wraplength=240)
self.moduleDateHdr = Label(moduleInfoFrame, text=_("date:"), state=DISABLED)
self.moduleDateHdr.grid(row=5, column=0, sticky=W)
self.moduleDateLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleDateLabel.grid(row=5, column=1, columnspan=3, sticky=W)
ToolTip(self.moduleDateLabel, text=_("Date of currently loaded module file (with parenthetical node when an update is available)."), wraplength=240)
self.moduleLicenseHdr = Label(moduleInfoFrame, text=_("license:"), state=DISABLED)
self.moduleLicenseHdr.grid(row=6, column=0, sticky=W)
self.moduleLicenseLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleLicenseLabel.grid(row=6, column=1, columnspan=3, sticky=W)
self.moduleImportsHdr = Label(moduleInfoFrame, text=_("imports:"), state=DISABLED)
self.moduleImportsHdr.grid(row=7, column=0, sticky=W)
self.moduleImportsLabel = Label(moduleInfoFrame, wraplength=600, justify="left")
self.moduleImportsLabel.grid(row=7, column=1, columnspan=3, sticky=W)
self.moduleEnableButton = Button(moduleInfoFrame, text=self.ENABLE, state=DISABLED, command=self.moduleEnable)
ToolTip(self.moduleEnableButton, text=_("Enable/disable plug in."), wraplength=240)
self.moduleEnableButton.grid(row=8, column=1, sticky=E)
self.moduleReloadButton = Button(moduleInfoFrame, text=_("Reload"), state=DISABLED, command=self.moduleReload)
ToolTip(self.moduleReloadButton, text=_("Reload/update plug in."), wraplength=240)
self.moduleReloadButton.grid(row=8, column=2, sticky=E)
self.moduleRemoveButton = Button(moduleInfoFrame, text=_("Remove"), state=DISABLED, command=self.moduleRemove)
ToolTip(self.moduleRemoveButton, text=_("Remove plug in from plug in table (does not erase the plug in's file)."), wraplength=240)
self.moduleRemoveButton.grid(row=8, column=3, sticky=E)
moduleInfoFrame.grid(row=2, column=0, columnspan=5, sticky=(N, S, E, W), padx=3, pady=3)
moduleInfoFrame.config(borderwidth=4, relief="groove")
okButton = Button(frame, text=_("Close"), command=self.ok)
ToolTip(okButton, text=_("Accept and changes (if any) and close dialog."), wraplength=240)
cancelButton = Button(frame, text=_("Cancel"), command=self.close)
ToolTip(cancelButton, text=_("Cancel changes (if any) and close dialog."), wraplength=240)
okButton.grid(row=3, column=3, sticky=(S,E), pady=3)
cancelButton.grid(row=3, column=4, sticky=(S,E), pady=3, padx=3)
enableDisableFrame = Frame(frame)
enableDisableFrame.grid(row=3, column=1, sticky=(S,W), pady=3)
enableAllButton = Button(enableDisableFrame, text=_("Enable All"), command=self.enableAll)
ToolTip(enableAllButton, text=_("Enable all plug ins."), wraplength=240)
disableAllButton = Button(enableDisableFrame, text=_("Disable All"), command=self.disableAll)
ToolTip(disableAllButton, text=_("Disable all plug ins."), wraplength=240)
enableAllButton.grid(row=1, column=1)
disableAllButton.grid(row=1, column=2)
self.loadTreeViews()
self.geometry("+{0}+{1}".format(dialogX+50,dialogY+100))
frame.grid(row=0, column=0, sticky=(N,S,E,W))
frame.columnconfigure(0, weight=0)
frame.columnconfigure(1, weight=1)
frame.rowconfigure(0, weight=1)
window = self.winfo_toplevel()
window.columnconfigure(0, weight=1)
window.rowconfigure(0, weight=1)
self.bind("<Return>", self.ok)
self.bind("<Escape>", self.close)
self.protocol("WM_DELETE_WINDOW", self.close)
self.grab_set()
self.wait_window(self)
def loadTreeViews(self):
self.selectedModule = None
# clear previous treeview entries
for previousNode in self.modulesView.get_children(""):
self.modulesView.delete(previousNode)
def loadSubtree(parentNode, moduleItems):
for moduleItem in sorted(moduleItems, key=lambda item: item[0]):
moduleInfo = moduleItem[1]
if parentNode or not moduleInfo.get("isImported"):
nodeName = moduleItem[0]
if parentNode:
nodeName = parentNode + GROUPSEP + nodeName
name = moduleInfo.get("name", nodeName)
node = self.modulesView.insert(parentNode, "end", nodeName, text=name)
self.modulesView.set(node, "author", moduleInfo.get("author"))
self.modulesView.set(node, "ver", moduleInfo.get("version"))
self.modulesView.set(node, "status", moduleInfo.get("status"))
self.modulesView.set(node, "date", moduleInfo.get("fileDate"))
if name in self.modulesWithNewerFileDates:
self.modulesView.set(node, "update", _("available"))
self.modulesView.set(node, "descr", moduleInfo.get("description"))
self.modulesView.set(node, "license", moduleInfo.get("license"))
if moduleInfo.get("imports"):
loadSubtree(node, [(importModuleInfo["name"],importModuleInfo)
for importModuleInfo in moduleInfo["imports"]])
loadSubtree("", self.pluginConfig.get("modules", {}).items())
# clear previous treeview entries
for previousNode in self.classesView.get_children(""):
self.classesView.delete(previousNode)
for i, classItem in enumerate(sorted(self.pluginConfig.get("classes", {}).items())):
className, moduleList = classItem
node = self.classesView.insert("", "end", className, text=className)
self.classesView.set(node, "modules", ', '.join(moduleList))
self.moduleSelect() # clear out prior selection
def ok(self, event=None):
if self.pluginConfigChanged:
PluginManager.pluginConfig = self.pluginConfig
PluginManager.pluginConfigChanged = True
PluginManager.reset() # force reloading of modules
if self.uiClassMethodsChanged or self.modelClassesChanged or self.disclosureSystemTypesChanged or self.hostSystemFeaturesChanged: # may require reloading UI
affectedItems = ""
if self.uiClassMethodsChanged:
affectedItems += _("menus of the user interface")
if self.modelClassesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("model objects of the processor")
if self.disclosureSystemTypesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("disclosure system types")
if self.hostSystemFeaturesChanged:
if affectedItems:
affectedItems += _(" and ")
affectedItems += _("host system features")
if messagebox.askyesno(_("User interface plug-in change"),
_("A change in plug-in class methods may have affected {0}. "
"Please restart Arelle to due to these changes. \n\n"
"Should Arelle restart itself now "
"(if there are any unsaved changes they would be lost!)?"
).format(affectedItems),
parent=self):
self.cntlr.uiThreadQueue.put((self.cntlr.quit, [None, True]))
self.close()
def close(self, event=None):
self.parent.focus_set()
self.destroy()
def moduleSelect(self, *args):
node = (self.modulesView.selection() or (None,))[0]
if node:
node = node.rpartition(GROUPSEP)[2] # drop leading path names for module name
moduleInfo = self.pluginConfig.get("modules", {}).get(node)
if moduleInfo:
self.selectedModule = node
name = moduleInfo["name"]
self.moduleNameLabel.config(text=name)
self.moduleAuthorHdr.config(state=ACTIVE)
self.moduleAuthorLabel.config(text=moduleInfo["author"])
self.moduleDescrHdr.config(state=ACTIVE)
self.moduleDescrLabel.config(text=moduleInfo["description"])
self.moduleClassesHdr.config(state=ACTIVE)
self.moduleClassesLabel.config(text=', '.join(moduleInfo["classMethods"]))
self.moduleUrlHdr.config(state=ACTIVE)
self.moduleUrlLabel.config(text=moduleInfo["moduleURL"])
self.moduleDateHdr.config(state=ACTIVE)
self.moduleDateLabel.config(text=moduleInfo["fileDate"] + " " +
(_("(an update is available)") if name in self.modulesWithNewerFileDates else ""))
self.moduleLicenseHdr.config(state=ACTIVE)
self.moduleLicenseLabel.config(text=moduleInfo["license"])
if moduleInfo.get("imports"):
self.moduleImportsHdr.config(state=ACTIVE)
_text = ", ".join(mi["name"] for mi in moduleInfo["imports"][:3])
if len(moduleInfo["imports"]) >= 3:
_text += ", ..."
self.moduleImportsLabel.config(text=_text)
_buttonState = DISABLED if moduleInfo.get("isImported") else ACTIVE
self.moduleEnableButton.config(state=_buttonState,
text={"enabled":self.DISABLE,
"disabled":self.ENABLE}[moduleInfo["status"]])
self.moduleReloadButton.config(state=_buttonState)
self.moduleRemoveButton.config(state=_buttonState)
else:
self.selectedModule = None
self.moduleNameLabel.config(text="")
self.moduleAuthorHdr.config(state=DISABLED)
self.moduleAuthorLabel.config(text="")
self.moduleDescrHdr.config(state=DISABLED)
self.moduleDescrLabel.config(text="")
self.moduleClassesHdr.config(state=DISABLED)
self.moduleClassesLabel.config(text="")
self.moduleUrlHdr.config(state=DISABLED)
self.moduleUrlLabel.config(text="")
self.moduleDateHdr.config(state=DISABLED)
self.moduleDateLabel.config(text="")
self.moduleLicenseHdr.config(state=DISABLED)
self.moduleLicenseLabel.config(text="")
self.moduleImportsHdr.config(state=DISABLED)
self.moduleImportsLabel.config(text="")
self.moduleEnableButton.config(state=DISABLED, text=self.ENABLE)
self.moduleReloadButton.config(state=DISABLED)
self.moduleRemoveButton.config(state=DISABLED)
def findLocally(self):
initialdir = self.cntlr.pluginDir # default plugin directory
if not self.cntlr.isMac: # can't navigate within app easily, always start in default directory
initialdir = self.cntlr.config.setdefault("pluginOpenDir", initialdir)
filename = self.cntlr.uiFileDialog("open",
parent=self,
title=_("Choose plug-in module file"),
initialdir=initialdir,
filetypes=[(_("Python files"), "*.py")],
defaultextension=".py")
if filename:
# check if a package is selected (any file in a directory containing an __init__.py
#if (os.path.basename(filename) == "__init__.py" and os.path.isdir(os.path.dirname(filename)) and
# os.path.isfile(filename)):
# filename = os.path.dirname(filename) # refer to the package instead
self.cntlr.config["pluginOpenDir"] = os.path.dirname(filename)
moduleInfo = PluginManager.moduleModuleInfo(filename)
self.loadFoundModuleInfo(moduleInfo, filename)
def findOnWeb(self):
url = DialogURL.askURL(self)
if url: # url is the in-cache or local file
moduleInfo = PluginManager.moduleModuleInfo(url)
self.cntlr.showStatus("") # clear web loading status
self.loadFoundModuleInfo(moduleInfo, url)
def loadFoundModuleInfo(self, moduleInfo, url):
if moduleInfo and moduleInfo.get("name"):
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
else:
messagebox.showwarning(_("Module is not itself a plug-in or in a directory with package __init__.py plug-in. "),
_("File does not itself contain a python program with an appropriate __pluginInfo__ declaration: \n\n{0}")
.format(url),
parent=self)
def checkIfImported(self, moduleInfo):
if moduleInfo.get("isImported"):
messagebox.showwarning(_("Plug-in is imported by a parent plug-in. "),
_("Plug-in has a parent, please request operation on the parent: \n\n{0}")
.format(moduleInfo.get("name")),
parent=self)
return True
return False
def removePluginConfigModuleInfo(self, name):
moduleInfo = self.pluginConfig["modules"].get(name)
if moduleInfo:
if self.checkIfImported(moduleInfo):
return;
def _removePluginConfigModuleInfo(moduleInfo):
_name = moduleInfo.get("name")
if _name:
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].get(classMethod)
if classMethods and _name in classMethods:
classMethods.remove(_name)
if not classMethods: # list has become unused
del self.pluginConfig["classes"][classMethod] # remove class
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
elif classMethod == "ModelObjectFactory.ElementSubstitutionClasses":
self.modelClassesChanged = True # model object factor classes changed
elif classMethod == "DisclosureSystem.Types":
self.disclosureSystemTypesChanged = True # disclosure system types changed
elif classMethod.startswith("Proxy."):
self.hostSystemFeaturesChanged = True # system features (e.g., proxy) changed
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_removePluginConfigModuleInfo(importModuleInfo)
self.pluginConfig["modules"].pop(_name, None)
_removePluginConfigModuleInfo(moduleInfo)
self.pluginConfigChanged = True
def addPluginConfigModuleInfo(self, moduleInfo):
if self.checkIfImported(moduleInfo):
return;
name = moduleInfo.get("name")
self.removePluginConfigModuleInfo(name) # remove any prior entry for this module
def _addPlugin(moduleInfo):
_name = moduleInfo.get("name")
if _name:
self.modulesWithNewerFileDates.discard(_name) # no longer has an update available
self.pluginConfig["modules"][_name] = moduleInfo
# add classes
for classMethod in moduleInfo["classMethods"]:
classMethods = self.pluginConfig["classes"].setdefault(classMethod, [])
if name not in classMethods:
classMethods.append(_name)
if classMethod.startswith("CntlrWinMain.Menu"):
self.uiClassMethodsChanged = True # may require reloading UI
elif classMethod == "ModelObjectFactory.ElementSubstitutionClasses":
self.modelClassesChanged = True # model object factor classes changed
elif classMethod == "DisclosureSystem.Types":
self.disclosureSystemTypesChanged = True # disclosure system types changed
elif classMethod.startswith("Proxy."):
self.hostSystemFeaturesChanged = True # system features (e.g., proxy) changed
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_addPlugin(importModuleInfo)
_addPlugin(moduleInfo)
self.pluginConfigChanged = True
def moduleEnable(self):
if self.selectedModule in self.pluginConfig["modules"]:
moduleInfo = self.pluginConfig["modules"][self.selectedModule]
if self.checkIfImported(moduleInfo):
return;
def _moduleEnable(moduleInfo):
if self.moduleEnableButton['text'] == self.ENABLE:
moduleInfo["status"] = "enabled"
elif self.moduleEnableButton['text'] == self.DISABLE:
moduleInfo["status"] = "disabled"
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_moduleEnable(importModuleInfo)
_moduleEnable(moduleInfo)
if self.moduleEnableButton['text'] == self.ENABLE:
self.moduleEnableButton['text'] = self.DISABLE
elif self.moduleEnableButton['text'] == self.DISABLE:
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
def moduleReload(self):
if self.selectedModule in self.pluginConfig["modules"]:
url = self.pluginConfig["modules"][self.selectedModule].get("moduleURL")
if url:
moduleInfo = PluginManager.moduleModuleInfo(url, reload=True)
if moduleInfo:
if self.checkIfImported(moduleInfo):
return;
self.addPluginConfigModuleInfo(moduleInfo)
self.loadTreeViews()
self.cntlr.showStatus(_("{0} reloaded").format(moduleInfo["name"]), clearAfter=5000)
else:
messagebox.showwarning(_("Module error"),
_("File or module cannot be reloaded: \n\n{0}")
.format(url),
parent=self)
def moduleRemove(self):
if self.selectedModule in self.pluginConfig["modules"]:
self.removePluginConfigModuleInfo(self.selectedModule)
self.pluginConfigChanged = True
self.loadTreeViews()
def enableAll(self):
self.enableDisableAll(True)
def disableAll(self):
self.enableDisableAll(False)
def enableDisableAll(self, doEnable):
for module in self.pluginConfig["modules"]:
if not module.get("isImported"):
moduleInfo = self.pluginConfig["modules"][module]
def _enableDisableAll(moduleInfo):
if doEnable:
moduleInfo["status"] = "enabled"
else:
moduleInfo["status"] = "disabled"
for importModuleInfo in moduleInfo.get("imports", EMPTYLIST):
_enableDisableAll(importModuleInfo)
_enableDisableAll(moduleInfo)
if doEnable:
self.moduleEnableButton['text'] = self.DISABLE
else:
self.moduleEnableButton['text'] = self.ENABLE
self.pluginConfigChanged = True
self.loadTreeViews()
