PA-757 - added function to get LE cert for new-style web app. by: Glenn, Filip, Nina

Author: Glenn Jones

pythonanywhere_core/website.py

@@ -1,6 +1,8 @@
 import getpass
 
 from pythonanywhere_core.base import call_api, get_api_endpoint
+from pythonanywhere_core.exceptions import PythonAnywhereApiException
+
 
 
 class Website:
@@ -70,6 +72,26 @@ def reload(self, domain_name: str) -> dict:
         )
         return response.json()
 
+    def auto_ssl(self, domain_name: str) -> dict:
+        """Creates and applies a Let's Encrypt certificate for ``domain_name``.
+        :param domain_name: domain name for website to apply the certificate to
+        :return: dictionary with response"""
+        response = call_api(
+            f"{self.api_endpoint}{domain_name}/ssl/",
+            "post",
+            json={"cert_type": "letsencrypt-auto-renew"}
+        )
+        return response.json()
+
+    def get_ssl_info(self, domain_name) -> dict:
+        """Get SSL certificate info"""
+        url = f"{self.api_endpoint}{domain_name}/ssl/"
+        response = call_api(url, "get")
+        if not response.ok:
+            raise PythonAnywhereApiException(f"GET SSL details via API failed, got {response}:{response.text}")
+
+        return response.json()
+
     def delete(self, domain_name: str) -> dict:
         """Deletes website with ``domain_name``.
         :param domain_name: domain name for website to delete

tests/test_website.py

@@ -5,6 +5,7 @@
 import responses
 
 from pythonanywhere_core.base import get_api_endpoint
+from pythonanywhere_core.exceptions import PythonAnywhereApiException
 from pythonanywhere_core.website import Website
 
 
@@ -112,3 +113,44 @@ def test_deletes_website(api_responses, domain_name, webapps_base_url):
     )
 
     assert Website().delete(domain_name=domain_name) == {}
+
+
+def test_sets_lets_encrypt_cert(api_responses, domain_name, webapps_base_url):
+    api_responses.add(
+        responses.POST,
+        url=f"{webapps_base_url}{domain_name}/ssl/",
+        body=json.dumps({"status": "OK"}),
+        status=200
+    )
+
+
+    assert Website().auto_ssl(domain_name=domain_name) == {"status": "OK"}
+
+    assert json.loads(api_responses.calls[0].request.body.decode()) == {
+        "cert_type": "letsencrypt-auto-renew"
+    }
+
+
+def test_returns_ssl_info(api_responses, domain_name, webapps_base_url):
+    api_responses.add(
+        responses.GET,
+        url=f"{webapps_base_url}{domain_name}/ssl/",
+        body=json.dumps({"status": "OK"}),
+        status=200
+    )
+
+    assert Website().get_ssl_info(domain_name=domain_name) == {"status": "OK"}
+
+
+def test_raises_if_ssl_info_does_not_return_200(api_responses, domain_name, webapps_base_url):
+    api_responses.add(
+        responses.GET,
+        url=f"{webapps_base_url}{domain_name}/ssl/",
+        status=404, body="nope"
+    )
+
+    with pytest.raises(PythonAnywhereApiException) as e:
+        Website().get_ssl_info(domain_name=domain_name)
+
+    assert "GET SSL details via API failed, got" in str(e.value)
+    assert "nope" in str(e.value)