dtlpy 1.115.44__py3-none-any.whl → 1.117.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (238) hide show
  1. dtlpy/__init__.py +491 -491
  2. dtlpy/__version__.py +1 -1
  3. dtlpy/assets/__init__.py +26 -26
  4. dtlpy/assets/code_server/config.yaml +2 -2
  5. dtlpy/assets/code_server/installation.sh +24 -24
  6. dtlpy/assets/code_server/launch.json +13 -13
  7. dtlpy/assets/code_server/settings.json +2 -2
  8. dtlpy/assets/main.py +53 -53
  9. dtlpy/assets/main_partial.py +18 -18
  10. dtlpy/assets/mock.json +11 -11
  11. dtlpy/assets/model_adapter.py +83 -83
  12. dtlpy/assets/package.json +61 -61
  13. dtlpy/assets/package_catalog.json +29 -29
  14. dtlpy/assets/package_gitignore +307 -307
  15. dtlpy/assets/service_runners/__init__.py +33 -33
  16. dtlpy/assets/service_runners/converter.py +96 -96
  17. dtlpy/assets/service_runners/multi_method.py +49 -49
  18. dtlpy/assets/service_runners/multi_method_annotation.py +54 -54
  19. dtlpy/assets/service_runners/multi_method_dataset.py +55 -55
  20. dtlpy/assets/service_runners/multi_method_item.py +52 -52
  21. dtlpy/assets/service_runners/multi_method_json.py +52 -52
  22. dtlpy/assets/service_runners/single_method.py +37 -37
  23. dtlpy/assets/service_runners/single_method_annotation.py +43 -43
  24. dtlpy/assets/service_runners/single_method_dataset.py +43 -43
  25. dtlpy/assets/service_runners/single_method_item.py +41 -41
  26. dtlpy/assets/service_runners/single_method_json.py +42 -42
  27. dtlpy/assets/service_runners/single_method_multi_input.py +45 -45
  28. dtlpy/assets/voc_annotation_template.xml +23 -23
  29. dtlpy/caches/base_cache.py +32 -32
  30. dtlpy/caches/cache.py +473 -473
  31. dtlpy/caches/dl_cache.py +201 -201
  32. dtlpy/caches/filesystem_cache.py +89 -89
  33. dtlpy/caches/redis_cache.py +84 -84
  34. dtlpy/dlp/__init__.py +20 -20
  35. dtlpy/dlp/cli_utilities.py +367 -367
  36. dtlpy/dlp/command_executor.py +764 -764
  37. dtlpy/dlp/dlp +1 -1
  38. dtlpy/dlp/dlp.bat +1 -1
  39. dtlpy/dlp/dlp.py +128 -128
  40. dtlpy/dlp/parser.py +651 -651
  41. dtlpy/entities/__init__.py +83 -83
  42. dtlpy/entities/analytic.py +347 -347
  43. dtlpy/entities/annotation.py +1879 -1879
  44. dtlpy/entities/annotation_collection.py +699 -699
  45. dtlpy/entities/annotation_definitions/__init__.py +20 -20
  46. dtlpy/entities/annotation_definitions/base_annotation_definition.py +100 -100
  47. dtlpy/entities/annotation_definitions/box.py +195 -195
  48. dtlpy/entities/annotation_definitions/classification.py +67 -67
  49. dtlpy/entities/annotation_definitions/comparison.py +72 -72
  50. dtlpy/entities/annotation_definitions/cube.py +204 -204
  51. dtlpy/entities/annotation_definitions/cube_3d.py +149 -149
  52. dtlpy/entities/annotation_definitions/description.py +32 -32
  53. dtlpy/entities/annotation_definitions/ellipse.py +124 -124
  54. dtlpy/entities/annotation_definitions/free_text.py +62 -62
  55. dtlpy/entities/annotation_definitions/gis.py +69 -69
  56. dtlpy/entities/annotation_definitions/note.py +139 -139
  57. dtlpy/entities/annotation_definitions/point.py +117 -117
  58. dtlpy/entities/annotation_definitions/polygon.py +182 -182
  59. dtlpy/entities/annotation_definitions/polyline.py +111 -111
  60. dtlpy/entities/annotation_definitions/pose.py +92 -92
  61. dtlpy/entities/annotation_definitions/ref_image.py +86 -86
  62. dtlpy/entities/annotation_definitions/segmentation.py +240 -240
  63. dtlpy/entities/annotation_definitions/subtitle.py +34 -34
  64. dtlpy/entities/annotation_definitions/text.py +85 -85
  65. dtlpy/entities/annotation_definitions/undefined_annotation.py +74 -74
  66. dtlpy/entities/app.py +220 -220
  67. dtlpy/entities/app_module.py +107 -107
  68. dtlpy/entities/artifact.py +174 -174
  69. dtlpy/entities/assignment.py +399 -399
  70. dtlpy/entities/base_entity.py +214 -214
  71. dtlpy/entities/bot.py +113 -113
  72. dtlpy/entities/codebase.py +292 -292
  73. dtlpy/entities/collection.py +38 -38
  74. dtlpy/entities/command.py +169 -169
  75. dtlpy/entities/compute.py +449 -449
  76. dtlpy/entities/dataset.py +1299 -1299
  77. dtlpy/entities/directory_tree.py +44 -44
  78. dtlpy/entities/dpk.py +470 -470
  79. dtlpy/entities/driver.py +235 -235
  80. dtlpy/entities/execution.py +397 -397
  81. dtlpy/entities/feature.py +124 -124
  82. dtlpy/entities/feature_set.py +152 -145
  83. dtlpy/entities/filters.py +798 -798
  84. dtlpy/entities/gis_item.py +107 -107
  85. dtlpy/entities/integration.py +184 -184
  86. dtlpy/entities/item.py +975 -959
  87. dtlpy/entities/label.py +123 -123
  88. dtlpy/entities/links.py +85 -85
  89. dtlpy/entities/message.py +175 -175
  90. dtlpy/entities/model.py +684 -684
  91. dtlpy/entities/node.py +1005 -1005
  92. dtlpy/entities/ontology.py +810 -803
  93. dtlpy/entities/organization.py +287 -287
  94. dtlpy/entities/package.py +657 -657
  95. dtlpy/entities/package_defaults.py +5 -5
  96. dtlpy/entities/package_function.py +185 -185
  97. dtlpy/entities/package_module.py +113 -113
  98. dtlpy/entities/package_slot.py +118 -118
  99. dtlpy/entities/paged_entities.py +299 -299
  100. dtlpy/entities/pipeline.py +624 -624
  101. dtlpy/entities/pipeline_execution.py +279 -279
  102. dtlpy/entities/project.py +394 -394
  103. dtlpy/entities/prompt_item.py +505 -505
  104. dtlpy/entities/recipe.py +301 -301
  105. dtlpy/entities/reflect_dict.py +102 -102
  106. dtlpy/entities/resource_execution.py +138 -138
  107. dtlpy/entities/service.py +974 -963
  108. dtlpy/entities/service_driver.py +117 -117
  109. dtlpy/entities/setting.py +294 -294
  110. dtlpy/entities/task.py +495 -495
  111. dtlpy/entities/time_series.py +143 -143
  112. dtlpy/entities/trigger.py +426 -426
  113. dtlpy/entities/user.py +118 -118
  114. dtlpy/entities/webhook.py +124 -124
  115. dtlpy/examples/__init__.py +19 -19
  116. dtlpy/examples/add_labels.py +135 -135
  117. dtlpy/examples/add_metadata_to_item.py +21 -21
  118. dtlpy/examples/annotate_items_using_model.py +65 -65
  119. dtlpy/examples/annotate_video_using_model_and_tracker.py +75 -75
  120. dtlpy/examples/annotations_convert_to_voc.py +9 -9
  121. dtlpy/examples/annotations_convert_to_yolo.py +9 -9
  122. dtlpy/examples/convert_annotation_types.py +51 -51
  123. dtlpy/examples/converter.py +143 -143
  124. dtlpy/examples/copy_annotations.py +22 -22
  125. dtlpy/examples/copy_folder.py +31 -31
  126. dtlpy/examples/create_annotations.py +51 -51
  127. dtlpy/examples/create_video_annotations.py +83 -83
  128. dtlpy/examples/delete_annotations.py +26 -26
  129. dtlpy/examples/filters.py +113 -113
  130. dtlpy/examples/move_item.py +23 -23
  131. dtlpy/examples/play_video_annotation.py +13 -13
  132. dtlpy/examples/show_item_and_mask.py +53 -53
  133. dtlpy/examples/triggers.py +49 -49
  134. dtlpy/examples/upload_batch_of_items.py +20 -20
  135. dtlpy/examples/upload_items_and_custom_format_annotations.py +55 -55
  136. dtlpy/examples/upload_items_with_modalities.py +43 -43
  137. dtlpy/examples/upload_segmentation_annotations_from_mask_image.py +44 -44
  138. dtlpy/examples/upload_yolo_format_annotations.py +70 -70
  139. dtlpy/exceptions.py +125 -125
  140. dtlpy/miscellaneous/__init__.py +20 -20
  141. dtlpy/miscellaneous/dict_differ.py +95 -95
  142. dtlpy/miscellaneous/git_utils.py +217 -217
  143. dtlpy/miscellaneous/json_utils.py +14 -14
  144. dtlpy/miscellaneous/list_print.py +105 -105
  145. dtlpy/miscellaneous/zipping.py +130 -130
  146. dtlpy/ml/__init__.py +20 -20
  147. dtlpy/ml/base_feature_extractor_adapter.py +27 -27
  148. dtlpy/ml/base_model_adapter.py +1287 -1230
  149. dtlpy/ml/metrics.py +461 -461
  150. dtlpy/ml/predictions_utils.py +274 -274
  151. dtlpy/ml/summary_writer.py +57 -57
  152. dtlpy/ml/train_utils.py +60 -60
  153. dtlpy/new_instance.py +252 -252
  154. dtlpy/repositories/__init__.py +56 -56
  155. dtlpy/repositories/analytics.py +85 -85
  156. dtlpy/repositories/annotations.py +916 -916
  157. dtlpy/repositories/apps.py +383 -383
  158. dtlpy/repositories/artifacts.py +452 -452
  159. dtlpy/repositories/assignments.py +599 -599
  160. dtlpy/repositories/bots.py +213 -213
  161. dtlpy/repositories/codebases.py +559 -559
  162. dtlpy/repositories/collections.py +332 -332
  163. dtlpy/repositories/commands.py +152 -152
  164. dtlpy/repositories/compositions.py +61 -61
  165. dtlpy/repositories/computes.py +439 -439
  166. dtlpy/repositories/datasets.py +1585 -1504
  167. dtlpy/repositories/downloader.py +1157 -923
  168. dtlpy/repositories/dpks.py +433 -433
  169. dtlpy/repositories/drivers.py +482 -482
  170. dtlpy/repositories/executions.py +815 -815
  171. dtlpy/repositories/feature_sets.py +256 -226
  172. dtlpy/repositories/features.py +255 -255
  173. dtlpy/repositories/integrations.py +484 -484
  174. dtlpy/repositories/items.py +912 -912
  175. dtlpy/repositories/messages.py +94 -94
  176. dtlpy/repositories/models.py +1000 -1000
  177. dtlpy/repositories/nodes.py +80 -80
  178. dtlpy/repositories/ontologies.py +511 -511
  179. dtlpy/repositories/organizations.py +525 -525
  180. dtlpy/repositories/packages.py +1941 -1941
  181. dtlpy/repositories/pipeline_executions.py +451 -451
  182. dtlpy/repositories/pipelines.py +640 -640
  183. dtlpy/repositories/projects.py +539 -539
  184. dtlpy/repositories/recipes.py +429 -399
  185. dtlpy/repositories/resource_executions.py +137 -137
  186. dtlpy/repositories/schema.py +120 -120
  187. dtlpy/repositories/service_drivers.py +213 -213
  188. dtlpy/repositories/services.py +1704 -1704
  189. dtlpy/repositories/settings.py +339 -339
  190. dtlpy/repositories/tasks.py +1477 -1477
  191. dtlpy/repositories/times_series.py +278 -278
  192. dtlpy/repositories/triggers.py +536 -536
  193. dtlpy/repositories/upload_element.py +257 -257
  194. dtlpy/repositories/uploader.py +661 -661
  195. dtlpy/repositories/webhooks.py +249 -249
  196. dtlpy/services/__init__.py +22 -22
  197. dtlpy/services/aihttp_retry.py +131 -131
  198. dtlpy/services/api_client.py +1786 -1785
  199. dtlpy/services/api_reference.py +40 -40
  200. dtlpy/services/async_utils.py +133 -133
  201. dtlpy/services/calls_counter.py +44 -44
  202. dtlpy/services/check_sdk.py +68 -68
  203. dtlpy/services/cookie.py +115 -115
  204. dtlpy/services/create_logger.py +156 -156
  205. dtlpy/services/events.py +84 -84
  206. dtlpy/services/logins.py +235 -235
  207. dtlpy/services/reporter.py +256 -256
  208. dtlpy/services/service_defaults.py +91 -91
  209. dtlpy/utilities/__init__.py +20 -20
  210. dtlpy/utilities/annotations/__init__.py +16 -16
  211. dtlpy/utilities/annotations/annotation_converters.py +269 -269
  212. dtlpy/utilities/base_package_runner.py +285 -264
  213. dtlpy/utilities/converter.py +1650 -1650
  214. dtlpy/utilities/dataset_generators/__init__.py +1 -1
  215. dtlpy/utilities/dataset_generators/dataset_generator.py +670 -670
  216. dtlpy/utilities/dataset_generators/dataset_generator_tensorflow.py +23 -23
  217. dtlpy/utilities/dataset_generators/dataset_generator_torch.py +21 -21
  218. dtlpy/utilities/local_development/__init__.py +1 -1
  219. dtlpy/utilities/local_development/local_session.py +179 -179
  220. dtlpy/utilities/reports/__init__.py +2 -2
  221. dtlpy/utilities/reports/figures.py +343 -343
  222. dtlpy/utilities/reports/report.py +71 -71
  223. dtlpy/utilities/videos/__init__.py +17 -17
  224. dtlpy/utilities/videos/video_player.py +598 -598
  225. dtlpy/utilities/videos/videos.py +470 -470
  226. {dtlpy-1.115.44.data → dtlpy-1.117.6.data}/scripts/dlp +1 -1
  227. dtlpy-1.117.6.data/scripts/dlp.bat +2 -0
  228. {dtlpy-1.115.44.data → dtlpy-1.117.6.data}/scripts/dlp.py +128 -128
  229. {dtlpy-1.115.44.dist-info → dtlpy-1.117.6.dist-info}/METADATA +186 -186
  230. dtlpy-1.117.6.dist-info/RECORD +239 -0
  231. {dtlpy-1.115.44.dist-info → dtlpy-1.117.6.dist-info}/WHEEL +1 -1
  232. {dtlpy-1.115.44.dist-info → dtlpy-1.117.6.dist-info}/licenses/LICENSE +200 -200
  233. tests/features/environment.py +551 -551
  234. dtlpy/assets/__pycache__/__init__.cpython-310.pyc +0 -0
  235. dtlpy-1.115.44.data/scripts/dlp.bat +0 -2
  236. dtlpy-1.115.44.dist-info/RECORD +0 -240
  237. {dtlpy-1.115.44.dist-info → dtlpy-1.117.6.dist-info}/entry_points.txt +0 -0
  238. {dtlpy-1.115.44.dist-info → dtlpy-1.117.6.dist-info}/top_level.txt +0 -0
