From e028c80a06a711c4397ce36c63b38f55ce1d867b Mon Sep 17 00:00:00 2001 From: Daniel Kahn Gillmor Date: Tue, 9 May 2023 21:59:09 -0400 Subject: [PATCH] Simplify FlagList sdproperty registration --- pgpy/packet/subpackets/signature.py | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/pgpy/packet/subpackets/signature.py b/pgpy/packet/subpackets/signature.py index c3c6d6c2..d1a74be3 100644 --- a/pgpy/packet/subpackets/signature.py +++ b/pgpy/packet/subpackets/signature.py @@ -100,17 +100,12 @@ class FlagList(Signature): def flags(self): return self._flags - @flags.register(list) - @flags.register(tuple) - def flags_list(self, val): + @flags.register + def flags_list(self, val: Union[list, tuple]): self._flags = list(val) - @flags.register(int) - @flags.register(CompressionAlgorithm) - @flags.register(HashAlgorithm) - @flags.register(PubKeyAlgorithm) - @flags.register(SymmetricKeyAlgorithm) - def flags_int(self, val): + @flags.register + def flags_int(self, val: int): if self.__flags__ is None: # pragma: no cover raise AttributeError("Error: __flags__ not set!")