geobox 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

geobox/file.py

@@ -2,13 +2,16 @@ from urllib.parse import urljoin
 from typing import Optional, Dict, List, Union, TYPE_CHECKING
 import os
 import mimetypes
+from geobox.exception import ValidationError
 import requests
 import sys
+from pathlib import Path
 
 from .base import Base
 from .enums import FileFormat, PublishFileType, InputGeomType, FileType
 from .utils import clean_data
 from .task import Task
+from .feature import Feature
 
 if TYPE_CHECKING:
     from . import GeoboxClient 
@@ -221,9 +224,9 @@ class File(Base):
             >>> from geobox import GeoboxClient
             >>> from geobox.file import File
             >>> client = GeoboxClient()
-            >>> files = File.get_file_by_name(client, name='test')
+            >>> files = File.get_files_by_name(client, name='test')
             or
-            >>> files = client.get_file_by_name(name='test')
+            >>> files = client.get_files_by_name(name='test')
         """
         return cls.get_files(api, q=f"name = '{name}'", user_id=user_id)
     
@@ -248,44 +251,34 @@ class File(Base):
             >>> client = GeoboxClient()
             >>> file_path = File.get_file(client, uuid="12345678-1234-5678-1234-567812345678")
         """
-        # Get the original filename from data or use uuid
-        if self.name:
-            filename = f"{self.name.split('.')[0]}" if len(self.name.split('.')) > 1 else f'{self.name}'
-        else:
-            filename = f'{self.uuid}'
-
         # If save_path is provided, check if it ends with a '/'
         if save_path and save_path.endswith('/'):
-            return f'{save_path}{filename}'
+            return f'{save_path}'
         
         if save_path and not save_path.endswith('/'):
             raise ValueError("save_path must end with a '/'")
         
-        return os.path.join(os.getcwd(), filename)
+        return os.getcwd()
     
 
-    def _get_file_ext(self, response: requests.Response) -> str:
+    def _get_file_name(self, response: requests.Response) -> str:
         """
-        Get the file extension of the response.
+        Get the file name from the response.
 
         Args:
             response (requests.Response): The response of the request.
 
         Returns:
-            str: The file extension of the response.
+            str: The file name 
         """
-        ext = ""
-        if 'Content-Disposition' in response.headers:
-            content_disposition = response.headers['Content-Disposition']
-            if 'filename=' in content_disposition:
-                filename = content_disposition.split('filename=')[-1].strip().strip('"')
-                ext = f".{filename.split('.')[-1]}"
+        if 'Content-Disposition' in response.headers and 'filename=' in response.headers['Content-Disposition']:
+            file_name = response.headers['Content-Disposition'].split('filename=')[-1].strip().strip('"')
 
-            else:
-                content_type = response.headers.get("Content-Type", "")
-                ext = mimetypes.guess_extension(content_type.split(";")[0])
+        else:
+            content_type = response.headers.get("Content-Type", "")
+            file_name = f'{self.name}.{mimetypes.guess_extension(content_type.split(";")[0])}'
 
-        return ext
+        return file_name
 
 
     def _create_progress_bar(self) -> 'tqdm':
@@ -338,20 +331,20 @@ class File(Base):
         save_path = self._get_save_path(save_path)
         os.makedirs(os.path.dirname(save_path), exist_ok=True)
         
-        with self.api.get(f"{self.endpoint}download/", stream=True) as response, \
-            open(save_path, 'wb') as f:
-            pbar = self._create_progress_bar() if progress_bar else None
-            for chunk in response.iter_content(chunk_size=8192):
-                f.write(chunk)
+        with self.api.get(f"{self.endpoint}download/", stream=True) as response:
+            file_name = self._get_file_name(response)
+            full_path = f"{save_path}/{file_name}"
+            with open(full_path, 'wb') as f:
+                pbar = self._create_progress_bar() if progress_bar else None
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+                    if pbar:
+                        pbar.update(len(chunk))
+                        pbar.refresh()
                 if pbar:
