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.
Files changed (243) hide show
  1. dtlpy/__init__.py +488 -488
  2. dtlpy/__version__.py +1 -1
  3. dtlpy/assets/__init__.py +26 -26
  4. dtlpy/assets/__pycache__/__init__.cpython-38.pyc +0 -0
  5. dtlpy/assets/code_server/config.yaml +2 -2
  6. dtlpy/assets/code_server/installation.sh +24 -24
  7. dtlpy/assets/code_server/launch.json +13 -13
  8. dtlpy/assets/code_server/settings.json +2 -2
  9. dtlpy/assets/main.py +53 -53
  10. dtlpy/assets/main_partial.py +18 -18
  11. dtlpy/assets/mock.json +11 -11
  12. dtlpy/assets/model_adapter.py +83 -83
  13. dtlpy/assets/package.json +61 -61
  14. dtlpy/assets/package_catalog.json +29 -29
  15. dtlpy/assets/package_gitignore +307 -307
  16. dtlpy/assets/service_runners/__init__.py +33 -33
  17. dtlpy/assets/service_runners/converter.py +96 -96
  18. dtlpy/assets/service_runners/multi_method.py +49 -49
  19. dtlpy/assets/service_runners/multi_method_annotation.py +54 -54
  20. dtlpy/assets/service_runners/multi_method_dataset.py +55 -55
  21. dtlpy/assets/service_runners/multi_method_item.py +52 -52
  22. dtlpy/assets/service_runners/multi_method_json.py +52 -52
  23. dtlpy/assets/service_runners/single_method.py +37 -37
  24. dtlpy/assets/service_runners/single_method_annotation.py +43 -43
  25. dtlpy/assets/service_runners/single_method_dataset.py +43 -43
  26. dtlpy/assets/service_runners/single_method_item.py +41 -41
  27. dtlpy/assets/service_runners/single_method_json.py +42 -42
  28. dtlpy/assets/service_runners/single_method_multi_input.py +45 -45
  29. dtlpy/assets/voc_annotation_template.xml +23 -23
  30. dtlpy/caches/base_cache.py +32 -32
  31. dtlpy/caches/cache.py +473 -473
  32. dtlpy/caches/dl_cache.py +201 -201
  33. dtlpy/caches/filesystem_cache.py +89 -89
  34. dtlpy/caches/redis_cache.py +84 -84
  35. dtlpy/dlp/__init__.py +20 -20
  36. dtlpy/dlp/cli_utilities.py +367 -367
  37. dtlpy/dlp/command_executor.py +764 -764
  38. dtlpy/dlp/dlp +1 -1
  39. dtlpy/dlp/dlp.bat +1 -1
  40. dtlpy/dlp/dlp.py +128 -128
  41. dtlpy/dlp/parser.py +651 -651
  42. dtlpy/entities/__init__.py +83 -83
  43. dtlpy/entities/analytic.py +311 -311
  44. dtlpy/entities/annotation.py +1879 -1879
  45. dtlpy/entities/annotation_collection.py +699 -699
  46. dtlpy/entities/annotation_definitions/__init__.py +20 -20
  47. dtlpy/entities/annotation_definitions/base_annotation_definition.py +100 -100
  48. dtlpy/entities/annotation_definitions/box.py +195 -195
  49. dtlpy/entities/annotation_definitions/classification.py +67 -67
  50. dtlpy/entities/annotation_definitions/comparison.py +72 -72
  51. dtlpy/entities/annotation_definitions/cube.py +204 -204
  52. dtlpy/entities/annotation_definitions/cube_3d.py +149 -149
  53. dtlpy/entities/annotation_definitions/description.py +32 -32
  54. dtlpy/entities/annotation_definitions/ellipse.py +124 -124
  55. dtlpy/entities/annotation_definitions/free_text.py +62 -62
  56. dtlpy/entities/annotation_definitions/gis.py +69 -69
  57. dtlpy/entities/annotation_definitions/note.py +139 -139
  58. dtlpy/entities/annotation_definitions/point.py +117 -117
  59. dtlpy/entities/annotation_definitions/polygon.py +182 -182
  60. dtlpy/entities/annotation_definitions/polyline.py +111 -111
  61. dtlpy/entities/annotation_definitions/pose.py +92 -92
  62. dtlpy/entities/annotation_definitions/ref_image.py +86 -86
  63. dtlpy/entities/annotation_definitions/segmentation.py +240 -240
  64. dtlpy/entities/annotation_definitions/subtitle.py +34 -34
  65. dtlpy/entities/annotation_definitions/text.py +85 -85
  66. dtlpy/entities/annotation_definitions/undefined_annotation.py +74 -74
  67. dtlpy/entities/app.py +220 -220
  68. dtlpy/entities/app_module.py +107 -107
  69. dtlpy/entities/artifact.py +174 -174
  70. dtlpy/entities/assignment.py +399 -399
  71. dtlpy/entities/base_entity.py +214 -214
  72. dtlpy/entities/bot.py +113 -113
  73. dtlpy/entities/codebase.py +296 -296
  74. dtlpy/entities/collection.py +38 -38
  75. dtlpy/entities/command.py +169 -169
  76. dtlpy/entities/compute.py +442 -442
  77. dtlpy/entities/dataset.py +1285 -1285
  78. dtlpy/entities/directory_tree.py +44 -44
  79. dtlpy/entities/dpk.py +470 -470
  80. dtlpy/entities/driver.py +222 -222
  81. dtlpy/entities/execution.py +397 -397
  82. dtlpy/entities/feature.py +124 -124
  83. dtlpy/entities/feature_set.py +145 -145
  84. dtlpy/entities/filters.py +641 -641
  85. dtlpy/entities/gis_item.py +107 -107
  86. dtlpy/entities/integration.py +184 -184
  87. dtlpy/entities/item.py +953 -953
  88. dtlpy/entities/label.py +123 -123
  89. dtlpy/entities/links.py +85 -85
  90. dtlpy/entities/message.py +175 -175
  91. dtlpy/entities/model.py +694 -691
  92. dtlpy/entities/node.py +1005 -1005
  93. dtlpy/entities/ontology.py +803 -803
  94. dtlpy/entities/organization.py +287 -287
  95. dtlpy/entities/package.py +657 -657
  96. dtlpy/entities/package_defaults.py +5 -5
  97. dtlpy/entities/package_function.py +185 -185
  98. dtlpy/entities/package_module.py +113 -113
  99. dtlpy/entities/package_slot.py +118 -118
  100. dtlpy/entities/paged_entities.py +290 -267
  101. dtlpy/entities/pipeline.py +593 -593
  102. dtlpy/entities/pipeline_execution.py +279 -279
  103. dtlpy/entities/project.py +394 -394
  104. dtlpy/entities/prompt_item.py +499 -499
  105. dtlpy/entities/recipe.py +301 -301
  106. dtlpy/entities/reflect_dict.py +102 -102
  107. dtlpy/entities/resource_execution.py +138 -138
  108. dtlpy/entities/service.py +958 -958
  109. dtlpy/entities/service_driver.py +117 -117
  110. dtlpy/entities/setting.py +294 -294
  111. dtlpy/entities/task.py +491 -491
  112. dtlpy/entities/time_series.py +143 -143
  113. dtlpy/entities/trigger.py +426 -426
  114. dtlpy/entities/user.py +118 -118
  115. dtlpy/entities/webhook.py +124 -124
  116. dtlpy/examples/__init__.py +19 -19
  117. dtlpy/examples/add_labels.py +135 -135
  118. dtlpy/examples/add_metadata_to_item.py +21 -21
  119. dtlpy/examples/annotate_items_using_model.py +65 -65
  120. dtlpy/examples/annotate_video_using_model_and_tracker.py +75 -75
  121. dtlpy/examples/annotations_convert_to_voc.py +9 -9
  122. dtlpy/examples/annotations_convert_to_yolo.py +9 -9
  123. dtlpy/examples/convert_annotation_types.py +51 -51
  124. dtlpy/examples/converter.py +143 -143
  125. dtlpy/examples/copy_annotations.py +22 -22
  126. dtlpy/examples/copy_folder.py +31 -31
  127. dtlpy/examples/create_annotations.py +51 -51
  128. dtlpy/examples/create_video_annotations.py +83 -83
  129. dtlpy/examples/delete_annotations.py +26 -26
  130. dtlpy/examples/filters.py +113 -113
  131. dtlpy/examples/move_item.py +23 -23
  132. dtlpy/examples/play_video_annotation.py +13 -13
  133. dtlpy/examples/show_item_and_mask.py +53 -53
  134. dtlpy/examples/triggers.py +49 -49
  135. dtlpy/examples/upload_batch_of_items.py +20 -20
  136. dtlpy/examples/upload_items_and_custom_format_annotations.py +55 -55
  137. dtlpy/examples/upload_items_with_modalities.py +43 -43
  138. dtlpy/examples/upload_segmentation_annotations_from_mask_image.py +44 -44
  139. dtlpy/examples/upload_yolo_format_annotations.py +70 -70
  140. dtlpy/exceptions.py +125 -125
  141. dtlpy/miscellaneous/__init__.py +20 -20
  142. dtlpy/miscellaneous/dict_differ.py +95 -95
  143. dtlpy/miscellaneous/git_utils.py +217 -217
  144. dtlpy/miscellaneous/json_utils.py +14 -14
  145. dtlpy/miscellaneous/list_print.py +105 -105
  146. dtlpy/miscellaneous/zipping.py +130 -130
  147. dtlpy/ml/__init__.py +20 -20
  148. dtlpy/ml/base_feature_extractor_adapter.py +27 -27
  149. dtlpy/ml/base_model_adapter.py +945 -940
  150. dtlpy/ml/metrics.py +461 -461
  151. dtlpy/ml/predictions_utils.py +274 -274
  152. dtlpy/ml/summary_writer.py +57 -57
  153. dtlpy/ml/train_utils.py +60 -60
  154. dtlpy/new_instance.py +252 -252
  155. dtlpy/repositories/__init__.py +56 -56
  156. dtlpy/repositories/analytics.py +85 -85
  157. dtlpy/repositories/annotations.py +916 -916
  158. dtlpy/repositories/apps.py +383 -383
  159. dtlpy/repositories/artifacts.py +452 -452
  160. dtlpy/repositories/assignments.py +599 -599
  161. dtlpy/repositories/bots.py +213 -213
  162. dtlpy/repositories/codebases.py +559 -559
  163. dtlpy/repositories/collections.py +332 -348
  164. dtlpy/repositories/commands.py +158 -158
  165. dtlpy/repositories/compositions.py +61 -61
  166. dtlpy/repositories/computes.py +434 -406
  167. dtlpy/repositories/datasets.py +1291 -1291
  168. dtlpy/repositories/downloader.py +895 -895
  169. dtlpy/repositories/dpks.py +433 -433
  170. dtlpy/repositories/drivers.py +266 -266
  171. dtlpy/repositories/executions.py +817 -817
  172. dtlpy/repositories/feature_sets.py +226 -226
  173. dtlpy/repositories/features.py +238 -238
  174. dtlpy/repositories/integrations.py +484 -484
  175. dtlpy/repositories/items.py +909 -915
  176. dtlpy/repositories/messages.py +94 -94
  177. dtlpy/repositories/models.py +877 -867
  178. dtlpy/repositories/nodes.py +80 -80
  179. dtlpy/repositories/ontologies.py +511 -511
  180. dtlpy/repositories/organizations.py +525 -525
  181. dtlpy/repositories/packages.py +1941 -1941
  182. dtlpy/repositories/pipeline_executions.py +448 -448
  183. dtlpy/repositories/pipelines.py +642 -642
  184. dtlpy/repositories/projects.py +539 -539
  185. dtlpy/repositories/recipes.py +399 -399
  186. dtlpy/repositories/resource_executions.py +137 -137
  187. dtlpy/repositories/schema.py +120 -120
  188. dtlpy/repositories/service_drivers.py +213 -213
  189. dtlpy/repositories/services.py +1704 -1704
  190. dtlpy/repositories/settings.py +339 -339
  191. dtlpy/repositories/tasks.py +1124 -1124
  192. dtlpy/repositories/times_series.py +278 -278
  193. dtlpy/repositories/triggers.py +536 -536
  194. dtlpy/repositories/upload_element.py +257 -257
  195. dtlpy/repositories/uploader.py +651 -651
  196. dtlpy/repositories/webhooks.py +249 -249
  197. dtlpy/services/__init__.py +22 -22
  198. dtlpy/services/aihttp_retry.py +131 -131
  199. dtlpy/services/api_client.py +1782 -1782
  200. dtlpy/services/api_reference.py +40 -40
  201. dtlpy/services/async_utils.py +133 -133
  202. dtlpy/services/calls_counter.py +44 -44
  203. dtlpy/services/check_sdk.py +68 -68
  204. dtlpy/services/cookie.py +115 -115
  205. dtlpy/services/create_logger.py +156 -156
  206. dtlpy/services/events.py +84 -84
  207. dtlpy/services/logins.py +235 -235
  208. dtlpy/services/reporter.py +256 -256
  209. dtlpy/services/service_defaults.py +91 -91
  210. dtlpy/utilities/__init__.py +20 -20
  211. dtlpy/utilities/annotations/__init__.py +16 -16
  212. dtlpy/utilities/annotations/annotation_converters.py +269 -269
  213. dtlpy/utilities/base_package_runner.py +264 -264
  214. dtlpy/utilities/converter.py +1650 -1650
  215. dtlpy/utilities/dataset_generators/__init__.py +1 -1
  216. dtlpy/utilities/dataset_generators/dataset_generator.py +670 -670
  217. dtlpy/utilities/dataset_generators/dataset_generator_tensorflow.py +23 -23
  218. dtlpy/utilities/dataset_generators/dataset_generator_torch.py +21 -21
  219. dtlpy/utilities/local_development/__init__.py +1 -1
  220. dtlpy/utilities/local_development/local_session.py +179 -179
  221. dtlpy/utilities/reports/__init__.py +2 -2
  222. dtlpy/utilities/reports/figures.py +343 -343
  223. dtlpy/utilities/reports/report.py +71 -71
  224. dtlpy/utilities/videos/__init__.py +17 -17
  225. dtlpy/utilities/videos/video_player.py +598 -598
  226. dtlpy/utilities/videos/videos.py +470 -470
  227. {dtlpy-1.113.10.data → dtlpy-1.114.13.data}/scripts/dlp +1 -1
  228. dtlpy-1.114.13.data/scripts/dlp.bat +2 -0
  229. {dtlpy-1.113.10.data → dtlpy-1.114.13.data}/scripts/dlp.py +128 -128
  230. {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/LICENSE +200 -200
  231. {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/METADATA +172 -172
  232. dtlpy-1.114.13.dist-info/RECORD +240 -0
  233. {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/WHEEL +1 -1
  234. tests/features/environment.py +551 -550
  235. dtlpy-1.113.10.data/scripts/dlp.bat +0 -2
  236. dtlpy-1.113.10.dist-info/RECORD +0 -244
  237. tests/assets/__init__.py +0 -0
  238. tests/assets/models_flow/__init__.py +0 -0
  239. tests/assets/models_flow/failedmain.py +0 -52
  240. tests/assets/models_flow/main.py +0 -62
  241. tests/assets/models_flow/main_model.py +0 -54
  242. {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/entry_points.txt +0 -0
  243. {dtlpy-1.113.10.dist-info → dtlpy-1.114.13.dist-info}/top_level.txt +0 -0
@@ -1,217 +1,217 @@
1
- import subprocess
2
- import logging
3
- import os
4
-
5
- logger = logging.getLogger(name='dtlpy')
6
-
7
-
8
- class GitUtils:
9
- """
10
- Performs git related methods
11
- """
12
-
13
- @staticmethod
14
- def git_included(path):
15
- """
16
- Get only included git repo files based on .gitignore file
17
-
18
- :param path: directory - str
19
- :return: list()
20
- """
21
- included_files = list()
22
-
23
- try:
24
- p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), 'ls-files'],
25
- stdout=subprocess.PIPE,
26
- stderr=subprocess.PIPE)
27
- output, err = p.communicate()
28
- string_output = str(output, 'utf-8')
29
- included_files = string_output.split('\n')
30
- except Exception:
31
- logging.warning('Error getting git info for git repository in: {}'.format(path))
32
- # include all files
33
- for r, d, f in os.walk(path):
34
- for folder in d:
35
- included_files.append(os.path.join(r, folder))
36
-
37
- return included_files
38
-
39
- @staticmethod
40
- def is_git_repo(path):
41
- """
42
- Check if directory is a git repo
43
-
44
- :param path: directory - str
45
- :return: True/False
46
- """
47
- try:
48
- p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'status'],
49
- stdout=subprocess.PIPE,
50
- stderr=subprocess.PIPE)
51
- output, err = p.communicate()
52
- if p.returncode != 0 and 'not a git repository' in str(err):
53
- response = False
54
- elif p.returncode == 0 and 'On branch' in str(output):
55
- response = True
56
- else:
57
- response = False
58
- except Exception:
59
- response = False
60
- logging.warning('Error getting git info for git repository in: {}'.format(path))
61
- return response
62
-
63
- @staticmethod
64
- def git_status(path):
65
- """
66
- Get git repository git status
67
-
68
- :param path: directory - str
69
- :return: String
70
- """
71
- status = dict()
72
- try:
73
- p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'status'],
74
- stdout=subprocess.PIPE,
75
- stderr=subprocess.PIPE)
76
- output, err = p.communicate()
77
- output_lines = str(output, 'utf-8').splitlines()
78
- branch = output_lines[0].replace('On branch ', '').strip()
79
-
80
- logs = GitUtils.git_log(path)
81
- status = {'branch': branch,
82
- 'commit_id': logs[0]['commit'],
83
- 'commit_author': logs[0]['Author'],
84
- 'commit_message': logs[0]['message']}
85
- except Exception:
86
- logging.warning('Error getting git info for git repository in: {}'.format(path))
87
-
88
- return status
89
-
90
- @staticmethod
91
- def git_url(path):
92
- """
93
- Get git remote url
94
-
95
- :param path: directory - str
96
- :return: String
97
- """
98
- url = ''
99
- try:
100
- p = subprocess.Popen(
101
- ['git', '--git-dir', os.path.join(path, '.git'), 'config', '--get', 'remote.origin.url'],
102
- stdout=subprocess.PIPE,
103
- stderr=subprocess.PIPE)
104
- output, err = p.communicate()
105
- url = str(output, 'utf-8').splitlines()[0]
106
-
107
- except Exception:
108
- logging.warning('Error getting git info for git repository in: {}'.format(path))
109
-
110
- return url
111
-
112
- @staticmethod
113
- def git_log(path):
114
- """
115
- Get git repository git log
116
-
117
- :param path: directory - str
118
- :return: log as list()
119
- """
120
- log = list()
121
- try:
122
- log_limit = 100
123
- p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'log'],
124
- stdout=subprocess.PIPE,
125
- stderr=subprocess.PIPE)
126
- output, err = p.communicate()
127
- string_output = str(output, 'utf-8').split('\ncommit')
128
- for output in string_output:
129
- output = output.split('\n')
130
- if output[0].startswith('commit'):
131
- output[0] = output[0].replace('commit', '')
132
- log_line = {
133
- 'commit': output[0].strip(),
134
- 'Author': output[1].replace('Author:', '').strip(),
135
- 'Date': output[2].replace('Date:', '').strip(),
136
- 'message': output[4].strip(),
137
- }
138
- log.append(log_line)
139
- log = log[0:log_limit]
140
- except Exception:
141
- logging.warning('Error getting git log for git repository in: {}'.format(path))
142
-
143
- return log
144
-
145
- @staticmethod
146
- def git_clone(path, git_url, **kwargs):
147
- """
148
- Clone git repo to local path
149
- :param path: `str` local path to clone to
150
- :param git_url: `str` git url to clone from
151
- :return `bool` for successful clone
152
- """
153
- cmd = ''
154
- if not os.path.isdir(path):
155
- os.makedirs(path)
156
- try:
157
- username = kwargs.get('username')
158
- password = kwargs.get('password')
159
- if username is not None and password is not None:
160
- if git_url.startswith('https://'):
161
- git_url = git_url.replace('https://', 'https://{}:{}@'.format(username, password))
162
- elif git_url.startswith('http://'):
163
- git_url = git_url.replace('http://', 'http://{}:{}@'.format(username, password))
164
- else:
165
- git_url = 'https://{}:{}@{}'.format(username, password, git_url)
166
- tag = kwargs.get('tag')
167
- branch = kwargs.get('branch')
168
- if tag is not None:
169
- branch_cmd = ['--branch', tag]
170
- elif branch is not None:
171
- branch_cmd = ['--branch', branch]
172
- else:
173
- branch_cmd = []
174
-
175
- cmd = ['git', 'clone'] + branch_cmd + [git_url, path]
176
- p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
177
- output, err = [std.decode() for std in p.communicate()]
178
- exit_code = p.returncode
179
- response = not exit_code
180
- if exit_code:
181
- logging.error('Error executing: {ps1} $ {cmd}\n{err}'.format(ps1=path, cmd=' '.join(cmd), err=err))
182
- except Exception:
183
- response = False
184
- logging.exception('Error cloning git with cmd: {}'.format(cmd))
185
- return response
186
-
187
- @staticmethod
188
- def git_command(path, cmd, show=False):
189
- """
190
- Execute command for a local git repo in path
191
- :param path: `str` local path for the git repo
192
- :param cmd: `str` or `list` of `str` specifying the command to run
193
- :param show: `bool` if True, prints the stdout of the process (default=False)
194
- :return: `bool` if command was successful or not
195
- """
196
- prev_dir = os.getcwd()
197
- if isinstance(cmd, str):
198
- cmd = cmd.split()
199
-
200
- try:
201
- os.chdir(path)
202
- p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
203
- output, err = p.communicate()
204
- output, err = output.decode(), err.decode()
205
- exit_code = p.returncode
206
- response = not exit_code
207
- # Print the process stdout / stderr
208
- if show:
209
- print(output)
210
- if exit_code:
211
- logging.error('Error executing: {ps1} $ {cmd}\n{err}'.format(ps1=path, cmd=' '.join(cmd), err=err))
212
- except Exception:
213
- response = False
214
- logging.critical('Error executing: {ps1} $ {cmd}'.format(ps1=path, cmd=' '.join(cmd)))
215
- finally:
216
- os.chdir(prev_dir)
217
- return response
1
+ import subprocess
2
+ import logging
3
+ import os
4
+
5
+ logger = logging.getLogger(name='dtlpy')
6
+
7
+
8
+ class GitUtils:
9
+ """
10
+ Performs git related methods
11
+ """
12
+
13
+ @staticmethod
14
+ def git_included(path):
15
+ """
16
+ Get only included git repo files based on .gitignore file
17
+
18
+ :param path: directory - str
19
+ :return: list()
20
+ """
21
+ included_files = list()
22
+
23
+ try:
24
+ p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), 'ls-files'],
25
+ stdout=subprocess.PIPE,
26
+ stderr=subprocess.PIPE)
27
+ output, err = p.communicate()
28
+ string_output = str(output, 'utf-8')
29
+ included_files = string_output.split('\n')
30
+ except Exception:
31
+ logging.warning('Error getting git info for git repository in: {}'.format(path))
32
+ # include all files
33
+ for r, d, f in os.walk(path):
34
+ for folder in d:
35
+ included_files.append(os.path.join(r, folder))
36
+
37
+ return included_files
38
+
39
+ @staticmethod
40
+ def is_git_repo(path):
41
+ """
42
+ Check if directory is a git repo
43
+
44
+ :param path: directory - str
45
+ :return: True/False
46
+ """
47
+ try:
48
+ p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'status'],
49
+ stdout=subprocess.PIPE,
50
+ stderr=subprocess.PIPE)
51
+ output, err = p.communicate()
52
+ if p.returncode != 0 and 'not a git repository' in str(err):
53
+ response = False
54
+ elif p.returncode == 0 and 'On branch' in str(output):
55
+ response = True
56
+ else:
57
+ response = False
58
+ except Exception:
59
+ response = False
60
+ logging.warning('Error getting git info for git repository in: {}'.format(path))
61
+ return response
62
+
63
+ @staticmethod
64
+ def git_status(path):
65
+ """
66
+ Get git repository git status
67
+
68
+ :param path: directory - str
69
+ :return: String
70
+ """
71
+ status = dict()
72
+ try:
73
+ p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'status'],
74
+ stdout=subprocess.PIPE,
75
+ stderr=subprocess.PIPE)
76
+ output, err = p.communicate()
77
+ output_lines = str(output, 'utf-8').splitlines()
78
+ branch = output_lines[0].replace('On branch ', '').strip()
79
+
80
+ logs = GitUtils.git_log(path)
81
+ status = {'branch': branch,
82
+ 'commit_id': logs[0]['commit'],
83
+ 'commit_author': logs[0]['Author'],
84
+ 'commit_message': logs[0]['message']}
85
+ except Exception:
86
+ logging.warning('Error getting git info for git repository in: {}'.format(path))
87
+
88
+ return status
89
+
90
+ @staticmethod
91
+ def git_url(path):
92
+ """
93
+ Get git remote url
94
+
95
+ :param path: directory - str
96
+ :return: String
97
+ """
98
+ url = ''
99
+ try:
100
+ p = subprocess.Popen(
101
+ ['git', '--git-dir', os.path.join(path, '.git'), 'config', '--get', 'remote.origin.url'],
102
+ stdout=subprocess.PIPE,
103
+ stderr=subprocess.PIPE)
104
+ output, err = p.communicate()
105
+ url = str(output, 'utf-8').splitlines()[0]
106
+
107
+ except Exception:
108
+ logging.warning('Error getting git info for git repository in: {}'.format(path))
109
+
110
+ return url
111
+
112
+ @staticmethod
113
+ def git_log(path):
114
+ """
115
+ Get git repository git log
116
+
117
+ :param path: directory - str
118
+ :return: log as list()
119
+ """
120
+ log = list()
121
+ try:
122
+ log_limit = 100
123
+ p = subprocess.Popen(['git', '--git-dir', os.path.join(path, '.git'), '--work-tree', path, 'log'],
124
+ stdout=subprocess.PIPE,
125
+ stderr=subprocess.PIPE)
126
+ output, err = p.communicate()
127
+ string_output = str(output, 'utf-8').split('\ncommit')
128
+ for output in string_output:
129
+ output = output.split('\n')
130
+ if output[0].startswith('commit'):
131
+ output[0] = output[0].replace('commit', '')
132
+ log_line = {
133
+ 'commit': output[0].strip(),
134
+ 'Author': output[1].replace('Author:', '').strip(),
135
+ 'Date': output[2].replace('Date:', '').strip(),
136
+ 'message': output[4].strip(),
137
+ }
138
+ log.append(log_line)
139
+ log = log[0:log_limit]
140
+ except Exception:
141
+ logging.warning('Error getting git log for git repository in: {}'.format(path))
142
+
143
+ return log
144
+
145
+ @staticmethod
146
+ def git_clone(path, git_url, **kwargs):
147
+ """
148
+ Clone git repo to local path
149
+ :param path: `str` local path to clone to
150
+ :param git_url: `str` git url to clone from
151
+ :return `bool` for successful clone
152
+ """
153
+ cmd = ''
154
+ if not os.path.isdir(path):
155
+ os.makedirs(path)
156
+ try:
157
+ username = kwargs.get('username')
158
+ password = kwargs.get('password')
159
+ if username is not None and password is not None:
160
+ if git_url.startswith('https://'):
161
+ git_url = git_url.replace('https://', 'https://{}:{}@'.format(username, password))
162
+ elif git_url.startswith('http://'):
163
+ git_url = git_url.replace('http://', 'http://{}:{}@'.format(username, password))
164
+ else:
165
+ git_url = 'https://{}:{}@{}'.format(username, password, git_url)
166
+ tag = kwargs.get('tag')
167
+ branch = kwargs.get('branch')
168
+ if tag is not None:
169
+ branch_cmd = ['--branch', tag]
170
+ elif branch is not None:
171
+ branch_cmd = ['--branch', branch]
172
+ else:
173
+ branch_cmd = []
174
+
175
+ cmd = ['git', 'clone'] + branch_cmd + [git_url, path]
176
+ p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
177
+ output, err = [std.decode() for std in p.communicate()]
178
+ exit_code = p.returncode
179
+ response = not exit_code
180
+ if exit_code:
181
+ logging.error('Error executing: {ps1} $ {cmd}\n{err}'.format(ps1=path, cmd=' '.join(cmd), err=err))
182
+ except Exception:
183
+ response = False
184
+ logging.exception('Error cloning git with cmd: {}'.format(cmd))
185
+ return response
186
+
187
+ @staticmethod
188
+ def git_command(path, cmd, show=False):
189
+ """
190
+ Execute command for a local git repo in path
191
+ :param path: `str` local path for the git repo
192
+ :param cmd: `str` or `list` of `str` specifying the command to run
193
+ :param show: `bool` if True, prints the stdout of the process (default=False)
194
+ :return: `bool` if command was successful or not
195
+ """
196
+ prev_dir = os.getcwd()
197
+ if isinstance(cmd, str):
198
+ cmd = cmd.split()
199
+
200
+ try:
201
+ os.chdir(path)
202
+ p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
203
+ output, err = p.communicate()
204
+ output, err = output.decode(), err.decode()
205
+ exit_code = p.returncode
206
+ response = not exit_code
207
+ # Print the process stdout / stderr
208
+ if show:
209
+ print(output)
210
+ if exit_code:
211
+ logging.error('Error executing: {ps1} $ {cmd}\n{err}'.format(ps1=path, cmd=' '.join(cmd), err=err))
212
+ except Exception:
213
+ response = False
214
+ logging.critical('Error executing: {ps1} $ {cmd}'.format(ps1=path, cmd=' '.join(cmd)))
215
+ finally:
216
+ os.chdir(prev_dir)
217
+ return response
@@ -1,14 +1,14 @@
1
- class JsonUtils:
2
- @staticmethod
3
- def get_if_absent(value, default=None):
4
- if value is None:
5
- if default is None:
6
- value_type = type(value)
7
- if value_type == dict:
8
- default = {}
9
- elif value_type == list:
10
- default = []
11
- else:
12
- default = ''
13
- value = default
14
- return value
1
+ class JsonUtils:
2
+ @staticmethod
3
+ def get_if_absent(value, default=None):
4
+ if value is None:
5
+ if default is None:
6
+ value_type = type(value)
7
+ if value_type == dict:
8
+ default = {}
9
+ elif value_type == list:
10
+ default = []
11
+ else:
12
+ default = ''
13
+ value = default
14
+ return value