geobox 1.3.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

geobox/mosaic.py

@@ -295,7 +295,7 @@ class Mosaic(Raster):
         return super().get_point(lat, lng)
 
 
-    def get_tile_render_url(self, x: int, y: int, z: int) -> str:
+    def get_render_png_url(self, x: int, y: int, z: int, **kwargs) -> str:
         """
         Get the tile render URL of the mosaic.
 
@@ -304,6 +304,15 @@ class Mosaic(Raster):
             y (int): The y coordinate of the tile.
             z (int): The zoom level of the tile.
 
+        Keyword Args:
+            indexes (str, optional): list of comma separated band indexes to be rendered. e.g. 1, 2, 3
+            nodata (int, optional)
+            expression (str, optional): band math expression. e.g. b1*b2+b3
+            rescale (List, optional): comma (',') separated Min,Max range. Can set multiple time for multiple bands.
+            color_formula (str, optional): Color formula. e.g. gamma R 0.5
+            colormap_name (str, optional)
+            colormap (str, optional): JSON encoded custom Colormap. e.g. {"0": "#ff0000", "1": "#00ff00"} or [[[0, 100], "#ff0000"], [[100, 200], "#00ff00"]]
+
         Returns:
             str: The tile render URL of the mosaic.
 
@@ -314,7 +323,7 @@ class Mosaic(Raster):
             >>> mosaic = Mosaic.get_mosaic(client, uuid="12345678-1234-5678-1234-567812345678")
             >>> mosaic.get_tile_render_url(x=1, y=1, z=1)
         """
-        return super().get_tile_render_url(x, y, z)
+        return super().get_render_png_url(x, y, z, **kwargs)
     
 
     def get_tile_png_url(self, x: int, y: int, z: int) -> str:

geobox/raster.py

@@ -3,6 +3,7 @@ from urllib.parse import urljoin, urlencode
 from typing import Optional, Dict, List, Optional, Union, TYPE_CHECKING
 import mimetypes
 import requests
+import sys
 
 from .base import Base
 from .utils import clean_data
@@ -179,7 +180,6 @@ class Raster(Base):
             name (str): The name of the raster.
             display_name (str): The display name of the raster.
             description (str): The description of the raster.
-            max_zoom (int): The max zoom of the raster.
 
         Returns:
             None
@@ -194,10 +194,9 @@ class Raster(Base):
         params = {
             'name': kwargs.get('name'),
             'display_name': kwargs.get('display_name'),
-            'description': kwargs.get('description'),
-            'max_zoom': kwargs.get('max_zoom'),
+            'description': kwargs.get('description')
         }
-        super()._update(self.endpoint, params)
+        return super()._update(self.endpoint, params)
     
 
     def delete(self) -> None:
@@ -236,7 +235,8 @@ class Raster(Base):
         return endpoint
     
     
-    def get_raster_info(self) -> Dict:
+    @property
+    def info(self) -> Dict:
         """
         Get the info of the raster.
 
@@ -248,7 +248,7 @@ class Raster(Base):
             >>> from geobox.raster import Raster
             >>> client = GeoboxClient()
             >>> raster = Raster.get_raster(client, uuid="12345678-1234-5678-1234-567812345678")
-            >>> raster.get_raster_info()
+            >>> raster.info
         """
         endpoint = urljoin(self.endpoint, 'info/')
         return self.api.get(endpoint)
@@ -323,53 +323,74 @@ class Raster(Base):
         
         Raises:
             ValueError: If save_path does not end with a '/'.
-        """
-        # 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}'
 
+        Example:
+            >>> from geobox import GeoboxClient
+            >>> from geobox.file import File
+            >>> from geobox import GeoboxClient
+            >>> client = GeoboxClient()
+            >>> file_path = File.get_file(client, uuid="12345678-1234-5678-1234-567812345678")
+        """
         # 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 and 'filename=' in response.headers['Content-Disposition']:
-            content_disposition = response.headers['Content-Disposition']
-            filename = content_disposition.split('filename=')[-1].strip().strip('"')
-            ext = f".{filename.split('.')[-1]}"
+            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])
+            file_name = f'{self.name}.{mimetypes.guess_extension(content_type.split(";")[0])}'
 