-                    pbar.update(len(chunk))
-                    pbar.refresh()
-            if pbar:
-                pbar.close()
+                    pbar.close()
 
-        final_path = os.path.abspath(save_path) + self._get_file_ext(response)
-        os.rename(os.path.abspath(save_path), os.path.abspath(final_path))
-        return os.path.abspath(final_path)
+        return os.path.abspath(full_path)
 
 
     def delete(self) -> None:
@@ -378,8 +371,8 @@ class File(Base):
                     input_layer: str = None, 
                     input_dataset: str = None, 
                     user_id: int = None, 
-                    input_srid: int = None, 
-                    file_encoding: str = "utf-8", 
+                    input_srid: int = Feature.BASE_SRID, 
+                    file_encoding: str = "UTF-8", 
                     replace_domain_codes_by_values: bool = False, 
                     report_errors: bool = True, 
                     as_terrain: bool = False) -> 'Task':
@@ -393,7 +386,7 @@ class File(Base):
             input_layer (str, optional): The name of the input layer.
             input_dataset (str, optional): The name of the input dataset.
             user_id (int, optional): Specific user. privileges required.
-            input_srid (int, optional): The SRID of the layer.
+            input_srid (int, optional): The SRID of the layer. default is: 3857
             file_encoding (str, optional): The encoding of the file. default is "utf-8".
             replace_domain_codes_by_values (bool, optional): Whether to replace domain codes by values. default is False.
             report_errors (bool, optional): Whether to report errors. default is True.
@@ -404,6 +397,7 @@ class File(Base):
 
         Raises:
             ValueError: If the publish_as is not a valid PublishFileType.
+            ValidationError: if the zipped file doesn't have any layers to publish.
 
         Example:
             >>> from geobox import GeoboxClient
@@ -416,30 +410,41 @@ class File(Base):
             ...                     input_layer='layer1', 
             ...                     input_dataset='dataset1', 
             ...                     input_srid=4326, 
