Run black on the codebase.

Author: UmanShahzad

ipinfo/cache/default.py

@@ -3,6 +3,7 @@
 """
 
 import cachetools
+
 from .interface import CacheInterface
 
 

ipinfo/details.py

@@ -2,6 +2,7 @@
 Details returned by the IPinfo service.
 """
 
+
 class Details:
     """Encapsulates data for single IP address."""
 
@@ -14,7 +15,7 @@ def __getattr__(self, attr):
         if attr in self.details:
             return self.details[attr]
         else:
-            raise AttributeError('{} is not a valid attribute of Details'.format(attr))
+            raise AttributeError("{} is not a valid attribute of Details".format(attr))
 
     @property
     def all(self):

ipinfo/exceptions.py

@@ -2,6 +2,8 @@
 Exceptions thrown by the IPinfo service.
 """
 
+
 class RequestQuotaExceededError(Exception):
     """Error indicating that users monthly request quota has been passed."""
+
     pass

ipinfo/handler.py

@@ -5,8 +5,10 @@
 import ipaddress
 import json
 import os
-import requests
 import sys
+
+import requests
+
 from .cache.default import DefaultCache
 from .details import Details
 from .exceptions import RequestQuotaExceededError
@@ -18,45 +20,49 @@ class Handler:
     and maintains access to cache.
     """
 
-    API_URL = 'https://ipinfo.io'
+    API_URL = "https://ipinfo.io"
     CACHE_MAXSIZE = 4096
     CACHE_TTL = 60 * 60 * 24
-    COUNTRY_FILE_DEFAULT = 'countries.json'
+    COUNTRY_FILE_DEFAULT = "countries.json"
     REQUEST_TIMEOUT_DEFAULT = 2
 
     def __init__(self, access_token=None, **kwargs):
         """Initialize the Handler object with country name list and the cache initialized."""
         self.access_token = access_token
-        self.countries = self._read_country_names(kwargs.get('countries_file'))
+        self.countries = self._read_country_names(kwargs.get("countries_file"))
 
-        self.request_options = kwargs.get('request_options', {})
-        if 'timeout' not in self.request_options:
-            self.request_options['timeout'] = self.REQUEST_TIMEOUT_DEFAULT
+        self.request_options = kwargs.get("request_options", {})
+        if "timeout" not in self.request_options:
+            self.request_options["timeout"] = self.REQUEST_TIMEOUT_DEFAULT
 
-        if 'cache' in kwargs:
-            self.cache = kwargs['cache']
+        if "cache" in kwargs:
+            self.cache = kwargs["cache"]
         else:
-            cache_options = kwargs.get('cache_options', {})
-            maxsize = cache_options.get('maxsize', self.CACHE_MAXSIZE)
-            ttl = cache_options.get('ttl', self.CACHE_TTL)
+            cache_options = kwargs.get("cache_options", {})
+            maxsize = cache_options.get("maxsize", self.CACHE_MAXSIZE)
+            ttl = cache_options.get("ttl", self.CACHE_TTL)
             self.cache = DefaultCache(maxsize, ttl, **cache_options)
 
     def getDetails(self, ip_address=None):
         """Get details for specified IP address as a Details object."""
         raw_details = self._requestDetails(ip_address)
-        raw_details['country_name'] = self.countries.get(raw_details.get('country'))
-        raw_details['ip_address'] = ipaddress.ip_address(raw_details.get('ip'))
-        raw_details['latitude'], raw_details['longitude'] = self._read_coords(raw_details.get('loc'))
+        raw_details["country_name"] = self.countries.get(raw_details.get("country"))
+        raw_details["ip_address"] = ipaddress.ip_address(raw_details.get("ip"))
+        raw_details["latitude"], raw_details["longitude"] = self._read_coords(
+            raw_details.get("loc")
+        )
         return Details(raw_details)
 
     def _requestDetails(self, ip_address=None):
         """Get IP address data by sending request to IPinfo API."""
         if ip_address not in self.cache:
             url = self.API_URL
             if ip_address:
-                url += '/' + ip_address
+                url += "/" + ip_address
 
-            response = requests.get(url, headers=self._get_headers(), **self.request_options)
+            response = requests.get(
+                url, headers=self._get_headers(), **self.request_options
+            )
             if response.status_code == 429:
                 raise RequestQuotaExceededError()
             response.raise_for_status()
@@ -67,26 +73,30 @@ def _requestDetails(self, ip_address=None):
     def _get_headers(self):
         """Built headers for request to IPinfo API."""
         headers = {
-            'user-agent': 'IPinfoClient/Python{version}/1.0'.format(version=sys.version_info[0]),
-            'accept': 'application/json'
+            "user-agent": "IPinfoClient/Python{version}/1.0".format(
+                version=sys.version_info[0]
+            ),
+            "accept": "application/json",
         }
 
         if self.access_token:
-            headers['authorization'] = 'Bearer {}'.format(self.access_token)
+            headers["authorization"] = "Bearer {}".format(self.access_token)
 
         return headers
 
     def _read_coords(self, location):
         lat, lon = None, None
-        coords = tuple(location.split(',')) if location else ''
+        coords = tuple(location.split(",")) if location else ""
         if len(coords) == 2 and coords[0] and coords[1]:
             lat, lon = coords[0], coords[1]
         return lat, lon
 
     def _read_country_names(self, countries_file=None):
         """Read list of countries from specified country file or default file."""
         if not countries_file:
-            countries_file = os.path.join(os.path.dirname(__file__), self.COUNTRY_FILE_DEFAULT)
+            countries_file = os.path.join(
+                os.path.dirname(__file__), self.COUNTRY_FILE_DEFAULT
+            )
         with open(countries_file) as f:
             countries_json = f.read()
 

tests/details_test.py

@@ -1,4 +1,5 @@
 import pytest
+
 from ipinfo.details import Details
 
 

tests/handler_test.py

@@ -1,7 +1,8 @@
+import ipaddress
+
 from ipinfo.cache.default import DefaultCache
 from ipinfo.details import Details
 from ipinfo.handler import Handler
-import ipaddress
 
 
 def test_init():