diff --git a/flanker/addresslib/drivers/dns_lookup.py b/flanker/addresslib/drivers/dns_lookup.py index b362508..349e97f 100644 --- a/flanker/addresslib/drivers/dns_lookup.py +++ b/flanker/addresslib/drivers/dns_lookup.py @@ -1,8 +1,12 @@ -import collections import dnsq +import sys +if sys.version_info.major == 3 and sys.version_info.minor >= 10: + from collections.abc import MutableMapping +else: + from collections import MutableMapping -class DNSLookup(collections.MutableMapping): +class DNSLookup(MutableMapping): """ DNSLookup has the same interface as a dict, but talks to a DNS server """ diff --git a/flanker/addresslib/drivers/redis_driver.py b/flanker/addresslib/drivers/redis_driver.py index 773b1c3..16b68d5 100644 --- a/flanker/addresslib/drivers/redis_driver.py +++ b/flanker/addresslib/drivers/redis_driver.py @@ -1,9 +1,13 @@ -import collections import os import redis +import sys +if sys.version_info.major == 3 and sys.version_info.minor >= 10: + from collections.abc import MutableMapping +else: + from collections import MutableMapping -class RedisCache(collections.MutableMapping): +class RedisCache(MutableMapping): """ RedisCache has the same interface as a dict, but talks to a redis server. """