-            ...                     file_encoding='utf-8')
+            ...                     file_encoding='UTF-8')
         """
         if not publish_as:
-            if self.file_type.value in ['GeoJSON', 'GPKG', 'DXF', 'GPX', 'Shapefile', 'KML', 'CSV', 'FileGDB']:
+            # checks the file format or file first layer format to dynamically set the publish_as
+            if self.file_type.value in ['GeoJSON', 'GPKG', 'DXF', 'GPX', 'Shapefile', 'KML', 'CSV', 'FileGDB'] or \
+                (self.file_type.value in ['Complex'] and self.layers and \
+                    FileType(self.layers[0]['format']).value in ['GeoJSON', 'GPKG', 'DXF', 'GPX', 'Shapefile', 'KML', 'CSV', 'FileGDB']):
                 publish_as = PublishFileType.VECTOR
-            elif self.file_type.value in ['GeoTIFF']:
+
+            elif self.file_type.value in ['GeoTIFF'] or \
+                    (self.file_type.value in ['Complex'] and self.layers and \
+                        FileType(self.layers[0]['format']).value in ['GeoTIFF']):
                 publish_as = PublishFileType.RASTER
-            elif self.file_type.value in ['GLB']:
+
+            elif self.file_type.value in ['GLB'] or \
+                    (self.file_type.value in ['Complex'] and self.layers and \
+                        FileType(self.layers[0]['format']).value in ['GLB']):
                 publish_as = PublishFileType.MODEL3D
 
+            else:
+                raise ValidationError('Unknown format')
+
         data = clean_data({
             "publish_as": publish_as.value if isinstance(publish_as, PublishFileType) else publish_as,
             "layer_name": name,
-            "input_layer": input_layer,
+            "input_layer": self.layers[0]['layer'] if not input_layer and self.layers else input_layer,
             "input_geom_type": input_geom_type.value if isinstance(input_geom_type, InputGeomType) else input_geom_type,
             "replace_domain_codes_by_values": replace_domain_codes_by_values,
-            "input_dataset": self.name if not input_dataset else input_dataset,
+            "input_dataset": self.layers[0]['dataset'] if not input_layer and self.layers else input_dataset,
             "user_id": user_id,
             "input_srid": input_srid,
             "file_encoding": file_encoding,
             "report_errors": report_errors,
             "as_terrain": as_terrain
         })
-
         endpoint = urljoin(self.endpoint, 'publish/')
         response = self.api.post(endpoint, data, is_json=False)
         task = Task.get_task(self.api, response.get('task_id'))

{geobox-1.2.1.dist-info → geobox-1.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geobox
-Version: 1.2.1
+Version: 1.2.2
 Summary: SDK for Geobox's APIs
 Author-email: Hamid Heydari <heydari.h62@gmail.com>
 License: MIT

{geobox-1.2.1.dist-info → geobox-1.2.2.dist-info}/RECORD

@@ -9,7 +9,7 @@ geobox/enums.py,sha256=jGs8GzyCzQMJ5go9owDKMiAwuWjDCpKPaNmQg2eiCaQ,6613
 geobox/exception.py,sha256=jvpnv0M2Ck1FpxHTL_aKYWxGvLnCQ3d9vOrMIktjw1U,1507
 geobox/feature.py,sha256=3Kbc1LjIkBt1YqRAry84BrR7qxx9BexvBB3YQ8D9mGk,18504
 geobox/field.py,sha256=2VxjeYgRwnDxHYpAsK0ASOCz8V0JmfTA3GqHDv3rfgQ,10526
-geobox/file.py,sha256=NqN9EoSGMiTev5vo_1QKciofcyoXonLoZz7B8hH_igk,19352
+geobox/file.py,sha256=Ula8r8pZaCmifsnUHDbaJujw53SLxUiCiBzVS358o88,19967
 geobox/log.py,sha256=ZTmVErhyAszf7M5YFxT5mqXNNDGPnRwXPeGLDS1G6PE,3582
 geobox/map.py,sha256=98P1gbB5U_eu71Hu6EQ0kL_ponbPAnoOCitjEa35q4I,31307
 geobox/model3d.py,sha256=qRYCx-q9LpGhdu5oz3av0uUoiDimuk4BvXXwMW5bo9Q,12061
@@ -30,8 +30,8 @@ geobox/vectorlayer.py,sha256=xnYsJei-bpKgM_EJlRbZ-bAIHdmvfU-VZ2t-NEEJCfc,49420
 geobox/version.py,sha256=0GLPhxCeEb2bAkdpPJWtXPXc1KP6kQ_TOMwLAL0ldo0,9374
 geobox/view.py,sha256=fRYlzNu4eGl6Zx9gPom47BkVE8DfWLj0bNlW2-u4TOU,37390
 geobox/workflow.py,sha256=6hKnSw4G0_ZlgmUb0g3fxT-UVsFbiYpF2FbEO5fpQv0,11606
-geobox-1.2.1.dist-info/licenses/LICENSE,sha256=AvFB7W94sJYKLDhBxLRshL3upexCOG8HQY_1JibB96w,1063
-geobox-1.2.1.dist-info/METADATA,sha256=toqzfxiUTgzmI9Vn5BAIN6v1D_4HGsT-naLcNP4F-28,2556
-geobox-1.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-geobox-1.2.1.dist-info/top_level.txt,sha256=ppXH8Bu2mlB-pLQ6lsoWEm2Gr6wZx1uzkhetsYA5ins,7
-geobox-1.2.1.dist-info/RECORD,,
+geobox-1.2.2.dist-info/licenses/LICENSE,sha256=AvFB7W94sJYKLDhBxLRshL3upexCOG8HQY_1JibB96w,1063
+geobox-1.2.2.dist-info/METADATA,sha256=RZiM4hs-22stMecCl19HntScQWrGeUiYN_YpyVPQUKc,2556
+geobox-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+geobox-1.2.2.dist-info/top_level.txt,sha256=ppXH8Bu2mlB-pLQ6lsoWEm2Gr6wZx1uzkhetsYA5ins,7
+geobox-1.2.2.dist-info/RECORD,,