-
Notifications
You must be signed in to change notification settings - Fork 536
/
tag_editor.py
126 lines (116 loc) · 4.82 KB
/
tag_editor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# king_phisher/client/dialogs/tag_editor.py
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following disclaimer
# in the documentation and/or other materials provided with the
# distribution.
# * Neither the name of the project nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
import functools
from king_phisher.client import gui_utilities
from king_phisher.client.widget import managers
import advancedhttpserver
from gi.repository import Gtk
from gi.repository import Pango
__all__ = ('TagEditorDialog',)
class TagEditorDialog(gui_utilities.GladeGObject):
"""
Display a dialog which can be used to edit the various tags that are
present on the remote server. This can be used to rename tags and modify
their descriptions.
"""
dependencies = gui_utilities.GladeDependencies(
children=(
'notebook',
'button_close',
'treeview_campaign_types',
'treeview_company_departments',
'treeview_industries'
)
)
top_gobject = 'dialog'
objects_persist = False
tag_tables = ('campaign_types', 'company_departments', 'industries')
def __init__(self, *args, **kwargs):
super(TagEditorDialog, self).__init__(*args, **kwargs)
self.popup_menus = {}
self.treeview_managers = {}
for tag_table in self.tag_tables:
treeview = self.gobjects['treeview_' + tag_table]
model = Gtk.ListStore(str, str, str)
treeview.set_model(model)
tvm = managers.TreeViewManager(
treeview,
cb_delete=functools.partial(self.delete_tag, tag_table),
cb_refresh=functools.partial(self.load_tags, tag_table)
)
name_renderer = Gtk.CellRendererText()
name_renderer.connect('edited', self.signal_renderer_edited, (tag_table, 1, 'name'))
name_renderer.set_property('editable', True)
description_renderer = Gtk.CellRendererText()
description_renderer.connect('edited', self.signal_renderer_edited, (tag_table, 2, 'description'))
description_renderer.set_property('editable', True)
description_renderer.set_property('ellipsize', Pango.EllipsizeMode.END)
tvm.set_column_titles(
('Name', 'Description'),
column_offset=1,
renderers=(name_renderer, description_renderer)
)
self.treeview_managers[tag_table] = tvm
self.popup_menus[tag_table] = tvm.get_popup_menu()
self.load_tags()
def delete_tag(self, tag_table, treeview, selection):
(model, tree_iter) = selection.get_selected()
if not tree_iter:
return
tag_id = model.get_value(tree_iter, 0)
if not gui_utilities.show_dialog_yes_no('Delete This Tag?', self.dialog, 'This action is irreversible.'):
return
self.application.rpc('db/table/delete', tag_table, tag_id)
self.load_tags(tag_table)
def load_tags(self, tag_tables=None):
if tag_tables is None:
tag_tables = self.tag_tables
elif isinstance(tag_tables, str):
tag_tables = (tag_tables,)
for tag_table in tag_tables:
model = self.gobjects['treeview_' + tag_table].get_model()
self.application.rpc.get_tag_model(tag_table, model=model)
def interact(self):
self.dialog.show_all()
self.dialog.run()
self.dialog.destroy()
def signal_renderer_edited(self, cell, path, property_value, details):
tag_table, store_id, property_name = details
model = self.gobjects['treeview_' + tag_table].get_model()
model_iter = model.get_iter(path)
tag_id = model.get_value(model_iter, 0)
try:
self.application.rpc('db/table/set', tag_table, tag_id, (property_name,), (property_value,))
except advancedhttpserver.RPCError:
gui_utilities.show_dialog_error('Failed To Modify', self.dialog, 'An error occurred while modifying the information.')
else:
model.set_value(model_iter, store_id, property_value)