Fix Github issue #6 (uncaught exception)

ericpignet · May 12, 2020 · 4c91d82 · 4c91d82
1 parent bdca16e
commit 4c91d82
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/custom_components/tplink_router/device_tracker.py b/custom_components/tplink_router/device_tracker.py
@@ -497,8 +497,11 @@ def _get_pub_key(self):
  ee = self._get_field_from_router_response(response.text, 'ee')
  nn = self._get_field_from_router_response(response.text, 'nn')
 
- e = int(ee, 16)
- n = int(nn, 16) #snipped for brevity
+ try:
+ e = int(ee, 16)
+ n = int(nn, 16) #snipped for brevity
+ except ValueError:
+ return False
 
  pubkey = construct((n, e))
  self.pubkey = PKCS1_v1_5.new(pubkey)
@@ -568,8 +571,8 @@ def _get_token(self):
 
  def _get_auth_tokens(self):
  """Retrieve auth tokens from the router."""
- _LOGGER.info("Retrieving PublicKey...")
 
+ _LOGGER.info("Retrieving PublicKey...")
  pubkey = self._get_pub_key()
  if not pubkey:
  return False