22
22
# Contact the authors at <mandos@fukt.bsnet.se>.
25
from __future__ import division, absolute_import, print_function, unicode_literals
25
from __future__ import (division, absolute_import, print_function,
86
87
self.proxy = proxy_object # Mandos Client proxy object
88
89
self.properties = dict()
89
self.proxy.connect_to_signal("PropertyChanged",
90
self.property_changed,
90
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
94
96
self.properties.update(
95
97
self.proxy.GetAll(client_interface,
96
98
dbus_interface = dbus.PROPERTIES_IFACE))
98
#XXX This break good super behaviour!
100
#XXX This breaks good super behaviour
99
101
# super(MandosClientPropertyCache, self).__init__(
100
102
# *args, **kwargs)
106
108
# Update properties dict with new value
107
109
self.properties[property] = value
111
def delete(self, *args, **kwargs):
112
self.property_changed_match.remove()
113
super(MandosClientPropertyCache, self).__init__(
110
117
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
154
161
if self.need_approval:
155
162
self.using_timer(True)
157
self.proxy.connect_to_signal("CheckerCompleted",
158
self.checker_completed,
161
self.proxy.connect_to_signal("CheckerStarted",
162
self.checker_started,
165
self.proxy.connect_to_signal("GotSecret",
169
self.proxy.connect_to_signal("NeedApproval",
173
self.proxy.connect_to_signal("Rejected",
164
self.match_objects = (
165
self.proxy.connect_to_signal("CheckerCompleted",
166
self.checker_completed,
169
self.proxy.connect_to_signal("CheckerStarted",
170
self.checker_started,
173
self.proxy.connect_to_signal("GotSecret",
177
self.proxy.connect_to_signal("NeedApproval",
181
self.proxy.connect_to_signal("Rejected",
185
#self.logger('Created client %s' % (self.properties["Name"]))
178
187
def property_changed(self, property=None, value=None):
179
188
super(self, MandosClientWidget).property_changed(property,
305
315
message = "Denial in %s. (a)pprove?"
306
316
message = message % unicode(timer).rsplit(".", 1)[0]
307
317
elif self.last_checker_failed:
308
timeout = datetime.timedelta(milliseconds
311
last_ok = isoformat_to_datetime(
312
max((self.properties["LastCheckedOK"]
313
or self.properties["Created"]),
314
self.properties["LastEnabled"]))
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
318
# When checker has failed, print a timer until client expires
319
expires = self.properties["Expires"]
321
timer = datetime.timedelta(0)
323
expires = datetime.datetime.strptime(expires,
324
'%Y-%m-%dT%H:%M:%S.%f')
325
timer = expires - datetime.datetime.utcnow()
316
326
message = ('A checker has failed! Time until client'
317
327
' gets disabled: %s'
318
328
% unicode(timer).rsplit(".", 1)[0])
337
347
self.update_hook()
339
349
def update_timer(self):
350
"""called by gobject. Will indefinitely loop until
351
gobject.source_remove() on tag is called"""
342
353
return True # Keep calling this
355
def delete(self, *args, **kwargs):
345
356
if self._update_timer_callback_tag is not None:
346
357
gobject.source_remove(self._update_timer_callback_tag)
347
358
self._update_timer_callback_tag = None
359
for match in self.match_objects:
361
self.match_objects = ()
348
362
if self.delete_hook is not None:
349
363
self.delete_hook(self)
364
return super(MandosClientWidget, self).delete(*args, **kwargs)
351
366
def render(self, maxcolrow, focus=False):
352
367
"""Render differently if we have focus.
359
374
This overrides the method from urwid.FlowWidget"""
361
self.proxy.Enable(dbus_interface = client_interface)
376
self.proxy.Enable(dbus_interface = client_interface,
363
self.proxy.Disable(dbus_interface = client_interface)
379
self.proxy.Disable(dbus_interface = client_interface,
365
382
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
366
dbus_interface = client_interface)
383
dbus_interface = client_interface,
368
386
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
369
dbus_interface = client_interface)
387
dbus_interface = client_interface,
370
389
elif key == "R" or key == "_" or key == "ctrl k":
371
390
self.server_proxy_object.RemoveClient(self.proxy
374
self.proxy.StartChecker(dbus_interface = client_interface)
394
self.proxy.StartChecker(dbus_interface = client_interface,
376
self.proxy.StopChecker(dbus_interface = client_interface)
397
self.proxy.StopChecker(dbus_interface = client_interface,
378
self.proxy.CheckedOK(dbus_interface = client_interface)
400
self.proxy.CheckedOK(dbus_interface = client_interface,
380
403
# elif key == "p" or key == "=":
381
404
# self.proxy.pause()
572
595
#self.log_message("Wrap mode: " + self.log_wrap)
574
597
def find_and_remove_client(self, path, name):
575
"""Find an client from its object path and remove it.
598
"""Find a client by its object path and remove it.
577
600
This is connected to the ClientRemoved signal from the
578
601
Mandos server object."""
580
603
client = self.clients_dict[path]
606
self.log_message("Unknown client %r (%r) removed", name,
584
self.remove_client(client, path)
586
611
def add_new_client(self, path):
587
612
client_proxy_object = self.bus.get_object(self.busname, path)