-        return ext
+        return file_name
 
 
-    def download(self, save_path: str = None) -> str:
+    def _create_progress_bar(self) -> 'tqdm':
+        """Creates a progress bar for the task."""
+        try:
+            from tqdm.auto import tqdm
+        except ImportError:
+            from .api import logger
+            logger.warning("[tqdm] extra is required to show the progress bar. install with: pip insatll geobox[tqdm]")
+            return None
+
+        return tqdm(unit="B", 
+                        total=int(self.size), 
+                        file=sys.stdout,
+                        dynamic_ncols=True,
+                        desc="Downloading",
+                        unit_scale=True,
+                        unit_divisor=1024, 
+                        ascii=True
+                )
+
+
+    def download(self, save_path: str = None, progress_bar: bool = True) -> str:
         """
         Download the raster.
 
         Args:
-            save_path (str): The path to save the raster.
-
+            save_path (str, optional): Path where the file should be saved. 
+                                    If not provided, it saves to the current working directory
+                                    using the original filename and appropriate extension.
+            progress_bar (bool, optional): Whether to show a progress bar. default: True
+        
         Returns:
             str: The path to save the raster.
 
@@ -390,23 +411,32 @@ class Raster(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:
-            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.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 get_content_file(self, save_path: str = None) -> str: 
+    def get_content_file(self, save_path: str = None, progress_bar: bool = True) -> str: 
         """
         Get Raster Content URL
 
         Args:
-            save_path (str): The path to save the raster.
-
+            save_path (str, optional): Path where the file should be saved. 
+                                    If not provided, it saves to the current working directory
+                                    using the original filename and appropriate extension.
+            progress_bar (bool, optional): Whether to show a progress bar. default: True
+        
         Returns:
             str: The path to save the raster.
 
