dtlpy 1.113.10__py3-none-any.whl → 1.114.13__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.
- dtlpy/__init__.py +488 -488
- dtlpy/__version__.py +1 -1
- dtlpy/assets/__init__.py +26 -26
- dtlpy/assets/__pycache__/__init__.cpython-38.pyc +0 -0
- dtlpy/assets/code_server/config.yaml +2 -2
- dtlpy/assets/code_server/installation.sh +24 -24
- dtlpy/assets/code_server/launch.json +13 -13
- dtlpy/assets/code_server/settings.json +2 -2
- dtlpy/assets/main.py +53 -53
- dtlpy/assets/main_partial.py +18 -18
- dtlpy/assets/mock.json +11 -11
- dtlpy/assets/model_adapter.py +83 -83
- dtlpy/assets/package.json +61 -61
- dtlpy/assets/package_catalog.json +29 -29
- dtlpy/assets/package_gitignore +307 -307
- dtlpy/assets/service_runners/__init__.py +33 -33
- dtlpy/assets/service_runners/converter.py +96 -96
- dtlpy/assets/service_runners/multi_method.py +49 -49
- dtlpy/assets/service_runners/multi_method_annotation.py +54 -54
- dtlpy/assets/service_runners/multi_method_dataset.py +55 -55
- dtlpy/assets/service_runners/multi_method_item.py +52 -52
- dtlpy/assets/service_runners/multi_method_json.py +52 -52
- dtlpy/assets/service_runners/single_method.py +37 -37
- dtlpy/assets/service_runners/single_method_annotation.py +43 -43
- dtlpy/assets/service_runners/single_method_dataset.py +43 -43
- dtlpy/assets/service_runners/single_method_item.py +41 -41
- dtlpy/assets/service_runners/single_method_json.py +42 -42
- dtlpy/assets/service_runners/single_method_multi_input.py +45 -45
- dtlpy/assets/voc_annotation_template.xml +23 -23
- dtlpy/caches/base_cache.py +32 -32
- dtlpy/caches/cache.py +473 -473
- dtlpy/caches/dl_cache.py +201 -201
- dtlpy/caches/filesystem_cache.py +89 -89
- dtlpy/caches/redis_cache.py +84 -84
- dtlpy/dlp/__init__.py +20 -20
- dtlpy/dlp/cli_utilities.py +367 -367
- dtlpy/dlp/command_executor.py +764 -764
- dtlpy/dlp/dlp +1 -1
- dtlpy/dlp/dlp.bat +1 -1
- dtlpy/dlp/dlp.py +128 -128
- dtlpy/dlp/parser.py +651 -651
- dtlpy/entities/__init__.py +83 -83
- dtlpy/entities/analytic.py +311 -311
- dtlpy/entities/annotation.py +1879 -1879
- dtlpy/entities/annotation_collection.py +699 -699
- dtlpy/entities/annotation_definitions/__init__.py +20 -20
- dtlpy/entities/annotation_definitions/base_annotation_definition.py +100 -100
- dtlpy/entities/annotation_definitions/box.py +195 -195
- dtlpy/entities/annotation_definitions/classification.py +67 -67
- dtlpy/entities/annotation_definitions/comparison.py +72 -72
- dtlpy/entities/annotation_definitions/cube.py +204 -204
- dtlpy/entities/annotation_definitions/cube_3d.py +149 -149
- dtlpy/entities/annotation_definitions/description.py +32 -32
- dtlpy/entities/annotation_definitions/ellipse.py +124 -124
- dtlpy/entities/annotation_definitions/free_text.py +62 -62
- dtlpy/entities/annotation_definitions/gis.py +69 -69
- dtlpy/entities/annotation_definitions/note.py +139 -139
- dtlpy/entities/annotation_definitions/point.py +117 -117
- dtlpy/entities/annotation_definitions/polygon.py +182 -182
- dtlpy/entities/annotation_definitions/polyline.py +111 -111
- dtlpy/entities/annotation_definitions/pose.py +92 -92
- dtlpy/entities/annotation_definitions/ref_image.py +86 -86
- dtlpy/entities/annotation_definitions/segmentation.py +240 -240
- dtlpy/entities/annotation_definitions/subtitle.py +34 -34
- dtlpy/entities/annotation_definitions/text.py +85 -85
- dtlpy/entities/annotation_definitions/undefined_annotation.py +74 -74
- dtlpy/entities/app.py +220 -220
- dtlpy/entities/app_module.py +107 -107
- dtlpy/entities/artifact.py +174 -174
- dtlpy/entities/assignment.py +399 -399
- dtlpy/entities/base_entity.py +214 -214
- dtlpy/entities/bot.py +113 -113
- dtlpy/entities/codebase.py +296 -296
- dtlpy/entities/collection.py +38 -38
- dtlpy/entities/command.py +169 -169
- dtlpy/entities/compute.py +442 -442
- dtlpy/entities/dataset.py +1285 -1285
- dtlpy/entities/directory_tree.py +44 -44
- dtlpy/entities/dpk.py +470 -470
- dtlpy/entities/driver.py +222 -222
- dtlpy/entities/execution.py +397 -397
- dtlpy/entities/feature.py +124 -124
- dtlpy/entities/feature_set.py +145 -145
- dtlpy/entities/filters.py +641 -641
- dtlpy/entities/gis_item.py +107 -107
- dtlpy/entities/integration.py +184 -184
- dtlpy/entities/item.py +953 -953
- dtlpy/entities/label.py +123 -123
- dtlpy/entities/links.py +85 -85
- dtlpy/entities/message.py +175 -175
- dtlpy/entities/model.py +694 -691
- dtlpy/entities/node.py +1005 -1005
- dtlpy/entities/ontology.py +803 -803
- dtlpy/entities/organization.py +287 -287
- dtlpy/entities/package.py +657 -657
- dtlpy/entities/package_defaults.py +5 -5
- dtlpy/entities/package_function.py +185 -185
- dtlpy/entities/package_module.py +113 -113
- dtlpy/entities/package_slot.py +118 -118
- dtlpy/entities/paged_entities.py +290 -267
- dtlpy/entities/pipeline.py +593 -593
- dtlpy/entities/pipeline_execution.py +279 -279
- dtlpy/entities/project.py +394 -394
- dtlpy/entities/prompt_item.py +499 -499
- dtlpy/entities/recipe.py +301 -301
- dtlpy/entities/reflect_dict.py +102 -102
- dtlpy/entities/resource_execution.py +138 -138
- dtlpy/entities/service.py +958 -958
- dtlpy/entities/service_driver.py +117 -117
- dtlpy/entities/setting.py +294 -294
- dtlpy/entities/task.py +491 -491
- dtlpy/entities/time_series.py +143 -143
- dtlpy/entities/trigger.py +426 -426
- dtlpy/entities/user.py +118 -118
- dtlpy/entities/webhook.py +124 -124
- dtlpy/examples/__init__.py +19 -19
- dtlpy/examples/add_labels.py +135 -135
- dtlpy/examples/add_metadata_to_item.py +21 -21
- dtlpy/examples/annotate_items_using_model.py +65 -65
- dtlpy/examples/annotate_video_using_model_and_tracker.py +75 -75
- dtlpy/examples/annotations_convert_to_voc.py +9 -9
- dtlpy/examples/annotations_convert_to_yolo.py +9 -9
- dtlpy/examples/convert_annotation_types.py +51 -51
- dtlpy/examples/converter.py +143 -143
- dtlpy/examples/copy_annotations.py +22 -22
- dtlpy/examples/copy_folder.py +31 -31
- dtlpy/examples/create_annotations.py +51 -51
- dtlpy/examples/create_video_annotations.py +83 -83
- dtlpy/examples/delete_annotations.py +26 -26
- dtlpy/examples/filters.py +113 -113
- dtlpy/examples/move_item.py +23 -23
- dtlpy/examples/play_video_annotation.py +13 -13
- dtlpy/examples/show_item_and_mask.py +53 -53
- dtlpy/examples/triggers.py +49 -49
- dtlpy/examples/upload_batch_of_items.py +20 -20
- dtlpy/examples/upload_items_and_custom_format_annotations.py +55 -55
- dtlpy/examples/upload_items_with_modalities.py +43 -43
- dtlpy/examples/upload_segmentation_annotations_from_mask_image.py +44 -44
- dtlpy/examples/upload_yolo_format_annotations.py +70 -70
- dtlpy/exceptions.py +125 -125
- dtlpy/miscellaneous/__init__.py +20 -20
- dtlpy/miscellaneous/dict_differ.py +95 -95
- dtlpy/miscellaneous/git_utils.py +217 -217
- dtlpy/miscellaneous/json_utils.py +14 -14
- dtlpy/miscellaneous/list_print.py +105 -105
- dtlpy/miscellaneous/zipping.py +130 -130
- dtlpy/ml/__init__.py +20 -20
- dtlpy/ml/base_feature_extractor_adapter.py +27 -27
- dtlpy/ml/base_model_adapter.py +945 -940
- dtlpy/ml/metrics.py +461 -461
- dtlpy/ml/predictions_utils.py +274 -274
- dtlpy/ml/summary_writer.py +57 -57
- dtlpy/ml/train_utils.py +60 -60
- dtlpy/new_instance.py +252 -252
- dtlpy/repositories/__init__.py +56 -56
- dtlpy/repositories/analytics.py +85 -85
- dtlpy/repositories/annotations.py +916 -916
- dtlpy/repositories/apps.py +383 -383
- dtlpy/repositories/artifacts.py +452 -452
- dtlpy/repositories/assignments.py +599 -599
- dtlpy/repositories/bots.py +213 -213
- dtlpy/repositories/codebases.py +559 -559
- dtlpy/repositories/collections.py +332 -348
- dtlpy/repositories/commands.py +158 -158
- dtlpy/repositories/compositions.py +61 -61
- dtlpy/repositories/computes.py +434 -406
- dtlpy/repositories/datasets.py +1291 -1291
- dtlpy/repositories/downloader.py +895 -895
- dtlpy/repositories/dpks.py +433 -433
- dtlpy/repositories/drivers.py +266 -266
- dtlpy/repositories/executions.py +817 -817
- dtlpy/repositories/feature_sets.py +226 -226
- dtlpy/repositories/features.py +238 -238
- dtlpy/repositories/integrations.py +484 -484
- dtlpy/repositories/items.py +909 -915
- dtlpy/repositories/messages.py +94 -94
- dtlpy/repositories/models.py +877 -867
- dtlpy/repositories/nodes.py +80 -80
- dtlpy/repositories/ontologies.py +511 -511
- dtlpy/repositories/organizations.py +525 -525
- dtlpy/repositories/packages.py +1941 -1941
- dtlpy/repositories/pipeline_executions.py +448 -448
- dtlpy/repositories/pipelines.py +642 -642
- dtlpy/repositories/projects.py +539 -539
- dtlpy/repositories/recipes.py +399 -399
- dtlpy/repositories/resource_executions.py +137 -137
- dtlpy/repositories/schema.py +120 -120
- dtlpy/repositories/service_drivers.py +213 -213
- dtlpy/repositories/services.py +1704 -1704
- dtlpy/repositories/settings.py +339 -339
- dtlpy/repositories/tasks.py +1124 -1124
- dtlpy/repositories/times_series.py +278 -278
- dtlpy/repositories/triggers.py +536 -536
- dtlpy/repositories/upload_element.py +257 -257
- dtlpy/repositories/uploader.py +651 -651
- dtlpy/repositories/webhooks.py +249 -249
- dtlpy/services/__init__.py +22 -22
- dtlpy/services/aihttp_retry.py +131 -131
- dtlpy/services/api_client.py +1782 -1782
- dtlpy/services/api_reference.py +40 -40
- dtlpy/services/async_utils.py +133 -133
- dtlpy/services/calls_counter.py +44 -44
- dtlpy/services/check_sdk.py +68 -68
- dtlpy/services/cookie.py +115 -115
- dtlpy/services/create_logger.py +156 -156
- dtlpy/services/events.py +84 -84
- dtlpy/services/logins.py +235 -235
- dtlpy/services/reporter.py +256 -256
- dtlpy/services/service_defaults.py +91 -91
- dtlpy/utilities/__init__.py +20 -20
- dtlpy/utilities/annotations/__init__.py +16 -16
- dtlpy/utilities/annotations/annotation_converters.py +269 -269
- dtlpy/utilities/base_package_runner.py +264 -264
- dtlpy/utilities/converter.py +1650 -1650
- dtlpy/utilities/dataset_generators/__init__.py +1 -1
- dtlpy/utilities/dataset_generators/dataset_generator.py +670 -670
- dtlpy/utilities/dataset_generators/dataset_generator_tensorflow.py +23 -23
- dtlpy/utilities/dataset_generators/dataset_generator_torch.py +21 -21
- dtlpy/utilities/local_development/__init__.py +1 -1
- dtlpy/utilities/local_development/local_session.py +179 -179
- dtlpy/utilities/reports/__init__.py +2 -2
- dtlpy/utilities/reports/figures.py +343 -343
- dtlpy/utilities/reports/report.py +71 -71
- dtlpy/utilities/videos/__init__.py +17 -17
- dtlpy/utilities/videos/video_player.py +598 -598
- dtlpy/utilities/videos/videos.py +470 -470
- {dtlpy-1.113.10.data → dtlpy-1.114.13.data}/scripts/dlp +1 -1
- dtlpy-1.114.13.data/scripts/dlp.bat +2 -0
- {dtlpy-1.113.10.data → dtlpy-1.114.13.data}/scripts/dlp.py +128 -128
- {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/LICENSE +200 -200
- {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/METADATA +172 -172
- dtlpy-1.114.13.dist-info/RECORD +240 -0
- {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/WHEEL +1 -1
- tests/features/environment.py +551 -550
- dtlpy-1.113.10.data/scripts/dlp.bat +0 -2
- dtlpy-1.113.10.dist-info/RECORD +0 -244
- tests/assets/__init__.py +0 -0
- tests/assets/models_flow/__init__.py +0 -0
- tests/assets/models_flow/failedmain.py +0 -52
- tests/assets/models_flow/main.py +0 -62
- tests/assets/models_flow/main_model.py +0 -54
- {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/entry_points.txt +0 -0
- {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/top_level.txt +0 -0
dtlpy/repositories/dpks.py
CHANGED
|
@@ -1,434 +1,434 @@
|
|
|
1
|
-
import json
|
|
2
|
-
import logging
|
|
3
|
-
import os
|
|
4
|
-
from typing import List, Optional
|
|
5
|
-
from pathlib import Path
|
|
6
|
-
|
|
7
|
-
from .. import exceptions, entities, services, miscellaneous, assets
|
|
8
|
-
from ..services.api_client import ApiClient
|
|
9
|
-
|
|
10
|
-
logger = logging.getLogger(name='dtlpy')
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
class Dpks:
|
|
14
|
-
def __init__(self, client_api: ApiClient, project: entities.Project = None):
|
|
15
|
-
self._client_api = client_api
|
|
16
|
-
self._project = project
|
|
17
|
-
|
|
18
|
-
@property
|
|
19
|
-
def project(self) -> Optional[entities.Project]:
|
|
20
|
-
return self._project
|
|
21
|
-
|
|
22
|
-
@project.setter
|
|
23
|
-
def project(self, project: entities.Project):
|
|
24
|
-
if not isinstance(project, entities.Project):
|
|
25
|
-
raise ValueError('Must input a valid Project entity')
|
|
26
|
-
self._project = project
|
|
27
|
-
|
|
28
|
-
def init(self, directory: str = None, name: str = None, description: str = None,
|
|
29
|
-
attributes: dict = None, icon: str = None, scope: str = None):
|
|
30
|
-
"""
|
|
31
|
-
Initialize a dpk project with the specified projects.
|
|
32
|
-
|
|
33
|
-
:param str directory: the directory where to initialize the project
|
|
34
|
-
:param str name: the name of the dpk.
|
|
35
|
-
:param str description: the description of the dpk.
|
|
36
|
-
:param str attributes: the attributes of the dpk.
|
|
37
|
-
:param str icon: the icon of the dpk.
|
|
38
|
-
:param str scope: the scope of the dpk.
|
|
39
|
-
|
|
40
|
-
** Example **
|
|
41
|
-
.. code-block:: python
|
|
42
|
-
dl.dpks.init(name='Hello World', description='A description of the dpk', attributes={
|
|
43
|
-
"Provider": "Dataloop",
|
|
44
|
-
"License": "",
|
|
45
|
-
"Category": "Model",
|
|
46
|
-
"Computer Vision": "Object Detection",
|
|
47
|
-
"Media Type": "Image"
|
|
48
|
-
},
|
|
49
|
-
icon='path_to_icon', scope='organization')
|
|
50
|
-
"""
|
|
51
|
-
if directory is None:
|
|
52
|
-
directory = os.getcwd()
|
|
53
|
-
dpk = entities.Dpk.from_json(_json={'name': miscellaneous.JsonUtils.get_if_absent(name),
|
|
54
|
-
'description': miscellaneous.JsonUtils.get_if_absent(description),
|
|
55
|
-
'attributes': miscellaneous.JsonUtils.get_if_absent(attributes),
|
|
56
|
-
'icon': miscellaneous.JsonUtils.get_if_absent(icon),
|
|
57
|
-
'scope': miscellaneous.JsonUtils.get_if_absent(scope, 'organization'),
|
|
58
|
-
'components': dict()
|
|
59
|
-
},
|
|
60
|
-
client_api=self._client_api)
|
|
61
|
-
dataloop_filepath = os.path.join(directory, assets.paths.APP_JSON_FILENAME)
|
|
62
|
-
with open(dataloop_filepath, 'w') as json_file:
|
|
63
|
-
json_file.write(json.dumps(dpk.to_json(), indent=4))
|
|
64
|
-
os.makedirs(os.path.join(directory, 'functions'), exist_ok=True)
|
|
65
|
-
os.makedirs(os.path.join(directory, 'panels'), exist_ok=True)
|
|
66
|
-
|
|
67
|
-
# noinspection PyMethodMayBeStatic
|
|
68
|
-
def pack(self, directory: str = None, name: str = None, subpaths_to_append: List[str] = None) -> str:
|
|
69
|
-
"""
|
|
70
|
-
:param str directory: optional - the project to pack, if not specified use the current project,
|
|
71
|
-
:param str name: optional - the name of the dpk file.
|
|
72
|
-
:param List[str] subpaths_to_append: optional - the files/directories to add to the dpk file.
|
|
73
|
-
(along with functions, panels and dataloop.json)
|
|
74
|
-
:return the path of the dpk file
|
|
75
|
-
|
|
76
|
-
**Example**
|
|
77
|
-
|
|
78
|
-
.. code-block:: python
|
|
79
|
-
filepath = dl.apps.pack(directory='/my-current-project', name='project-name')
|
|
80
|
-
"""
|
|
81
|
-
# create/get .dataloop dir
|
|
82
|
-
cwd = os.getcwd()
|
|
83
|
-
dl_dir = os.path.join(cwd, '.dataloop')
|
|
84
|
-
if not os.path.isdir(dl_dir):
|
|
85
|
-
os.mkdir(dl_dir)
|
|
86
|
-
if directory is None:
|
|
87
|
-
directory = cwd
|
|
88
|
-
|
|
89
|
-
# get dpk name
|
|
90
|
-
if name is None:
|
|
91
|
-
name = os.path.basename(directory)
|
|
92
|
-
|
|
93
|
-
with open(os.path.join(directory, 'dataloop.json'), 'r') as file:
|
|
94
|
-
app_json = json.load(file)
|
|
95
|
-
version = app_json.get('version', None)
|
|
96
|
-
if version is None:
|
|
97
|
-
logger.warning('No Version specified, setting to 1.0.0')
|
|
98
|
-
version = '1.0.0'
|
|
99
|
-
# create/get dist folder
|
|
100
|
-
dpk_filename = os.path.join(dl_dir, '{}_{}.dpk'.format(name, version))
|
|
101
|
-
|
|
102
|
-
if not os.path.isdir(directory):
|
|
103
|
-
raise ValueError('Not a directory: {}'.format(directory))
|
|
104
|
-
if subpaths_to_append is None:
|
|
105
|
-
subpaths_to_append = []
|
|
106
|
-
|
|
107
|
-
try:
|
|
108
|
-
directory = os.path.abspath(directory)
|
|
109
|
-
# create zipfile
|
|
110
|
-
miscellaneous.Zipping.zip_directory(zip_filename=dpk_filename,
|
|
111
|
-
directory=directory,
|
|
112
|
-
ignore_directories=['artifacts'])
|
|
113
|
-
return dpk_filename
|
|
114
|
-
except Exception:
|
|
115
|
-
logger.error('Error when packing:')
|
|
116
|
-
raise
|
|
117
|
-
|
|
118
|
-
def _build_entities_from_response(self, response_items) -> miscellaneous.List[entities.Dpk]:
|
|
119
|
-
pool = self._client_api.thread_pools(pool_name='entity.create')
|
|
120
|
-
jobs = [None for _ in range(len(response_items))]
|
|
121
|
-
# return triggers list
|
|
122
|
-
for i_item, item in enumerate(response_items):
|
|
123
|
-
jobs[i_item] = pool.submit(entities.Dpk._protected_from_json,
|
|
124
|
-
**{'client_api': self._client_api,
|
|
125
|
-
'_json': item,
|
|
126
|
-
'project': self._project})
|
|
127
|
-
# get all results
|
|
128
|
-
results = [j.result() for j in jobs]
|
|
129
|
-
# log errors
|
|
130
|
-
_ = [logger.warning(r[1]) for r in results if r[0] is False]
|
|
131
|
-
# return good jobs
|
|
132
|
-
items = miscellaneous.List([r[1] for r in results if r[0] is True])
|
|
133
|
-
return items
|
|
134
|
-
|
|
135
|
-
def pull(self, dpk: entities.Dpk = None, dpk_id: str = None, dpk_name: str = None, local_path=None) -> str:
|
|
136
|
-
"""
|
|
137
|
-
Pulls the app from the platform as dpk file.
|
|
138
|
-
|
|
139
|
-
Note: you must pass either dpk_name or dpk_id to the function.
|
|
140
|
-
:param str dpk: DPK entity.
|
|
141
|
-
:param str dpk_id: the name of the dpk.
|
|
142
|
-
:param str dpk_name: the id of the dpk.
|
|
143
|
-
:param local_path: the path where you want to install the dpk file.
|
|
144
|
-
:return local path where the package pull
|
|
145
|
-
|
|
146
|
-
**Example**
|
|
147
|
-
..code-block:: python
|
|
148
|
-
path = dl.dpks.pull(dpk_name='my-app')
|
|
149
|
-
"""
|
|
150
|
-
if dpk is None:
|
|
151
|
-
dpk = self.get(dpk_id=dpk_id, dpk_name=dpk_name)
|
|
152
|
-
if local_path is None:
|
|
153
|
-
local_path = os.path.join(
|
|
154
|
-
services.service_defaults.DATALOOP_PATH,
|
|
155
|
-
"dpks",
|
|
156
|
-
dpk.name,
|
|
157
|
-
str(dpk.version))
|
|
158
|
-
|
|
159
|
-
dpk.codebases.unpack(codebase=dpk.codebase, local_path=local_path)
|
|
160
|
-
return local_path
|
|
161
|
-
|
|
162
|
-
def __get_by_name(self, dpk_name: str, dpk_version: str = None):
|
|
163
|
-
filters = entities.Filters(field='name',
|
|
164
|
-
values=dpk_name,
|
|
165
|
-
resource=entities.FiltersResource.DPK,
|
|
166
|
-
use_defaults=False)
|
|
167
|
-
dpks = self.list(filters=filters)
|
|
168
|
-
# only latest version returns so should be only one
|
|
169
|
-
if dpks.items_count == 0:
|
|
170
|
-
raise exceptions.PlatformException(error='404',
|
|
171
|
-
message='Dpk not found. Name: {}'.format(dpk_name))
|
|
172
|
-
elif dpks.items_count > 1:
|
|
173
|
-
raise exceptions.PlatformException(
|
|
174
|
-
error='400',
|
|
175
|
-
message='More than one dpk found by the name of: {}'.format(dpk_name))
|
|
176
|
-
dpk: entities.Dpk = dpks.items[0]
|
|
177
|
-
|
|
178
|
-
########################
|
|
179
|
-
# get specific version #
|
|
180
|
-
########################
|
|
181
|
-
if dpk_version is not None and dpk.version != dpk_version:
|
|
182
|
-
filters = entities.Filters(field='version',
|
|
183
|
-
values=dpk_version,
|
|
184
|
-
resource=entities.FiltersResource.DPK,
|
|
185
|
-
use_defaults=False)
|
|
186
|
-
dpk_v = dpk._get_revision_pages(filters=filters)
|
|
187
|
-
if dpk_v.items_count == 0:
|
|
188
|
-
raise exceptions.PlatformException(
|
|
189
|
-
error='404',
|
|
190
|
-
message=f'Dpk version not found. Name: {dpk_name}, version: {dpk_version}')
|
|
191
|
-
elif dpk_v.items_count > 1:
|
|
192
|
-
# should never be here - more than one with same name and version
|
|
193
|
-
raise exceptions.PlatformException(
|
|
194
|
-
error='400',
|
|
195
|
-
message=f'More than one dpk found with name: {dpk_name}, version: {dpk_version}')
|
|
196
|
-
dpk = dpk_v.items[0]
|
|
197
|
-
return dpk
|
|
198
|
-
|
|
199
|
-
def publish(
|
|
200
|
-
self,
|
|
201
|
-
dpk: entities.Dpk = None,
|
|
202
|
-
ignore_max_file_size: bool = False,
|
|
203
|
-
manifest_filepath='dataloop.json',
|
|
204
|
-
local_path: str = None
|
|
205
|
-
) -> entities.Dpk:
|
|
206
|
-
"""
|
|
207
|
-
Upload a dpk entity to the dataloop platform.
|
|
208
|
-
|
|
209
|
-
:param entities.Dpk dpk: Optional. The DPK entity to publish. If None, a new DPK is created
|
|
210
|
-
from the manifest file.
|
|
211
|
-
:param bool ignore_max_file_size: Optional. If True, the maximum file size check is ignored
|
|
212
|
-
during the packaging of the codebase.
|
|
213
|
-
:param str manifest_filepath: Optional. Path to the manifest file. Can be absolute or relative.
|
|
214
|
-
Defaults to 'dataloop.json'
|
|
215
|
-
:param str local_path: Optional. The path where the dpk files are located.
|
|
216
|
-
|
|
217
|
-
:return the published dpk
|
|
218
|
-
:rtype dl.entities.Dpk
|
|
219
|
-
|
|
220
|
-
**Example**
|
|
221
|
-
|
|
222
|
-
.. code-block:: python
|
|
223
|
-
published_dpk = dl.dpks.publish()
|
|
224
|
-
"""
|
|
225
|
-
manifest_path = Path(manifest_filepath).resolve()
|
|
226
|
-
|
|
227
|
-
if dpk is None:
|
|
228
|
-
if not manifest_path.exists():
|
|
229
|
-
raise FileNotFoundError(f'{manifest_filepath} file must exist in order to publish a dpk')
|
|
230
|
-
with open(manifest_filepath, 'r') as f:
|
|
231
|
-
json_file = json.load(f)
|
|
232
|
-
dpk = entities.Dpk.from_json(_json=json_file,
|
|
233
|
-
client_api=self._client_api,
|
|
234
|
-
project=self.project)
|
|
235
|
-
|
|
236
|
-
if not dpk.context:
|
|
237
|
-
dpk.context = {}
|
|
238
|
-
if 'project' not in dpk.context:
|
|
239
|
-
if not self.project:
|
|
240
|
-
raise exceptions.PlatformException('400', 'project id must be provided in the context')
|
|
241
|
-
dpk.context['project'] = self.project.id
|
|
242
|
-
if 'org' not in dpk.context and dpk.scope == 'organization':
|
|
243
|
-
if not self.project:
|
|
244
|
-
raise exceptions.PlatformException('400', 'org id must be provided in the context')
|
|
245
|
-
dpk.context['org'] = self.project.org['id']
|
|
246
|
-
|
|
247
|
-
if self.project and self.project.id != dpk.context['project']:
|
|
248
|
-
logger.warning("the project id that provide different from the dpk project id")
|
|
249
|
-
|
|
250
|
-
if local_path is None:
|
|
251
|
-
if manifest_filepath == 'dataloop.json':
|
|
252
|
-
local_path = os.getcwd()
|
|
253
|
-
else:
|
|
254
|
-
local_path = os.path.dirname(manifest_filepath)
|
|
255
|
-
if dpk.codebase is None:
|
|
256
|
-
dpk.codebase = self.project.codebases.pack(directory=local_path,
|
|
257
|
-
name=dpk.display_name,
|
|
258
|
-
extension='dpk',
|
|
259
|
-
ignore_directories=['artifacts'],
|
|
260
|
-
ignore_max_file_size=ignore_max_file_size)
|
|
261
|
-
|
|
262
|
-
success_pack, response_pack = self._client_api.gen_request(req_type='post',
|
|
263
|
-
json_req=dpk.to_json(),
|
|
264
|
-
path='/app-registry')
|
|
265
|
-
if not success_pack:
|
|
266
|
-
raise exceptions.PlatformException(error=response_pack)
|
|
267
|
-
|
|
268
|
-
return entities.Dpk.from_json(response_pack.json(), self._client_api, dpk.project)
|
|
269
|
-
|
|
270
|
-
def update(self, dpk: entities.Dpk) -> entities.Dpk:
|
|
271
|
-
"""
|
|
272
|
-
Update the dpk entity in the platform.
|
|
273
|
-
|
|
274
|
-
Note: the update will update only attributes, displayName, Icon, description and pipeline template preview
|
|
275
|
-
|
|
276
|
-
:param entities.Dpk dpk: the dpk entity to update.
|
|
277
|
-
:return the updated dpk entity.
|
|
278
|
-
:rtype entities.Dpk
|
|
279
|
-
"""
|
|
280
|
-
success, response = self._client_api.gen_request(req_type='patch',
|
|
281
|
-
path='/app-registry/{}'.format(dpk.id),
|
|
282
|
-
json_req=dpk.to_json())
|
|
283
|
-
if not success:
|
|
284
|
-
raise exceptions.PlatformException(response)
|
|
285
|
-
res = response.json()
|
|
286
|
-
logger.info(res.get('message'))
|
|
287
|
-
return entities.Dpk.from_json(res.get('dpk'), self._client_api, self._project)
|
|
288
|
-
|
|
289
|
-
def delete(self, dpk_id: str) -> bool:
|
|
290
|
-
"""
|
|
291
|
-
Delete the dpk from the app store.
|
|
292
|
-
|
|
293
|
-
Note: after removing the dpk, you cant get it again, it's advised to pull it first.
|
|
294
|
-
|
|
295
|
-
:param str dpk_id: the id of the dpk.
|
|
296
|
-
:return whether the operation ran successfully
|
|
297
|
-
:rtype bool
|
|
298
|
-
"""
|
|
299
|
-
success, response = self._client_api.gen_request(req_type='delete', path=f'/app-registry/{dpk_id}')
|
|
300
|
-
if success:
|
|
301
|
-
logger.info(f'Deleted dpk: {dpk_id} successfully')
|
|
302
|
-
else:
|
|
303
|
-
raise exceptions.PlatformException(response)
|
|
304
|
-
return success
|
|
305
|
-
|
|
306
|
-
def list(self, filters: entities.Filters = None) -> entities.PagedEntities:
|
|
307
|
-
"""
|
|
308
|
-
List the available dpks.
|
|
309
|
-
|
|
310
|
-
:param entities.Filters filters: the filters to apply on the list
|
|
311
|
-
:return a paged entity representing the list of dpks.
|
|
312
|
-
|
|
313
|
-
** Example **
|
|
314
|
-
.. code-block:: python
|
|
315
|
-
dpks = dl.dpks.list()
|
|
316
|
-
"""
|
|
317
|
-
if filters is None:
|
|
318
|
-
filters = entities.Filters(resource=entities.FiltersResource.DPK)
|
|
319
|
-
elif not isinstance(filters, entities.Filters):
|
|
320
|
-
raise ValueError('Unknown filters type: {!r}'.format(type(filters)))
|
|
321
|
-
elif filters.resource != entities.FiltersResource.DPK:
|
|
322
|
-
raise ValueError('Filters resource must to be FiltersResource.DPK. Got: {!r}'.format(filters.resource))
|
|
323
|
-
|
|
324
|
-
if self._project is not None:
|
|
325
|
-
filters.add(field='context.project', values=self._project.id)
|
|
326
|
-
|
|
327
|
-
paged = entities.PagedEntities(items_repository=self,
|
|
328
|
-
filters=filters,
|
|
329
|
-
page_offset=filters.page,
|
|
330
|
-
page_size=filters.page_size,
|
|
331
|
-
client_api=self._client_api)
|
|
332
|
-
paged.get_page()
|
|
333
|
-
return paged
|
|
334
|
-
|
|
335
|
-
def _list(self, filters: entities.Filters):
|
|
336
|
-
url = '/app-registry/query'
|
|
337
|
-
|
|
338
|
-
# request
|
|
339
|
-
success, response = self._client_api.gen_request(req_type='post',
|
|
340
|
-
path=url,
|
|
341
|
-
json_req=filters.prepare())
|
|
342
|
-
if not success:
|
|
343
|
-
raise exceptions.PlatformException(response)
|
|
344
|
-
return response.json()
|
|
345
|
-
|
|
346
|
-
def get_revisions(self, dpk_id: str, version: str):
|
|
347
|
-
"""
|
|
348
|
-
Get the revision of a specific dpk.
|
|
349
|
-
|
|
350
|
-
:param str dpk_id: the id of the dpk.
|
|
351
|
-
:param str version: the version of the dpk.
|
|
352
|
-
:return the entity of the dpk
|
|
353
|
-
:rtype entities.Dpk
|
|
354
|
-
|
|
355
|
-
** Example **
|
|
356
|
-
..coed-block:: python
|
|
357
|
-
dpk = dl.dpks.get_revisions(dpk_id='id', version='1.0.0')
|
|
358
|
-
"""
|
|
359
|
-
if dpk_id is None or version is None:
|
|
360
|
-
raise ValueError('You must provide both dpk_id and version')
|
|
361
|
-
url = '/app-registry/{}/revisions/{}'.format(dpk_id, version)
|
|
362
|
-
|
|
363
|
-
# request
|
|
364
|
-
success, response = self._client_api.gen_request(req_type='get',
|
|
365
|
-
path=url)
|
|
366
|
-
if not success:
|
|
367
|
-
raise exceptions.PlatformException(response)
|
|
368
|
-
|
|
369
|
-
dpk = entities.Dpk.from_json(_json=response.json(),
|
|
370
|
-
client_api=self._client_api,
|
|
371
|
-
project=self._project,
|
|
372
|
-
is_fetched=False)
|
|
373
|
-
return dpk
|
|
374
|
-
|
|
375
|
-
def get(self, dpk_name: str = None, dpk_version: str = None, dpk_id: str = None) -> entities.Dpk:
|
|
376
|
-
"""
|
|
377
|
-
Get a specific dpk from the platform.
|
|
378
|
-
|
|
379
|
-
Note: you must pass either dpk_id or dpk_name.
|
|
380
|
-
|
|
381
|
-
:param str dpk_id: the id of the dpk to get.
|
|
382
|
-
:param str dpk_name: the name of the dpk to get.
|
|
383
|
-
:param str dpk_version: options - to get a specific dpk version
|
|
384
|
-
:return the entity of the dpk
|
|
385
|
-
:rtype entities.Dpk
|
|
386
|
-
|
|
387
|
-
** Example **
|
|
388
|
-
..coed-block:: python
|
|
389
|
-
dpk = dl.dpks.get(dpk_name='name')
|
|
390
|
-
"""
|
|
391
|
-
if dpk_id is None and dpk_name is None:
|
|
392
|
-
raise ValueError('You must provide an identifier, either dpk_id or dpk_name')
|
|
393
|
-
if dpk_id is not None:
|
|
394
|
-
url = '/app-registry/{}'.format(dpk_id)
|
|
395
|
-
|
|
396
|
-
# request
|
|
397
|
-
success, response = self._client_api.gen_request(req_type='get',
|
|
398
|
-
path=url)
|
|
399
|
-
if not success:
|
|
400
|
-
raise exceptions.PlatformException(response)
|
|
401
|
-
|
|
402
|
-
dpk = entities.Dpk.from_json(_json=response.json(),
|
|
403
|
-
client_api=self._client_api,
|
|
404
|
-
project=self._project,
|
|
405
|
-
is_fetched=False)
|
|
406
|
-
else:
|
|
407
|
-
dpk = self.__get_by_name(dpk_name=dpk_name, dpk_version=dpk_version)
|
|
408
|
-
|
|
409
|
-
return dpk
|
|
410
|
-
|
|
411
|
-
def get_previews(self, dpk: entities.Dpk):
|
|
412
|
-
"""
|
|
413
|
-
Get the preview of a specific dpk.
|
|
414
|
-
|
|
415
|
-
:param entities.Dpk dpk: the dpk entity to get the preview for.
|
|
416
|
-
:return the preview of the dpk's templates
|
|
417
|
-
:rtype dict
|
|
418
|
-
|
|
419
|
-
** Example **
|
|
420
|
-
..coed-block:: python
|
|
421
|
-
res = dl.dpks.get_previews(dpk=dpk)
|
|
422
|
-
"""
|
|
423
|
-
url = '/app-registry/{}/previews'.format(dpk.id)
|
|
424
|
-
|
|
425
|
-
# request
|
|
426
|
-
success, response = self._client_api.gen_request(
|
|
427
|
-
req_type='get',
|
|
428
|
-
path=url
|
|
429
|
-
)
|
|
430
|
-
|
|
431
|
-
if not success:
|
|
432
|
-
raise exceptions.PlatformException(response)
|
|
433
|
-
|
|
1
|
+
import json
|
|
2
|
+
import logging
|
|
3
|
+
import os
|
|
4
|
+
from typing import List, Optional
|
|
5
|
+
from pathlib import Path
|
|
6
|
+
|
|
7
|
+
from .. import exceptions, entities, services, miscellaneous, assets
|
|
8
|
+
from ..services.api_client import ApiClient
|
|
9
|
+
|
|
10
|
+
logger = logging.getLogger(name='dtlpy')
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
class Dpks:
|
|
14
|
+
def __init__(self, client_api: ApiClient, project: entities.Project = None):
|
|
15
|
+
self._client_api = client_api
|
|
16
|
+
self._project = project
|
|
17
|
+
|
|
18
|
+
@property
|
|
19
|
+
def project(self) -> Optional[entities.Project]:
|
|
20
|
+
return self._project
|
|
21
|
+
|
|
22
|
+
@project.setter
|
|
23
|
+
def project(self, project: entities.Project):
|
|
24
|
+
if not isinstance(project, entities.Project):
|
|
25
|
+
raise ValueError('Must input a valid Project entity')
|
|
26
|
+
self._project = project
|
|
27
|
+
|
|
28
|
+
def init(self, directory: str = None, name: str = None, description: str = None,
|
|
29
|
+
attributes: dict = None, icon: str = None, scope: str = None):
|
|
30
|
+
"""
|
|
31
|
+
Initialize a dpk project with the specified projects.
|
|
32
|
+
|
|
33
|
+
:param str directory: the directory where to initialize the project
|
|
34
|
+
:param str name: the name of the dpk.
|
|
35
|
+
:param str description: the description of the dpk.
|
|
36
|
+
:param str attributes: the attributes of the dpk.
|
|
37
|
+
:param str icon: the icon of the dpk.
|
|
38
|
+
:param str scope: the scope of the dpk.
|
|
39
|
+
|
|
40
|
+
** Example **
|
|
41
|
+
.. code-block:: python
|
|
42
|
+
dl.dpks.init(name='Hello World', description='A description of the dpk', attributes={
|
|
43
|
+
"Provider": "Dataloop",
|
|
44
|
+
"License": "",
|
|
45
|
+
"Category": "Model",
|
|
46
|
+
"Computer Vision": "Object Detection",
|
|
47
|
+
"Media Type": "Image"
|
|
48
|
+
},
|
|
49
|
+
icon='path_to_icon', scope='organization')
|
|
50
|
+
"""
|
|
51
|
+
if directory is None:
|
|
52
|
+
directory = os.getcwd()
|
|
53
|
+
dpk = entities.Dpk.from_json(_json={'name': miscellaneous.JsonUtils.get_if_absent(name),
|
|
54
|
+
'description': miscellaneous.JsonUtils.get_if_absent(description),
|
|
55
|
+
'attributes': miscellaneous.JsonUtils.get_if_absent(attributes),
|
|
56
|
+
'icon': miscellaneous.JsonUtils.get_if_absent(icon),
|
|
57
|
+
'scope': miscellaneous.JsonUtils.get_if_absent(scope, 'organization'),
|
|
58
|
+
'components': dict()
|
|
59
|
+
},
|
|
60
|
+
client_api=self._client_api)
|
|
61
|
+
dataloop_filepath = os.path.join(directory, assets.paths.APP_JSON_FILENAME)
|
|
62
|
+
with open(dataloop_filepath, 'w') as json_file:
|
|
63
|
+
json_file.write(json.dumps(dpk.to_json(), indent=4))
|
|
64
|
+
os.makedirs(os.path.join(directory, 'functions'), exist_ok=True)
|
|
65
|
+
os.makedirs(os.path.join(directory, 'panels'), exist_ok=True)
|
|
66
|
+
|
|
67
|
+
# noinspection PyMethodMayBeStatic
|
|
68
|
+
def pack(self, directory: str = None, name: str = None, subpaths_to_append: List[str] = None) -> str:
|
|
69
|
+
"""
|
|
70
|
+
:param str directory: optional - the project to pack, if not specified use the current project,
|
|
71
|
+
:param str name: optional - the name of the dpk file.
|
|
72
|
+
:param List[str] subpaths_to_append: optional - the files/directories to add to the dpk file.
|
|
73
|
+
(along with functions, panels and dataloop.json)
|
|
74
|
+
:return the path of the dpk file
|
|
75
|
+
|
|
76
|
+
**Example**
|
|
77
|
+
|
|
78
|
+
.. code-block:: python
|
|
79
|
+
filepath = dl.apps.pack(directory='/my-current-project', name='project-name')
|
|
80
|
+
"""
|
|
81
|
+
# create/get .dataloop dir
|
|
82
|
+
cwd = os.getcwd()
|
|
83
|
+
dl_dir = os.path.join(cwd, '.dataloop')
|
|
84
|
+
if not os.path.isdir(dl_dir):
|
|
85
|
+
os.mkdir(dl_dir)
|
|
86
|
+
if directory is None:
|
|
87
|
+
directory = cwd
|
|
88
|
+
|
|
89
|
+
# get dpk name
|
|
90
|
+
if name is None:
|
|
91
|
+
name = os.path.basename(directory)
|
|
92
|
+
|
|
93
|
+
with open(os.path.join(directory, 'dataloop.json'), 'r') as file:
|
|
94
|
+
app_json = json.load(file)
|
|
95
|
+
version = app_json.get('version', None)
|
|
96
|
+
if version is None:
|
|
97
|
+
logger.warning('No Version specified, setting to 1.0.0')
|
|
98
|
+
version = '1.0.0'
|
|
99
|
+
# create/get dist folder
|
|
100
|
+
dpk_filename = os.path.join(dl_dir, '{}_{}.dpk'.format(name, version))
|
|
101
|
+
|
|
102
|
+
if not os.path.isdir(directory):
|
|
103
|
+
raise ValueError('Not a directory: {}'.format(directory))
|
|
104
|
+
if subpaths_to_append is None:
|
|
105
|
+
subpaths_to_append = []
|
|
106
|
+
|
|
107
|
+
try:
|
|
108
|
+
directory = os.path.abspath(directory)
|
|
109
|
+
# create zipfile
|
|
110
|
+
miscellaneous.Zipping.zip_directory(zip_filename=dpk_filename,
|
|
111
|
+
directory=directory,
|
|
112
|
+
ignore_directories=['artifacts'])
|
|
113
|
+
return dpk_filename
|
|
114
|
+
except Exception:
|
|
115
|
+
logger.error('Error when packing:')
|
|
116
|
+
raise
|
|
117
|
+
|
|
118
|
+
def _build_entities_from_response(self, response_items) -> miscellaneous.List[entities.Dpk]:
|
|
119
|
+
pool = self._client_api.thread_pools(pool_name='entity.create')
|
|
120
|
+
jobs = [None for _ in range(len(response_items))]
|
|
121
|
+
# return triggers list
|
|
122
|
+
for i_item, item in enumerate(response_items):
|
|
123
|
+
jobs[i_item] = pool.submit(entities.Dpk._protected_from_json,
|
|
124
|
+
**{'client_api': self._client_api,
|
|
125
|
+
'_json': item,
|
|
126
|
+
'project': self._project})
|
|
127
|
+
# get all results
|
|
128
|
+
results = [j.result() for j in jobs]
|
|
129
|
+
# log errors
|
|
130
|
+
_ = [logger.warning(r[1]) for r in results if r[0] is False]
|
|
131
|
+
# return good jobs
|
|
132
|
+
items = miscellaneous.List([r[1] for r in results if r[0] is True])
|
|
133
|
+
return items
|
|
134
|
+
|
|
135
|
+
def pull(self, dpk: entities.Dpk = None, dpk_id: str = None, dpk_name: str = None, local_path=None) -> str:
|
|
136
|
+
"""
|
|
137
|
+
Pulls the app from the platform as dpk file.
|
|
138
|
+
|
|
139
|
+
Note: you must pass either dpk_name or dpk_id to the function.
|
|
140
|
+
:param str dpk: DPK entity.
|
|
141
|
+
:param str dpk_id: the name of the dpk.
|
|
142
|
+
:param str dpk_name: the id of the dpk.
|
|
143
|
+
:param local_path: the path where you want to install the dpk file.
|
|
144
|
+
:return local path where the package pull
|
|
145
|
+
|
|
146
|
+
**Example**
|
|
147
|
+
..code-block:: python
|
|
148
|
+
path = dl.dpks.pull(dpk_name='my-app')
|
|
149
|
+
"""
|
|
150
|
+
if dpk is None:
|
|
151
|
+
dpk = self.get(dpk_id=dpk_id, dpk_name=dpk_name)
|
|
152
|
+
if local_path is None:
|
|
153
|
+
local_path = os.path.join(
|
|
154
|
+
services.service_defaults.DATALOOP_PATH,
|
|
155
|
+
"dpks",
|
|
156
|
+
dpk.name,
|
|
157
|
+
str(dpk.version))
|
|
158
|
+
|
|
159
|
+
dpk.codebases.unpack(codebase=dpk.codebase, local_path=local_path)
|
|
160
|
+
return local_path
|
|
161
|
+
|
|
162
|
+
def __get_by_name(self, dpk_name: str, dpk_version: str = None):
|
|
163
|
+
filters = entities.Filters(field='name',
|
|
164
|
+
values=dpk_name,
|
|
165
|
+
resource=entities.FiltersResource.DPK,
|
|
166
|
+
use_defaults=False)
|
|
167
|
+
dpks = self.list(filters=filters)
|
|
168
|
+
# only latest version returns so should be only one
|
|
169
|
+
if dpks.items_count == 0:
|
|
170
|
+
raise exceptions.PlatformException(error='404',
|
|
171
|
+
message='Dpk not found. Name: {}'.format(dpk_name))
|
|
172
|
+
elif dpks.items_count > 1:
|
|
173
|
+
raise exceptions.PlatformException(
|
|
174
|
+
error='400',
|
|
175
|
+
message='More than one dpk found by the name of: {}'.format(dpk_name))
|
|
176
|
+
dpk: entities.Dpk = dpks.items[0]
|
|
177
|
+
|
|
178
|
+
########################
|
|
179
|
+
# get specific version #
|
|
180
|
+
########################
|
|
181
|
+
if dpk_version is not None and dpk.version != dpk_version:
|
|
182
|
+
filters = entities.Filters(field='version',
|
|
183
|
+
values=dpk_version,
|
|
184
|
+
resource=entities.FiltersResource.DPK,
|
|
185
|
+
use_defaults=False)
|
|
186
|
+
dpk_v = dpk._get_revision_pages(filters=filters)
|
|
187
|
+
if dpk_v.items_count == 0:
|
|
188
|
+
raise exceptions.PlatformException(
|
|
189
|
+
error='404',
|
|
190
|
+
message=f'Dpk version not found. Name: {dpk_name}, version: {dpk_version}')
|
|
191
|
+
elif dpk_v.items_count > 1:
|
|
192
|
+
# should never be here - more than one with same name and version
|
|
193
|
+
raise exceptions.PlatformException(
|
|
194
|
+
error='400',
|
|
195
|
+
message=f'More than one dpk found with name: {dpk_name}, version: {dpk_version}')
|
|
196
|
+
dpk = dpk_v.items[0]
|
|
197
|
+
return dpk
|
|
198
|
+
|
|
199
|
+
def publish(
|
|
200
|
+
self,
|
|
201
|
+
dpk: entities.Dpk = None,
|
|
202
|
+
ignore_max_file_size: bool = False,
|
|
203
|
+
manifest_filepath='dataloop.json',
|
|
204
|
+
local_path: str = None
|
|
205
|
+
) -> entities.Dpk:
|
|
206
|
+
"""
|
|
207
|
+
Upload a dpk entity to the dataloop platform.
|
|
208
|
+
|
|
209
|
+
:param entities.Dpk dpk: Optional. The DPK entity to publish. If None, a new DPK is created
|
|
210
|
+
from the manifest file.
|
|
211
|
+
:param bool ignore_max_file_size: Optional. If True, the maximum file size check is ignored
|
|
212
|
+
during the packaging of the codebase.
|
|
213
|
+
:param str manifest_filepath: Optional. Path to the manifest file. Can be absolute or relative.
|
|
214
|
+
Defaults to 'dataloop.json'
|
|
215
|
+
:param str local_path: Optional. The path where the dpk files are located.
|
|
216
|
+
|
|
217
|
+
:return the published dpk
|
|
218
|
+
:rtype dl.entities.Dpk
|
|
219
|
+
|
|
220
|
+
**Example**
|
|
221
|
+
|
|
222
|
+
.. code-block:: python
|
|
223
|
+
published_dpk = dl.dpks.publish()
|
|
224
|
+
"""
|
|
225
|
+
manifest_path = Path(manifest_filepath).resolve()
|
|
226
|
+
|
|
227
|
+
if dpk is None:
|
|
228
|
+
if not manifest_path.exists():
|
|
229
|
+
raise FileNotFoundError(f'{manifest_filepath} file must exist in order to publish a dpk')
|
|
230
|
+
with open(manifest_filepath, 'r') as f:
|
|
231
|
+
json_file = json.load(f)
|
|
232
|
+
dpk = entities.Dpk.from_json(_json=json_file,
|
|
233
|
+
client_api=self._client_api,
|
|
234
|
+
project=self.project)
|
|
235
|
+
|
|
236
|
+
if not dpk.context:
|
|
237
|
+
dpk.context = {}
|
|
238
|
+
if 'project' not in dpk.context:
|
|
239
|
+
if not self.project:
|
|
240
|
+
raise exceptions.PlatformException('400', 'project id must be provided in the context')
|
|
241
|
+
dpk.context['project'] = self.project.id
|
|
242
|
+
if 'org' not in dpk.context and dpk.scope == 'organization':
|
|
243
|
+
if not self.project:
|
|
244
|
+
raise exceptions.PlatformException('400', 'org id must be provided in the context')
|
|
245
|
+
dpk.context['org'] = self.project.org['id']
|
|
246
|
+
|
|
247
|
+
if self.project and self.project.id != dpk.context['project']:
|
|
248
|
+
logger.warning("the project id that provide different from the dpk project id")
|
|
249
|
+
|
|
250
|
+
if local_path is None:
|
|
251
|
+
if manifest_filepath == 'dataloop.json':
|
|
252
|
+
local_path = os.getcwd()
|
|
253
|
+
else:
|
|
254
|
+
local_path = os.path.dirname(manifest_filepath)
|
|
255
|
+
if dpk.codebase is None:
|
|
256
|
+
dpk.codebase = self.project.codebases.pack(directory=local_path,
|
|
257
|
+
name=dpk.display_name,
|
|
258
|
+
extension='dpk',
|
|
259
|
+
ignore_directories=['artifacts'],
|
|
260
|
+
ignore_max_file_size=ignore_max_file_size)
|
|
261
|
+
|
|
262
|
+
success_pack, response_pack = self._client_api.gen_request(req_type='post',
|
|
263
|
+
json_req=dpk.to_json(),
|
|
264
|
+
path='/app-registry')
|
|
265
|
+
if not success_pack:
|
|
266
|
+
raise exceptions.PlatformException(error=response_pack)
|
|
267
|
+
|
|
268
|
+
return entities.Dpk.from_json(response_pack.json(), self._client_api, dpk.project)
|
|
269
|
+
|
|
270
|
+
def update(self, dpk: entities.Dpk) -> entities.Dpk:
|
|
271
|
+
"""
|
|
272
|
+
Update the dpk entity in the platform.
|
|
273
|
+
|
|
274
|
+
Note: the update will update only attributes, displayName, Icon, description and pipeline template preview
|
|
275
|
+
|
|
276
|
+
:param entities.Dpk dpk: the dpk entity to update.
|
|
277
|
+
:return the updated dpk entity.
|
|
278
|
+
:rtype entities.Dpk
|
|
279
|
+
"""
|
|
280
|
+
success, response = self._client_api.gen_request(req_type='patch',
|
|
281
|
+
path='/app-registry/{}'.format(dpk.id),
|
|
282
|
+
json_req=dpk.to_json())
|
|
283
|
+
if not success:
|
|
284
|
+
raise exceptions.PlatformException(response)
|
|
285
|
+
res = response.json()
|
|
286
|
+
logger.info(res.get('message'))
|
|
287
|
+
return entities.Dpk.from_json(res.get('dpk'), self._client_api, self._project)
|
|
288
|
+
|
|
289
|
+
def delete(self, dpk_id: str) -> bool:
|
|
290
|
+
"""
|
|
291
|
+
Delete the dpk from the app store.
|
|
292
|
+
|
|
293
|
+
Note: after removing the dpk, you cant get it again, it's advised to pull it first.
|
|
294
|
+
|
|
295
|
+
:param str dpk_id: the id of the dpk.
|
|
296
|
+
:return whether the operation ran successfully
|
|
297
|
+
:rtype bool
|
|
298
|
+
"""
|
|
299
|
+
success, response = self._client_api.gen_request(req_type='delete', path=f'/app-registry/{dpk_id}')
|
|
300
|
+
if success:
|
|
301
|
+
logger.info(f'Deleted dpk: {dpk_id} successfully')
|
|
302
|
+
else:
|
|
303
|
+
raise exceptions.PlatformException(response)
|
|
304
|
+
return success
|
|
305
|
+
|
|
306
|
+
def list(self, filters: entities.Filters = None) -> entities.PagedEntities:
|
|
307
|
+
"""
|
|
308
|
+
List the available dpks.
|
|
309
|
+
|
|
310
|
+
:param entities.Filters filters: the filters to apply on the list
|
|
311
|
+
:return a paged entity representing the list of dpks.
|
|
312
|
+
|
|
313
|
+
** Example **
|
|
314
|
+
.. code-block:: python
|
|
315
|
+
dpks = dl.dpks.list()
|
|
316
|
+
"""
|
|
317
|
+
if filters is None:
|
|
318
|
+
filters = entities.Filters(resource=entities.FiltersResource.DPK)
|
|
319
|
+
elif not isinstance(filters, entities.Filters):
|
|
320
|
+
raise ValueError('Unknown filters type: {!r}'.format(type(filters)))
|
|
321
|
+
elif filters.resource != entities.FiltersResource.DPK:
|
|
322
|
+
raise ValueError('Filters resource must to be FiltersResource.DPK. Got: {!r}'.format(filters.resource))
|
|
323
|
+
|
|
324
|
+
if self._project is not None:
|
|
325
|
+
filters.add(field='context.project', values=self._project.id)
|
|
326
|
+
|
|
327
|
+
paged = entities.PagedEntities(items_repository=self,
|
|
328
|
+
filters=filters,
|
|
329
|
+
page_offset=filters.page,
|
|
330
|
+
page_size=filters.page_size,
|
|
331
|
+
client_api=self._client_api)
|
|
332
|
+
paged.get_page()
|
|
333
|
+
return paged
|
|
334
|
+
|
|
335
|
+
def _list(self, filters: entities.Filters):
|
|
336
|
+
url = '/app-registry/query'
|
|
337
|
+
|
|
338
|
+
# request
|
|
339
|
+
success, response = self._client_api.gen_request(req_type='post',
|
|
340
|
+
path=url,
|
|
341
|
+
json_req=filters.prepare())
|
|
342
|
+
if not success:
|
|
343
|
+
raise exceptions.PlatformException(response)
|
|
344
|
+
return response.json()
|
|
345
|
+
|
|
346
|
+
def get_revisions(self, dpk_id: str, version: str):
|
|
347
|
+
"""
|
|
348
|
+
Get the revision of a specific dpk.
|
|
349
|
+
|
|
350
|
+
:param str dpk_id: the id of the dpk.
|
|
351
|
+
:param str version: the version of the dpk.
|
|
352
|
+
:return the entity of the dpk
|
|
353
|
+
:rtype entities.Dpk
|
|
354
|
+
|
|
355
|
+
** Example **
|
|
356
|
+
..coed-block:: python
|
|
357
|
+
dpk = dl.dpks.get_revisions(dpk_id='id', version='1.0.0')
|
|
358
|
+
"""
|
|
359
|
+
if dpk_id is None or version is None:
|
|
360
|
+
raise ValueError('You must provide both dpk_id and version')
|
|
361
|
+
url = '/app-registry/{}/revisions/{}'.format(dpk_id, version)
|
|
362
|
+
|
|
363
|
+
# request
|
|
364
|
+
success, response = self._client_api.gen_request(req_type='get',
|
|
365
|
+
path=url)
|
|
366
|
+
if not success:
|
|
367
|
+
raise exceptions.PlatformException(response)
|
|
368
|
+
|
|
369
|
+
dpk = entities.Dpk.from_json(_json=response.json(),
|
|
370
|
+
client_api=self._client_api,
|
|
371
|
+
project=self._project,
|
|
372
|
+
is_fetched=False)
|
|
373
|
+
return dpk
|
|
374
|
+
|
|
375
|
+
def get(self, dpk_name: str = None, dpk_version: str = None, dpk_id: str = None) -> entities.Dpk:
|
|
376
|
+
"""
|
|
377
|
+
Get a specific dpk from the platform.
|
|
378
|
+
|
|
379
|
+
Note: you must pass either dpk_id or dpk_name.
|
|
380
|
+
|
|
381
|
+
:param str dpk_id: the id of the dpk to get.
|
|
382
|
+
:param str dpk_name: the name of the dpk to get.
|
|
383
|
+
:param str dpk_version: options - to get a specific dpk version
|
|
384
|
+
:return the entity of the dpk
|
|
385
|
+
:rtype entities.Dpk
|
|
386
|
+
|
|
387
|
+
** Example **
|
|
388
|
+
..coed-block:: python
|
|
389
|
+
dpk = dl.dpks.get(dpk_name='name')
|
|
390
|
+
"""
|
|
391
|
+
if dpk_id is None and dpk_name is None:
|
|
392
|
+
raise ValueError('You must provide an identifier, either dpk_id or dpk_name')
|
|
393
|
+
if dpk_id is not None:
|
|
394
|
+
url = '/app-registry/{}'.format(dpk_id)
|
|
395
|
+
|
|
396
|
+
# request
|
|
397
|
+
success, response = self._client_api.gen_request(req_type='get',
|
|
398
|
+
path=url)
|
|
399
|
+
if not success:
|
|
400
|
+
raise exceptions.PlatformException(response)
|
|
401
|
+
|
|
402
|
+
dpk = entities.Dpk.from_json(_json=response.json(),
|
|
403
|
+
client_api=self._client_api,
|
|
404
|
+
project=self._project,
|
|
405
|
+
is_fetched=False)
|
|
406
|
+
else:
|
|
407
|
+
dpk = self.__get_by_name(dpk_name=dpk_name, dpk_version=dpk_version)
|
|
408
|
+
|
|
409
|
+
return dpk
|
|
410
|
+
|
|
411
|
+
def get_previews(self, dpk: entities.Dpk):
|
|
412
|
+
"""
|
|
413
|
+
Get the preview of a specific dpk.
|
|
414
|
+
|
|
415
|
+
:param entities.Dpk dpk: the dpk entity to get the preview for.
|
|
416
|
+
:return the preview of the dpk's templates
|
|
417
|
+
:rtype dict
|
|
418
|
+
|
|
419
|
+
** Example **
|
|
420
|
+
..coed-block:: python
|
|
421
|
+
res = dl.dpks.get_previews(dpk=dpk)
|
|
422
|
+
"""
|
|
423
|
+
url = '/app-registry/{}/previews'.format(dpk.id)
|
|
424
|
+
|
|
425
|
+
# request
|
|
426
|
+
success, response = self._client_api.gen_request(
|
|
427
|
+
req_type='get',
|
|
428
|
+
path=url
|
|
429
|
+
)
|
|
430
|
+
|
|
431
|
+
if not success:
|
|
432
|
+
raise exceptions.PlatformException(response)
|
|
433
|
+
|
|
434
434
|
return response.json()
|