@@ -1,640 +1,640 @@
1
- import logging
2
- from .. import entities, repositories, exceptions, miscellaneous, _api_reference
3
- from ..services.api_client import ApiClient
4
-
5
- logger = logging.getLogger(name="dtlpy")
6
-
7
-
8
- def generate_basic_pipeline():
9
- return dict(name="", projectId="", nodes=[], connections=[], startNodes=[], variables=[])
10
-
11
-
12
- class Pipelines:
13
- """
14
- Pipelines Repository
15
-
16
- The Pipelines class allows users to manage pipelines and their properties. See our documentation for more information on `pipelines <https://dataloop.ai/docs/pipelines-overview>`_.
17
- """
18
-
19
- def __init__(self, client_api: ApiClient, project: entities.Project = None):
20
- self._client_api = client_api
21
- self._project = project
22
-
23
- ############
24
- # entities #
25
- ############
26
- @property
27
- def project(self) -> entities.Project:
28
- if self._project is None:
29
- try:
30
- self._project = repositories.Projects(client_api=self._client_api).get()
31
- except exceptions.NotFound:
32
- raise exceptions.PlatformException(
33
- error="2001",
34
- message='Missing "project". need to set a Project entity or use project.pipelines repository',
35
- )
36
- return self._project
37
-
38
- @project.setter
39
- def project(self, project: entities.Project):
40
- if not isinstance(project, entities.Project):
41
- raise ValueError("Must input a valid Project entity")
42
- self._project = project
43
-
44
- ###########
45
- # methods #
46
- ###########
47
- @property
48
- def platform_url(self):
49
- return self._client_api._get_resource_url("projects/{}/pipelines".format(self.project.id))
50
-
51
- def open_in_web(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, pipeline_name: str = None):
52
- """
53
- Open the pipeline in web platform.
54
-
55
- **prerequisites**: Must be *owner* or *developer* to use this method.
56
-
57
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
58
- :param str pipeline_id: pipeline id
59
- :param str pipeline_name: pipeline name
60
-
61
- **Example**:
62
-
63
- .. code-block:: python
64
-
65
- project.pipelines.open_in_web(pipeline_id='pipeline_id')
66
- """
67
- if pipeline_name is not None:
68
- pipeline = self.get(pipeline_name=pipeline_name)
69
- if pipeline is not None:
70
- pipeline.open_in_web()
71
- elif pipeline_id is not None:
72
- self._client_api._open_in_web(url=self.platform_url + "/" + str(pipeline_id))
73
- else:
74
- self._client_api._open_in_web(url=self.platform_url)
75
-
76
- @_api_reference.add(path="/pipelines/{pipelineId}", method="get")
77
- def get(self, pipeline_name=None, pipeline_id=None, fetch=None) -> entities.Pipeline:
78
- """
79
- Get Pipeline object to use in your code.
80
-
81
- **prerequisites**: You must be an *owner* or *developer* to use this method.
82
-
83
- You must provide at least ONE of the following params: pipeline_name, pipeline_id.
84
-
85
- :param str pipeline_id: pipeline id
86
- :param str pipeline_name: pipeline name
87
- :param fetch: optional - fetch entity from platform, default taken from cookie
88
- :return: Pipeline object
89
- :rtype: dtlpy.entities.pipeline.Pipeline
90
-
91
- **Example**:
92
-
93
- .. code-block:: python
94
-
95
- pipeline = project.pipelines.get(pipeline_id='pipeline_id')
96
- """
97
- if fetch is None:
98
- fetch = self._client_api.fetch_entities
99
-
100
- if pipeline_name is None and pipeline_id is None:
101
- raise exceptions.PlatformException(error="400", message="Must provide an identifier in inputs")
102
- elif fetch:
103
- if pipeline_id is not None:
104
- success, response = self._client_api.gen_request(
105
- req_type="get", path="/pipelines/{}".format(pipeline_id)
106
- )
107
- if not success:
108
- raise exceptions.PlatformException(response)
109
- pipeline = entities.Pipeline.from_json(
110
- client_api=self._client_api, _json=response.json(), project=self._project
111
- )
112
- if pipeline_name is not None and pipeline.name != pipeline_name:
113
- logger.warning(
114
- "Mismatch found in pipeline.get: pipeline_name is different then pipeline.name:"
115
- " {!r} != {!r}".format(pipeline_name, pipeline.name)
116
- )
117
- elif pipeline_name is not None:
118
- filters = entities.Filters(
119
- field="name", values=pipeline_name, resource=entities.FiltersResource.PIPELINE, use_defaults=False
120
- )
121
- if self._project is not None:
122
- filters.add(field="projectId", values=self._project.id)
123
- pipelines = self.list(filters=filters)
124
- if pipelines.items_count == 0:
125
- raise exceptions.PlatformException(
126
- error="404", message="Pipeline not found. Name: {}".format(pipeline_name)
127
- )
128
- elif pipelines.items_count > 1:
129
- raise exceptions.PlatformException(
130
- error="400",
131
- message="More than one pipelines found by the name of: {} "
132
- "Please get pipeline from a project entity".format(pipeline_name),
133
- )
134
- pipeline = pipelines.items[0]
135
- else:
136
- raise exceptions.PlatformException(
137
- error="400",
138
- message="No checked-out pipeline was found, must checkout or provide an identifier in inputs",
139
- )
140
- else:
141
- pipeline = entities.Pipeline.from_json(
142
- _json={"id": pipeline_id, "name": pipeline_name},
143
- client_api=self._client_api,
144
- project=self._project,
145
- is_fetched=False,
146
- )
147
-
148
- return pipeline
149
-
150
- def _build_entities_from_response(self, response_items) -> miscellaneous.List[entities.Pipeline]:
151
- pool = self._client_api.thread_pools(pool_name="entity.create")
152
- jobs = [None for _ in range(len(response_items))]
153
-
154
- for i_pipeline, pipeline in enumerate(response_items):
155
- jobs[i_pipeline] = pool.submit(
156
- entities.Pipeline._protected_from_json,
157
- **{"client_api": self._client_api, "_json": pipeline, "project": self._project}
158
- )
159
-
160
- # get all results
161
- # noinspection PyUnresolvedReferences
162
- results = [j.result() for j in jobs]
163
- # log errors
164
- _ = [logger.warning(r[1]) for r in results if r[0] is False]
165
- # return good jobs
166
- pipelines = miscellaneous.List([r[1] for r in results if r[0] is True])
167
- return pipelines
168
-
169
- def _list(self, filters: entities.Filters):
170
- url = "/pipelines/query"
171
-
172
- # request
173
- success, response = self._client_api.gen_request(req_type="post", path=url, json_req=filters.prepare())
174
- if not success:
175
- raise exceptions.PlatformException(response)
176
- return response.json()
177
-
178
- @_api_reference.add(path="/pipelines/query", method="post")
179
- def list(self, filters: entities.Filters = None, project_id: str = None) -> entities.PagedEntities:
180
- """
181
- List project pipelines.
182
-
183
- **prerequisites**: You must be an *owner* or *developer* to use this method.
184
-
185
- :param dtlpy.entities.filters.Filters filters: Filters entity or a dictionary containing filters parameters
186
- :param str project_id: project id
187
- :return: Paged entity
188
- :rtype: dtlpy.entities.paged_entities.PagedEntities
189
-
190
- **Example**:
191
-
192
- .. code-block:: python
193
-
194
- pipelines = project.pipelines.list()
195
- """
196
- if filters is None:
197
- filters = entities.Filters(resource=entities.FiltersResource.PIPELINE)
198
- # assert type filters
199
- elif not isinstance(filters, entities.Filters):
200
- raise exceptions.PlatformException(error="400", message="Unknown filters type: {!r}".format(type(filters)))
201
- if filters.resource != entities.FiltersResource.PIPELINE:
202
- raise exceptions.PlatformException(
203
- error="400",
204
- message="Filters resource must to be FiltersResource.PIPELINE. Got: {!r}".format(filters.resource),
205
- )
206
-
207
- if project_id is None and self._project is not None:
208
- project_id = self._project.id
209
-
210
- if project_id is not None:
211
- filters.add(field="projectId", values=project_id)
212
-
213
- paged = entities.PagedEntities(
214
- items_repository=self,
215
- filters=filters,
216
- page_offset=filters.page,
217
- page_size=filters.page_size,
218
- project_id=project_id,
219
- client_api=self._client_api,
220
- )
221
- paged.get_page()
222
- return paged
223
-
224
- def _name_validation(self, name: str):
225
- url = "/piper-misc/naming/packages/{}".format(name)
226
-
227
- # request
228
- success, response = self._client_api.gen_request(req_type="get", path=url)
229
- if not success:
230
- raise exceptions.PlatformException(response)
231
-
232
- # @_api_reference.add(path='/pipelines/{pipelineId}', method='delete')
233
- def delete(self, pipeline: entities.Pipeline = None, pipeline_name: str = None, pipeline_id: str = None):
234
- """
235
- Delete Pipeline object.
236
-
237
- **prerequisites**: You must be an *owner* or *developer* to use this method.
238
-
239
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
240
- :param str pipeline_id: pipeline id
241
- :param str pipeline_name: pipeline name
242
- :return: True if success
243
- :rtype: bool
244
-
245
- **Example**:
246
-
247
- .. code-block:: python
248
-
249
- is_deleted = project.pipelines.delete(pipeline_id='pipeline_id')
250
- """
251
- # get id and name
252
- if pipeline_id is None:
253
- if pipeline is None:
254
- pipeline = self.get(pipeline_name=pipeline_name)
255
- pipeline_id = pipeline.id
256
-
257
- # request
258
- success, response = self._client_api.gen_request(req_type="delete", path="/pipelines/{}".format(pipeline_id))
259
-
260
- # exception handling
261
- if not success:
262
- raise exceptions.PlatformException(response)
263
-
264
- # return results
265
- return True
266
-
267
- @_api_reference.add(path="/pipelines/{pipelineId}/settings", method="patch")
268
- def update_settings(self, pipeline: entities.Pipeline, settings: entities.PipelineSettings):
269
- """
270
- Update pipeline settings.
271
-
272
- **prerequisites**: You must be an *owner* or *developer* to use this method.
273
-
274
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
275
- :param dtlpy.entities.pipeline.PipelineSettings settings: settings entity
276
- :return: Pipeline object
277
- :rtype: dtlpy.entities.pipeline.Pipeline
278
-
279
- **Example**:
280
-
281
- .. code-block:: python
282
-
283
- pipeline = project.pipelines.update_settings(pipeline='pipeline_entity', settings=dl.PipelineSettings(keep_triggers_active=True))
284
- """
285
- # payload
286
- payload = {"settings": settings.to_json()}
287
-
288
- # request
289
- success, response = self._client_api.gen_request(
290
- req_type="patch", path="/pipelines/{}".format(pipeline.id), json_req=payload
291
- )
292
- if not success:
293
- raise exceptions.PlatformException(response)
294
-
295
- # return entity
296
- return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
297
-
298
- def __update_variables(self, pipeline: entities.Pipeline):
299
- pipeline_json = pipeline.to_json()
300
- variables = pipeline_json.get("variables", list())
301
-
302
- for var in variables:
303
- if var.get("reference", None) is None:
304
- var["reference"] = pipeline.id
305
-
306
- # payload
307
- payload = {"variables": variables}
308
-
309
- # request
310
- success, response = self._client_api.gen_request(
311
- req_type="patch", path="/pipelines/{}/variables".format(pipeline.id), json_req=payload
312
- )
313
- if not success:
314
- raise exceptions.PlatformException(response)
315
-
316
- # return entity
317
- return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
318
-
319
- @_api_reference.add(path="/pipelines/{pipelineId}", method="patch")
320
- def update(self, pipeline: entities.Pipeline = None) -> entities.Pipeline:
321
- """
322
- Update pipeline changes to platform.
323
-
324
- **prerequisites**: You must be an *owner* or *developer* to use this method.
325
-
326
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
327
- :return: Pipeline object
328
- :rtype: dtlpy.entities.pipeline.Pipeline
329
-
330
- **Example**:
331
-
332
- .. code-block:: python
333
-
334
- pipeline = project.pipelines.update(pipeline='pipeline_entity')
335
- """
336
- # payload
337
- payload = pipeline.to_json()
338
-
339
- # update settings
340
- if pipeline.settings_changed():
341
- new_pipeline = self.update_settings(pipeline=pipeline, settings=pipeline.settings)
342
- payload["settings"] = new_pipeline.to_json().get("settings", payload.get("settings"))
343
-
344
- # update variables
345
- if pipeline.variables_changed():
346
- new_pipeline = self.__update_variables(pipeline=pipeline)
347
- payload["variables"] = new_pipeline.to_json().get("variables", payload.get("variables"))
348
-
349
- success, response = self._client_api.gen_request(
350
- req_type="patch", path="/pipelines/{}".format(pipeline.id), json_req=payload
351
- )
352
-
353
- # exception handling
354
- if not success:
355
- raise exceptions.PlatformException(response)
356
-
357
- # return entity
358
- return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
359
-
360
- @_api_reference.add(path="/pipelines", method="post")
361
- def create(self, name: str = None, project_id: str = None, pipeline_json: dict = None) -> entities.Pipeline:
362
- """
363
- Create a new pipeline.
364
-
365
- **prerequisites**: You must be an *owner* or *developer* to use this method.
366
-
367
- :param str name: pipeline name
368
- :param str project_id: project id
369
- :param dict pipeline_json: json containing the pipeline fields
370
- :return: Pipeline object
371
- :rtype: dtlpy.entities.pipeline.Pipeline
372
-
373
- **Example**:
374
-
375
- .. code-block:: python
376
-
377
- pipeline = project.pipelines.create(name='pipeline_name')
378
- """
379
- if pipeline_json is None:
380
- pipeline_json = generate_basic_pipeline()
381
-
382
- if name is not None:
383
- pipeline_json["name"] = name
384
-
385
- if project_id is not None:
386
- pipeline_json["projectId"] = project_id
387
- else:
388
- if not pipeline_json.get("projectId", None):
389
- pipeline_json["projectId"] = self.project.id
390
-
391
- success, response = self._client_api.gen_request(req_type="post", path="/pipelines", json_req=pipeline_json)
392
- if success:
393
- pipeline = entities.Pipeline.from_json(
394
- client_api=self._client_api, _json=response.json(), project=self.project
395
- )
396
- else:
397
- raise exceptions.PlatformException(response)
398
- assert isinstance(pipeline, entities.Pipeline)
399
- return pipeline
400
-
401
- @_api_reference.add(path="/pipelines/{pipelineId}/install", method="post")
402
- def install(self, pipeline: entities.Pipeline = None, resume_option: entities.PipelineResumeOption = None):
403
- """
404
- Install (start) a pipeline.
405
-
406
- **prerequisites**: You must be an *owner* or *developer* to use this method.
407
-
408
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
409
- :param dtlpy.entities.pipeline.PipelineResumeOption resume_option: optional - resume pipeline method (what to do with existing cycles)
410
- :return: Composition object
411
-
412
- **Example**:
413
-
414
- .. code-block:: python
415
-
416
- project.pipelines.install(pipeline='pipeline_entity')
417
- """
418
-
419
- payload = {}
420
- if resume_option:
421
- payload["resumeOption"] = resume_option
422
-
423
- success, response = self._client_api.gen_request(
424
- req_type="post", path="/pipelines/{}/install".format(pipeline.id), json_req=payload
425
- )
426
-
427
- if not success:
428
- raise exceptions.PlatformException(response)
429
-
430
- return entities.Pipeline.from_json(client_api=self._client_api, _json=response.json(), project=self.project)
431
-
432
- @_api_reference.add(path="/pipelines/{pipelineId}/uninstall", method="post")
433
- def pause(self, pipeline: entities.Pipeline = None, keep_triggers_active: bool = None):
434
- """
435
- Pause a pipeline.
436
-
437
- **prerequisites**: You must be an *owner* or *developer* to use this method.
438
-
439
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
440
- :param bool keep_triggers_active: Do we want the triggers to stay active and collect events
441
- :return: Composition object
442
-
443
- **Example**:
444
-
445
- .. code-block:: python
446
-
447
- project.pipelines.pause(pipeline='pipeline_entity')
448
- """
449
-
450
- payload = {}
451
- if keep_triggers_active is not None:
452
- payload["keepTriggersActive"] = keep_triggers_active
453
-
454
- success, response = self._client_api.gen_request(
455
- req_type="post", path="/pipelines/{}/uninstall".format(pipeline.id), json_req=payload
456
- )
457
-
458
- if not success:
459
- raise exceptions.PlatformException(response)
460
-
461
- @_api_reference.add(path="/pipelines/{pipelineId}/reset", method="post")
462
- def reset(
463
- self,
464
- pipeline: entities.Pipeline = None,
465
- pipeline_id: str = None,
466
- pipeline_name: str = None,
467
- stop_if_running: bool = False,
468
- ):
469
- """
470
- Reset pipeline counters.
471
-
472
- **prerequisites**: You must be an *owner* or *developer* to use this method.
473
-
474
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity - optional
475
- :param str pipeline_id: pipeline_id - optional
476
- :param str pipeline_name: pipeline_name - optional
477
- :param bool stop_if_running: If the pipeline is installed it will stop the pipeline and reset the counters.
478
- :return: bool
479
-
480
- **Example**:
481
-
482
- .. code-block:: python
483
-
484
- success = project.pipelines.reset(pipeline='pipeline_entity')
485
- """
486
-
487
- if pipeline_id is None:
488
- if pipeline is None:
489
- if pipeline_name is not None:
490
- pipeline = self.get(pipeline_name=pipeline_name)
491
- else:
492
- raise exceptions.PlatformException(
493
- "400", "Must provide one of pipeline, pipeline_id or pipeline_name"
494
- )
495
- pipeline_id = pipeline.id
496
-
497
- if stop_if_running is True:
498
- if pipeline is None:
499
- pipeline = self.get(pipeline_id=pipeline_id)
500
- pipeline.pause()
501
-
502
- success, response = self._client_api.gen_request(
503
- req_type="post", path="/pipelines/{}/reset".format(pipeline_id)
504
- )
505
-
506
- if not success:
507
- raise exceptions.PlatformException(response)
508
-
509
- return True
510
-
511
- @_api_reference.add(path="/pipelines/{id}/statistics", method="get")
512
- def stats(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, pipeline_name: str = None):
513
- """
514
- Get pipeline counters.
515
-
516
- **prerequisites**: You must be an *owner* or *developer* to use this method.
517
-
518
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity - optional
519
- :param str pipeline_id: pipeline_id - optional
520
- :param str pipeline_name: pipeline_name - optional
521
- :return: PipelineStats
522
- :rtype: dtlpy.entities.pipeline.PipelineStats
523
-
524
- **Example**:
525
-
526
- .. code-block:: python
527
-
528
- pipeline_stats = project.pipelines.stats(pipeline='pipeline_entity')
529
- """
530
-
531
- if pipeline_id is None:
532
- if pipeline is None:
533
- if pipeline_name is not None:
534
- pipeline = self.get(pipeline_name=pipeline_name)
535
- else:
536
- raise exceptions.PlatformException(
537
- "400", "Must provide one of pipeline, pipeline_id or pipeline_name"
538
- )
539
- pipeline_id = pipeline.id
540
-
541
- success, response = self._client_api.gen_request(
542
- req_type="get", path="/pipelines/{}/statistics".format(pipeline_id)
543
- )
544
-
545
- if not success:
546
- raise exceptions.PlatformException(response)
547
-
548
- return entities.PipelineStats.from_json(_json=response.json())
549
-
550
- def execute(
551
- self,
552
- pipeline: entities.Pipeline = None,
553
- pipeline_id: str = None,
554
- pipeline_name: str = None,
555
- execution_input=None,
556
- ):
557
- """
558
- Execute a pipeline and return the pipeline execution as an object.
559
-
560
- **prerequisites**: You must be an *owner* or *developer* to use this method.
561
-
562
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
563
- :param str pipeline_id: pipeline id
564
- :param str pipeline_name: pipeline name
565
- :param execution_input: list of the dl.FunctionIO or dict of pipeline input - example {'item': 'item_id'}
566
- :return: entities.PipelineExecution object
567
- :rtype: dtlpy.entities.pipeline_execution.PipelineExecution
568
-
569
- **Example**:
570
-
571
- .. code-block:: python
572
-
573
- pipeline_execution= project.pipelines.execute(pipeline='pipeline_entity', execution_input= {'item': 'item_id'} )
574
- """
575
-
576
- if pipeline is None and pipeline_id is None:
577
- raise ValueError("Pipeline or pipeline_id must be provided")
578
-
579
- if pipeline is None:
580
- pipeline = self.get(pipeline_id=pipeline_id)
581
-
582
- execution = repositories.PipelineExecutions(
583
- pipeline=pipeline, client_api=self._client_api, project=self._project
584
- ).create(pipeline_id=pipeline.id, execution_input=execution_input, test_mode=False)
585
- return execution
586
-
587
- def test(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, execution_input=None):
588
- """
589
- Execute a pipeline in test mode and return the pipeline execution as an object.
590
-
591
- **prerequisites**: You must be an *owner* or *developer* to use this method.
592
-
593
- :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
594
- :param str pipeline_id: pipeline id
595
- :param execution_input: list of the dl.FunctionIO or dict of pipeline input - example {'item': 'item_id'}
596
- :return: entities.PipelineExecution object
597
- :rtype: dtlpy.entities.pipeline_execution.PipelineExecution
598
-
599
- **Example**:
600
-
601
- .. code-block:: python
602
-
603
- pipeline_execution= project.pipelines.test(pipeline='pipeline_entity', execution_input= {'item': 'item_id'} )
604
- """
605
- if pipeline is None and pipeline_id is None:
606
- raise ValueError("Pipeline or pipeline_id must be provided")
607
-
608
- if pipeline is None:
609
- pipeline = self.get(pipeline_id=pipeline_id)
610
-
611
- execution = repositories.PipelineExecutions(
612
- pipeline=pipeline, client_api=self._client_api, project=self._project
613
- ).create(pipeline_id=pipeline.id, execution_input=execution_input, test_mode=True)
614
- return execution
615
-
616
- @_api_reference.add(path="/pipelines/validate", method="post")
617
- def validate(self, pipeline_json: dict):
618
- """
619
- Validate a pipeline configuration.
620
-
621
- **prerequisites**: You must be an *owner* or *developer* to use this method.
622
-
623
- :param dict pipeline_json: pipeline configuration as JSON dictionary
624
- :return: Validation result
625
- :rtype: dict
626
-
627
- **Example**:
628
-
629
- .. code-block:: python
630
-
631
- validation_result = project.pipelines.validate(pipeline_json={'name': 'my_pipeline', 'nodes': []})
632
- """
633
- success, response = self._client_api.gen_request(
634
- req_type="post", path="/pipelines/validate", json_req=pipeline_json
635
- )
636
-
637
- if not success:
638
- raise exceptions.PlatformException(response)
639
-
640
- return response.json()
1
+ import logging
2
+ from .. import entities, repositories, exceptions, miscellaneous, _api_reference
3
+ from ..services.api_client import ApiClient
4
+
5
+ logger = logging.getLogger(name="dtlpy")
6
+
7
+
8
+ def generate_basic_pipeline():
9
+ return dict(name="", projectId="", nodes=[], connections=[], startNodes=[], variables=[])
10
+
11
+
12
+ class Pipelines:
13
+ """
14
+ Pipelines Repository
15
+
16
+ The Pipelines class allows users to manage pipelines and their properties. See our documentation for more information on `pipelines <https://dataloop.ai/docs/pipelines-overview>`_.
17
+ """
18
+
19
+ def __init__(self, client_api: ApiClient, project: entities.Project = None):
20
+ self._client_api = client_api
21
+ self._project = project
22
+
23
+ ############
24
+ # entities #
25
+ ############
26
+ @property
27
+ def project(self) -> entities.Project:
28
+ if self._project is None:
29
+ try:
30
+ self._project = repositories.Projects(client_api=self._client_api).get()
31
+ except exceptions.NotFound:
32
+ raise exceptions.PlatformException(
33
+ error="2001",
34
+ message='Missing "project". need to set a Project entity or use project.pipelines repository',
35
+ )
36
+ return self._project
37
+
38
+ @project.setter
39
+ def project(self, project: entities.Project):
40
+ if not isinstance(project, entities.Project):
41
+ raise ValueError("Must input a valid Project entity")
42
+ self._project = project
43
+
44
+ ###########
45
+ # methods #
46
+ ###########
47
+ @property
48
+ def platform_url(self):
49
+ return self._client_api._get_resource_url("projects/{}/pipelines".format(self.project.id))
50
+
51
+ def open_in_web(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, pipeline_name: str = None):
52
+ """
53
+ Open the pipeline in web platform.
54
+
55
+ **prerequisites**: Must be *owner* or *developer* to use this method.
56
+
57
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
58
+ :param str pipeline_id: pipeline id
59
+ :param str pipeline_name: pipeline name
60
+
61
+ **Example**:
62
+
63
+ .. code-block:: python
64
+
65
+ project.pipelines.open_in_web(pipeline_id='pipeline_id')
66
+ """
67
+ if pipeline_name is not None:
68
+ pipeline = self.get(pipeline_name=pipeline_name)
69
+ if pipeline is not None:
70
+ pipeline.open_in_web()
71
+ elif pipeline_id is not None:
72
+ self._client_api._open_in_web(url=self.platform_url + "/" + str(pipeline_id))
73
+ else:
74
+ self._client_api._open_in_web(url=self.platform_url)
75
+
76
+ @_api_reference.add(path="/pipelines/{pipelineId}", method="get")
77
+ def get(self, pipeline_name=None, pipeline_id=None, fetch=None) -> entities.Pipeline:
78
+ """
79
+ Get Pipeline object to use in your code.
80
+
81
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
82
+
83
+ You must provide at least ONE of the following params: pipeline_name, pipeline_id.
84
+
85
+ :param str pipeline_id: pipeline id
86
+ :param str pipeline_name: pipeline name
87
+ :param fetch: optional - fetch entity from platform, default taken from cookie
88
+ :return: Pipeline object
89
+ :rtype: dtlpy.entities.pipeline.Pipeline
90
+
91
+ **Example**:
92
+
93
+ .. code-block:: python
94
+
95
+ pipeline = project.pipelines.get(pipeline_id='pipeline_id')
96
+ """
97
+ if fetch is None:
98
+ fetch = self._client_api.fetch_entities
99
+
100
+ if pipeline_name is None and pipeline_id is None:
101
+ raise exceptions.PlatformException(error="400", message="Must provide an identifier in inputs")
102
+ elif fetch:
103
+ if pipeline_id is not None:
104
+ success, response = self._client_api.gen_request(
105
+ req_type="get", path="/pipelines/{}".format(pipeline_id)
106
+ )
107
+ if not success:
108
+ raise exceptions.PlatformException(response)
109
+ pipeline = entities.Pipeline.from_json(
110
+ client_api=self._client_api, _json=response.json(), project=self._project
111
+ )
112
+ if pipeline_name is not None and pipeline.name != pipeline_name:
113
+ logger.warning(
114
+ "Mismatch found in pipeline.get: pipeline_name is different then pipeline.name:"
115
+ " {!r} != {!r}".format(pipeline_name, pipeline.name)
116
+ )
117
+ elif pipeline_name is not None:
118
+ filters = entities.Filters(
119
+ field="name", values=pipeline_name, resource=entities.FiltersResource.PIPELINE, use_defaults=False
120
+ )
121
+ if self._project is not None:
122
+ filters.add(field="projectId", values=self._project.id)
123
+ pipelines = self.list(filters=filters)
124
+ if pipelines.items_count == 0:
125
+ raise exceptions.PlatformException(
126
+ error="404", message="Pipeline not found. Name: {}".format(pipeline_name)
127
+ )
128
+ elif pipelines.items_count > 1:
129
+ raise exceptions.PlatformException(
130
+ error="400",
131
+ message="More than one pipelines found by the name of: {} "
132
+ "Please get pipeline from a project entity".format(pipeline_name),
133
+ )
134
+ pipeline = pipelines.items[0]
135
+ else:
136
+ raise exceptions.PlatformException(
137
+ error="400",
138
+ message="No checked-out pipeline was found, must checkout or provide an identifier in inputs",
139
+ )
140
+ else:
141
+ pipeline = entities.Pipeline.from_json(
142
+ _json={"id": pipeline_id, "name": pipeline_name},
143
+ client_api=self._client_api,
144
+ project=self._project,
145
+ is_fetched=False,
146
+ )
147
+
148
+ return pipeline
149
+
150
+ def _build_entities_from_response(self, response_items) -> miscellaneous.List[entities.Pipeline]:
151
+ pool = self._client_api.thread_pools(pool_name="entity.create")
152
+ jobs = [None for _ in range(len(response_items))]
153
+
154
+ for i_pipeline, pipeline in enumerate(response_items):
155
+ jobs[i_pipeline] = pool.submit(
156
+ entities.Pipeline._protected_from_json,
157
+ **{"client_api": self._client_api, "_json": pipeline, "project": self._project}
158
+ )
159
+
160
+ # get all results
161
+ # noinspection PyUnresolvedReferences
162
+ results = [j.result() for j in jobs]
163
+ # log errors
164
+ _ = [logger.warning(r[1]) for r in results if r[0] is False]
165
+ # return good jobs
166
+ pipelines = miscellaneous.List([r[1] for r in results if r[0] is True])
167
+ return pipelines
168
+
169
+ def _list(self, filters: entities.Filters):
170
+ url = "/pipelines/query"
171
+
172
+ # request
173
+ success, response = self._client_api.gen_request(req_type="post", path=url, json_req=filters.prepare())
174
+ if not success:
175
+ raise exceptions.PlatformException(response)
176
+ return response.json()
177
+
178
+ @_api_reference.add(path="/pipelines/query", method="post")
179
+ def list(self, filters: entities.Filters = None, project_id: str = None) -> entities.PagedEntities:
180
+ """
181
+ List project pipelines.
182
+
183
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
184
+
185
+ :param dtlpy.entities.filters.Filters filters: Filters entity or a dictionary containing filters parameters
186
+ :param str project_id: project id
187
+ :return: Paged entity
188
+ :rtype: dtlpy.entities.paged_entities.PagedEntities
189
+
190
+ **Example**:
191
+
192
+ .. code-block:: python
193
+
194
+ pipelines = project.pipelines.list()
195
+ """
196
+ if filters is None:
197
+ filters = entities.Filters(resource=entities.FiltersResource.PIPELINE)
198
+ # assert type filters
199
+ elif not isinstance(filters, entities.Filters):
200
+ raise exceptions.PlatformException(error="400", message="Unknown filters type: {!r}".format(type(filters)))
201
+ if filters.resource != entities.FiltersResource.PIPELINE:
202
+ raise exceptions.PlatformException(
203
+ error="400",
204
+ message="Filters resource must to be FiltersResource.PIPELINE. Got: {!r}".format(filters.resource),
205
+ )
206
+
207
+ if project_id is None and self._project is not None:
208
+ project_id = self._project.id
209
+
210
+ if project_id is not None:
211
+ filters.add(field="projectId", values=project_id)
212
+
213
+ paged = entities.PagedEntities(
214
+ items_repository=self,
215
+ filters=filters,
216
+ page_offset=filters.page,
217
+ page_size=filters.page_size,
218
+ project_id=project_id,
219
+ client_api=self._client_api,
220
+ )
221
+ paged.get_page()
222
+ return paged
223
+
224
+ def _name_validation(self, name: str):
225
+ url = "/piper-misc/naming/packages/{}".format(name)
226
+
227
+ # request
228
+ success, response = self._client_api.gen_request(req_type="get", path=url)
229
+ if not success:
230
+ raise exceptions.PlatformException(response)
231
+
232
+ # @_api_reference.add(path='/pipelines/{pipelineId}', method='delete')
233
+ def delete(self, pipeline: entities.Pipeline = None, pipeline_name: str = None, pipeline_id: str = None):
234
+ """
235
+ Delete Pipeline object.
236
+
237
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
238
+
239
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
240
+ :param str pipeline_id: pipeline id
241
+ :param str pipeline_name: pipeline name
242
+ :return: True if success
243
+ :rtype: bool
244
+
245
+ **Example**:
246
+
247
+ .. code-block:: python
248
+
249
+ is_deleted = project.pipelines.delete(pipeline_id='pipeline_id')
250
+ """
251
+ # get id and name
252
+ if pipeline_id is None:
253
+ if pipeline is None:
254
+ pipeline = self.get(pipeline_name=pipeline_name)
255
+ pipeline_id = pipeline.id
256
+
257
+ # request
258
+ success, response = self._client_api.gen_request(req_type="delete", path="/pipelines/{}".format(pipeline_id))
259
+
260
+ # exception handling
261
+ if not success:
262
+ raise exceptions.PlatformException(response)
263
+
264
+ # return results
265
+ return True
266
+
267
+ @_api_reference.add(path="/pipelines/{pipelineId}/settings", method="patch")
268
+ def update_settings(self, pipeline: entities.Pipeline, settings: entities.PipelineSettings):
269
+ """
270
+ Update pipeline settings.
271
+
272
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
273
+
274
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
275
+ :param dtlpy.entities.pipeline.PipelineSettings settings: settings entity
276
+ :return: Pipeline object
277
+ :rtype: dtlpy.entities.pipeline.Pipeline
278
+
279
+ **Example**:
280
+
281
+ .. code-block:: python
282
+
283
+ pipeline = project.pipelines.update_settings(pipeline='pipeline_entity', settings=dl.PipelineSettings(keep_triggers_active=True))
284
+ """
285
+ # payload
286
+ payload = {"settings": settings.to_json()}
287
+
288
+ # request
289
+ success, response = self._client_api.gen_request(
290
+ req_type="patch", path="/pipelines/{}".format(pipeline.id), json_req=payload
291
+ )
292
+ if not success:
293
+ raise exceptions.PlatformException(response)
294
+
295
+ # return entity
296
+ return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
297
+
298
+ def __update_variables(self, pipeline: entities.Pipeline):
299
+ pipeline_json = pipeline.to_json()
300
+ variables = pipeline_json.get("variables", list())
301
+
302
+ for var in variables:
303
+ if var.get("reference", None) is None:
304
+ var["reference"] = pipeline.id
305
+
306
+ # payload
307
+ payload = {"variables": variables}
308
+
309
+ # request
310
+ success, response = self._client_api.gen_request(
311
+ req_type="patch", path="/pipelines/{}/variables".format(pipeline.id), json_req=payload
312
+ )
313
+ if not success:
314
+ raise exceptions.PlatformException(response)
315
+
316
+ # return entity
317
+ return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
318
+
319
+ @_api_reference.add(path="/pipelines/{pipelineId}", method="patch")
320
+ def update(self, pipeline: entities.Pipeline = None) -> entities.Pipeline:
321
+ """
322
+ Update pipeline changes to platform.
323
+
324
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
325
+
326
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
327
+ :return: Pipeline object
328
+ :rtype: dtlpy.entities.pipeline.Pipeline
329
+
330
+ **Example**:
331
+
332
+ .. code-block:: python
333
+
334
+ pipeline = project.pipelines.update(pipeline='pipeline_entity')
335
+ """
336
+ # payload
337
+ payload = pipeline.to_json()
338
+
339
+ # update settings
340
+ if pipeline.settings_changed():
341
+ new_pipeline = self.update_settings(pipeline=pipeline, settings=pipeline.settings)
342
+ payload["settings"] = new_pipeline.to_json().get("settings", payload.get("settings"))
343
+
344
+ # update variables
345
+ if pipeline.variables_changed():
346
+ new_pipeline = self.__update_variables(pipeline=pipeline)
347
+ payload["variables"] = new_pipeline.to_json().get("variables", payload.get("variables"))
348
+
349
+ success, response = self._client_api.gen_request(
350
+ req_type="patch", path="/pipelines/{}".format(pipeline.id), json_req=payload
351
+ )
352
+
353
+ # exception handling
354
+ if not success:
355
+ raise exceptions.PlatformException(response)
356
+
357
+ # return entity
358
+ return entities.Pipeline.from_json(_json=response.json(), client_api=self._client_api, project=self._project)
359
+
360
+ @_api_reference.add(path="/pipelines", method="post")
361
+ def create(self, name: str = None, project_id: str = None, pipeline_json: dict = None) -> entities.Pipeline:
362
+ """
363
+ Create a new pipeline.
364
+
365
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
366
+
367
+ :param str name: pipeline name
368
+ :param str project_id: project id
369
+ :param dict pipeline_json: json containing the pipeline fields
370
+ :return: Pipeline object
371
+ :rtype: dtlpy.entities.pipeline.Pipeline
372
+
373
+ **Example**:
374
+
375
+ .. code-block:: python
376
+
377
+ pipeline = project.pipelines.create(name='pipeline_name')
378
+ """
379
+ if pipeline_json is None:
380
+ pipeline_json = generate_basic_pipeline()
381
+
382
+ if name is not None:
383
+ pipeline_json["name"] = name
384
+
385
+ if project_id is not None:
386
+ pipeline_json["projectId"] = project_id
387
+ else:
388
+ if not pipeline_json.get("projectId", None):
389
+ pipeline_json["projectId"] = self.project.id
390
+
391
+ success, response = self._client_api.gen_request(req_type="post", path="/pipelines", json_req=pipeline_json)
392
+ if success:
393
+ pipeline = entities.Pipeline.from_json(
394
+ client_api=self._client_api, _json=response.json(), project=self.project
395
+ )
396
+ else:
397
+ raise exceptions.PlatformException(response)
398
+ assert isinstance(pipeline, entities.Pipeline)
399
+ return pipeline
400
+
401
+ @_api_reference.add(path="/pipelines/{pipelineId}/install", method="post")
402
+ def install(self, pipeline: entities.Pipeline = None, resume_option: entities.PipelineResumeOption = None):
403
+ """
404
+ Install (start) a pipeline.
405
+
406
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
407
+
408
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
409
+ :param dtlpy.entities.pipeline.PipelineResumeOption resume_option: optional - resume pipeline method (what to do with existing cycles)
410
+ :return: Composition object
411
+
412
+ **Example**:
413
+
414
+ .. code-block:: python
415
+
416
+ project.pipelines.install(pipeline='pipeline_entity')
417
+ """
418
+
419
+ payload = {}
420
+ if resume_option:
421
+ payload["resumeOption"] = resume_option
422
+
423
+ success, response = self._client_api.gen_request(
424
+ req_type="post", path="/pipelines/{}/install".format(pipeline.id), json_req=payload
425
+ )
426
+
427
+ if not success:
428
+ raise exceptions.PlatformException(response)
429
+
430
+ return entities.Pipeline.from_json(client_api=self._client_api, _json=response.json(), project=self.project)
431
+
432
+ @_api_reference.add(path="/pipelines/{pipelineId}/uninstall", method="post")
433
+ def pause(self, pipeline: entities.Pipeline = None, keep_triggers_active: bool = None):
434
+ """
435
+ Pause a pipeline.
436
+
437
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
438
+
439
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
440
+ :param bool keep_triggers_active: Do we want the triggers to stay active and collect events
441
+ :return: Composition object
442
+
443
+ **Example**:
444
+
445
+ .. code-block:: python
446
+
447
+ project.pipelines.pause(pipeline='pipeline_entity')
448
+ """
449
+
450
+ payload = {}
451
+ if keep_triggers_active is not None:
452
+ payload["keepTriggersActive"] = keep_triggers_active
453
+
454
+ success, response = self._client_api.gen_request(
455
+ req_type="post", path="/pipelines/{}/uninstall".format(pipeline.id), json_req=payload
456
+ )
457
+
458
+ if not success:
459
+ raise exceptions.PlatformException(response)
460
+
461
+ @_api_reference.add(path="/pipelines/{pipelineId}/reset", method="post")
462
+ def reset(
463
+ self,
464
+ pipeline: entities.Pipeline = None,
465
+ pipeline_id: str = None,
466
+ pipeline_name: str = None,
467
+ stop_if_running: bool = False,
468
+ ):
469
+ """
470
+ Reset pipeline counters.
471
+
472
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
473
+
474
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity - optional
475
+ :param str pipeline_id: pipeline_id - optional
476
+ :param str pipeline_name: pipeline_name - optional
477
+ :param bool stop_if_running: If the pipeline is installed it will stop the pipeline and reset the counters.
478
+ :return: bool
479
+
480
+ **Example**:
481
+
482
+ .. code-block:: python
483
+
484
+ success = project.pipelines.reset(pipeline='pipeline_entity')
485
+ """
486
+
487
+ if pipeline_id is None:
488
+ if pipeline is None:
489
+ if pipeline_name is not None:
490
+ pipeline = self.get(pipeline_name=pipeline_name)
491
+ else:
492
+ raise exceptions.PlatformException(
493
+ "400", "Must provide one of pipeline, pipeline_id or pipeline_name"
494
+ )
495
+ pipeline_id = pipeline.id
496
+
497
+ if stop_if_running is True:
498
+ if pipeline is None:
499
+ pipeline = self.get(pipeline_id=pipeline_id)
500
+ pipeline.pause()
501
+
502
+ success, response = self._client_api.gen_request(
503
+ req_type="post", path="/pipelines/{}/reset".format(pipeline_id)
504
+ )
505
+
506
+ if not success:
507
+ raise exceptions.PlatformException(response)
508
+
509
+ return True
510
+
511
+ @_api_reference.add(path="/pipelines/{id}/statistics", method="get")
512
+ def stats(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, pipeline_name: str = None):
513
+ """
514
+ Get pipeline counters.
515
+
516
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
517
+
518
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity - optional
519
+ :param str pipeline_id: pipeline_id - optional
520
+ :param str pipeline_name: pipeline_name - optional
521
+ :return: PipelineStats
522
+ :rtype: dtlpy.entities.pipeline.PipelineStats
523
+
524
+ **Example**:
525
+
526
+ .. code-block:: python
527
+
528
+ pipeline_stats = project.pipelines.stats(pipeline='pipeline_entity')
529
+ """
530
+
531
+ if pipeline_id is None:
532
+ if pipeline is None:
533
+ if pipeline_name is not None:
534
+ pipeline = self.get(pipeline_name=pipeline_name)
535
+ else:
536
+ raise exceptions.PlatformException(
537
+ "400", "Must provide one of pipeline, pipeline_id or pipeline_name"
538
+ )
539
+ pipeline_id = pipeline.id
540
+
541
+ success, response = self._client_api.gen_request(
542
+ req_type="get", path="/pipelines/{}/statistics".format(pipeline_id)
543
+ )
544
+
545
+ if not success:
546
+ raise exceptions.PlatformException(response)
547
+
548
+ return entities.PipelineStats.from_json(_json=response.json())
549
+
550
+ def execute(
551
+ self,
552
+ pipeline: entities.Pipeline = None,
553
+ pipeline_id: str = None,
554
+ pipeline_name: str = None,
555
+ execution_input=None,
556
+ ):
557
+ """
558
+ Execute a pipeline and return the pipeline execution as an object.
559
+
560
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
561
+
562
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
563
+ :param str pipeline_id: pipeline id
564
+ :param str pipeline_name: pipeline name
565
+ :param execution_input: list of the dl.FunctionIO or dict of pipeline input - example {'item': 'item_id'}
566
+ :return: entities.PipelineExecution object
567
+ :rtype: dtlpy.entities.pipeline_execution.PipelineExecution
568
+
569
+ **Example**:
570
+
571
+ .. code-block:: python
572
+
573
+ pipeline_execution= project.pipelines.execute(pipeline='pipeline_entity', execution_input= {'item': 'item_id'} )
574
+ """
575
+
576
+ if pipeline is None and pipeline_id is None:
577
+ raise ValueError("Pipeline or pipeline_id must be provided")
578
+
579
+ if pipeline is None:
580
+ pipeline = self.get(pipeline_id=pipeline_id)
581
+
582
+ execution = repositories.PipelineExecutions(
583
+ pipeline=pipeline, client_api=self._client_api, project=self._project
584
+ ).create(pipeline_id=pipeline.id, execution_input=execution_input, test_mode=False)
585
+ return execution
586
+
587
+ def test(self, pipeline: entities.Pipeline = None, pipeline_id: str = None, execution_input=None):
588
+ """
589
+ Execute a pipeline in test mode and return the pipeline execution as an object.
590
+
591
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
592
+
593
+ :param dtlpy.entities.pipeline.Pipeline pipeline: pipeline entity
594
+ :param str pipeline_id: pipeline id
595
+ :param execution_input: list of the dl.FunctionIO or dict of pipeline input - example {'item': 'item_id'}
596
+ :return: entities.PipelineExecution object
597
+ :rtype: dtlpy.entities.pipeline_execution.PipelineExecution
598
+
599
+ **Example**:
600
+
601
+ .. code-block:: python
602
+
603
+ pipeline_execution= project.pipelines.test(pipeline='pipeline_entity', execution_input= {'item': 'item_id'} )
604
+ """
605
+ if pipeline is None and pipeline_id is None:
606
+ raise ValueError("Pipeline or pipeline_id must be provided")
607
+
608
+ if pipeline is None:
609
+ pipeline = self.get(pipeline_id=pipeline_id)
610
+
611
+ execution = repositories.PipelineExecutions(
612
+ pipeline=pipeline, client_api=self._client_api, project=self._project
613
+ ).create(pipeline_id=pipeline.id, execution_input=execution_input, test_mode=True)
614
+ return execution
615
+
616
+ @_api_reference.add(path="/pipelines/validate", method="post")
617
+ def validate(self, pipeline_json: dict):
618
+ """
619
+ Validate a pipeline configuration.
620
+
621
+ **prerequisites**: You must be an *owner* or *developer* to use this method.
622
+
623
+ :param dict pipeline_json: pipeline configuration as JSON dictionary
624
+ :return: Validation result
625
+ :rtype: dict
626
+
627
+ **Example**:
628
+
629
+ .. code-block:: python
630
+
631
+ validation_result = project.pipelines.validate(pipeline_json={'name': 'my_pipeline', 'nodes': []})
632
+ """
633
+ success, response = self._client_api.gen_request(
634
+ req_type="post", path="/pipelines/validate", json_req=pipeline_json
635
+ )
636
+
637
+ if not success:
638
+ raise exceptions.PlatformException(response)
639
+
640
+ return response.json()