feat: include enw types

VorTECHsa · Sep 5, 2023 · e0ee5ef · e0ee5ef
1 parent 73f100a
commit e0ee5ef
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 2 deletions.
diff --git a/vortexasdk/api/shared_types.py b/vortexasdk/api/shared_types.py
@@ -160,3 +160,9 @@ class Scrubber(BaseModel):
     tag: Optional[str] = None
     scrubber: Optional[str] = None
     planned: Optional[bool] = None
+
+
+class VesselClassEntry:
+    id: Optional[str] = None
+    layer: Optional[str] = None
+    label: Optional[str] = None
diff --git a/vortexasdk/api/vessel.py b/vortexasdk/api/vessel.py
@@ -10,6 +10,7 @@
     Tag,
     Scrubber,
     Flag,
+    VesselClassEntry,
 )
 
 
@@ -46,6 +47,7 @@ class Vessel(Node):
     current_product_type: Optional[List] = None
 
     vessel_class: Optional[str] = None
+    classes: Optional[List[VesselClassEntry]] = None
     vessel_status: Optional[str] = None
 
     corporate_entities: Optional[List[VesselEntityCorporateEntity]] = None
@@ -81,6 +83,7 @@ class VesselEntity(IDName):
     dwt: Optional[int] = None
 
     vessel_class: Optional[str] = None
+    classes: Optional[List[VesselClassEntry]] = None
     corporate_entities: Optional[
         List[VesselEntityCorporateEntityWithConfidence]
     ] = None

diff --git a/vortexasdk/api/vessel_availability.py b/vortexasdk/api/vessel_availability.py
@@ -2,7 +2,11 @@
 from typing import List, Optional
 
 
-from vortexasdk.api.shared_types import EntityWithListLayer, ISODate
+from vortexasdk.api.shared_types import (
+    EntityWithListLayer,
+    ISODate,
+    VesselClassEntry,
+)
 
 
 class DeclaredDestination(BaseModel):
@@ -42,6 +46,7 @@ class VesselAvailability(BaseModel):
     available_at: Optional[ISODate] = None
     evaluated_at: Optional[ISODate] = None
     vessel_class: Optional[str] = None
+    vessel_class_hierarchy: Optional[List[VesselClassEntry]] = None
     vessel_declared_destination: Optional[List[DeclaredDestination]] = None
     vessel_dwt: Optional[int] = None
     vessel_cubic_capacity: Optional[int] = None

diff --git a/vortexasdk/api/voyages.py b/vortexasdk/api/voyages.py
@@ -12,6 +12,7 @@
     ISODate,
     Scrubber,
     Tag,
+    VesselClassEntry,
 )
 
 
@@ -49,6 +50,7 @@ class VoyagesVesselEntity(BaseModel):
     name: Optional[str] = None
     dead_weight: Optional[int] = None
     vessel_class: Optional[str] = None
+    classes: Optional[List[VesselClassEntry]] = None
     imo: Optional[int] = None
     mmsi: Optional[int] = None
     call_sign: Optional[str] = None

diff --git a/vortexasdk/version.py b/vortexasdk/version.py
@@ -1 +1 @@
-__version__ = "0.53.2"
+__version__ = "0.53.4"