earthengine-api 1.5.9rc2__tar.gz → 1.5.10rc0__tar.gz

{earthengine_api-1.5.9rc2/earthengine_api.egg-info → earthengine_api-1.5.10rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: earthengine-api
-Version: 1.5.9rc2
+Version: 1.5.10rc0
 Summary: Earth Engine Python API
 Author-email: Google LLC <noreply@google.com>
 License: Apache-2.0

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0/earthengine_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: earthengine-api
-Version: 1.5.9rc2
+Version: 1.5.10rc0
 Summary: Earth Engine Python API
 Author-email: Google LLC <noreply@google.com>
 License: Apache-2.0

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0}/ee/__init__.py

@@ -1,6 +1,6 @@
 """The EE Python library."""
 
-__version__ = '1.5.9rc2'
+__version__ = '1.5.10rc0'
 
 # Using lowercase function naming to match the JavaScript names.
 # pylint: disable=g-bad-name

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0}/ee/cli/commands.py

@@ -2049,6 +2049,9 @@ class ProjectConfigSetCommand:
   def run(
       self, args: argparse.Namespace, config: utils.CommandLineConfig
   ) -> None:
+    if not args.max_concurrent_exports:
+      raise ValueError('Flag --max_concurrent_exports must be set.')
+
     if args.max_concurrent_exports < 0:
       raise ValueError('"max_concurrent_exports" must be >= 0.')
 

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0}/ee/data.py

@@ -2331,16 +2331,17 @@ def create_assets(
     asset_ids: Sequence[str], asset_type: str, mk_parents: bool
 ) -> None:
   """Creates the specified assets if they do not exist."""
-  for asset_id in asset_ids:
+  ids = [convert_asset_id_to_asset_name(asset_id) for asset_id in asset_ids]
+  for asset_id in ids:
     if getInfo(asset_id):
       print('Asset %s already exists.' % asset_id)
       continue
     if mk_parents:
       parts = asset_id.split('/')
       # We don't need to create the namespace and the user's/project's folder.
-      if len(parts) > 2:
-        path = parts[0] + '/' + parts[1] + '/'
-        for part in parts[2:-1]:
+      if len(parts) > 3:
+        path = '/'.join(parts[0:3]) + '/'
+        for part in parts[3:-1]:
           path += part
           if getInfo(path) is None:
             createAsset({'type': ASSET_TYPE_FOLDER_CLOUD}, path)

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0}/ee/tests/data_test.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python3
 """Test for the ee.data module."""
 
+import json
+from typing import Any, Optional
 from unittest import mock
 
+import googleapiclient
 import httplib2
 import requests
 
@@ -14,6 +17,25 @@ from ee import featurecollection
 from ee import image
 
 
+def NotFoundError() -> googleapiclient.errors.HttpError:
+  """Creates a mock HttpError with a 404 status code."""
+  resp = httplib2.Response({'status': '404', 'reason': 'Not Found'})
+  content = json.dumps({'error': {'code': 404, 'message': 'Not Found'}}).encode(
+      'utf-8'
+  )
+  return googleapiclient.errors.HttpError(resp, content)
+
+
+def NewFolderAsset(
+    name: str, quota: Optional[dict[str, int]] = None
+) -> dict[str, Any]:
+  return {
+      'type': 'FOLDER',
+      'name': name,
+      'quota': quota or {},
+  }
+
+
 class DataTest(unittest.TestCase):
 
   def setUp(self):
@@ -183,6 +205,66 @@ class DataTest(unittest.TestCase):
       asset = mock_create_asset.call_args.kwargs['body']
       self.assertEqual(asset, {'type': 'FOLDER'})
 
+  def testCreateAssets(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_name = 'projects/some-project/assets/some-asset'
+      cloud_api_resource.projects().assets().get().execute.side_effect = (
+          NotFoundError()
+      )
+      ee.data.create_assets([asset_name], 'FOLDER', False)
+      mock_create_asset = cloud_api_resource.projects().assets().create
+      mock_create_asset.assert_called_once_with(
+          parent='projects/some-project',
+          assetId='some-asset',
+          body={'type': 'FOLDER'},
+          prettyPrint=False,
+      )
+
+  def testCreateAssets_empty(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      ee.data.create_assets([], 'FOLDER', False)
+      mock_create_asset = cloud_api_resource.projects().assets().create
+      mock_create_asset.assert_not_called()
+
+  def testCreateAssets_noOpIfAssetExists(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_name = 'projects/some-project/assets/some-asset'
+      cloud_api_resource.projects().assets().get.execute.return_value = (
+          NewFolderAsset(asset_name)
+      )
+      ee.data.create_assets([asset_name], 'FOLDER', False)
+      mock_create_asset = cloud_api_resource.projects().assets().create
+      mock_create_asset.assert_not_called()
+
+  def testCreateAssets_withParents(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_name = 'projects/some-project/assets/foo/bar'
+      cloud_api_resource.projects().assets().get().execute.side_effect = (
+          NotFoundError()
+      )
+      ee.data.create_assets([asset_name], 'FOLDER', True)
+      mock_create_asset = cloud_api_resource.projects().assets().create
+      mock_create_asset.assert_has_calls([
+          mock.call(
+              parent='projects/some-project',
+              assetId='foo',
+              body={'type': 'FOLDER'},
+              prettyPrint=False,
+          ),
+          mock.call().execute(num_retries=5),
+          mock.call(
+              parent='projects/some-project',
+              assetId='foo/bar',
+              body={'type': 'FOLDER'},
+              prettyPrint=False,
+          ),
+          mock.call().execute(num_retries=5),
+      ])
+
   def testStartIngestion(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):

{earthengine_api-1.5.9rc2 → earthengine_api-1.5.10rc0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "earthengine-api"
-version = "1.5.9rc2"
+version = "1.5.10rc0"
 description = "Earth Engine Python API"
 readme = "README.md"
 requires-python = ">=3.9"