def __init__(self, system_bus, mainloop): # Init dbus service. dbus.service.Object.__init__(self, system_bus, DSC_SERVICE_PATH) # Init. self.mainloop = mainloop self.pkg_cache = AptCache() self.exit_flag = False self.simulate = False self.all_upgrade_pkg_names = [] self.download_dir = "/var/cache/apt/archives" self.is_upgrading = False self.in_update_list = False self.apt_action_pool = AptActionPool(self.pkg_cache) self.apt_action_pool.start() global_event.register_event('parse-download-error', self.send_parse_download_error) global_event.register_event("action-start", self.action_start) global_event.register_event("action-update", self.action_update) global_event.register_event("action-finish", self.action_finish) global_event.register_event("action-failed", self.action_failed) global_event.register_event("download-start", self.download_start) global_event.register_event("download-update", self.download_update) global_event.register_event("download-finish", self.download_finish) global_event.register_event("download-stop", self.download_stop) global_event.register_event("download-error", self.download_failed) global_event.register_event("update-list-start", self.update_list_start) global_event.register_event("update-list-finish", self.update_list_finish) global_event.register_event("update-list-failed", self.update_list_failed) global_event.register_event("update-list-update", self.update_list_update) self.packages_status = {} self.exit_manager = ExitManager(self.mainloop, self.is_update_list_running, self.is_download_action_running, self.is_apt_action_running, self.have_exit_request) self.exit_manager.start() self.set_download_dir('/var/cache/apt/archives') self.download_manager = DownloadManager(global_event, verbose=True)
def __init__(self, system_bus, mainloop): # Init dbus service. dbus.service.Object.__init__(self, system_bus, DSC_SERVICE_PATH) # Init. self.mainloop = mainloop self.pkg_cache = AptCache() self.exit_flag = False self.simulate = False self.all_upgrade_pkg_names = [] self.download_dir = "/var/cache/apt/archives" self.is_upgrading = False self.in_update_list = False self.apt_action_pool = AptActionPool(self.pkg_cache) self.apt_action_pool.start() global_event.register_event( 'parse-download-error', self.send_parse_download_error) global_event.register_event("action-start", self.action_start) global_event.register_event("action-update", self.action_update) global_event.register_event("action-finish", self.action_finish) global_event.register_event("action-failed", self.action_failed) global_event.register_event("download-start", self.download_start) global_event.register_event("download-update", self.download_update) global_event.register_event("download-finish", self.download_finish) global_event.register_event("download-stop", self.download_stop) global_event.register_event("download-error", self.download_failed) global_event.register_event( "update-list-start", self.update_list_start) global_event.register_event( "update-list-finish", self.update_list_finish) global_event.register_event( "update-list-failed", self.update_list_failed) global_event.register_event( "update-list-update", self.update_list_update) self.packages_status = {} self.exit_manager = ExitManager( self.mainloop, self.is_update_list_running, self.is_download_action_running, self.is_apt_action_running, self.have_exit_request) self.exit_manager.start() self.set_download_dir('/var/cache/apt/archives') self.download_manager = DownloadManager(global_event, verbose=True)
class PackageManager(dbus.service.Object): ''' docs ''' def __init__(self, system_bus, mainloop): log("init dbus") # Init dbus service. dbus.service.Object.__init__(self, system_bus, DSC_SERVICE_PATH) # Init. self.mainloop = mainloop self.pkg_cache = AptCache() self.exit_flag = False self.simulate = False self.all_upgrade_pkg_names = [] self.is_upgrading = False self.in_update_list = False self.apt_action_pool = AptActionPool(self.pkg_cache) self.apt_action_pool.start() global_event.register_event('parse-download-error', self.send_parse_download_error) global_event.register_event("action-start", self.action_start) global_event.register_event("action-update", self.action_update) global_event.register_event("action-finish", self.action_finish) global_event.register_event("action-failed", self.action_failed) global_event.register_event("download-start", self.download_start) global_event.register_event("download-update", self.download_update) global_event.register_event("download-finish", self.download_finish) global_event.register_event("download-stop", self.download_stop) global_event.register_event("download-error", self.download_failed) global_event.register_event("update-list-start", self.update_list_start) global_event.register_event("update-list-finish", self.update_list_finish) global_event.register_event("update-list-failed", self.update_list_failed) global_event.register_event("update-list-update", self.update_list_update) self.packages_status = {} self.exit_manager = ExitManager( self.mainloop, self.is_update_list_running, self.is_download_action_running, self.is_apt_action_running, self.have_exit_request) self.exit_manager.start() log("init finish") self.set_download_dir('/var/cache/apt/archives') self.init_download_manager(5) def download_start(self, pkg_name, action_type): utils.set_running_lock(True) self.update_signal([("download-start", (pkg_name, action_type))]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_update(self, pkg_name, action_type, data): self.update_signal([("download-update", (pkg_name, action_type, data))]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_stop(self, pkg_name, action_type): self.update_signal([("download-stop", (pkg_name, action_type))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_failed(self, pkg_name, action_type, e): logger.error("%s download failed with %s" % (pkg_name, e)) utils.set_running_lock(False) self.update_signal([("download-failed", (pkg_name, action_type, e))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_finish(self, action_id, action_type, all_pkg_names): utils.set_running_lock(False) self.update_signal([("download-finish", (action_id, action_type))]) if action_type == ACTION_INSTALL: self.apt_action_pool.add_install_action(all_pkg_names) elif action_type == ACTION_UPGRADE: self.start_upgrade(all_pkg_names, action_id) self.is_upgrading = False self.exit_manager.check() def action_start(self, signal_content): utils.set_running_lock(True) self.update_signal([("action-start", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_update(self, signal_content): self.update_signal([("action-update", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_finish(self, signal_content): utils.set_running_lock(False) pkg_name, action_type, pkg_info_list = signal_content if action_type == ACTION_INSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status(pkg_name, self.pkg_cache.PKG_STATUS_INSTALLED) elif action_type == ACTION_UPGRADE: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status(pkg_name, self.pkg_cache.PKG_STATUS_UPGRADED) elif action_type == ACTION_UNINSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status(pkg_name, self.pkg_cache.PKG_STATUS_UNINSTALLED) self.update_signal([("action-finish", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def action_failed(self, signal_content): utils.set_running_lock(False) self.update_signal([("action-failed", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def is_update_list_running(self): return self.in_update_list def is_download_action_running(self): return len(self.download_manager.fetch_files_dict) > 0 def is_apt_action_running(self): return len(self.apt_action_pool.active_mission_list) + len(self.apt_action_pool.wait_mission_list) > 0 def have_exit_request(self): return self.exit_flag def update_list_start(self): self.in_update_list = True self.update_signal([("update-list-start", "")]) def update_list_finish(self): self.update_signal([("update-list-merge", "")]) self.pkg_cache.open(apb.OpProgress()) db_build.BuildSoftwareDB(self.pkg_cache) self.update_signal([("update-list-finish", "")]) self.in_update_list = False self.exit_manager.check() def update_list_failed(self): self.in_update_list = False self.update_signal([("update-list-failed", "")]) self.exit_manager.check() def update_list_update(self, percent, status_message, speed_str): self.update_signal([("update-list-update", (percent, status_message, speed_str))]) def handle_dbus_reply(self, *reply): log("%s (reply): %s" % (self.module_dbus_name, str(reply))) def handle_dbus_error(self, *error): log("%s (error): %s" % (self.module_dbus_name, str(error))) def send_parse_download_error(self, pkg_name, action_type): self.update_signal([("parse-download-error", (pkg_name, action_type))]) def get_unique_id(self): return str(uuid.uuid4()) def get_real_pkg_name(self, pkg_name): if pkg_name in self.pkg_cache: return pkg_name elif (pkg_name + ":i386") in self.pkg_cache: return pkg_name + ":i386" else: return None def add_upgrade_download_with_new_policy(self, pkg_names, action_type): action_id = '%s_%s' % (self.get_unique_id(), action_type) self.update_signal([("ready-download-start", (action_id, action_type))]) real_pkg_dict, not_in_cache = parse_pkg.get_real_pkg_dict(self.pkg_cache, pkg_names) if not_in_cache: self.update_signal([("pkgs-not-in-cache", (json.dumps(not_in_cache), action_type))]) else: (all_change_pkgs, mark_failed_pkg_dict, marked_delete_sys_pkgs ) = parse_pkg.get_changes_pkgs(self.pkg_cache, real_pkg_dict) if mark_failed_pkg_dict: self.update_signal([("pkgs-mark-failed", (json.dumps(mark_failed_pkg_dict.keys()), action_type))]) elif marked_delete_sys_pkgs: self.update_signal([("marked-delete-system-pkgs", (json.dumps(marked_delete_sys_pkgs), action_type))]) else: self.update_signal([("ready-download-finish", (action_id, action_type))]) download_pkg_infos = parse_pkg.check_pkg_download_info(all_change_pkgs) if download_pkg_infos[0] == DOWNLOAD_STATUS_ERROR: self.update_signal([("pkgs-parse-download-error", (str(download_pkg_infos[1]), action_type))]) elif download_pkg_infos[0] == DOWNLOAD_STATUS_NOTNEED: self.start_upgrade(pkg_names, action_id) else: (names, download_urls, download_hash_infos, pkg_sizes) = download_pkg_infos self.all_upgrade_pkg_names = all_change_pkgs self.download_manager.add_download( action_id, action_type, download_urls, all_pkg_names=pkg_names, all_change_pkgs=all_change_pkgs, file_save_dir=self.download_dir, ) def add_download(self, pkg_name, action_type, simulate=False): pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) self.update_signal([("ready-download-finish", (pkg_name, action_type))]) if pkg_infos[0] == DOWNLOAD_STATUS_NOTNEED: self.download_finish(pkg_name, action_type, [pkg_name,]) print "Don't need download" elif pkg_infos[0] == DOWNLOAD_STATUS_ERROR: self.update_signal([("parse-download-error", (pkg_name, action_type))]) print "Download error" else: (names, download_urls, download_hash_infos, pkg_sizes) = pkg_infos self.download_manager.add_download( pkg_name, action_type, download_urls, all_pkg_names=[pkg_name,], file_save_dir=self.download_dir) def start_upgrade(self, pkg_names, action_id): self.apt_action_pool.add_multi_upgrade_mission(pkg_names, action_id) def del_source_list_d(self): white_list_path = os.path.join(get_parent_dir(__file__), 'white_list.txt') if os.path.exists(white_list_path): with open(white_list_path) as fp: for line in fp: line = line.strip() if os.path.exists(line): os.remove(line) def get_desktops(self, pkg_name): desktops = [] try: pkg_obj = self.pkg_cache[pkg_name] for f in pkg_obj.installed_files: if f.endswith(".desktop"): desktops.append(f) except: pass return desktops @dbus.service.method(DSC_SERVICE_NAME, in_signature='', out_signature='as') def read_no_notify_config(self, no_notify_config_path): if os.path.exists(no_notify_config_path): no_notify_config_str = read_file(no_notify_config_path) try: no_notify_config = eval(no_notify_config_str) if type(no_notify_config).__name__ != "list": no_notify_config = [] except Exception: no_notify_config = [] return no_notify_config else: return [] @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def add_no_notify_pkg(self, info): pkg_name, path = info no_notify_config = self.read_no_notify_config(path) if pkg_name not in no_notify_config: pkg_name = str(pkg_name) no_notify_config.append(pkg_name) write_file(path, str(no_notify_config)) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_no_notify_pkg(self, info): pkg_name, path = info pkg_name = str(pkg_name) path = str(path) no_notify_config = self.read_no_notify_config(path) if pkg_name in no_notify_config: write_file(path, str(filter(lambda config_pkg_name: config_pkg_name != pkg_name, no_notify_config))) @dbus.service.method(DSC_SERVICE_NAME, in_signature="i", out_signature="") def init_download_manager(self, number): self.download_manager = DownloadManager(global_event, number, verbose=True) @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="") def set_download_dir(self, local_dir): apt_pkg.config.set("Dir::Cache::Archives", local_dir) self.download_dir = local_dir @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="ai") def get_download_size(self, pkg_name): total_size = 0 pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) if pkg_infos[0] == DOWNLOAD_STATUS_NOTNEED: total_size = get_pkg_own_size(self.pkg_cache, pkg_name) size_flag = PKG_SIZE_OWN elif pkg_infos[0] == DOWNLOAD_STATUS_ERROR: total_size = -1 size_flag = PKG_SIZE_ERROR else: (names, download_urls, download_hash_infos, pkg_sizes) = pkg_infos for size in pkg_sizes: total_size += size size_flag = PKG_SIZE_DOWNLOAD return [size_flag, total_size] @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def get_upgrade_download_size(self, pkg_names): real_pkg_dict, not_in_cache = parse_pkg.get_real_pkg_dict(self.pkg_cache, pkg_names) if not_in_cache: return ["-1", json.dumps([])] else: (all_change_pkgs, mark_failed_pkg_dict, marked_delete_sys_pkgs ) = parse_pkg.get_changes_pkgs(self.pkg_cache, real_pkg_dict) download_pkg_infos = parse_pkg.check_pkg_download_info(all_change_pkgs) change_pkg_names = [pkg.name for pkg in all_change_pkgs] if download_pkg_infos[0] == DOWNLOAD_STATUS_ERROR: return ["-1", json.dumps([])] elif download_pkg_infos[0] == DOWNLOAD_STATUS_NOTNEED: return ["0", json.dumps(change_pkg_names)] else: (names, download_urls, download_hash_infos, pkg_sizes) = download_pkg_infos total_size = 0 for size in pkg_sizes: total_size += size return [str(total_size), json.dumps(change_pkg_names)] @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="ai") def clean_download_cache(self): '''Clean download cache.''' # get action packages. remain_pkgs = [] for (pkg_name, info_dict) in self.download_manager.fetch_files_dict.items(): remain_pkgs.append(pkg_name) for (pkg_name, info_dict) in self.apt_action_pool.install_action_dict.items(): remain_pkgs.append(pkg_name) for (pkg_name, info_dict) in self.apt_action_pool.upgrade_action_dict.items(): remain_pkgs.append(pkg_name) # Get depend packages. remain_pkgs_paths = [] for pkg_name in remain_pkgs: result = get_pkg_dependence_file_path(self.pkg_cache, pkg_name) if not result: remain_pkgs_paths += result # Init clean size. packageNum = 0 cleanSize = 0 # Delete cache directory. cache_archive_dir = get_cache_archive_dir() if os.path.exists(cache_archive_dir): for root, folder, files in os.walk(cache_archive_dir): for file_name in files: path = os.path.join(root, file_name) if path.endswith(".deb") and (path not in remain_pkgs_paths): packageNum += 1 cleanSize += os.path.getsize(path) remove_file(path) return [packageNum, cleanSize] @dbus.service.method(DSC_SERVICE_NAME, in_signature="b", out_signature="") def say_hello(self, simulate): # Set exit_flag with False to prevent backend exit, # this just useful that frontend start again before backend exit. log("Say hello from frontend.") self.exit_flag = False self.simulate = simulate @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def request_quit(self): # Set exit flag. self.exit_flag = True self.exit_manager.check() self.update_signal([("frontend-quit", "")]) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def change_source_list(self, repo_urls): ubuntu_repo_url, deepin_repo_url = repo_urls new_source_list_content = get_source_list_contents(ubuntu_repo_url, deepin_repo_url) os.system('cp %s %s.save' % (SOURCE_LIST, SOURCE_LIST)) with open(SOURCE_LIST, 'w') as fp: fp.write(new_source_list_content) @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="(sas)") # def request_upgrade_pkgs(self): // old api name def RequestUpgradeStatus(self): """ Get upgrade status @return: Return (status_code, [json.dumps(pkg_name, pkg_version)]), status code is in below constants: - normal: no dpkg action, just return packages that need to upgrade; - upgrading: in upgrading packages; - cache-updating: in updating apt cache; """ if self.in_update_list: return ("cache-updating", []) elif self.is_upgrading: return ("upgrading", []) else: cache_upgrade_pkgs = self.pkg_cache.get_upgrade_pkgs() return ("normal", cache_upgrade_pkgs) @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="s") def request_uninstall_pkgs(self): return json.dumps(self.pkg_cache.get_uninstall_pkgs()) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_install_version(self, pkg_names): pkg_versions = [] for pkg_name in pkg_names: try: version = self.pkg_cache[pkg_name].versions[0].version pkg_versions.append(version) except: try: version = self.pkg_cache[pkg_name+":i386"].versions[0].version pkg_versions.append(version) except: self.update_signal([("pkg-not-in-cache", pkg_name)]) return pkg_versions @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="as") def is_pkg_in_cache(self, pkg_name): result = [] try: self.pkg_cache[pkg_name] result.append(pkg_name) except: try: pkg_name += ":i386" self.pkg_cache[pkg_name] result.append(pkg_name) except: pass return result @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_uninstall_version(self, pkg_names): return self.pkg_cache.get_pkgs_uninstall_version(pkg_names) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def install_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: ThreadMethod(self.add_download, (real_pkg_name, ACTION_INSTALL, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg(self, pkg_name, purge): real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.apt_action_pool.add_uninstall_action(real_pkg_name, self.simulate, purge) else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg_from_desktop(self, desktop_path, purge): ThreadMethod(self.uninstall_pkg_from_desktop_thread, (desktop_path, purge)).start() def uninstall_pkg_from_desktop_thread(self, path, purge): pkg_name = self.get_pkg_name_from_path(path) if pkg_name: self.uninstall_pkg(pkg_name, purge) else: global_event.emit("action-failed", (path, ACTION_UNINSTALL, [], "no package found for path: %s" % path)) @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def get_pkg_name_from_path(self, path): result = subprocess.check_output(['dpkg', '-S', path]).strip().split(":") if len(result) == 2 and result[1].strip() == path.strip(): return result[0].strip() else: return "" @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def upgrade_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.update_signal([("ready-download-start", (real_pkg_name, ACTION_UPGRADE))]) ThreadMethod(self.add_download, (real_pkg_name, ACTION_UPGRADE, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def upgrade_pkgs_with_new_policy(self, pkg_names): ThreadMethod(self.add_upgrade_download_with_new_policy, (pkg_names, ACTION_UPGRADE)).start() @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def stop_download_pkg(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def cancel_upgrade_download(self): if getattr(self, 'upgrade_id'): self.download_manager.stop_wait_download(self.upgrade_id) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_wait_missions(self, pkg_infos): self.apt_action_pool.remove_wait_missions(pkg_infos) @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_wait_downloads(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def request_pkg_short_desc(self, pkg_name): return self.pkg_cache.get_pkg_short_desc(pkg_name) @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="as") def request_status(self): download_status = { ACTION_INSTALL : [], ACTION_UPGRADE : []} for (pkg_name, info_dict) in self.download_manager.fetch_files_dict.items(): if info_dict["action_type"] == ACTION_INSTALL: download_status[ACTION_INSTALL].append((pkg_name, info_dict["status"])) elif info_dict["action_type"] == ACTION_UPGRADE: download_status[ACTION_UPGRADE].append((pkg_name, info_dict["status"])) action_status = { ACTION_INSTALL : [], ACTION_UPGRADE : [], ACTION_UNINSTALL : []} for (pkg_name, info_dict) in self.apt_action_pool.install_action_dict.items(): action_status[ACTION_INSTALL].append((pkg_name, info_dict["status"])) for (pkg_name, info_dict) in self.apt_action_pool.upgrade_action_dict.items(): action_status[ACTION_UPGRADE].append((pkg_name, info_dict["status"])) for (pkg_name, info_dict) in self.apt_action_pool.uninstall_action_dict.items(): action_status[ACTION_UNINSTALL].append((pkg_name, info_dict["status"])) return [str(download_status), str(action_status)] @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def start_update_list(self): log("start update list...") self.del_source_list_d() self.pkg_cache.open(apb.OpProgress()) self.apt_action_pool.add_update_list_mission() log("start update list done") @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="ab") def request_pkgs_install_status(self, pkg_names): _status = [] for pkg in pkg_names: _status.append(self.pkg_cache.is_pkg_installed(pkg)) return _status @dbus.service.method(DSC_SERVICE_NAME, in_signature="(si)", out_signature="") def set_pkg_status(self, pkg_status): pkg_name, status = pkg_status self.pkg_cache.set_pkg_status(pkg_name, pkg_status) @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="i") def get_pkg_installed(self, pkg_name): if self.is_pkg_in_cache(pkg_name): if self.pkg_cache.is_pkg_installed(pkg_name): return 1 else: return 0 else: return -1 @dbus.service.method(DSC_SERVICE_NAME, in_signature="ss", out_signature="s") def get_pkg_start_status(self, pkg_name, start_pkg_names): is_current_installed = self.get_pkg_installed(pkg_name) if is_current_installed == -1: return "unknown" elif is_current_installed == 0: return "uninstalled" else: desktops = self.get_desktops(pkg_name) names = start_pkg_names.split(",") for name in names: if self.get_pkg_installed(name) == 1: desktops += self.get_desktops(name) return json.dumps(desktops) @dbus.service.signal(DSC_SERVICE_NAME, signature='a(sv)') # Use below command for test: # dbus-monitor --system "type='signal', interface='com.linuxdeepin.softwarecenter'" def update_signal(self, message): # The signal is emitted when this method exits # You can have code here if you wish pass
class PackageManager(dbus.service.Object): ''' docs ''' def invoked_log(func): @functools.wraps(func) def wrapper(*args, **kw): info = "dbus method invoked: %s, arguments: %s" % (func.__name__, str(args[1:])) logger.debug(info) return func(*args, **kw) return wrapper def __init__(self, system_bus, mainloop): # Init dbus service. dbus.service.Object.__init__(self, system_bus, DSC_SERVICE_PATH) # Init. self.mainloop = mainloop self.pkg_cache = AptCache() self.exit_flag = False self.simulate = False self.all_upgrade_pkg_names = [] self.download_dir = "/var/cache/apt/archives" self.is_upgrading = False self.in_update_list = False self.apt_action_pool = AptActionPool(self.pkg_cache) self.apt_action_pool.start() global_event.register_event('parse-download-error', self.send_parse_download_error) global_event.register_event("action-start", self.action_start) global_event.register_event("action-update", self.action_update) global_event.register_event("action-finish", self.action_finish) global_event.register_event("action-failed", self.action_failed) global_event.register_event("download-start", self.download_start) global_event.register_event("download-update", self.download_update) global_event.register_event("download-finish", self.download_finish) global_event.register_event("download-stop", self.download_stop) global_event.register_event("download-error", self.download_failed) global_event.register_event("update-list-start", self.update_list_start) global_event.register_event("update-list-finish", self.update_list_finish) global_event.register_event("update-list-failed", self.update_list_failed) global_event.register_event("update-list-update", self.update_list_update) self.packages_status = {} self.exit_manager = ExitManager(self.mainloop, self.is_update_list_running, self.is_download_action_running, self.is_apt_action_running, self.have_exit_request) self.exit_manager.start() self.set_download_dir('/var/cache/apt/archives') self.download_manager = DownloadManager(global_event, verbose=True) def download_start(self, pkg_name, action_type): utils.set_running_lock(True) self.update_signal([("download-start", (pkg_name, action_type))]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_update(self, pkg_name, action_type, data): self.update_signal([("download-update", (pkg_name, action_type, data)) ]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_stop(self, pkg_name, action_type): self.update_signal([("download-stop", (pkg_name, action_type))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_failed(self, pkg_name, action_type, e): logger.error("%s download failed with %s" % (pkg_name, e)) utils.set_running_lock(False) self.update_signal([("download-failed", (pkg_name, action_type, e))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_finish(self, action_id, action_type, all_pkg_names): utils.set_running_lock(False) self.update_signal([("download-finish", (action_id, action_type))]) if action_type == ACTION_INSTALL: self.apt_action_pool.add_install_action(all_pkg_names) elif action_type == ACTION_UPGRADE: self.start_upgrade(all_pkg_names, action_id) self.is_upgrading = False self.exit_manager.check() def action_start(self, signal_content): utils.set_running_lock(True) self.update_signal([("action-start", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_update(self, signal_content): self.update_signal([("action-update", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_finish(self, signal_content): utils.set_running_lock(False) pkg_name, action_type, pkg_info_list = signal_content if action_type == ACTION_INSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_INSTALLED) elif action_type == ACTION_UPGRADE: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_UPGRADED) elif action_type == ACTION_UNINSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_UNINSTALLED) self.update_signal([("action-finish", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def action_failed(self, signal_content): utils.set_running_lock(False) self.update_signal([("action-failed", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def is_update_list_running(self): return self.in_update_list def is_download_action_running(self): return len(self.download_manager.download_task_info) > 0 def is_apt_action_running(self): return len(self.apt_action_pool.active_mission_list) + len( self.apt_action_pool.wait_mission_list) > 0 def have_exit_request(self): return self.exit_flag def update_list_start(self): self.in_update_list = True self.update_signal([("update-list-start", "")]) def update_list_finish(self): self.update_signal([("update-list-merge", "")]) self.pkg_cache.open(apb.OpProgress()) db_build.BuildSoftwareDB(self.pkg_cache) self.update_signal([("update-list-finish", "")]) self.in_update_list = False self.exit_manager.check() def update_list_failed(self): self.in_update_list = False self.update_signal([("update-list-failed", "")]) self.exit_manager.check() def update_list_update(self, percent, status_message, speed_str): self.update_signal([("update-list-update", (percent, status_message, speed_str))]) def handle_dbus_reply(self, *reply): logger.info("%s (reply): %s" % (self.module_dbus_name, str(reply))) def handle_dbus_error(self, *error): logger.info("%s (error): %s" % (self.module_dbus_name, str(error))) def send_parse_download_error(self, pkg_name, action_type): self.update_signal([("parse-download-error", (pkg_name, action_type))]) def get_unique_id(self): return str(uuid.uuid4()) def get_real_pkg_name(self, pkg_name): if pkg_name in self.pkg_cache: return pkg_name elif (pkg_name + ":i386") in self.pkg_cache: return pkg_name + ":i386" else: return None def dist_upgrade(self): action_type = ACTION_UPGRADE action_id = '%s_%s' % (self.get_unique_id(), action_type) self.update_signal([("ready-download-start", (action_id, action_type)) ]) dist_upgrade_changes = self.pkg_cache.get_dist_upgrade_changes() # detect and prevent the system components from being uninstalled pkg_in_white_list = [] download_pkgs = [] self.all_upgrade_pkg_names = [] for pkg in dist_upgrade_changes["delete"]: name = pkg.name if name in SYS_PKG_WHITE_LIST: pkg_in_white_list.append(name) if len(pkg_in_white_list) > 0: self.update_signal([("remove-pkgs-in-white-list", (json.dumps(pkg_in_white_list), action_type)) ]) logger.error("found remove system packages: %s" % json.dumps(pkg_in_white_list)) else: for status in dist_upgrade_changes: for pkg in dist_upgrade_changes[status]: self.all_upgrade_pkg_names.append(pkg.name) if status == "delete" or status == "keep": continue if not parse_pkg.pkg_file_has_exist(pkg): download_pkgs.append(pkg) self.update_signal([("ready-download-finish", (action_id, action_type))]) if len(download_pkgs) == 0: global_event.emit("download-finish", action_id, action_type, self.all_upgrade_pkg_names) else: names = [] download_urls = [] download_hash_infos = [] pkg_sizes = [] for pkg in download_pkgs: names.append(pkg.name) download_urls.append(pkg.candidate.uri) download_hash_infos.append(pkg.candidate.md5) pkg_sizes.append(pkg.candidate.size) self.download_manager.add_download( action_id, action_type, download_urls, download_sizes=pkg_sizes, download_md5s=download_hash_infos, all_task_names=names, all_change_pkgs=self.all_upgrade_pkg_names, file_save_dir=self.download_dir) def add_download(self, pkg_name, action_type, simulate=False): installed = self.get_pkg_installed(pkg_name) if installed == 1: self.update_signal([("pkg-installed", (pkg_name, action_type))]) return elif installed == -1: self.update_signal([("pkg-not-in-cache", (pkg_name, action_type))]) return pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) status = pkg_infos["status"] self.update_signal([("ready-download-finish", (pkg_name, action_type)) ]) if status == DOWNLOAD_STATUS_NOTNEED: self.download_finish(pkg_name, action_type, [ pkg_name, ]) elif status == DOWNLOAD_STATUS_ERROR: self.update_signal([("parse-download-error", (pkg_name, action_type))]) else: names = [] download_urls = [] download_hash_infos = [] pkg_sizes = [] for info in pkg_infos["info"]: names.append(info["name"]) download_urls.append(info["url"]) download_hash_infos.append(info["hash"]) pkg_sizes.append(info["size"]) self.download_manager.add_download( pkg_name, action_type, download_urls, download_sizes=pkg_sizes, download_md5s=download_hash_infos, all_task_names=[pkg_name], file_save_dir=self.download_dir) def start_upgrade(self, pkg_names, action_id): self.apt_action_pool.add_multi_upgrade_mission(pkg_names, action_id) def del_source_list_d(self): white_list_path = os.path.join(get_parent_dir(__file__), 'white_list.txt') if os.path.exists(white_list_path): with open(white_list_path) as fp: for line in fp: line = line.strip() if os.path.exists(line): os.remove(line) def get_desktops(self, pkg_name): desktops = [] try: pkg_obj = self.pkg_cache[pkg_name] for f in pkg_obj.installed_files: if f.endswith(".desktop"): desktops.append(f) except: pass return desktops @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature='', out_signature='as') def read_no_notify_config(self, no_notify_config_path): if os.path.exists(no_notify_config_path): no_notify_config_str = read_file(no_notify_config_path) try: no_notify_config = eval(no_notify_config_str) if type(no_notify_config).__name__ != "list": no_notify_config = [] except Exception: no_notify_config = [] return no_notify_config else: return [] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def add_no_notify_pkg(self, info): pkg_name, path = info no_notify_config = self.read_no_notify_config(path) if pkg_name not in no_notify_config: pkg_name = str(pkg_name) no_notify_config.append(pkg_name) write_file(path, str(no_notify_config)) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_no_notify_pkg(self, info): pkg_name, path = info pkg_name = str(pkg_name) path = str(path) no_notify_config = self.read_no_notify_config(path) if pkg_name in no_notify_config: write_file( path, str( filter(lambda config_pkg_name: config_pkg_name != pkg_name, no_notify_config))) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="i", out_signature="") def init_download_manager(self, number): pass @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="") def set_download_dir(self, local_dir): apt_pkg.config.set("Dir::Cache::Archives", local_dir) self.download_dir = local_dir @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="ai") def get_download_size(self, pkg_name): total_size = 0 pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) status = pkg_infos["status"] if status == DOWNLOAD_STATUS_NOTNEED: total_size = get_pkg_own_size(self.pkg_cache, pkg_name) size_flag = PKG_SIZE_OWN elif status == DOWNLOAD_STATUS_ERROR: total_size = -1 size_flag = PKG_SIZE_ERROR else: for info in pkg_infos["info"]: total_size += info["size"] size_flag = PKG_SIZE_DOWNLOAD return [size_flag, total_size] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="ai") def clean_download_cache(self): '''Clean download cache.''' # get action packages. remain_pkgs = [] for pkg_name in self.download_manager.task_name_to_id.keys(): remain_pkgs.append(pkg_name) for pkg_name in self.apt_action_pool.install_action_dict.keys(): remain_pkgs.append(pkg_name) for pkg_name in self.apt_action_pool.upgrade_action_dict.keys(): remain_pkgs.append(pkg_name) # Get depend packages. remain_pkgs_paths = [] for pkg_name in remain_pkgs: result = get_pkg_dependence_file_path(self.pkg_cache, pkg_name) if not result: remain_pkgs_paths += result # Init clean size. packageNum = 0 cleanSize = 0 # Delete cache directory. cache_archive_dir = get_cache_archive_dir() if os.path.exists(cache_archive_dir): for root, folder, files in os.walk(cache_archive_dir): for file_name in files: path = os.path.join(root, file_name) if path.endswith(".deb") and (path not in remain_pkgs_paths): packageNum += 1 cleanSize += os.path.getsize(path) remove_file(path) return [packageNum, cleanSize] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="b", out_signature="") def say_hello(self, simulate): # Set exit_flag with False to prevent backend exit, # this just useful that frontend start again before backend exit. self.exit_flag = False self.simulate = simulate @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def request_quit(self): # Set exit flag. self.exit_flag = True self.exit_manager.check() self.update_signal([("frontend-quit", "")]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def change_source_list(self, repo_urls): ubuntu_repo_url, deepin_repo_url = repo_urls new_source_list_content = get_source_list_contents( ubuntu_repo_url, deepin_repo_url) os.system('cp %s %s.save' % (SOURCE_LIST, SOURCE_LIST)) with open(SOURCE_LIST, 'w') as fp: fp.write(new_source_list_content) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="(ss)") def RequestUpgradeStatus(self): if self.in_update_list: return ("cache-updating", json.dumps([])) elif self.is_upgrading: return ("upgrading", json.dumps([])) else: dist_upgrade_changes = self.pkg_cache.get_dist_upgrade_changes() return_infos = [] for status in dist_upgrade_changes: for pkg in dist_upgrade_changes[status]: return_infos.append( dict(status=status, name=pkg.name, version=pkg.candidate.version, size=pkg.candidate.size, downloaded=parse_pkg.pkg_file_has_exist(pkg))) return ("normal", json.dumps(return_infos)) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="a(si)", out_signature="") def RemoveWaitMissions(self, pkg_infos): self.apt_action_pool.remove_wait_missions(pkg_infos) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="s") def request_uninstall_pkgs(self): return json.dumps(self.pkg_cache.get_uninstall_pkgs()) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_install_version(self, pkg_names): pkg_versions = [] for pkg_name in pkg_names: try: version = self.pkg_cache[pkg_name].versions[0].version pkg_versions.append(version) except: try: version = self.pkg_cache[pkg_name + ":i386"].versions[0].version pkg_versions.append(version) except: self.update_signal([("pkg-not-in-cache", pkg_name)]) return pkg_versions @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="as") def is_pkg_in_cache(self, pkg_name): result = [] try: self.pkg_cache[pkg_name] result.append(pkg_name) except: try: pkg_name += ":i386" self.pkg_cache[pkg_name] result.append(pkg_name) except: pass return result @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_uninstall_version(self, pkg_names): return self.pkg_cache.get_pkgs_uninstall_version(pkg_names) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def install_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: ThreadMethod( self.add_download, (real_pkg_name, ACTION_INSTALL, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg(self, pkg_name, purge): real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.apt_action_pool.add_uninstall_action(real_pkg_name, self.simulate, purge) else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg_from_desktop(self, desktop_path, purge): ThreadMethod(self.uninstall_pkg_from_desktop_thread, (desktop_path, purge)).start() def uninstall_pkg_from_desktop_thread(self, path, purge): pkg_name = self.get_pkg_name_from_path(path) if pkg_name: self.uninstall_pkg(pkg_name, purge) else: global_event.emit("action-failed", (path, ACTION_UNINSTALL, [], "no package found for path: %s" % path)) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def get_pkg_name_from_path(self, path): pkg_name = utils.file_path_to_package_name(path) if pkg_name == "" and path.endswith(".desktop"): desktop_name = os.path.split(path)[1] path = utils.desktop_name_to_desktop_path(desktop_name) pkg_name = utils.file_path_to_package_name(path) return pkg_name @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def upgrade_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.update_signal([("ready-download-start", (real_pkg_name, ACTION_UPGRADE))]) ThreadMethod( self.add_download, (real_pkg_name, ACTION_UPGRADE, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def DistUpgrade(self): ThreadMethod(self.dist_upgrade).start() @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def stop_download_pkg(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def cancel_upgrade_download(self): if getattr(self, 'upgrade_id'): self.download_manager.stop_wait_download(self.upgrade_id) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_wait_downloads(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def request_pkg_short_desc(self, pkg_name): return self.pkg_cache.get_pkg_short_desc(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="as") def request_status(self): download_status = {ACTION_INSTALL: [], ACTION_UPGRADE: []} for info_dict in self.download_manager.download_task_info.values(): pkg_name = info_dict["task_name"] if info_dict["action_type"] == ACTION_INSTALL: download_status[ACTION_INSTALL].append( (pkg_name, info_dict["status"])) elif info_dict["action_type"] == ACTION_UPGRADE: download_status[ACTION_UPGRADE].append( (pkg_name, info_dict["status"])) action_status = { ACTION_INSTALL: [], ACTION_UPGRADE: [], ACTION_UNINSTALL: [] } for info_dict in self.apt_action_pool.install_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_INSTALL].append( (pkg_name, info_dict["status"])) for info_dict in self.apt_action_pool.upgrade_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_UPGRADE].append( (pkg_name, info_dict["status"])) for info_dict in self.apt_action_pool.uninstall_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_UNINSTALL].append( (pkg_name, info_dict["status"])) return [str(download_status), str(action_status)] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def start_update_list(self): self.del_source_list_d() self.pkg_cache.open(apb.OpProgress()) self.apt_action_pool.add_update_list_mission() @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="ab") def request_pkgs_install_status(self, pkg_names): _status = [] for pkg in pkg_names: _status.append(self.pkg_cache.is_pkg_installed(pkg)) return _status @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="(si)", out_signature="") def set_pkg_status(self, pkg_status): pkg_name, status = pkg_status self.pkg_cache.set_pkg_status(pkg_name, pkg_status) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="i") def get_pkg_installed(self, pkg_name): if self.is_pkg_in_cache(pkg_name): if self.pkg_cache.is_pkg_installed(pkg_name): return 1 else: return 0 else: return -1 @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="ss", out_signature="s") def get_pkg_start_status(self, pkg_name, start_pkg_names): is_current_installed = self.get_pkg_installed(pkg_name) if is_current_installed == -1: return "unknown" elif is_current_installed == 0: return "uninstalled" else: desktops = self.get_desktops(pkg_name) names = start_pkg_names.split(",") for name in names: if self.get_pkg_installed(name) == 1: desktops += self.get_desktops(name) return json.dumps(desktops) @dbus.service.signal(DSC_SERVICE_NAME, signature='a(sv)') # Use below command for test: # dbus-monitor --system "type='signal', interface='com.linuxdeepin.softwarecenter'" def update_signal(self, message): # The signal is emitted when this method exits # You can have code here if you wish info = str(message) logger.debug("dbus signal emit: %s" % info)
class PackageManager(dbus.service.Object): ''' docs ''' def invoked_log(func): @functools.wraps(func) def wrapper(*args, **kw): info = "dbus method invoked: %s, arguments: %s" % ( func.__name__, str(args[1:])) logger.debug(info) return func(*args, **kw) return wrapper def __init__(self, system_bus, mainloop): # Init dbus service. dbus.service.Object.__init__(self, system_bus, DSC_SERVICE_PATH) # Init. self.mainloop = mainloop self.pkg_cache = AptCache() self.exit_flag = False self.simulate = False self.all_upgrade_pkg_names = [] self.download_dir = "/var/cache/apt/archives" self.is_upgrading = False self.in_update_list = False self.apt_action_pool = AptActionPool(self.pkg_cache) self.apt_action_pool.start() global_event.register_event( 'parse-download-error', self.send_parse_download_error) global_event.register_event("action-start", self.action_start) global_event.register_event("action-update", self.action_update) global_event.register_event("action-finish", self.action_finish) global_event.register_event("action-failed", self.action_failed) global_event.register_event("download-start", self.download_start) global_event.register_event("download-update", self.download_update) global_event.register_event("download-finish", self.download_finish) global_event.register_event("download-stop", self.download_stop) global_event.register_event("download-error", self.download_failed) global_event.register_event( "update-list-start", self.update_list_start) global_event.register_event( "update-list-finish", self.update_list_finish) global_event.register_event( "update-list-failed", self.update_list_failed) global_event.register_event( "update-list-update", self.update_list_update) self.packages_status = {} self.exit_manager = ExitManager( self.mainloop, self.is_update_list_running, self.is_download_action_running, self.is_apt_action_running, self.have_exit_request) self.exit_manager.start() self.set_download_dir('/var/cache/apt/archives') self.download_manager = DownloadManager(global_event, verbose=True) def download_start(self, pkg_name, action_type): utils.set_running_lock(True) self.update_signal([("download-start", (pkg_name, action_type))]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_update(self, pkg_name, action_type, data): self.update_signal([("download-update", ( pkg_name, action_type, data))]) if action_type == ACTION_UPGRADE: self.is_upgrading = True def download_stop(self, pkg_name, action_type): self.update_signal([("download-stop", (pkg_name, action_type))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_failed(self, pkg_name, action_type, e): logger.error("%s download failed with %s" % (pkg_name, e)) utils.set_running_lock(False) self.update_signal([("download-failed", (pkg_name, action_type, e))]) self.exit_manager.check() if action_type == ACTION_UPGRADE: self.is_upgrading = False def download_finish(self, action_id, action_type, all_pkg_names): utils.set_running_lock(False) self.update_signal([("download-finish", (action_id, action_type))]) if action_type == ACTION_INSTALL: self.apt_action_pool.add_install_action(all_pkg_names) elif action_type == ACTION_UPGRADE: self.start_upgrade(all_pkg_names, action_id) self.is_upgrading = False self.exit_manager.check() def action_start(self, signal_content): utils.set_running_lock(True) self.update_signal([("action-start", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_update(self, signal_content): self.update_signal([("action-update", signal_content)]) if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = True def action_finish(self, signal_content): utils.set_running_lock(False) pkg_name, action_type, pkg_info_list = signal_content if action_type == ACTION_INSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_INSTALLED) elif action_type == ACTION_UPGRADE: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_UPGRADED) elif action_type == ACTION_UNINSTALL: for pkg_info in pkg_info_list: self.pkg_cache.set_pkg_status( pkg_name, self.pkg_cache.PKG_STATUS_UNINSTALLED) self.update_signal([("action-finish", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def action_failed(self, signal_content): utils.set_running_lock(False) self.update_signal([("action-failed", signal_content)]) self.exit_manager.check() if signal_content[1] == ACTION_UPGRADE: self.is_upgrading = False def is_update_list_running(self): return self.in_update_list def is_download_action_running(self): return len(self.download_manager.download_task_info) > 0 def is_apt_action_running(self): return len(self.apt_action_pool.active_mission_list) + len( self.apt_action_pool.wait_mission_list) > 0 def have_exit_request(self): return self.exit_flag def update_list_start(self): self.in_update_list = True self.update_signal([("update-list-start", "")]) def update_list_finish(self): self.update_signal([("update-list-merge", "")]) self.pkg_cache.open(apb.OpProgress()) db_build.BuildSoftwareDB(self.pkg_cache) self.update_signal([("update-list-finish", "")]) self.in_update_list = False self.exit_manager.check() def update_list_failed(self): self.in_update_list = False self.update_signal([("update-list-failed", "")]) self.exit_manager.check() def update_list_update(self, percent, status_message, speed_str): self.update_signal([( "update-list-update", (percent, status_message, speed_str))]) def handle_dbus_reply(self, *reply): logger.info("%s (reply): %s" % (self.module_dbus_name, str(reply))) def handle_dbus_error(self, *error): logger.info("%s (error): %s" % (self.module_dbus_name, str(error))) def send_parse_download_error(self, pkg_name, action_type): self.update_signal([("parse-download-error", (pkg_name, action_type))]) def get_unique_id(self): return str(uuid.uuid4()) def get_real_pkg_name(self, pkg_name): if pkg_name in self.pkg_cache: return pkg_name elif (pkg_name + ":i386") in self.pkg_cache: return pkg_name + ":i386" else: return None def dist_upgrade(self): action_type = ACTION_UPGRADE action_id = '%s_%s' % (self.get_unique_id(), action_type) self.update_signal([( "ready-download-start", (action_id, action_type))]) dist_upgrade_changes = self.pkg_cache.get_dist_upgrade_changes() # detect and prevent the system components from being uninstalled pkg_in_white_list = [] download_pkgs = [] self.all_upgrade_pkg_names = [] for pkg in dist_upgrade_changes["delete"]: name = pkg.name if name in SYS_PKG_WHITE_LIST: pkg_in_white_list.append(name) if len(pkg_in_white_list) > 0: self.update_signal([( "remove-pkgs-in-white-list", (json.dumps(pkg_in_white_list), action_type))]) logger.error("found remove system packages: %s" % json.dumps(pkg_in_white_list)) else: for status in dist_upgrade_changes: for pkg in dist_upgrade_changes[status]: self.all_upgrade_pkg_names.append(pkg.name) if status == "delete" or status == "keep": continue if not parse_pkg.pkg_file_has_exist(pkg): download_pkgs.append(pkg) self.update_signal([("ready-download-finish", (action_id, action_type))]) if len(download_pkgs) == 0: global_event.emit("download-finish", action_id, action_type, self.all_upgrade_pkg_names) else: names = [] download_urls = [] download_hash_infos = [] pkg_sizes = [] for pkg in download_pkgs: names.append(pkg.name) download_urls.append(pkg.candidate.uri) download_hash_infos.append(pkg.candidate.md5) pkg_sizes.append(pkg.candidate.size) self.download_manager.add_download( action_id, action_type, download_urls, download_sizes=pkg_sizes, download_md5s=download_hash_infos, all_task_names=names, all_change_pkgs=self.all_upgrade_pkg_names, file_save_dir=self.download_dir ) def add_download(self, pkg_name, action_type, simulate=False): installed = self.get_pkg_installed(pkg_name) if installed == 1: self.update_signal([("pkg-installed", (pkg_name, action_type))]) return elif installed == -1: self.update_signal([("pkg-not-in-cache", (pkg_name, action_type))]) return pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) status = pkg_infos["status"] self.update_signal([("ready-download-finish", (pkg_name, action_type))]) if status == DOWNLOAD_STATUS_NOTNEED: self.download_finish(pkg_name, action_type, [pkg_name,]) elif status == DOWNLOAD_STATUS_ERROR: self.update_signal([("parse-download-error", (pkg_name, action_type))]) else: names = [] download_urls = [] download_hash_infos = [] pkg_sizes = [] for info in pkg_infos["info"]: names.append(info["name"]) download_urls.append(info["url"]) download_hash_infos.append(info["hash"]) pkg_sizes.append(info["size"]) self.download_manager.add_download( pkg_name, action_type, download_urls, download_sizes=pkg_sizes, download_md5s=download_hash_infos, all_task_names=[pkg_name], file_save_dir=self.download_dir ) def start_upgrade(self, pkg_names, action_id): self.apt_action_pool.add_multi_upgrade_mission(pkg_names, action_id) def del_source_list_d(self): white_list_path = os.path.join(get_parent_dir(__file__), 'white_list.txt') if os.path.exists(white_list_path): with open(white_list_path) as fp: for line in fp: line = line.strip() if os.path.exists(line): os.remove(line) def get_desktops(self, pkg_name): desktops = [] try: pkg_obj = self.pkg_cache[pkg_name] for f in pkg_obj.installed_files: if f.endswith(".desktop"): desktops.append(f) except: pass return desktops @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature='', out_signature='as') def read_no_notify_config(self, no_notify_config_path): if os.path.exists(no_notify_config_path): no_notify_config_str = read_file(no_notify_config_path) try: no_notify_config = eval(no_notify_config_str) if type(no_notify_config).__name__ != "list": no_notify_config = [] except Exception: no_notify_config = [] return no_notify_config else: return [] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def add_no_notify_pkg(self, info): pkg_name, path = info no_notify_config = self.read_no_notify_config(path) if pkg_name not in no_notify_config: pkg_name = str(pkg_name) no_notify_config.append(pkg_name) write_file(path, str(no_notify_config)) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_no_notify_pkg(self, info): pkg_name, path = info pkg_name = str(pkg_name) path = str(path) no_notify_config = self.read_no_notify_config(path) if pkg_name in no_notify_config: write_file(path, str(filter( lambda config_pkg_name: config_pkg_name != pkg_name, no_notify_config))) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="i", out_signature="") def init_download_manager(self, number): pass @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="") def set_download_dir(self, local_dir): apt_pkg.config.set("Dir::Cache::Archives", local_dir) self.download_dir = local_dir @invoked_log @dbus.service.method( DSC_SERVICE_NAME, in_signature="s", out_signature="ai") def get_download_size(self, pkg_name): total_size = 0 pkg_infos = get_pkg_download_info(self.pkg_cache, pkg_name) status = pkg_infos["status"] if status == DOWNLOAD_STATUS_NOTNEED: total_size = get_pkg_own_size(self.pkg_cache, pkg_name) size_flag = PKG_SIZE_OWN elif status == DOWNLOAD_STATUS_ERROR: total_size = -1 size_flag = PKG_SIZE_ERROR else: for info in pkg_infos["info"]: total_size += info["size"] size_flag = PKG_SIZE_DOWNLOAD return [size_flag, total_size] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="ai") def clean_download_cache(self): '''Clean download cache.''' # get action packages. remain_pkgs = [] for pkg_name in self.download_manager.task_name_to_id.keys(): remain_pkgs.append(pkg_name) for pkg_name in self.apt_action_pool.install_action_dict.keys(): remain_pkgs.append(pkg_name) for pkg_name in self.apt_action_pool.upgrade_action_dict.keys(): remain_pkgs.append(pkg_name) # Get depend packages. remain_pkgs_paths = [] for pkg_name in remain_pkgs: result = get_pkg_dependence_file_path(self.pkg_cache, pkg_name) if not result: remain_pkgs_paths += result # Init clean size. packageNum = 0 cleanSize = 0 # Delete cache directory. cache_archive_dir = get_cache_archive_dir() if os.path.exists(cache_archive_dir): for root, folder, files in os.walk(cache_archive_dir): for file_name in files: path = os.path.join(root, file_name) if path.endswith(".deb") and ( path not in remain_pkgs_paths): packageNum += 1 cleanSize += os.path.getsize(path) remove_file(path) return [packageNum, cleanSize] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="b", out_signature="") def say_hello(self, simulate): # Set exit_flag with False to prevent backend exit, # this just useful that frontend start again before backend exit. self.exit_flag = False self.simulate = simulate @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def request_quit(self): # Set exit flag. self.exit_flag = True self.exit_manager.check() self.update_signal([("frontend-quit", "")]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def change_source_list(self, repo_urls): ubuntu_repo_url, deepin_repo_url = repo_urls new_source_list_content = get_source_list_contents( ubuntu_repo_url, deepin_repo_url) os.system('cp %s %s.save' % (SOURCE_LIST, SOURCE_LIST)) with open(SOURCE_LIST, 'w') as fp: fp.write(new_source_list_content) @invoked_log @dbus.service.method( DSC_SERVICE_NAME, in_signature="", out_signature="(ss)") def RequestUpgradeStatus(self): if self.in_update_list: return ("cache-updating", json.dumps([])) elif self.is_upgrading: return ("upgrading", json.dumps([])) else: dist_upgrade_changes = self.pkg_cache.get_dist_upgrade_changes() return_infos = [] for status in dist_upgrade_changes: for pkg in dist_upgrade_changes[status]: return_infos.append(dict( status=status, name=pkg.name, version=pkg.candidate.version, size=pkg.candidate.size, downloaded=parse_pkg.pkg_file_has_exist(pkg) )) return ("normal", json.dumps(return_infos)) @invoked_log @dbus.service.method( DSC_SERVICE_NAME, in_signature="a(si)", out_signature="") def RemoveWaitMissions(self, pkg_infos): self.apt_action_pool.remove_wait_missions(pkg_infos) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="s") def request_uninstall_pkgs(self): return json.dumps(self.pkg_cache.get_uninstall_pkgs()) @invoked_log @dbus.service.method( DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_install_version(self, pkg_names): pkg_versions = [] for pkg_name in pkg_names: try: version = self.pkg_cache[pkg_name].versions[0].version pkg_versions.append(version) except: try: version = self.pkg_cache[pkg_name+":i386"].versions[0].version pkg_versions.append(version) except: self.update_signal([("pkg-not-in-cache", pkg_name)]) return pkg_versions @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="as") def is_pkg_in_cache(self, pkg_name): result = [] try: self.pkg_cache[pkg_name] result.append(pkg_name) except: try: pkg_name += ":i386" self.pkg_cache[pkg_name] result.append(pkg_name) except: pass return result @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="as") def request_pkgs_uninstall_version(self, pkg_names): return self.pkg_cache.get_pkgs_uninstall_version(pkg_names) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def install_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: ThreadMethod(self.add_download, (real_pkg_name, ACTION_INSTALL, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg(self, pkg_name, purge): real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.apt_action_pool.add_uninstall_action(real_pkg_name, self.simulate, purge) else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="sb", out_signature="") def uninstall_pkg_from_desktop(self, desktop_path, purge): ThreadMethod(self.uninstall_pkg_from_desktop_thread, (desktop_path, purge)).start() def uninstall_pkg_from_desktop_thread(self, path, purge): pkg_name = self.get_pkg_name_from_path(path) if pkg_name: self.uninstall_pkg(pkg_name, purge) else: global_event.emit("action-failed", (path, ACTION_UNINSTALL, [], "no package found for path: %s" % path)) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def get_pkg_name_from_path(self, path): pkg_name = utils.file_path_to_package_name(path) if pkg_name == "" and path.endswith(".desktop"): desktop_name = os.path.split(path)[1] path = utils.desktop_name_to_desktop_path(desktop_name) pkg_name = utils.file_path_to_package_name(path) return pkg_name @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def upgrade_pkg(self, pkg_names): for pkg_name in pkg_names: real_pkg_name = self.get_real_pkg_name(pkg_name) if real_pkg_name: self.update_signal([("ready-download-start", (real_pkg_name, ACTION_UPGRADE))]) ThreadMethod(self.add_download, (real_pkg_name, ACTION_UPGRADE, self.simulate)).start() else: self.update_signal([("pkg-not-in-cache", pkg_name)]) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def DistUpgrade(self): ThreadMethod(self.dist_upgrade).start() @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def stop_download_pkg(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def cancel_upgrade_download(self): if getattr(self, 'upgrade_id'): self.download_manager.stop_wait_download(self.upgrade_id) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="") def remove_wait_downloads(self, pkg_names): for pkg_name in pkg_names: self.download_manager.stop_wait_download(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="s") def request_pkg_short_desc(self, pkg_name): return self.pkg_cache.get_pkg_short_desc(pkg_name) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="as") def request_status(self): download_status = { ACTION_INSTALL : [], ACTION_UPGRADE : []} for info_dict in self.download_manager.download_task_info.values(): pkg_name = info_dict["task_name"] if info_dict["action_type"] == ACTION_INSTALL: download_status[ACTION_INSTALL].append((pkg_name, info_dict["status"])) elif info_dict["action_type"] == ACTION_UPGRADE: download_status[ACTION_UPGRADE].append((pkg_name, info_dict["status"])) action_status = { ACTION_INSTALL : [], ACTION_UPGRADE : [], ACTION_UNINSTALL : []} for info_dict in self.apt_action_pool.install_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_INSTALL].append((pkg_name, info_dict["status"])) for info_dict in self.apt_action_pool.upgrade_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_UPGRADE].append((pkg_name, info_dict["status"])) for info_dict in self.apt_action_pool.uninstall_action_dict.values(): pkg_name = info_dict["task_name"] action_status[ACTION_UNINSTALL].append((pkg_name, info_dict["status"])) return [str(download_status), str(action_status)] @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="", out_signature="") def start_update_list(self): self.del_source_list_d() self.pkg_cache.open(apb.OpProgress()) self.apt_action_pool.add_update_list_mission() @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="as", out_signature="ab") def request_pkgs_install_status(self, pkg_names): _status = [] for pkg in pkg_names: _status.append(self.pkg_cache.is_pkg_installed(pkg)) return _status @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="(si)", out_signature="") def set_pkg_status(self, pkg_status): pkg_name, status = pkg_status self.pkg_cache.set_pkg_status(pkg_name, pkg_status) @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="s", out_signature="i") def get_pkg_installed(self, pkg_name): if self.is_pkg_in_cache(pkg_name): if self.pkg_cache.is_pkg_installed(pkg_name): return 1 else: return 0 else: return -1 @invoked_log @dbus.service.method(DSC_SERVICE_NAME, in_signature="ss", out_signature="s") def get_pkg_start_status(self, pkg_name, start_pkg_names): is_current_installed = self.get_pkg_installed(pkg_name) if is_current_installed == -1: return "unknown" elif is_current_installed == 0: return "uninstalled" else: desktops = self.get_desktops(pkg_name) names = start_pkg_names.split(",") for name in names: if self.get_pkg_installed(name) == 1: desktops += self.get_desktops(name) return json.dumps(desktops) @dbus.service.signal(DSC_SERVICE_NAME, signature='a(sv)') # Use below command for test: # dbus-monitor --system "type='signal', interface='com.linuxdeepin.softwarecenter'" def update_signal(self, message): # The signal is emitted when this method exits # You can have code here if you wish info = str(message) logger.debug("dbus signal emit: %s" % info)
if cache.is_pkg_upgradable(pkg_name): pkg.mark_upgrade() elif not cache.is_pkg_installed(pkg_name): pkg.mark_install() # Get package information. pkgs = sorted(cache.get_changes(), key=lambda pkg: pkg.name) cache._depcache.init() file_paths.append( os.path.join(cache_archive_dir, get_filename(pkg.candidate))) for pkg in pkgs: file_paths.append( os.path.join(cache_archive_dir, get_filename(pkg.candidate))) return file_paths if __name__ == "__main__": from apt_cache import AptCache cache = AptCache() # deb_package = debfile.DebPackage("/test/Download/geany_1.22+dfsg-2_amd64.deb", cache) # print deb_package.VERSION_NONE, deb_package.VERSION_OUTDATED, deb_package.VERSION_SAME, deb_package.VERSION_NEWER # print deb_package.compare_to_version_in_cache() # print deb_package.check() # print deb_package.check_breaks_existing_packages() # print deb_package.check_conflicts() for path in get_pkg_dependence_file_path(cache, "kingsoft-office"): print path