From 09bea010bc405e846219a6254af24e99e7733ad5 Mon Sep 17 00:00:00 2001 From: arenekosreal <17194552+arenekosreal@users.noreply.github.com> Date: Fri, 20 Sep 2024 22:19:50 +0800 Subject: [PATCH] Make ruff happy --- src/crx_repo/client.py | 40 +++++++++++++++++++++++++--------------- 1 file changed, 25 insertions(+), 15 deletions(-) diff --git a/src/crx_repo/client.py b/src/crx_repo/client.py index f3b4fa0..5385f44 100644 --- a/src/crx_repo/client.py +++ b/src/crx_repo/client.py @@ -105,21 +105,31 @@ async def _do_download(self): _logger.error("Failed to build because async operation timeout.") _logger.debug("Checking checksums of extension %s...", self.extension_id) sha256_hash = hash_calculator.hexdigest() - if update.sha256 is not None and sha256_hash != update.sha256: - _logger.error( - "SHA256 checksum of %s mismatch. Removing file.", - self.extension_id, - ) - extension_path.unlink() - else: - if update.sha256 is None: - _logger.warning("No sha256 is provied for this extension file, still keeping file.") - else: - _logger.info( - "SHA256 checksum of %s match. Keeping file.", - self.extension_id, - ) - _ = extension_path.rename(extension_path.parent / extension_path.stem) + self._after_extension_downloaded(update, sha256_hash, extension_path) + + def _after_extension_downloaded( + self, + update: UpdateInfo, + sha256_hash: str, + extension_path: Path, + ): + if update.sha256 is not None and sha256_hash != update.sha256: + _logger.error( + "SHA256 checksum of %s mismatch. Removing file.", + self.extension_id, + ) + extension_path.unlink() + else: + if update.sha256 is None: + _logger.warning( + "No sha256 is provied for this extension file, still keeping file." + ) + else: + _logger.info( + "SHA256 checksum of %s match. Keeping file.", + self.extension_id, + ) + _ = extension_path.rename(extension_path.parent / extension_path.stem) async def _check_update( self,