Skip to content

Commit e5a3008

Browse files
Ken LippoldKen Lippold
authored andcommitted
Added optional ID argument to create methods.
1 parent ff28bb5 commit e5a3008

File tree

11 files changed

+28
-1
lines changed

11 files changed

+28
-1
lines changed

src/hydroserverpy/api/services/etl/data_connection.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,12 @@ def create(
5151
transformer_settings: dict = ...,
5252
loader_type: str = ...,
5353
loader_settings: dict = ...,
54+
uid: Optional[UUID] = None
5455
) -> "DataConnection":
5556
"""Create a new data connection."""
5657

5758
body = {
59+
"id": normalize_uuid(uid),
5860
"name": name,
5961
"type": data_connection_type,
6062
"workspaceId": normalize_uuid(workspace),

src/hydroserverpy/api/services/etl/orchestration_system.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,12 @@ def create(
3939
name: str,
4040
orchestration_system_type: str,
4141
workspace: Optional[Union["Workspace", UUID, str]] = None,
42+
uid: Optional[UUID] = None
4243
) -> "OrchestrationSystem":
4344
"""Create a new orchestration system."""
4445

4546
body = {
47+
"id": normalize_uuid(uid),
4648
"name": name,
4749
"type": orchestration_system_type,
4850
"workspaceId": normalize_uuid(workspace),

src/hydroserverpy/api/services/etl/task.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,10 +89,12 @@ def create(
8989
interval: Optional[int] = None,
9090
interval_period: Optional[str] = None,
9191
mappings: List[dict] = Field(default_factory=list),
92+
uid: Optional[UUID] = None
9293
) -> "Task":
9394
"""Create a new ETL task."""
9495

9596
body = {
97+
"id": normalize_uuid(uid),
9698
"name": name,
9799
"workspaceId": normalize_uuid(workspace),
98100
"dataConnectionId": normalize_uuid(data_connection),

src/hydroserverpy/api/services/iam/workspace.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,17 @@ def list(
3737
is_associated=is_associated,
3838
)
3939

40-
def create(self, name: str, is_private: bool, **_) -> "Workspace":
40+
def create(
41+
self,
42+
name: str,
43+
is_private: bool,
44+
uid: Optional[UUID] = None,
45+
**_
46+
) -> "Workspace":
4147
"""Create a new workspace."""
4248

4349
return super().create(
50+
id=normalize_uuid(uid),
4451
name=name,
4552
is_private=is_private,
4653
)

src/hydroserverpy/api/services/sta/datastream.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,10 +113,12 @@ def create(
113113
result_end_time: Optional[datetime] = None,
114114
is_private: bool = False,
115115
is_visible: bool = True,
116+
uid: Optional[UUID] = None,
116117
) -> "Datastream":
117118
"""Create a new datastream."""
118119

119120
body = {
121+
"id": normalize_uuid(uid),
120122
"name": name,
121123
"description": description,
122124
"thingId": normalize_uuid(thing),

src/hydroserverpy/api/services/sta/observed_property.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,12 @@ def create(
4646
observed_property_type: str,
4747
code: str,
4848
workspace: Optional[Union["Workspace", UUID, str]] = None,
49+
uid: Optional[UUID] = None
4950
) -> "ObservedProperty":
5051
"""Create a new observed property."""
5152

5253
body = {
54+
"id": normalize_uuid(uid),
5355
"name": name,
5456
"definition": definition,
5557
"description": description,

src/hydroserverpy/api/services/sta/processing_level.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,12 @@ def create(
4242
definition: Optional[str] = None,
4343
explanation: Optional[str] = None,
4444
workspace: Optional[Union["Workspace", UUID, str]] = None,
45+
uid: Optional[UUID] = None,
4546
) -> "ProcessingLevel":
4647
"""Create a new processing level."""
4748

4849
body = {
50+
"id": normalize_uuid(uid),
4951
"code": code,
5052
"definition": definition,
5153
"explanation": explanation,

src/hydroserverpy/api/services/sta/result_qualifier.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,12 @@ def create(
3737
code: str,
3838
description: Optional[str] = None,
3939
workspace: Optional[Union["Workspace", UUID, str]] = None,
40+
uid: Optional[UUID] = None,
4041
) -> "ResultQualifier":
4142
"""Create a new result qualifier."""
4243

4344
body = {
45+
"id": normalize_uuid(uid),
4446
"code": code,
4547
"description": description,
4648
"workspaceId": normalize_uuid(workspace),

src/hydroserverpy/api/services/sta/sensor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,12 @@ def create(
5454
method_link: Optional[str] = None,
5555
method_code: Optional[str] = None,
5656
workspace: Optional[Union["Workspace", UUID, str]] = None,
57+
uid: Optional[UUID] = None
5758
) -> "Sensor":
5859
"""Create a new sensor."""
5960

6061
body = {
62+
"id": normalize_uuid(uid),
6163
"name": name,
6264
"description": description,
6365
"encodingType": encoding_type,

src/hydroserverpy/api/services/sta/thing.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,12 @@ def create(
6868
county: Optional[str] = None,
6969
country: Optional[str] = None,
7070
data_disclaimer: Optional[str] = None,
71+
uid: Optional[UUID] = None,
7172
) -> "Thing":
7273
"""Create a new thing."""
7374

7475
body = {
76+
"id": normalize_uuid(uid),
7577
"name": name,
7678
"description": description,
7779
"samplingFeatureType": sampling_feature_type,

0 commit comments

Comments
 (0)