class ManagerDeviceList(DeviceList): def __init__(self, adapter: Optional[str] = None, inst: Optional["Blueman"] = None) -> None: cr = Gtk.CellRendererText() cr.props.ellipsize = Pango.EllipsizeMode.END tabledata: List[ListDataDict] = [ # device picture { "id": "device_surface", "type": str, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "celldata_func": (self._set_cell_data, None) }, # device caption { "id": "caption", "type": str, "renderer": cr, "render_attrs": { "markup": 1 }, "view_props": { "expand": True } }, { "id": "rssi_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": { "spacing": 0 }, "celldata_func": (self._set_cell_data, "rssi") }, { "id": "lq_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": { "spacing": 0 }, "celldata_func": (self._set_cell_data, "lq") }, { "id": "tpl_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": { "spacing": 0 }, "celldata_func": (self._set_cell_data, "tpl") }, { "id": "alias", "type": str }, # used for quick access instead of device.GetProperties { "id": "connected", "type": bool }, # used for quick access instead of device.GetProperties { "id": "paired", "type": bool }, # used for quick access instead of device.GetProperties { "id": "trusted", "type": bool }, # used for quick access instead of device.GetProperties { "id": "objpush", "type": bool }, # used to set Send File button { "id": "rssi", "type": float }, { "id": "lq", "type": float }, { "id": "tpl", "type": float }, { "id": "icon_info", "type": Gtk.IconInfo }, { "id": "cell_fader", "type": CellFade }, { "id": "row_fader", "type": TreeRowFade }, { "id": "levels_visible", "type": bool }, { "id": "initial_anim", "type": bool }, ] super().__init__(adapter, tabledata) self.set_name("ManagerDeviceList") self.set_headers_visible(False) self.props.has_tooltip = True self.Blueman = inst self.Config = Config("org.blueman.general") self.Config.connect('changed', self._on_settings_changed) # Set the correct sorting self._on_settings_changed(self.Config, "sort-by") self._on_settings_changed(self.Config, "sort-type") self.connect("query-tooltip", self.tooltip_query) self.tooltip_row: Optional[Gtk.TreePath] = None self.tooltip_col: Optional[Gtk.TreeViewColumn] = None self.connect("popup-menu", self._on_popup_menu) self.connect("button_press_event", self.on_event_clicked) self.connect("button_release_event", self.on_event_clicked) self.menu: Optional[ManagerDeviceMenu] = None self.connect("drag_data_received", self.drag_recv) self.connect("drag-motion", self.drag_motion) Gtk.Widget.drag_dest_set(self, Gtk.DestDefaults.ALL, [], Gdk.DragAction.COPY | Gdk.DragAction.DEFAULT) Gtk.Widget.drag_dest_add_uri_targets(self) self.set_search_equal_func(self.search_func) self._faderhandlers: Dict[str, int] = {} def _on_settings_changed(self, settings: Config, key: str) -> None: if key in ('sort-by', 'sort-order'): sort_by = settings['sort-by'] sort_order = settings['sort-order'] if sort_order == 'ascending': sort_type = Gtk.SortType.ASCENDING else: sort_type = Gtk.SortType.DESCENDING column_id = self.ids.get(sort_by) if column_id: self.liststore.set_sort_column_id(column_id, sort_type) def on_icon_theme_changed(self, _icon_them: Gtk.IconTheme) -> None: for row in self.liststore: device = self.get(row.iter, "device")["device"] self.row_setup_event(row.iter, device) def search_func(self, model: Gtk.TreeModel, column: int, key: str, tree_iter: Gtk.TreeIter) -> bool: row = self.get(tree_iter, "caption") if key.lower() in row["caption"].lower(): return False logging.info(f"{model} {column} {key} {tree_iter}") return True def drag_recv(self, _widget: Gtk.Widget, context: Gdk.DragContext, x: int, y: int, selection: Gtk.SelectionData, _info: int, time: int) -> None: uris = list(selection.get_uris()) context.finish(True, False, time) path = self.get_path_at_pos(x, y) if path: tree_iter = self.get_iter(path[0]) assert tree_iter is not None device = self.get(tree_iter, "device")["device"] command = f"blueman-sendto --device={device['Address']}" launch(command, paths=uris, name=_("File Sender")) context.finish(True, False, time) else: context.finish(False, False, time) def drag_motion(self, _widget: Gtk.Widget, drag_context: Gdk.DragContext, x: int, y: int, timestamp: int) -> bool: result = self.get_path_at_pos(x, y) if result is not None: path = result[0] assert path is not None if not self.selection.path_is_selected(path): tree_iter = self.get_iter(path) assert tree_iter is not None has_obj_push = self._has_objpush( self.get(tree_iter, "device")["device"]) if has_obj_push: Gdk.drag_status(drag_context, Gdk.DragAction.COPY, timestamp) self.set_cursor(path) return True else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False return False else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False def _on_popup_menu(self, _widget: Gtk.Widget) -> bool: if self.Blueman is None: return False if self.menu is None: self.menu = ManagerDeviceMenu(self.Blueman) window = self.get_window() assert window is not None selected = self.selected() assert selected is not None rect = self.get_cell_area(self.liststore.get_path(selected), self.get_column(1)) self.menu.popup_at_rect(window, rect, Gdk.Gravity.CENTER, Gdk.Gravity.NORTH) return True def on_event_clicked(self, _widget: Gtk.Widget, event: Gdk.Event) -> bool: if event.type not in (Gdk.EventType._2BUTTON_PRESS, Gdk.EventType.BUTTON_PRESS): return False path = self.get_path_at_pos(int(cast(Gdk.EventButton, event).x), int(cast(Gdk.EventButton, event).y)) if path is None: return False assert path[0] is not None row = self.get(path[0], "device", "connected") if not row: return False if self.Blueman is None: return False if self.menu is None: self.menu = ManagerDeviceMenu(self.Blueman) if event.type == Gdk.EventType._2BUTTON_PRESS and cast( Gdk.EventButton, event).button == 1: if self.menu.show_generic_connect_calc(row["device"]['UUIDs']): self.menu.generic_connect(None, device=row["device"], connect=not row["connected"]) if event.type == Gdk.EventType.BUTTON_PRESS and cast( Gdk.EventButton, event).button == 3: self.menu.popup_at_pointer(event) return False def get_icon_info(self, icon_name: str, size: int = 48, fallback: bool = True) -> Optional[Gtk.IconInfo]: if icon_name is None and not fallback: return None elif icon_name is None and fallback: icon_name = "image-missing" icon_info = self.icon_theme.lookup_icon_for_scale( icon_name, size, self.get_scale_factor(), Gtk.IconLookupFlags.FORCE_SIZE) return icon_info def make_device_icon(self, icon_info: Gtk.IconInfo, is_paired: bool = False, is_trusted: bool = False) -> cairo.Surface: window = self.get_window() scale = self.get_scale_factor() target = icon_info.load_surface(window) ctx = cairo.Context(target) if is_paired: _icon_info = self.get_icon_info("dialog-password", 16, False) assert _icon_info is not None paired_surface = _icon_info.load_surface(window) ctx.set_source_surface(paired_surface, 1 / scale, 1 / scale) ctx.paint_with_alpha(0.8) if is_trusted: _icon_info = self.get_icon_info("blueman-trust", 16, False) assert _icon_info is not None trusted_surface = _icon_info.load_surface(window) assert isinstance(target, cairo.ImageSurface) assert isinstance(trusted_surface, cairo.ImageSurface) height = target.get_height() mini_height = trusted_surface.get_height() y = height / scale - mini_height / scale - 1 / scale ctx.set_source_surface(trusted_surface, 1 / scale, y) ctx.paint_with_alpha(0.8) return target def device_remove_event(self, device: Device) -> None: tree_iter = self.find_device(device) assert tree_iter is not None row_fader = self.get(tree_iter, "row_fader")["row_fader"] super().device_remove_event(device) self._faderhandlers.update({ device.get_object_path(): row_fader.connect("animation-finished", self.__on_fader_finished, device) }) row_fader.thaw() self.emit("device-selected", None, None) row_fader.animate(start=row_fader.get_state(), end=0.0, duration=400) def __on_fader_finished(self, fader: TreeRowFade, device: Device) -> None: fader.disconnect(self._faderhandlers.pop(device.get_object_path())) fader.freeze() def device_add_event(self, device: Device) -> None: self.add_device(device) @staticmethod def make_caption(name: str, klass: str, address: str) -> str: return "<span size='x-large'>%(0)s</span>\n<span size='small'>%(1)s</span>\n<i>%(2)s</i>" \ % {"0": html.escape(name), "1": klass, "2": address} @staticmethod def get_device_class(device: Device) -> str: klass = get_minor_class(device['Class']) if klass != _("Uncategorized"): return klass else: return get_major_class(device['Class']) def row_setup_event(self, tree_iter: Gtk.TreeIter, device: Device) -> None: if not self.get(tree_iter, "initial_anim")["initial_anim"]: model = self.props.model assert model is not None cell_fader = CellFade(self, model.get_path(tree_iter), [2, 3, 4]) row_fader = TreeRowFade(self, model.get_path(tree_iter)) has_objpush = self._has_objpush(device) self.set(tree_iter, row_fader=row_fader, cell_fader=cell_fader, levels_visible=False, objpush=has_objpush) cell_fader.freeze() def on_finished(fader: TreeRowFade) -> None: fader.disconnect(faderhandler) fader.freeze() faderhandler = row_fader.connect("animation-finished", on_finished) row_fader.set_state(0.0) row_fader.animate(start=0.0, end=1.0, duration=500) self.set(tree_iter, initial_anim=True) klass = get_minor_class(device['Class']) # Bluetooth >= 4 devices use Appearance property appearance = device["Appearance"] if klass != _("Uncategorized") and klass != _("Unknown"): description = klass elif klass == _("Unknown") and appearance: description = gatt_appearance_to_name(appearance) else: description = get_major_class(device['Class']) icon_info = self.get_icon_info(device["Icon"], 48, False) caption = self.make_caption(device['Alias'], description, device['Address']) self.set(tree_iter, caption=caption, icon_info=icon_info, alias=device['Alias']) try: self.row_update_event(tree_iter, "Trusted", device['Trusted']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Paired", device['Paired']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Connected", device["Connected"]) except Exception as e: logging.exception(e) def row_update_event(self, tree_iter: Gtk.TreeIter, key: str, value: Any) -> None: logging.info(f"{key} {value}") if key == "Trusted": if value: self.set(tree_iter, trusted=True) else: self.set(tree_iter, trusted=False) elif key == "Paired": if value: self.set(tree_iter, paired=True) else: self.set(tree_iter, paired=False) elif key == "Alias": device = self.get(tree_iter, "device")["device"] c = self.make_caption(value, self.get_device_class(device), device['Address']) self.set(tree_iter, caption=c, alias=value) elif key == "UUIDs": device = self.get(tree_iter, "device")["device"] has_objpush = self._has_objpush(device) self.set(tree_iter, objpush=has_objpush) elif key == "Connected": self.set(tree_iter, connected=value) def level_setup_event(self, row_ref: Gtk.TreeRowReference, device: Device, cinfo: Optional[conn_info]) -> None: if not row_ref.valid(): return tree_iter = self.get_iter(row_ref.get_path()) assert tree_iter is not None row = self.get(tree_iter, "levels_visible", "cell_fader", "rssi", "lq", "tpl") if cinfo is not None: # cinfo init may fail for bluetooth devices version 4 and up # FIXME Workaround is horrible and we should show something better if cinfo.failed: rssi_perc = tpl_perc = lq_perc = 100.0 else: try: rssi = float(cinfo.get_rssi()) except ConnInfoReadError: rssi = 0 try: lq = float(cinfo.get_lq()) except ConnInfoReadError: lq = 0 try: tpl = float(cinfo.get_tpl()) except ConnInfoReadError: tpl = 0 rssi_perc = 50 + (rssi / 127 / 2 * 100) tpl_perc = 50 + (tpl / 127 / 2 * 100) lq_perc = lq / 255 * 100 if lq_perc < 10: lq_perc = 10 if rssi_perc < 10: rssi_perc = 10 if tpl_perc < 10: tpl_perc = 10 if not row["levels_visible"]: logging.info("animating up") self.set(tree_iter, levels_visible=True) fader = row["cell_fader"] fader.thaw() fader.set_state(0.0) fader.animate(start=0.0, end=1.0, duration=400) def on_finished(fader: CellFade) -> None: fader.freeze() fader.disconnect(faderhandler) faderhandler = fader.connect("animation-finished", on_finished) w = 14 * self.get_scale_factor() h = 48 * self.get_scale_factor() to_store = {} if round(row["rssi"], -1) != round(rssi_perc, -1): icon_name = "blueman-rssi-%d.png" % round(rssi_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file_at_scale( os.path.join(PIXMAP_PATH, icon_name), w, h, True) to_store.update({"rssi": rssi_perc, "rssi_pb": icon}) if round(row["lq"], -1) != round(lq_perc, -1): icon_name = "blueman-lq-%d.png" % round(lq_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file_at_scale( os.path.join(PIXMAP_PATH, icon_name), w, h, True) to_store.update({"lq": lq_perc, "lq_pb": icon}) if round(row["tpl"], -1) != round(tpl_perc, -1): icon_name = "blueman-tpl-%d.png" % round(tpl_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file_at_scale( os.path.join(PIXMAP_PATH, icon_name), w, h, True) to_store.update({"tpl": tpl_perc, "tpl_pb": icon}) if to_store: self.set(tree_iter, **to_store) else: if row["levels_visible"]: logging.info("animating down") self.set(tree_iter, levels_visible=False, rssi=-1, lq=-1, tpl=-1) fader = row["cell_fader"] fader.thaw() fader.set_state(1.0) fader.animate(start=fader.get_state(), end=0.0, duration=400) def on_finished(fader: CellFade) -> None: fader.disconnect(faderhandler) fader.freeze() if row_ref.valid(): assert tree_iter is not None # https://github.com/python/mypy/issues/2608 self.set(tree_iter, rssi_pb=None, lq_pb=None, tpl_pb=None) faderhandler = fader.connect("animation-finished", on_finished) def tooltip_query(self, _tw: Gtk.Widget, x: int, y: int, _kb: bool, tooltip: Gtk.Tooltip) -> bool: path = self.get_path_at_pos(x, y) if path is not None: if path[0] != self.tooltip_row or path[1] != self.tooltip_col: self.tooltip_row = path[0] self.tooltip_col = path[1] return False if path[1] == self.columns["device_surface"]: tree_iter = self.get_iter(path[0]) assert tree_iter is not None row = self.get(tree_iter, "trusted", "paired") trusted = row["trusted"] paired = row["paired"] if trusted and paired: tooltip.set_markup(_("<b>Trusted and Paired</b>")) elif paired: tooltip.set_markup(_("<b>Paired</b>")) elif trusted: tooltip.set_markup(_("<b>Trusted</b>")) else: return False self.tooltip_row = path[0] self.tooltip_col = path[1] return True if path[1] == self.columns["tpl_pb"] \ or path[1] == self.columns["lq_pb"] \ or path[1] == self.columns["rssi_pb"]: tree_iter = self.get_iter(path[0]) assert tree_iter is not None dt = self.get(tree_iter, "connected")["connected"] if dt: rssi = self.get(tree_iter, "rssi")["rssi"] lq = self.get(tree_iter, "lq")["lq"] tpl = self.get(tree_iter, "tpl")["tpl"] if rssi < 30: rssi_state = _("Poor") elif rssi < 40: rssi_state = _("Sub-optimal") elif rssi < 60: rssi_state = _("Optimal") elif rssi < 70: rssi_state = _("Much") else: rssi_state = _("Too much") if tpl < 30: tpl_state = _("Low") elif tpl < 40: tpl_state = _("Sub-optimal") elif tpl < 60: tpl_state = _("Optimal") elif tpl < 70: tpl_state = _("High") else: tpl_state = _("Very High") tooltip_template: str = "" if path[1] == self.columns["tpl_pb"]: tooltip_template = \ "<b>Connected</b>\nReceived Signal Strength: %(rssi)u%% <i>(%(rssi_state)s)</i>\n" \ "Link Quality: %(lq)u%%\n<b>Transmit Power Level: %(tpl)u%%</b> <i>(%(tpl_state)s)</i>" elif path[1] == self.columns["lq_pb"]: tooltip_template = \ "<b>Connected</b>\nReceived Signal Strength: %(rssi)u%% <i>(%(rssi_state)s)</i>\n" \ "<b>Link Quality: %(lq)u%%</b>\nTransmit Power Level: %(tpl)u%% <i>(%(tpl_state)s)</i>" elif path[1] == self.columns["rssi_pb"]: tooltip_template = \ "<b>Connected</b>\n<b>Received Signal Strength: %(rssi)u%%</b> <i>(%(rssi_state)s)</i>\n" \ "Link Quality: %(lq)u%%\nTransmit Power Level: %(tpl)u%% <i>(%(tpl_state)s)</i>" state_dict = { "rssi_state": rssi_state, "rssi": rssi, "lq": lq, "tpl": tpl, "tpl_state": tpl_state } tooltip.set_markup(tooltip_template % state_dict) self.tooltip_row = path[0] self.tooltip_col = path[1] return True return False def _has_objpush(self, device: Device) -> bool: if device is None: return False for uuid in device["UUIDs"]: if ServiceUUID(uuid).short_uuid == OBEX_OBJPUSH_SVCLASS_ID: return True return False def _set_cell_data(self, _col: Gtk.TreeViewColumn, cell: Gtk.CellRenderer, _model: Gtk.TreeModel, tree_iter: Gtk.TreeIter, data: Optional[str]) -> None: if data is None: row = self.get(tree_iter, "icon_info", "trusted", "paired") surface = self.make_device_icon(row["icon_info"], row["paired"], row["trusted"]) cell.set_property("surface", surface) elif data in ("rssi", "lq", "tpl"): window = self.get_window() scale = self.get_scale_factor() pb = self.get(tree_iter, data + "_pb")[data + "_pb"] if pb: surface = Gdk.cairo_surface_create_from_pixbuf( pb, scale, window) cell.set_property("surface", surface) else: cell.set_property("surface", None) def add_device(self, device: Device) -> None: if "Name" not in device and self.Config["hide-unnamed"]: logging.info(f"Hiding unnamed device: {device['Address']}") return super().add_device(device)
class ManagerDeviceList(DeviceList): def __init__(self, adapter: Optional[str] = None, inst: Optional["Blueman"] = None) -> None: cr = Gtk.CellRendererText() cr.props.ellipsize = Pango.EllipsizeMode.END tabledata: List[ListDataDict] = [ # device picture {"id": "device_surface", "type": str, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "celldata_func": (self._set_cell_data, None)}, # device caption {"id": "caption", "type": str, "renderer": cr, "render_attrs": {"markup": 1}, "view_props": {"expand": True}}, {"id": "battery_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": {"spacing": 0}, "celldata_func": (self._set_cell_data, "battery")}, {"id": "rssi_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": {"spacing": 0}, "celldata_func": (self._set_cell_data, "rssi")}, {"id": "lq_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": {"spacing": 0}, "celldata_func": (self._set_cell_data, "lq")}, {"id": "tpl_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "view_props": {"spacing": 0}, "celldata_func": (self._set_cell_data, "tpl")}, {"id": "alias", "type": str}, # used for quick access instead of device.GetProperties {"id": "connected", "type": bool}, # used for quick access instead of device.GetProperties {"id": "paired", "type": bool}, # used for quick access instead of device.GetProperties {"id": "trusted", "type": bool}, # used for quick access instead of device.GetProperties {"id": "objpush", "type": bool}, # used to set Send File button {"id": "battery", "type": float}, {"id": "rssi", "type": float}, {"id": "lq", "type": float}, {"id": "tpl", "type": float}, {"id": "icon_info", "type": Gtk.IconInfo}, {"id": "cell_fader", "type": CellFade}, {"id": "row_fader", "type": TreeRowFade}, {"id": "initial_anim", "type": bool}, {"id": "blocked", "type": bool} ] super().__init__(adapter, tabledata) self.set_name("ManagerDeviceList") self.set_headers_visible(False) self.props.has_tooltip = True self.Blueman = inst self._monitored_devices: Set[str] = set() self.manager.connect_signal("battery-created", self.on_battery_created) self.manager.connect_signal("battery-removed", self.on_battery_removed) self._batteries: Dict[str, Battery] = {} self.Config = Config("org.blueman.general") self.Config.connect('changed', self._on_settings_changed) # Set the correct sorting self._on_settings_changed(self.Config, "sort-by") self._on_settings_changed(self.Config, "sort-type") self.connect("query-tooltip", self.tooltip_query) self.tooltip_row: Optional[Gtk.TreePath] = None self.tooltip_col: Optional[Gtk.TreeViewColumn] = None self.connect("popup-menu", self._on_popup_menu) self.connect("button_press_event", self.on_event_clicked) self.connect("button_release_event", self.on_event_clicked) self.menu: Optional[ManagerDeviceMenu] = None self.connect("drag_data_received", self.drag_recv) self.connect("drag-motion", self.drag_motion) Gtk.Widget.drag_dest_set(self, Gtk.DestDefaults.ALL, [], Gdk.DragAction.COPY | Gdk.DragAction.DEFAULT) Gtk.Widget.drag_dest_add_uri_targets(self) self.set_search_equal_func(self.search_func) self.filter.set_visible_func(self.filter_func) def _on_settings_changed(self, settings: Config, key: str) -> None: if key in ('sort-by', 'sort-order'): sort_by = settings['sort-by'] sort_order = settings['sort-order'] if sort_order == 'ascending': sort_type = Gtk.SortType.ASCENDING else: sort_type = Gtk.SortType.DESCENDING column_id = self.ids.get(sort_by) if column_id: self.liststore.set_sort_column_id(column_id, sort_type) def on_icon_theme_changed(self, _icon_them: Gtk.IconTheme) -> None: for row in self.liststore: device = self.get(row.iter, "device")["device"] self.row_setup_event(row.iter, device) def on_battery_created(self, _manager: Manager, obj_path: str) -> None: if obj_path not in self._batteries: battery_proxy = Battery(obj_path=obj_path) self._batteries[obj_path] = battery_proxy logging.debug(f"{obj_path} {battery_proxy['Percentage']}") def on_battery_removed(self, _manager: Manager, obj_path: str) -> None: if obj_path in self._batteries: battery = self._batteries.pop(obj_path) battery.destroy() def search_func(self, model: Gtk.TreeModel, column: int, key: str, tree_iter: Gtk.TreeIter) -> bool: row = self.get(tree_iter, "caption") if key.lower() in row["caption"].lower(): return False logging.info(f"{model} {column} {key} {tree_iter}") return True def filter_func(self, _model: Gtk.TreeModel, tree_iter: Gtk.TreeIter, _data: Any) -> bool: no_name = self.get(tree_iter, "no_name")["no_name"] if no_name and self.Config["hide-unnamed"]: logging.debug("Hiding unnamed device") return False else: return True def drag_recv(self, _widget: Gtk.Widget, context: Gdk.DragContext, x: int, y: int, selection: Gtk.SelectionData, _info: int, time: int) -> None: uris = list(selection.get_uris()) context.finish(True, False, time) path = self.get_path_at_pos(x, y) if path: tree_iter = self.get_iter(path[0]) assert tree_iter is not None device = self.get(tree_iter, "device")["device"] command = f"blueman-sendto --device={device['Address']}" launch(command, paths=uris, name=_("File Sender")) context.finish(True, False, time) else: context.finish(False, False, time) def drag_motion(self, _widget: Gtk.Widget, drag_context: Gdk.DragContext, x: int, y: int, timestamp: int) -> bool: result = self.get_path_at_pos(x, y) if result is not None: path = result[0] assert path is not None path = self.filter.convert_path_to_child_path(path) if path is None: return False if not self.selection.path_is_selected(path): tree_iter = self.get_iter(path) assert tree_iter is not None has_obj_push = self._has_objpush(self.get(tree_iter, "device")["device"]) if has_obj_push: Gdk.drag_status(drag_context, Gdk.DragAction.COPY, timestamp) self.set_cursor(path) return True else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False return False else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False def _on_popup_menu(self, _widget: Gtk.Widget) -> bool: if self.Blueman is None: return False if self.menu is None: self.menu = ManagerDeviceMenu(self.Blueman) window = self.get_window() assert window is not None selected = self.selected() assert selected is not None rect = self.get_cell_area(self.liststore.get_path(selected), self.get_column(1)) self.menu.popup_at_rect(window, rect, Gdk.Gravity.CENTER, Gdk.Gravity.NORTH) return True def on_event_clicked(self, _widget: Gtk.Widget, event: Gdk.Event) -> bool: if event.type not in (Gdk.EventType._2BUTTON_PRESS, Gdk.EventType.BUTTON_PRESS): return False posdata = self.get_path_at_pos(int(cast(Gdk.EventButton, event).x), int(cast(Gdk.EventButton, event).y)) if posdata is None: return False else: path = posdata[0] assert path is not None childpath = self.filter.convert_path_to_child_path(path) assert childpath is not None row = self.get(childpath, "device", "connected") if not row: return False if self.Blueman is None: return False if self.menu is None: self.menu = ManagerDeviceMenu(self.Blueman) if event.type == Gdk.EventType._2BUTTON_PRESS and cast(Gdk.EventButton, event).button == 1: if self.menu.show_generic_connect_calc(row["device"]['UUIDs']): if row["connected"]: self.menu.disconnect_service(row["device"]) else: self.menu.connect_service(row["device"]) if event.type == Gdk.EventType.BUTTON_PRESS and cast(Gdk.EventButton, event).button == 3: self.menu.popup_at_pointer(event) return False def get_icon_info(self, icon_name: str, size: int = 48, fallback: bool = True) -> Optional[Gtk.IconInfo]: if icon_name is None and not fallback: return None elif icon_name is None and fallback: icon_name = "image-missing" icon_info = self.icon_theme.lookup_icon_for_scale(icon_name, size, self.get_scale_factor(), Gtk.IconLookupFlags.FORCE_SIZE) return icon_info def make_device_icon(self, icon_info: Gtk.IconInfo, is_paired: bool = False, is_trusted: bool = False, is_blocked: bool = False) -> cairo.Surface: window = self.get_window() scale = self.get_scale_factor() target = icon_info.load_surface(window) ctx = cairo.Context(target) if is_paired: _icon_info = self.get_icon_info("blueman-paired-emblem", 16, False) assert _icon_info is not None paired_surface = _icon_info.load_surface(window) ctx.set_source_surface(paired_surface, 1 / scale, 1 / scale) ctx.paint_with_alpha(0.8) if is_trusted: _icon_info = self.get_icon_info("blueman-trusted-emblem", 16, False) assert _icon_info is not None trusted_surface = _icon_info.load_surface(window) assert isinstance(target, cairo.ImageSurface) assert isinstance(trusted_surface, cairo.ImageSurface) height = target.get_height() mini_height = trusted_surface.get_height() y = height / scale - mini_height / scale - 1 / scale ctx.set_source_surface(trusted_surface, 1 / scale, y) ctx.paint_with_alpha(0.8) if is_blocked: _icon_info = self.get_icon_info("blueman-blocked-emblem", 16, False) assert _icon_info is not None blocked_surface = _icon_info.load_surface(window) assert isinstance(target, cairo.ImageSurface) assert isinstance(blocked_surface, cairo.ImageSurface) width = target.get_width() mini_width = blocked_surface.get_width() ctx.set_source_surface(blocked_surface, (width - mini_width - 1) / scale, 1 / scale) ctx.paint_with_alpha(0.8) return target def device_remove_event(self, device: Device) -> None: tree_iter = self.find_device(device) assert tree_iter is not None iter_set, _child_tree_iter = self.filter.convert_child_iter_to_iter(tree_iter) if iter_set: row_fader = self.get(tree_iter, "row_fader")["row_fader"] self._prepare_fader(row_fader, lambda: self.__fader_finished(device)) row_fader.animate(start=row_fader.get_state(), end=0.0, duration=400) def __fader_finished(self, device: Device) -> None: super().device_remove_event(device) self.emit("device-selected", None, None) def device_add_event(self, device: Device) -> None: self.add_device(device) @staticmethod def make_caption(name: str, klass: str, address: str) -> str: return "<span size='x-large'>%(0)s</span>\n<span size='small'>%(1)s</span>\n<i>%(2)s</i>" \ % {"0": html.escape(name), "1": klass, "2": address} @staticmethod def get_device_class(device: Device) -> str: klass = get_minor_class(device['Class']) if klass != _("Uncategorized"): return klass else: return get_major_class(device['Class']) def row_setup_event(self, tree_iter: Gtk.TreeIter, device: Device) -> None: if not self.get(tree_iter, "initial_anim")["initial_anim"]: assert self.liststore is not None child_path = self.liststore.get_path(tree_iter) result = self.filter.convert_child_path_to_path(child_path) if child_path is not None: cell_fader = CellFade(self, child_path, [2, 3, 4, 5]) row_fader = TreeRowFade(self, child_path) self.set(tree_iter, row_fader=row_fader, cell_fader=cell_fader) cell_fader.freeze() if result is not None: self._prepare_fader(row_fader).animate(start=0.0, end=1.0, duration=500) self.set(tree_iter, initial_anim=True) else: self.set(tree_iter, initial_anim=False) has_objpush = self._has_objpush(device) klass = get_minor_class(device['Class']) # Bluetooth >= 4 devices use Appearance property appearance = device["Appearance"] if klass != _("Uncategorized") and klass != _("Unknown"): description = klass elif klass == _("Unknown") and appearance: description = gatt_appearance_to_name(appearance) else: description = get_major_class(device['Class']) icon_info = self.get_icon_info(device["Icon"], 48, False) caption = self.make_caption(device['Alias'], description, device['Address']) self.set(tree_iter, caption=caption, icon_info=icon_info, alias=device['Alias'], objpush=has_objpush) try: self.row_update_event(tree_iter, "Trusted", device['Trusted']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Paired", device['Paired']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Connected", device["Connected"]) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Blocked", device["Blocked"]) except Exception as e: logging.exception(e) if device["Connected"]: self._monitor_power_levels(tree_iter, device) def _monitor_power_levels(self, tree_iter: Gtk.TreeIter, device: Device) -> None: if device["Address"] in self._monitored_devices: return assert self.Adapter is not None cinfo = conn_info(device["Address"], os.path.basename(self.Adapter.get_object_path())) try: cinfo.init() except ConnInfoReadError: logging.warning("Failed to get power levels, probably a LE device.") model = self.liststore assert isinstance(model, Gtk.TreeModel) r = Gtk.TreeRowReference.new(model, model.get_path(tree_iter)) self._update_power_levels(tree_iter, device, cinfo) GLib.timeout_add(1000, self._check_power_levels, r, cinfo, device["Address"]) self._monitored_devices.add(device["Address"]) def _check_power_levels(self, row_ref: Gtk.TreeRowReference, cinfo: conn_info, address: str) -> bool: if not row_ref.valid(): logging.warning("stopping monitor (row does not exist)") cinfo.deinit() self._monitored_devices.remove(address) return False tree_iter = self.get_iter(row_ref.get_path()) assert tree_iter is not None device = self.get(tree_iter, "device")["device"] if device["Connected"]: self._update_power_levels(tree_iter, device, cinfo) return True else: cinfo.deinit() self._disable_power_levels(tree_iter) self._monitored_devices.remove(address) return False def row_update_event(self, tree_iter: Gtk.TreeIter, key: str, value: Any) -> None: logging.info(f"{key} {value}") if key == "Trusted": if value: self.set(tree_iter, trusted=True) else: self.set(tree_iter, trusted=False) elif key == "Paired": if value: self.set(tree_iter, paired=True) else: self.set(tree_iter, paired=False) elif key == "Alias": device = self.get(tree_iter, "device")["device"] c = self.make_caption(value, self.get_device_class(device), device['Address']) self.set(tree_iter, caption=c, alias=value) elif key == "UUIDs": device = self.get(tree_iter, "device")["device"] has_objpush = self._has_objpush(device) self.set(tree_iter, objpush=has_objpush) elif key == "Connected": self.set(tree_iter, connected=value) if value: self._monitor_power_levels(tree_iter, self.get(tree_iter, "device")["device"]) else: self._disable_power_levels(tree_iter) elif key == "Name": self.set(tree_iter, no_name=False) self.filter.refilter() elif key == "Blocked": self.set(tree_iter, blocked=value) def _update_power_levels(self, tree_iter: Gtk.TreeIter, device: Device, cinfo: conn_info) -> None: row = self.get(tree_iter, "cell_fader", "battery", "rssi", "lq", "tpl") bars = {} obj_path = device.get_object_path() if obj_path in self._batteries: bars["battery"] = self._batteries[obj_path]["Percentage"] # cinfo init may fail for bluetooth devices version 4 and up # FIXME Workaround is horrible and we should show something better if cinfo.failed: bars.update({"rssi": 100.0, "tpl": 100.0, "lq": 100.0}) else: try: bars["rssi"] = max(50 + float(cinfo.get_rssi()) / 127 * 50, 10) except ConnInfoReadError: bars["rssi"] = 50 try: bars["lq"] = max(float(cinfo.get_lq()) / 255 * 100, 10) except ConnInfoReadError: bars["lq"] = 10 try: bars["tpl"] = max(50 + float(cinfo.get_tpl()) / 127 * 50, 10) except ConnInfoReadError: bars["tpl"] = 50 if row["battery"] == row["rssi"] == row["tpl"] == row["lq"] == 0: self._prepare_fader(row["cell_fader"]).animate(start=0.0, end=1.0, duration=400) w = 14 * self.get_scale_factor() h = 48 * self.get_scale_factor() for (name, perc) in bars.items(): if round(row[name], -1) != round(perc, -1): icon_name = f"blueman-{name}-{int(round(perc, -1))}.png" icon = GdkPixbuf.Pixbuf.new_from_file_at_scale(os.path.join(PIXMAP_PATH, icon_name), w, h, True) self.set(tree_iter, **{name: perc, f"{name}_pb": icon}) def _disable_power_levels(self, tree_iter: Gtk.TreeIter) -> None: row = self.get(tree_iter, "cell_fader", "battery", "rssi", "lq", "tpl") if row["battery"] == row["rssi"] == row["tpl"] == row["lq"] == 0: return self.set(tree_iter, battery=0, rssi=0, lq=0, tpl=0) self._prepare_fader(row["cell_fader"], lambda: self.set(tree_iter, battery_pb=None, rssi_pb=None, lq_pb=None, tpl_pb=None)).animate(start=1.0, end=0.0, duration=400) def _prepare_fader(self, fader: AnimBase, callback: Optional[Callable[[], None]] = None) -> AnimBase: def on_finished(finished_fader: AnimBase) -> None: finished_fader.disconnect(handler) finished_fader.freeze() if callback: callback() fader.thaw() handler = fader.connect("animation-finished", on_finished) return fader def tooltip_query(self, _tw: Gtk.Widget, x: int, y: int, _kb: bool, tooltip: Gtk.Tooltip) -> bool: path = self.get_path_at_pos(x, y) if path is None: return False if path[0] != self.tooltip_row or path[1] != self.tooltip_col: self.tooltip_row = path[0] self.tooltip_col = path[1] return False if path[1] == self.columns["device_surface"]: tree_iter = self.get_iter(path[0]) assert tree_iter is not None row = self.get(tree_iter, "trusted", "paired", "blocked") trusted = row["trusted"] paired = row["paired"] blocked = row["blocked"] str_list = [] if trusted: str_list.append(_("Trusted")) if paired: str_list.append(_("Paired")) if blocked: str_list.append(_("Blocked")) text = ", ".join(str_list) if text: tooltip.set_markup(f"<b>{text}</b>") else: return False self.tooltip_row = path[0] self.tooltip_col = path[1] return True elif path[1] == self.columns["battery_pb"] \ or path[1] == self.columns["tpl_pb"] \ or path[1] == self.columns["lq_pb"] \ or path[1] == self.columns["rssi_pb"]: tree_iter = self.get_iter(path[0]) assert tree_iter is not None dt = self.get(tree_iter, "connected")["connected"] if not dt: return False lines = [_("<b>Connected</b>")] battery = self.get(tree_iter, "battery")["battery"] rssi = self.get(tree_iter, "rssi")["rssi"] lq = self.get(tree_iter, "lq")["lq"] tpl = self.get(tree_iter, "tpl")["tpl"] if battery != 0: if path[1] == self.columns["battery_pb"]: lines.append(f"<b>Battery: {int(battery)}%</b>") else: lines.append(f"Battery: {int(battery)}%") if rssi != 0: if rssi < 30: rssi_state = _("Poor") elif rssi < 40: rssi_state = _("Sub-optimal") elif rssi < 60: rssi_state = _("Optimal") elif rssi < 70: rssi_state = _("Much") else: rssi_state = _("Too much") if path[1] == self.columns["rssi_pb"]: lines.append(_("<b>Received Signal Strength: %(rssi)u%%</b> <i>(%(rssi_state)s)</i>") % {"rssi": rssi, "rssi_state": rssi_state}) else: lines.append(_("Received Signal Strength: %(rssi)u%% <i>(%(rssi_state)s)</i>") % {"rssi": rssi, "rssi_state": rssi_state}) if lq != 0: if path[1] == self.columns["lq_pb"]: lines.append(_("<b>Link Quality: %(lq)u%%</b>") % {"lq": lq}) else: lines.append(_("Link Quality: %(lq)u%%") % {"lq": lq}) if tpl != 0: if tpl < 30: tpl_state = _("Low") elif tpl < 40: tpl_state = _("Sub-optimal") elif tpl < 60: tpl_state = _("Optimal") elif tpl < 70: tpl_state = _("High") else: tpl_state = _("Very High") if path[1] == self.columns["tpl_pb"]: lines.append(_("<b>Transmit Power Level: %(tpl)u%%</b> <i>(%(tpl_state)s)</i>") % {"tpl": tpl, "tpl_state": tpl_state}) else: lines.append(_("Transmit Power Level: %(tpl)u%% <i>(%(tpl_state)s)</i>") % {"tpl": tpl, "tpl_state": tpl_state}) tooltip.set_markup("\n".join(lines)) self.tooltip_row = path[0] self.tooltip_col = path[1] return True return False def _has_objpush(self, device: Device) -> bool: if device is None: return False for uuid in device["UUIDs"]: if ServiceUUID(uuid).short_uuid == OBEX_OBJPUSH_SVCLASS_ID: return True return False def _set_cell_data(self, _col: Gtk.TreeViewColumn, cell: Gtk.CellRenderer, model: Gtk.TreeModelFilter, tree_iter: Gtk.TreeIter, data: Optional[str]) -> None: tree_iter = model.convert_iter_to_child_iter(tree_iter) if data is None: row = self.get(tree_iter, "icon_info", "trusted", "paired", "blocked") surface = self.make_device_icon(row["icon_info"], row["paired"], row["trusted"], row["blocked"]) cell.set_property("surface", surface) else: window = self.get_window() scale = self.get_scale_factor() pb = self.get(tree_iter, data + "_pb")[data + "_pb"] if pb: surface = Gdk.cairo_surface_create_from_pixbuf(pb, scale, window) cell.set_property("surface", surface) else: cell.set_property("surface", None)
class ManagerDeviceList(DeviceList): def __init__(self, adapter=None, inst=None): cr = Gtk.CellRendererText() cr.props.ellipsize = Pango.EllipsizeMode.END tabledata = [ # device picture {"id": "device_surface", "type": str, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {}, "celldata_func": (self._set_device_cell_data, None)}, # device caption {"id": "caption", "type": str, "renderer": cr, "render_attrs": {"markup": 1}, "view_props": {"expand": True}}, {"id": "rssi_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {"pixbuf": 2}, "view_props": {"spacing": 0}}, {"id": "lq_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {"pixbuf": 3}, "view_props": {"spacing": 0}}, {"id": "tpl_pb", "type": GdkPixbuf.Pixbuf, "renderer": Gtk.CellRendererPixbuf(), "render_attrs": {"pixbuf": 4}, "view_props": {"spacing": 0}}, {"id": "alias", "type": str}, # used for quick access instead of device.GetProperties {"id": "connected", "type": bool}, # used for quick access instead of device.GetProperties {"id": "paired", "type": bool}, # used for quick access instead of device.GetProperties {"id": "trusted", "type": bool}, # used for quick access instead of device.GetProperties {"id": "objpush", "type": bool}, # used to set Send File button {"id": "rssi", "type": float}, {"id": "lq", "type": float}, {"id": "tpl", "type": float}, {"id": "icon_info", "type": Gtk.IconInfo}, {"id": "cell_fader", "type": GObject.TYPE_PYOBJECT}, {"id": "row_fader", "type": GObject.TYPE_PYOBJECT}, {"id": "levels_visible", "type": bool}, {"id": "initial_anim", "type": bool}, ] super().__init__(adapter, tabledata) self.set_name("ManagerDeviceList") self.set_headers_visible(False) self.props.has_tooltip = True self.Blueman = inst self.Config = Config("org.blueman.general") self.Config.connect('changed', self._on_settings_changed) # Set the correct sorting self._on_settings_changed(self.Config, "sort-by") self._on_settings_changed(self.Config, "sort-type") self.connect("query-tooltip", self.tooltip_query) self.tooltip_row = None self.tooltip_col = None self.connect("button_press_event", self.on_event_clicked) self.connect("button_release_event", self.on_event_clicked) self.menu = None self.connect("drag_data_received", self.drag_recv) self.connect("drag-motion", self.drag_motion) Gtk.Widget.drag_dest_set(self, Gtk.DestDefaults.ALL, [], Gdk.DragAction.COPY | Gdk.DragAction.DEFAULT) Gtk.Widget.drag_dest_add_uri_targets(self) self.set_search_equal_func(self.search_func, None) def _on_settings_changed(self, settings, key): if key in ('sort-by', 'sort-order'): sort_by = settings['sort-by'] sort_order = settings['sort-order'] if sort_order == 'ascending': sort_type = Gtk.SortType.ASCENDING else: sort_type = Gtk.SortType.DESCENDING column_id = self.ids.get(sort_by) if column_id: self.liststore.set_sort_column_id(column_id, sort_type) def on_icon_theme_changed(self, widget): for row in self.liststore: device = self.get(row.iter, "device")["device"] self.row_setup_event(row.iter, device) def do_device_found(self, device): tree_iter = self.find_device(device) if tree_iter: anim = TreeRowColorFade(self, self.props.model.get_path(tree_iter), Gdk.RGBA(0, 0, 1, 1)) anim.animate(start=0.8, end=1.0) def search_func(self, model, column, key, tree_iter): row = self.get(tree_iter, "caption") if key.lower() in row["caption"].lower(): return False logging.info("%s %s %s %s" % (model, column, key, tree_iter)) return True def drag_recv(self, widget, context, x, y, selection, target_type, time): uris = list(selection.get_uris()) context.finish(True, False, time) path = self.get_path_at_pos(x, y) if path: tree_iter = self.get_iter(path[0]) device = self.get(tree_iter, "device")["device"] command = "blueman-sendto --device=%s" % device['Address'] launch(command, uris, False, "blueman", _("File Sender")) context.finish(True, False, time) else: context.finish(False, False, time) return True def drag_motion(self, widget, drag_context, x, y, timestamp): result = self.get_path_at_pos(x, y) if result is not None: path = result[0] if not self.selection.path_is_selected(path): tree_iter = self.get_iter(path) has_obj_push = self._has_objpush(self.get(tree_iter, "device")["device"]) if has_obj_push: Gdk.drag_status(drag_context, Gdk.DragAction.COPY, timestamp) self.set_cursor(path) return True else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False else: Gdk.drag_status(drag_context, Gdk.DragAction.DEFAULT, timestamp) return False def on_event_clicked(self, widget, event): if event.type not in (Gdk.EventType._2BUTTON_PRESS, Gdk.EventType.BUTTON_PRESS): return path = self.get_path_at_pos(int(event.x), int(event.y)) if path is None: return row = self.get(path[0], "device", "connected") if not row: return if self.Blueman is None: return if self.menu is None: self.menu = ManagerDeviceMenu(self.Blueman) if event.type == Gdk.EventType._2BUTTON_PRESS and event.button == 1: if self.menu.show_generic_connect_calc(row["device"]['UUIDs']): self.menu.generic_connect(item=None, device=row["device"], connect=not row["connected"]) if event.type == Gdk.EventType.BUTTON_PRESS and event.button == 3: self.menu.popup(None, None, None, None, event.button, event.time) def get_icon_info(self, icon_name, size=48, fallback=True): if icon_name is None and not fallback: return None elif icon_name is None and fallback: icon_name = "image-missing" icon_info = self.icon_theme.lookup_icon_for_scale(icon_name, size, self.get_scale_factor(), Gtk.IconLookupFlags.FORCE_SIZE) return icon_info def make_device_icon(self, icon_info, is_paired=False, is_trusted=False): window = self.get_window() scale = self.get_scale_factor() target = icon_info.load_surface(window) ctx = cairo.Context(target) if is_paired: icon_info = self.get_icon_info("dialog-password", 16, False) paired_surface = icon_info.load_surface(window) ctx.set_source_surface(paired_surface, 1 / scale, 1 / scale) ctx.paint_with_alpha(0.8) if is_trusted: icon_info = self.get_icon_info("blueman-trust", 16, False) trusted_surface = icon_info.load_surface(window) height = target.get_height() mini_height = trusted_surface.get_height() y = height / scale - mini_height / scale - 1 / scale ctx.set_source_surface(trusted_surface, 1 / scale, y) ctx.paint_with_alpha(0.8) return target def device_remove_event(self, device): tree_iter = self.find_device(device) row_fader = self.get(tree_iter, "row_fader")["row_fader"] row_fader.connect("animation-finished", self.__on_fader_finished, device, tree_iter) row_fader.thaw() self.emit("device-selected", None, None) row_fader.animate(start=row_fader.get_state(), end=0.0, duration=400) def __on_fader_finished(self, fader, device, tree_iter): fader.disconnect_by_func(self.__on_fader_finished) fader.freeze() super().device_remove_event(device) def device_add_event(self, device): self.add_device(device) def make_caption(self, name, klass, address): return "<span size='x-large'>%(0)s</span>\n<span size='small'>%(1)s</span>\n<i>%(2)s</i>" \ % {"0": html.escape(name), "1": klass.capitalize(), "2": address} def get_device_class(self, device): klass = get_minor_class(device['Class']) if klass != "uncategorized": return get_minor_class(device['Class'], True) else: return get_major_class(device['Class']) def row_setup_event(self, tree_iter, device): if not self.get(tree_iter, "initial_anim")["initial_anim"]: cell_fader = CellFade(self, self.props.model.get_path(tree_iter), [2, 3, 4]) row_fader = TreeRowFade(self, self.props.model.get_path(tree_iter)) has_objpush = self._has_objpush(device) self.set(tree_iter, row_fader=row_fader, cell_fader=cell_fader, levels_visible=False, objpush=has_objpush) cell_fader.freeze() def on_finished(fader): fader.disconnect_by_func(on_finished) fader.freeze() row_fader.connect("animation-finished", on_finished) row_fader.set_state(0.0) row_fader.animate(start=0.0, end=1.0, duration=500) self.set(tree_iter, initial_anim=True) klass = get_minor_class(device['Class']) # Bluetooth >= 4 devices use Appearance property appearance = device["Appearance"] if klass != "uncategorized" and klass != "unknown": # get translated version description = get_minor_class(device['Class'], True).capitalize() elif klass == "unknown" and appearance: description = gatt_appearance_to_name(appearance) else: description = get_major_class(device['Class']).capitalize() icon_info = self.get_icon_info(device["Icon"], 48, False) caption = self.make_caption(device['Alias'], description, device['Address']) self.set(tree_iter, caption=caption, icon_info=icon_info, alias=device['Alias']) try: self.row_update_event(tree_iter, "Trusted", device['Trusted']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Paired", device['Paired']) except Exception as e: logging.exception(e) try: self.row_update_event(tree_iter, "Connected", device["Connected"]) except Exception as e: logging.exception(e) def row_update_event(self, tree_iter, key, value): logging.info("%s %s" % (key, value)) if key == "Trusted": if value: self.set(tree_iter, trusted=True) else: self.set(tree_iter, trusted=False) elif key == "Paired": if value: self.set(tree_iter, paired=True) else: self.set(tree_iter, paired=False) elif key == "Alias": device = self.get(tree_iter, "device")["device"] c = self.make_caption(value, self.get_device_class(device), device['Address']) self.set(tree_iter, caption=c, alias=value) elif key == "UUIDs": device = self.get(tree_iter, "device")["device"] has_objpush = self._has_objpush(device) self.set(tree_iter, objpush=has_objpush) elif key == "Connected": self.set(tree_iter, connected=value) def level_setup_event(self, row_ref, device, cinfo): if not row_ref.valid(): return tree_iter = self.get_iter(row_ref.get_path()) row = self.get(tree_iter, "levels_visible", "cell_fader", "rssi", "lq", "tpl") if cinfo is not None: # cinfo init may fail for bluetooth devices version 4 and up # FIXME Workaround is horrible and we should show something better if cinfo.failed: rssi_perc = tpl_perc = lq_perc = 100.0 else: try: rssi = float(cinfo.get_rssi()) except ConnInfoReadError: rssi = 0 try: lq = float(cinfo.get_lq()) except ConnInfoReadError: lq = 0 try: tpl = float(cinfo.get_tpl()) except ConnInfoReadError: tpl = 0 rssi_perc = 50 + (rssi / 127 / 2 * 100) tpl_perc = 50 + (tpl / 127 / 2 * 100) lq_perc = lq / 255 * 100 if lq_perc < 10: lq_perc = 10 if rssi_perc < 10: rssi_perc = 10 if tpl_perc < 10: tpl_perc = 10 if not row["levels_visible"]: logging.info("animating up") self.set(tree_iter, levels_visible=True) fader = row["cell_fader"] fader.thaw() fader.set_state(0.0) fader.animate(start=0.0, end=1.0, duration=400) def on_finished(fader): fader.freeze() fader.disconnect_by_func(on_finished) fader.connect("animation-finished", on_finished) to_store = {} if round(row["rssi"], -1) != round(rssi_perc, -1): icon_name = "blueman-rssi-%d.png" % round(rssi_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file(os.path.join(PIXMAP_PATH, icon_name)) to_store.update({"rssi": rssi_perc, "rssi_pb": icon}) if round(row["lq"], -1) != round(lq_perc, -1): icon_name = "blueman-lq-%d.png" % round(lq_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file(os.path.join(PIXMAP_PATH, icon_name)) to_store.update({"lq": lq_perc, "lq_pb": icon}) if round(row["tpl"], -1) != round(tpl_perc, -1): icon_name = "blueman-tpl-%d.png" % round(tpl_perc, -1) icon = GdkPixbuf.Pixbuf.new_from_file(os.path.join(PIXMAP_PATH, icon_name)) to_store.update({"tpl": tpl_perc, "tpl_pb": icon}) if to_store: self.set(tree_iter, **to_store) else: if row["levels_visible"]: logging.info("animating down") self.set(tree_iter, levels_visible=False, rssi=-1, lq=-1, tpl=-1) fader = row["cell_fader"] fader.thaw() fader.set_state(1.0) fader.animate(start=fader.get_state(), end=0.0, duration=400) def on_finished(fader): fader.disconnect_by_func(on_finished) fader.freeze() if row_ref.valid(): self.set(tree_iter, rssi_pb=None, lq_pb=None, tpl_pb=None) fader.connect("animation-finished", on_finished) def tooltip_query(self, tw, x, y, kb, tooltip): path = self.get_path_at_pos(x, y) if path is not None: if path[0] != self.tooltip_row or path[1] != self.tooltip_col: self.tooltip_row = path[0] self.tooltip_col = path[1] return False if path[1] == self.columns["device_surface"]: tree_iter = self.get_iter(path[0]) row = self.get(tree_iter, "trusted", "paired") trusted = row["trusted"] paired = row["paired"] if trusted and paired: tooltip.set_markup(_("<b>Trusted and Paired</b>")) elif paired: tooltip.set_markup(_("<b>Paired</b>")) elif trusted: tooltip.set_markup(_("<b>Trusted</b>")) else: return False self.tooltip_row = path[0] self.tooltip_col = path[1] return True if path[1] == self.columns["tpl_pb"] \ or path[1] == self.columns["lq_pb"] \ or path[1] == self.columns["rssi_pb"]: tree_iter = self.get_iter(path[0]) dt = self.get(tree_iter, "connected")["connected"] if dt: rssi = self.get(tree_iter, "rssi")["rssi"] lq = self.get(tree_iter, "lq")["lq"] tpl = self.get(tree_iter, "tpl")["tpl"] if rssi < 30: rssi_state = _("Poor") elif rssi < 40: rssi_state = _("Sub-optimal") elif rssi < 60: rssi_state = _("Optimal") elif rssi < 70: rssi_state = _("Much") else: rssi_state = _("Too much") if tpl < 30: tpl_state = _("Low") elif tpl < 40: tpl_state = _("Sub-optimal") elif tpl < 60: tpl_state = _("Optimal") elif tpl < 70: tpl_state = _("High") else: tpl_state = _("Very High") tooltip_template: str = "" if path[1] == self.columns["tpl_pb"]: tooltip_template = \ "<b>Connected</b>\nReceived Signal Strength: %(rssi)u%% <i>(%(rssi_state)s)</i>\n" \ "Link Quality: %(lq)u%%\n<b>Transmit Power Level: %(tpl)u%%</b> <i>(%(tpl_state)s)</i>" elif path[1] == self.columns["lq_pb"]: tooltip_template = \ "<b>Connected</b>\nReceived Signal Strength: %(rssi)u%% <i>(%(rssi_state)s)</i>\n" \ "<b>Link Quality: %(lq)u%%</b>\nTransmit Power Level: %(tpl)u%% <i>(%(tpl_state)s)</i>" elif path[1] == self.columns["rssi_pb"]: tooltip_template = \ "<b>Connected</b>\n<b>Received Signal Strength: %(rssi)u%%</b> <i>(%(rssi_state)s)</i>\n" \ "Link Quality: %(lq)u%%\nTransmit Power Level: %(tpl)u%% <i>(%(tpl_state)s)</i>" state_dict = {"rssi_state": rssi_state, "rssi": rssi, "lq": lq, "tpl": tpl, "tpl_state": tpl_state} tooltip.set_markup(tooltip_template % state_dict) self.tooltip_row = path[0] self.tooltip_col = path[1] return True return False def _has_objpush(self, device): if device is None: return False for uuid in device["UUIDs"]: if ServiceUUID(uuid).short_uuid == OBEX_OBJPUSH_SVCLASS_ID: return True return False def _set_device_cell_data(self, col, cell, model, tree_iter, data): row = self.get(tree_iter, "icon_info", "trusted", "paired") surface = self.make_device_icon(row["icon_info"], row["paired"], row["trusted"]) cell.set_property("surface", surface)