@@ -427,17 +457,23 @@ class Raster(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}content/", stream=True) as response, \
-            open(save_path, 'wb') as f:
-            for chunk in response.iter_content(chunk_size=8192):
-                f.write(chunk)
+        with self.api.get(f"{self.endpoint}content/", 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.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 get_tile_render_url(self, x: int, y: int, z: int) -> str:
+    def get_render_png_url(self, x: int, y: int, z: int, **kwargs) -> str:
         """
         Get the PNG URL of the raster.
 
@@ -446,6 +482,15 @@ class Raster(Base):
             y (int): The y coordinate of the tile.
             z (int): The zoom level of the tile.
 
+        Keyword Args:
+            indexes (str, optional): list of comma separated band indexes to be rendered. e.g. 1, 2, 3
+            nodata (int, optional)
+            expression (str, optional): band math expression. e.g. b1*b2+b3
+            rescale (List, optional): comma (',') separated Min,Max range. Can set multiple time for multiple bands.
+            color_formula (str, optional): Color formula. e.g. gamma R 0.5
+            colormap_name (str, optional)
+            colormap (str, optional): JSON encoded custom Colormap. e.g. {"0": "#ff0000", "1": "#00ff00"} or [[[0, 100], "#ff0000"], [[100, 200], "#00ff00"]]
+
         Returns:
             str: The PNG Render URL of the raster.
 
@@ -456,7 +501,18 @@ class Raster(Base):
             >>> raster = Raster.get_raster(client, uuid="12345678-1234-5678-1234-567812345678")
             >>> raster.get_tile_render_url(x=10, y=20, z=1)
         """
+        params = clean_data({
+            'indexes': kwargs.get('indexes'),
+            'nodata': kwargs.get('nodata'),
+            'expression': kwargs.get('expression'),
+            'rescale': kwargs.get('rescale'),
+            'color_formula': kwargs.get('color_formula'),
+            'colormap_name': kwargs.get('colormap_name'),
+            'colormap': kwargs.get('colormap')
+        })
+        query_string = urlencode(clean_data(params))
         endpoint = urljoin(self.api.base_url, f'{self.endpoint}render/{z}/{x}/{y}.png')
+        endpoint = urljoin(endpoint, f'?{query_string}')
         return endpoint
     
 

{geobox-1.3.0.dist-info → geobox-1.3.1.dist-info}/METADATA

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

{geobox-1.3.0.dist-info → geobox-1.3.1.dist-info}/RECORD

@@ -13,10 +13,10 @@ geobox/file.py,sha256=Ula8r8pZaCmifsnUHDbaJujw53SLxUiCiBzVS358o88,19967
 geobox/log.py,sha256=ZTmVErhyAszf7M5YFxT5mqXNNDGPnRwXPeGLDS1G6PE,3582
 geobox/map.py,sha256=RjuhbOVQN-OBwjBtO46AosB_Ta348EA1nExBdnxH3nw,31252
 geobox/model3d.py,sha256=qRYCx-q9LpGhdu5oz3av0uUoiDimuk4BvXXwMW5bo9Q,12061
-geobox/mosaic.py,sha256=RmMyvzt6C2Fd8a0nJX5gUOGJnO3MrCWgwmYfhJ9d_Jw,23025
+geobox/mosaic.py,sha256=feBRsAgvhl0tKoghfZiQsxFntS7skAnzEYYuU4GKbP0,23672
 geobox/plan.py,sha256=_ZV9F6loG92uQZGJl_9T08Kg85g3hnODmpccSkTYVdw,11193
 geobox/query.py,sha256=1guO9O9DAEb8oj_iMiW3PhEDgMVMAN9ccEp19dnwAWE,24209
-geobox/raster.py,sha256=DkvajG2jWDlGFbS2ywv4XCVOyfR82ouUMT3tdIE1PsU,26437
+geobox/raster.py,sha256=06L95vTFdbOwykDxo6XDRkNIDJKMpFcsjbFg75WGLLA,29177
 geobox/route.py,sha256=cKZTTHBHM8woWbhHIpCvrRX_doJcTkGktouee0z7oG0,2723
 geobox/scene.py,sha256=Sz2tiyJk-bXYjktfcp1d2gGrW3gt2T4D1FAMoRHOCng,11369
 geobox/settings.py,sha256=rGRdM18Vo7xnjfZXPLRMbKeoVC_lZmzkNRPS0SQv05o,5660
@@ -30,8 +30,8 @@ geobox/vectorlayer.py,sha256=1xjhWpvb1hFhLRei2hwU7IY5KxDKzynZMb3fbIDTMZE,51092
 geobox/version.py,sha256=0GLPhxCeEb2bAkdpPJWtXPXc1KP6kQ_TOMwLAL0ldo0,9374
 geobox/view.py,sha256=sIi6Mi7NUAX5gN83cpubD3l7AcZ-1g5kG2NKHeVeW-g,37518
 geobox/workflow.py,sha256=6hKnSw4G0_ZlgmUb0g3fxT-UVsFbiYpF2FbEO5fpQv0,11606
-geobox-1.3.0.dist-info/licenses/LICENSE,sha256=AvFB7W94sJYKLDhBxLRshL3upexCOG8HQY_1JibB96w,1063
-geobox-1.3.0.dist-info/METADATA,sha256=UW0-KY-DuDA4blEJeWwoP05fxSMQ-yOa7AjfIs0dzzg,2556
-geobox-1.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-geobox-1.3.0.dist-info/top_level.txt,sha256=ppXH8Bu2mlB-pLQ6lsoWEm2Gr6wZx1uzkhetsYA5ins,7
-geobox-1.3.0.dist-info/RECORD,,
+geobox-1.3.1.dist-info/licenses/LICENSE,sha256=AvFB7W94sJYKLDhBxLRshL3upexCOG8HQY_1JibB96w,1063
+geobox-1.3.1.dist-info/METADATA,sha256=mK4pFtm6hb68Yy5t_t1E6HCi3z9M5W18fV_N-MOYmgM,2556
+geobox-1.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+geobox-1.3.1.dist-info/top_level.txt,sha256=ppXH8Bu2mlB-pLQ6lsoWEm2Gr6wZx1uzkhetsYA5ins,7
+geobox-1.3.1.dist-info/RECORD,,