Your IP : 18.220.13.15
import dbus
import dbus.exceptions
import time
from dbus.mainloop.glib import DBusGMainLoop
from gi.repository import GLib, GObject
from .exceptions import TunedAdminDBusException
__all__ = ["DBusController"]
class DBusController(object):
def __init__(self, bus_name, interface_name, object_name, debug = False):
self._bus_name = bus_name
self._interface_name = interface_name
self._object_name = object_name
self._proxy = None
self._interface = None
self._debug = debug
self._main_loop = None
self._action = None
self._on_exit_action = None
self._ret = True
self._exit = False
self._exception = None
def _init_proxy(self):
try:
if self._proxy is None:
DBusGMainLoop(set_as_default=True)
self._main_loop = GLib.MainLoop()
bus = dbus.SystemBus()
self._proxy = bus.get_object(self._bus_name, self._object_name)
self._interface = dbus.Interface(self._proxy, dbus_interface = self._interface_name)
except dbus.exceptions.DBusException:
raise TunedAdminDBusException("Cannot talk to TuneD daemon via DBus. Is TuneD daemon running?")
def _idle(self):
if self._action is not None:
# This may (and very probably will) run in child thread, so catch and pass exceptions to the main thread
try:
self._action_exit_code = self._action(*self._action_args, **self._action_kwargs)
except TunedAdminDBusException as e:
self._exception = e
self._exit = True
if self._exit:
if self._on_exit_action is not None:
self._on_exit_action(*self._on_exit_action_args,
**self._on_exit_action_kwargs)
self._main_loop.quit()
return False
else:
time.sleep(1)
return True
def set_on_exit_action(self, action, *args, **kwargs):
self._on_exit_action = action
self._on_exit_action_args = args
self._on_exit_action_kwargs = kwargs
def set_action(self, action, *args, **kwargs):
self._action = action
self._action_args = args
self._action_kwargs = kwargs
def run(self):
self._exception = None
GLib.idle_add(self._idle)
self._main_loop.run()
# Pass exception happened in child thread to the caller
if self._exception is not None:
raise self._exception
return self._ret
def _call(self, method_name, *args, **kwargs):
self._init_proxy()
try:
method = self._interface.get_dbus_method(method_name)
return method(*args, timeout=40)
except dbus.exceptions.DBusException as dbus_exception:
err_str = "DBus call to TuneD daemon failed"
if self._debug:
err_str += " (%s)" % str(dbus_exception)
raise TunedAdminDBusException(err_str)
def set_signal_handler(self, signal, cb):
self._init_proxy()
self._proxy.connect_to_signal(signal, cb)
def is_running(self):
return self._call("is_running")
def start(self):
return self._call("start")
def stop(self):
return self._call("stop")
def profiles(self):
return self._call("profiles")
def profiles2(self):
return self._call("profiles2")
def profile_info(self, profile_name):
return self._call("profile_info", profile_name)
def log_capture_start(self, log_level, timeout):
return self._call("log_capture_start", log_level, timeout)
def log_capture_finish(self, token):
return self._call("log_capture_finish", token)
def active_profile(self):
return self._call("active_profile")
def profile_mode(self):
return self._call("profile_mode")
def post_loaded_profile(self):
return self._call("post_loaded_profile")
def switch_profile(self, new_profile):
if new_profile == "":
return (False, "No profile specified")
return self._call("switch_profile", new_profile)
def auto_profile(self):
return self._call("auto_profile")
def recommend_profile(self):
return self._call("recommend_profile")
def verify_profile(self):
return self._call("verify_profile")
def verify_profile_ignore_missing(self):
return self._call("verify_profile_ignore_missing")
def off(self):
return self._call("disable")
def get_plugins(self):
"""Return dict with plugin names and their hints
Return:
dictionary -- {plugin_name: {parameter_name: default_value}}
"""
return self._call("get_all_plugins")
def get_plugin_documentation(self, plugin_name):
"""Return docstring of plugin's class"""
return self._call("get_plugin_documentation", plugin_name)
def get_plugin_hints(self, plugin_name):
"""Return dictionary with parameters of plugin and their hints
Parameters:
plugin_name -- name of plugin
Return:
dictionary -- {parameter_name: hint}
"""
return self._call("get_plugin_hints", plugin_name)
def instance_acquire_devices(self, devices, instance):
return self._call("instance_acquire_devices", devices, instance)
def get_instances(self, plugin_name):
return self._call("get_instances", plugin_name)
def instance_get_devices(self, instance):
return self._call("instance_get_devices", instance)
def exit(self, ret):
self.set_action(None)
self._ret = ret
self._exit = True
return ret