feat: adds check_domain_availability endpoint (#17)

Author: becelot

hostingde/domain/domain.py

@@ -1,5 +1,6 @@
-from typing import Optional, List
+from typing import Optional, List, Union
 
+from hostingde.domain.requests.check_availability import CheckAvailabilityRequest, CheckAvailabilityResponse
 from hostingde.domain.requests.register_domain import RegisterDomainRequest
 from hostingde.paginator import HostingDePaginator
 from hostingde.hostingde import HostingDeCore
@@ -49,6 +50,21 @@ def jobs_find(
 
         return self._iter(uri, Job, filter, limit, sort)
 
+    def check_domain_name_availability(self, domain_names: Union[str, List[str]]) -> List[CheckAvailabilityResponse]:
+        uri = self.build_uri('domainStatus')
+
+        if type(domain_names) == str:
+            domain_names = [domain_names]
+
+        response = self._request(
+            uri,
+            CheckAvailabilityRequest(domain_names=domain_names)
+        )
+
+        data = response.json().get('responses', [])
+
+        return list(map(lambda x: self._instance(CheckAvailabilityResponse, x), data))
+
     def list_domains(
         self,
         limit: Optional[int] = None,
@@ -129,4 +145,4 @@ def register_domain(
         if not asynchronous and domain.id is not None:
             JobWaiter(self, domain.id, 'domainCreate').wait()
 
-        return domain
+        return domain

hostingde/domain/requests/check_availability.py

@@ -0,0 +1,35 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import List, Optional
+
+from marshmallow_enum import EnumField
+
+from hostingde.model import Model
+
+
+class CheckAvailabilityStatus(Enum):
+    alreadyRegistered = 'alreadyRegistered'
+    registered = 'registered'
+    nameContainsForbiddenCharacter = 'nameContainsForbiddenCharacter'
+    available = 'available'
+    suffixDoesNotExist = 'suffixDoesNotExist'
+    suffixCannotBeRegistered = 'suffixCannotBeRegistered'
+    canNotCheck = 'canNotCheck'
+    unknown = 'unknown'
+
+
+@dataclass
+class CheckAvailabilityRequest(Model):
+    domain_names: List[str]
+
+
+@dataclass
+class CheckAvailabilityResponse(Model):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+    domain_name: str
+    domain_name_unicode: str
+    domain_suffix: str
+    transfer_method: Optional[str]
+    status: CheckAvailabilityStatus = EnumField(CheckAvailabilityStatus)