feat(python): update collections import for 3.10

mailgun · Feb 21, 2023 · 590f9d9 · 590f9d9
1 parent 2bb021e
commit 590f9d9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git 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
@@ -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.
     """