From: Bob Halley Date: Wed, 20 May 2020 14:18:56 +0000 (-0700) Subject: flake8 linting X-Git-Tag: v2.0.0rc1~181 X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4587be94f22099a195904c30bfff51fe53c1c87e;p=thirdparty%2Fdnspython.git flake8 linting --- diff --git a/dns/dnssec.py b/dns/dnssec.py index dee290d0..f775086f 100644 --- a/dns/dnssec.py +++ b/dns/dnssec.py @@ -105,7 +105,7 @@ def key_id(key): """ rdata = _to_rdata(key, None) - if key.algorithm == RSAMD5: + if key.algorithm == Algorithm.RSAMD5: return (rdata[-3] << 8) + rdata[-2] else: total = 0 @@ -196,46 +196,46 @@ def _find_candidate_keys(keys, rrsig): def _is_rsa(algorithm): - return algorithm in (RSAMD5, RSASHA1, - RSASHA1NSEC3SHA1, RSASHA256, - RSASHA512) + return algorithm in (Algorithm.RSAMD5, Algorithm.RSASHA1, + Algorithm.RSASHA1NSEC3SHA1, Algorithm.RSASHA256, + Algorithm.RSASHA512) def _is_dsa(algorithm): - return algorithm in (DSA, DSANSEC3SHA1) + return algorithm in (Algorithm.DSA, Algorithm.DSANSEC3SHA1) def _is_ecdsa(algorithm): - return algorithm in (ECDSAP256SHA256, ECDSAP384SHA384) + return algorithm in (Algorithm.ECDSAP256SHA256, Algorithm.ECDSAP384SHA384) def _is_eddsa(algorithm): - return algorithm in (ED25519, ED448) + return algorithm in (Algorithm.ED25519, Algorithm.ED448) def _is_gost(algorithm): - return algorithm == ECCGOST + return algorithm == Algorithm.ECCGOST def _is_md5(algorithm): - return algorithm == RSAMD5 + return algorithm == Algorithm.RSAMD5 def _is_sha1(algorithm): - return algorithm in (DSA, RSASHA1, - DSANSEC3SHA1, RSASHA1NSEC3SHA1) + return algorithm in (Algorithm.DSA, Algorithm.RSASHA1, + Algorithm.DSANSEC3SHA1, Algorithm.RSASHA1NSEC3SHA1) def _is_sha256(algorithm): - return algorithm in (RSASHA256, ECDSAP256SHA256) + return algorithm in (Algorithm.RSASHA256, Algorithm.ECDSAP256SHA256) def _is_sha384(algorithm): - return algorithm == ECDSAP384SHA384 + return algorithm == Algorithm.ECDSAP384SHA384 def _is_sha512(algorithm): - return algorithm == RSASHA512 + return algorithm == Algorithm.RSASHA512 def _make_hash(algorithm): @@ -249,9 +249,9 @@ def _make_hash(algorithm): return hashes.SHA384() if _is_sha512(algorithm): return hashes.SHA512() - if algorithm == ED25519: + if algorithm == Algorithm.ED25519: return hashes.SHA512() - if algorithm == ED448: + if algorithm == Algorithm.ED448: return hashes.SHAKE256(114) raise ValidationFailure('unknown hash for algorithm %u' % algorithm) @@ -358,7 +358,7 @@ def _validate_rrsig(rrset, rrsig, keys, origin=None, now=None): _bytes_to_long(sig_s)) elif _is_ecdsa(rrsig.algorithm): keyptr = candidate_key.key - if rrsig.algorithm == ECDSAP256SHA256: + if rrsig.algorithm == Algorithm.ECDSAP256SHA256: curve = ec.SECP256R1() octets = 32 else: @@ -380,7 +380,7 @@ def _validate_rrsig(rrset, rrsig, keys, origin=None, now=None): elif _is_eddsa(rrsig.algorithm): keyptr = candidate_key.key - if rrsig.algorithm == ED25519: + if rrsig.algorithm == Algorithm.ED25519: loader = ed25519.Ed25519PublicKey else: loader = ed448.Ed448PublicKey diff --git a/dns/opcode.py b/dns/opcode.py index 509916e9..5a76326a 100644 --- a/dns/opcode.py +++ b/dns/opcode.py @@ -104,4 +104,4 @@ def is_update(flags): Returns a ``bool``. """ - return from_flags(flags) == UPDATE + return from_flags(flags) == Opcode.UPDATE diff --git a/dns/rdataclass.py b/dns/rdataclass.py index e7c957be..46744eab 100644 --- a/dns/rdataclass.py +++ b/dns/rdataclass.py @@ -50,7 +50,7 @@ class RdataClass(dns.enum.IntEnum): globals().update(RdataClass.__members__) -_metaclasses = {NONE, ANY} +_metaclasses = {RdataClass.NONE, RdataClass.ANY} class UnknownRdataclass(dns.exception.DNSException): diff --git a/dns/rdatatype.py b/dns/rdatatype.py index ba2e2615..0a669ebc 100644 --- a/dns/rdatatype.py +++ b/dns/rdatatype.py @@ -116,9 +116,10 @@ _registered_by_value = {} globals().update(RdataType.__members__) -_metatypes = {OPT} +_metatypes = {RdataType.OPT} -_singletons = {SOA, NXT, DNAME, NSEC, CNAME} +_singletons = {RdataType.SOA, RdataType.NXT, RdataType.DNAME, + RdataType.NSEC, RdataType.CNAME} class UnknownRdatatype(dns.exception.DNSException): diff --git a/dns/rdtypes/ANY/CDNSKEY.py b/dns/rdtypes/ANY/CDNSKEY.py index aae8bd77..72253183 100644 --- a/dns/rdtypes/ANY/CDNSKEY.py +++ b/dns/rdtypes/ANY/CDNSKEY.py @@ -16,7 +16,7 @@ # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. import dns.rdtypes.dnskeybase -from dns.rdtypes.dnskeybase import SEP, REVOKE, ZONE +from dns.rdtypes.dnskeybase import SEP, REVOKE, ZONE # noqa: F401 class CDNSKEY(dns.rdtypes.dnskeybase.DNSKEYBase): diff --git a/dns/rdtypes/ANY/DNSKEY.py b/dns/rdtypes/ANY/DNSKEY.py index f5c918b0..2ee37988 100644 --- a/dns/rdtypes/ANY/DNSKEY.py +++ b/dns/rdtypes/ANY/DNSKEY.py @@ -16,7 +16,7 @@ # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. import dns.rdtypes.dnskeybase -from dns.rdtypes.dnskeybase import SEP, REVOKE, ZONE +from dns.rdtypes.dnskeybase import SEP, REVOKE, ZONE # noqa: F401 class DNSKEY(dns.rdtypes.dnskeybase.DNSKEYBase): diff --git a/dns/rdtypes/dnskeybase.py b/dns/rdtypes/dnskeybase.py index 3b35a0c6..1099e394 100644 --- a/dns/rdtypes/dnskeybase.py +++ b/dns/rdtypes/dnskeybase.py @@ -24,7 +24,7 @@ import dns.dnssec import dns.rdata # wildcard import -__all__ = ["SEP", "REVOKE", "ZONE"] +__all__ = ["SEP", "REVOKE", "ZONE"] # noqa: F822 class Flag(enum.IntFlag): SEP = 0x0001