cecil 0.0.18__tar.gz → 0.0.20__tar.gz

{cecil-0.0.18 → cecil-0.0.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cecil
-Version: 0.0.18
+Version: 0.0.20
 Summary: Python SDK for Cecil Earth
 License-Expression: MIT
 License-File: LICENSE.txt

{cecil-0.0.18 → cecil-0.0.20}/src/cecil/client.py

@@ -17,10 +17,12 @@ from .errors import (
 )
 from .models import (
     AOI,
+    AOIRecord,
     AOICreate,
     DataRequest,
     DataRequestCreate,
     OrganisationCreate,
+    OrganisationSettings,
     RecoverAPIKey,
     RecoverAPIKeyRequest,
     RotateAPIKey,
@@ -53,9 +55,9 @@ class Client:
         res = self._get(url=f"/v0/aois/{id}")
         return AOI(**res)
 
-    def list_aois(self) -> List[AOI]:
+    def list_aois(self) -> List[AOIRecord]:
         res = self._get(url="/v0/aois")
-        return [AOI(**record) for record in res["records"]]
+        return [AOIRecord(**record) for record in res["records"]]
 
     def create_data_request(self, aoi_id: str, dataset_id: str) -> DataRequest:
         res = self._post(
@@ -160,6 +162,23 @@ class Client:
         res = self._get(url="/v0/users")
         return [User(**record) for record in res["records"]]
 
+    def get_organisation_settings(self) -> OrganisationSettings:
+        res = self._get(url="/v0/organisation/settings")
+        return OrganisationSettings(**res)
+
+    def update_organisation_settings(
+        self,
+        *,
+        monthly_data_request_limit,
+    ) -> OrganisationSettings:
+        res = self._post(
+            url="/v0/organisation/settings",
+            model=OrganisationSettings(
+                monthly_data_request_limit=monthly_data_request_limit,
+            ),
+        )
+        return OrganisationSettings(**res)
+
     def _request(self, method: str, url: str, skip_auth=False, **kwargs) -> Dict:
 
         if skip_auth is False:

{cecil-0.0.18 → cecil-0.0.20}/src/cecil/models.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Dict
+from typing import Dict, Optional
 
 from pydantic import BaseModel, ConfigDict, SecretStr
 from pydantic.alias_generators import to_camel
@@ -15,6 +15,15 @@ class AOI(BaseModel):
     created_by: str
 
 
+class AOIRecord(BaseModel):
+    model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
+    id: str
+    name: str
+    hectares: float
+    created_at: datetime.datetime
+    created_by: str
+
+
 class AOICreate(BaseModel):
     model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
     name: str
@@ -41,6 +50,11 @@ class OrganisationCreate(BaseModel):
     name: str
 
 
+class OrganisationSettings(BaseModel):
+    model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
+    monthly_data_request_limit: Optional[int] = None
+
+
 class RecoverAPIKey(BaseModel):
     model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
     message: str

cecil-0.0.20/src/cecil/version.py

@@ -0,0 +1 @@
+__version__ = "0.0.20"

cecil-0.0.18/src/cecil/version.py

@@ -1 +0,0 @@
-__version__ = "0.0.18"