add null check

Buxdehuda · Feb 5, 2024 · 5710e2e · 5710e2e
1 parent de5c51f
commit 5710e2e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/certbotstratoapi.py b/certbotstratoapi.py
@@ -154,12 +154,13 @@ def get_package_id(self) -> None:
                 return
             else:
                 # Old page layout: still relevant?
-                customer_link = package_element.find_next('a', class_='customer-link')['href']
-                query_parameters = urllib.parse.parse_qs(customer_link)
-                if 'cID' in query_parameters:
-                    self.package_id = query_parameters['cID'][0]
-                    print(f'INFO: strato package id (cID): {self.package_id}')
-                    return
+                customer_link = package_element.find_next('a', class_='customer-link')
+                if customer_link is not None and customer_link.has_attr('href'):
+                    query_parameters = urllib.parse.parse_qs(customer_link['href'])
+                    if 'cID' in query_parameters:
+                        self.package_id = query_parameters['cID'][0]
+                        print(f'INFO: strato package id (cID): {self.package_id}')
+                        return
 
         print(f'ERROR: Domain {self.second_level_domain_name} not '
             'found in strato packages. Using fallback cID=1')