first pass

client.py

@@ -0,0 +1,86 @@
+from typing import List
+import requests
+
+from model import *
+
+
+class NeonAPI:
+    def __init__(
+        self, api_key: str, *, base_url: str = "https://console.neon.tech/api/v2/"
+    ):
+        self.api_key = api_key
+        self.base_url = base_url
+        self.session = requests.Session()
+
+    def _request(self, method: str, path: str, **kwargs):
+        # Set HTTP headers for outgoing requests.
+        headers = kwargs.pop("headers", {})
+        headers["Authorization"] = f"Bearer {self.api_key}"
+        headers["Accept"] = "application/json"
+        headers["Content-Type"] = "application/json"
+
+        return self.session.request(
+            method, self.base_url + path, headers=headers, **kwargs
+        )
+
+
+class NeonClient(NeonAPI):
+    def __init__(self, api_key: str, **kwargs):
+        super().__init__(api_key, **kwargs)
+        # self.api_keys = APIKeyResource(self)
+
+    def me(self) -> CurrentUserInfoResponse:
+        """Get information about the user."""
+
+        r = self._request("GET", "users/me")
+        r.raise_for_status()
+
+        return CurrentUserInfoResponse(**r.json())
+
+    def get_api_keys(self) -> List[ApiKeysListResponseItem]:
+        """Get a list of API keys."""
+
+        r = self._request("GET", "api_keys")
+        r.raise_for_status()
+
+        return [ApiKeysListResponseItem(**item) for item in r.json()]
+
+    def create_api_key(self, name: str) -> ApiKeyCreateResponse:
+        """Create a new API key."""
+
+        r = self._request("POST", "api_keys", json={"key_name": name})
+        r.raise_for_status()
+
+        return ApiKeyCreateResponse(**r.json())
+
+    def delete_api_key(self, key_id: str) -> ApiKeyRevokeResponse:
+        """Delete an API key."""
+
+        r = self._request("DELETE", f"api_keys/{key_id}")
+        r.raise_for_status()
+
+        return ApiKeyRevokeResponse(**r.json())
+
+    def get_projects(self) -> List[ProjectListItem]:
+        """Get a list of projects."""
+
+        r = self._request("GET", "projects")
+        r.raise_for_status()
+
+        return [ProjectListItem(**item) for item in r.json()["projects"]]
+
+    def create_project(self, project_id: str) -> ProjectCreateRequest:
+        """Create a new project."""
+
+        r = self._request("POST", "projects", json={"project": {"name": project_id}})
+        r.raise_for_status()
+
+        return ProjectCreateRequest(**r.json())
+
+    def delete_project(self, project_id: str) -> ProjectUpdateRequest:
+        """Delete a project."""
+
+        r = self._request("DELETE", f"projects/{project_id}")
+        r.raise_for_status()
+
+        return ProjectUpdateRequest(**r.json())