diff --git a/tna_utilities/url.py b/tna_utilities/url.py index beb5c28..3ab8954 100644 --- a/tna_utilities/url.py +++ b/tna_utilities/url.py @@ -38,13 +38,13 @@ def parameter_exists(self, parameter) -> bool: def parameter_values(self, parameter: str) -> list: """ Get the values associated with a parameter in the query parameters. - Raises an AttributeError if the parameter does not exist. + Raises a KeyError if the parameter does not exist. """ for key, values in self.args: if key == parameter: return values - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def add_parameter( self, parameter: str, values: str | int | list | None = None @@ -87,7 +87,7 @@ def remove_parameter(self, parameter: str) -> "QueryStringTransformer": if key == parameter: self.args.remove((key, vals)) return self - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def is_value_in_parameter(self, parameter: str, value: str | int) -> bool: """ @@ -98,7 +98,7 @@ def is_value_in_parameter(self, parameter: str, value: str | int) -> bool: for key, values in self.args: if key == parameter: return str(value) in values - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def add_parameter_value( self, parameter: str, value: str | int @@ -113,7 +113,7 @@ def add_parameter_value( if str(value) not in values: values.append(str(value)) return self - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def toggle_parameter_value( self, parameter: str, value: str | int @@ -131,7 +131,7 @@ def toggle_parameter_value( else: self.add_parameter_value(parameter, value) return self - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def remove_parameter_value( self, parameter: str, value: str | int @@ -146,7 +146,7 @@ def remove_parameter_value( if str(value) in values: values.remove(str(value)) return self - raise AttributeError(f"Parameter '{parameter}' does not exist") + raise KeyError(f"Parameter '{parameter}' does not exist") def get_query_string(self) -> str: """