From 23f36252635115bfb23c4610526cd0352fb9ab4c Mon Sep 17 00:00:00 2001 From: Wolf Vollprecht Date: Tue, 6 Aug 2024 11:27:02 +0200 Subject: [PATCH] fix type checkign --- src/rattler_build_conda_compat/modify_recipe.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/rattler_build_conda_compat/modify_recipe.py b/src/rattler_build_conda_compat/modify_recipe.py index e8aed5a..1874aa7 100644 --- a/src/rattler_build_conda_compat/modify_recipe.py +++ b/src/rattler_build_conda_compat/modify_recipe.py @@ -5,7 +5,7 @@ import io import logging import re -from typing import TYPE_CHECKING, Any, Literal +from typing import TYPE_CHECKING, Any, Literal, Set import requests from ruamel.yaml import YAML @@ -18,6 +18,8 @@ logger = logging.getLogger(__name__) +HashType = Literal["md5", "sha256"] + yaml = YAML() yaml.preserve_quotes = True yaml.width = 4096 @@ -81,7 +83,7 @@ def __init__(self, message: str = "Could not update version") -> None: class Hash: - def __init__(self, hash_type: Literal["md5", "sha256"], hash_value: str) -> None: + def __init__(self, hash_type: HashType, hash_value: str) -> None: self.hash_type = hash_type self.hash_value = hash_value @@ -105,9 +107,9 @@ def update_hash(source: Source, url: str, hash_: Hash | None) -> None: * `url` - The URL to download and hash (if no hash is provided). * `hash_` - The hash to use. If not provided, the file will be downloaded and `sha256` hashed. """ - hash_type = hash_.hash_type if hash_ is not None else "sha256" + hash_type : HashType = hash_.hash_type if hash_ is not None else "sha256" # delete all old hashes that we are not updating - all_hash_types = {"md5", "sha256"} + all_hash_types: Set[HashType] = {"md5", "sha256"} for key in all_hash_types - {hash_type}: if key in source: del source[key]