fal 1.11.0__py3-none-any.whl → 1.11.1__py3-none-any.whl

fal/_fal_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.11.0'
-__version_tuple__ = version_tuple = (1, 11, 0)
+__version__ = version = '1.11.1'
+__version_tuple__ = version_tuple = (1, 11, 1)

fal/cli/_utils.py

@@ -25,7 +25,7 @@ def get_app_data_from_toml(app_name):
         raise ValueError(f"App {app_name} not found in pyproject.toml")
 
     try:
-        app_ref = app_data["ref"]
+        app_ref = app_data.pop("ref")
     except KeyError:
         raise ValueError(f"App {app_name} does not have a ref key in pyproject.toml")
 
@@ -33,8 +33,11 @@ def get_app_data_from_toml(app_name):
     project_root, _ = find_project_root(None)
     app_ref = str(project_root / app_ref)
 
-    app_auth = app_data.get("auth", "private")
-    app_deployment_strategy = app_data.get("deployment_strategy", "recreate")
-    app_no_scale = app_data.get("no_scale", False)
+    app_auth = app_data.pop("auth", "private")
+    app_deployment_strategy = app_data.pop("deployment_strategy", "recreate")
+    app_no_scale = app_data.pop("no_scale", False)
+
+    if len(app_data) > 0:
+        raise ValueError(f"Found unexpected keys in pyproject.toml: {app_data}")
 
     return app_ref, app_auth, app_deployment_strategy, app_no_scale

fal/toolkit/utils/download_utils.py

@@ -126,11 +126,12 @@ def download_file(
     *,
     force: bool = False,
     request_headers: dict[str, str] | None = None,
+    filesize_limit: int | None = None,
 ) -> Path:
     """Downloads a file from the specified URL to the target directory.
 
     The function downloads the file from the given URL and saves it in the specified
-    target directory.
+    target directory, provided it is below the given filesize limit.
 
     It also checks whether the local file already exists and whether its content length
     matches the expected content length from the remote file. If the local file already
@@ -151,6 +152,8 @@ def download_file(
             Defaults to `False`.
         request_headers: A dictionary containing additional headers to be included in
             the HTTP request. Defaults to `None`.
+        filesize_limit: An integer specifying the maximum downloadable size,
+            in megabytes. Defaults to `None`.
 
 
     Returns:
@@ -160,12 +163,22 @@ def download_file(
         ValueError: If the provided `file_name` contains a forward slash ('/').
         DownloadError: If an error occurs during the download process.
     """
+    ONE_MB = 1024**2
+
     try:
-        file_name = _get_remote_file_properties(url, request_headers)[0]
+        file_name, expected_filesize = _get_remote_file_properties(url, request_headers)
     except Exception as e:
-        print(f"GOt error: {e}")
+        print(f"Got error: {e}")
         raise DownloadError(f"Failed to get remote file properties for {url}") from e
 
+    expected_filesize_mb = expected_filesize / ONE_MB
+
+    if filesize_limit is not None and expected_filesize_mb > filesize_limit:
+        raise DownloadError(
+            f"""File to be downloaded is of size {expected_filesize_mb},
+                which is over the limit of {filesize_limit}"""
+        )
+
     if "/" in file_name:
         raise ValueError(f"File name '{file_name}' cannot contain a slash.")
 
@@ -194,7 +207,10 @@ def download_file(
 
     try:
         _download_file_python(
-            url=url, target_path=target_path, request_headers=request_headers
+            url=url,
+            target_path=target_path,
+            request_headers=request_headers,
+            filesize_limit=filesize_limit,
         )
     except Exception as e:
         msg = f"Failed to download {url} to {target_path}"
@@ -207,7 +223,10 @@ def download_file(
 
 
 def _download_file_python(
-    url: str, target_path: Path | str, request_headers: dict[str, str] | None = None
+    url: str,
+    target_path: Path | str,
+    request_headers: dict[str, str] | None = None,
+    filesize_limit: int | None = None,
 ) -> Path:
     """Download a file from a given URL and save it to a specified path using a
     Python interface.
@@ -217,6 +236,8 @@ def _download_file_python(
         target_path: The path where the downloaded file will be saved.
         request_headers: A dictionary containing additional headers to be included in
             the HTTP request. Defaults to `None`.
+        filesize_limit: A integer value specifying how many megabytes can be
+            downloaded at maximum. Defaults to `None`.
 
     Returns:
         The path where the downloaded file has been saved.
@@ -233,7 +254,10 @@ def _download_file_python(
             file_path = temp_file.name
 
             for progress, total_size in _stream_url_data_to_file(
-                url, temp_file.name, request_headers=request_headers
+                url,
+                temp_file.name,
+                request_headers=request_headers,
+                filesize_limit=filesize_limit,
             ):
                 if total_size:
                     progress_msg = f"Downloading {url} ... {progress:.2%}"
@@ -261,6 +285,7 @@ def _stream_url_data_to_file(
     file_path: str,
     chunk_size_in_mb: int = 64,
     request_headers: dict[str, str] | None = None,
+    filesize_limit: int | None = None,
 ):
     """Download data from a URL and stream it to a file.
 
@@ -277,6 +302,8 @@ def _stream_url_data_to_file(
             Defaults to 64.
         request_headers: A dictionary containing additional headers to be included in
             the HTTP request. Defaults to `None`.
+        filesize_limit: An integer specifying how many megabytes can be
+            downloaded at maximum. Defaults to `None`.
 
     Yields:
         A tuple containing two elements:
@@ -300,6 +327,11 @@ def _stream_url_data_to_file(
             f_stream.write(data)
 
             received_size = f_stream.tell()
+            if filesize_limit is not None and received_size > filesize_limit:
+                raise DownloadError(
+                    f"""Attempted to download more data {received_size}
+                        than the set limit of {filesize_limit}"""
+                )
 
             if total_size:
                 progress = received_size / total_size

{fal-1.11.0.dist-info → fal-1.11.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fal
-Version: 1.11.0
+Version: 1.11.1
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8

{fal-1.11.0.dist-info → fal-1.11.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=NYxfUyIQwdONRcmiATXsc3iD0kuKImoMYD3H1Vdv9OM,513
+fal/_fal_version.py,sha256=YfyxoPRBHFNEDWZGG9EWRRtidnKlEjmjHUQe05P3IMI,513
 fal/_serialization.py,sha256=rD2YiSa8iuzCaZohZwN_MPEB-PpSKbWRDeaIDpTEjyY,7653
 fal/_version.py,sha256=EBGqrknaf1WygENX-H4fBefLvHryvJBBGtVJetaB0NY,266
 fal/api.py,sha256=jqmQfhRvZMYpWWvbTfARxjywP_72c314pnLPTb5dGwA,44755
@@ -20,7 +20,7 @@ fal/auth/__init__.py,sha256=MXwS5zyY1SYJWEkc6s39et73Dkg3cDJg1ZwxRhXNj4c,4704
 fal/auth/auth0.py,sha256=rSG1mgH-QGyKfzd7XyAaj1AYsWt-ho8Y_LZ-FUVWzh4,5421
 fal/auth/local.py,sha256=sndkM6vKpeVny6NHTacVlTbiIFqaksOmw0Viqs_RN1U,1790
 fal/cli/__init__.py,sha256=padK4o0BFqq61kxAA1qQ0jYr2SuhA2mf90B3AaRkmJA,37
-fal/cli/_utils.py,sha256=45G0LEz2bW-69MUQKPdatVE_CBC2644gC-V0qdNEsco,1252
+fal/cli/_utils.py,sha256=reMvQm4gcDs3mfvFELzFXnsqAjeurfTBuOfKY8M4yGM,1364
 fal/cli/api.py,sha256=-rl50A00CxqVZtDh0iZmpCHMFY0jZySaumbPCe3MSoQ,2090
 fal/cli/apps.py,sha256=KhweXdLdd1wZquhAMVwgl38dZAyFns-j3ZdIZLOLGWg,11504
 fal/cli/auth.py,sha256=--MhfHGwxmtHbRkGioyn1prKn_U-pBzbz0G_QeZou-U,1352
@@ -68,7 +68,7 @@ fal/toolkit/image/nsfw_filter/inference.py,sha256=BhIPF_zxRLetThQYxDDF0sdx9VRwvu
 fal/toolkit/image/nsfw_filter/model.py,sha256=63mu8D15z_IosoRUagRLGHy6VbLqFmrG-yZqnu2vVm4,457
 fal/toolkit/image/nsfw_filter/requirements.txt,sha256=3Pmrd0Ny6QAeBqUNHCgffRyfaCARAPJcfSCX5cRYpbM,37
 fal/toolkit/utils/__init__.py,sha256=CrmM9DyCz5-SmcTzRSm5RaLgxy3kf0ZsSEN9uhnX2Xo,97
-fal/toolkit/utils/download_utils.py,sha256=Ju-rvQ0oqFkARONRyrfIGVaqthjlxcFIb-qg5SE7yBw,18419
+fal/toolkit/utils/download_utils.py,sha256=NgOMNs-bQGSg3gWnu123BgZitJgJrvtRexIefTMuylY,19739
 fal/toolkit/utils/retry.py,sha256=mHcQvvNIpu-Hi29P1HXSZuyvolRd48dMaJToqzlG0NY,1353
 openapi_fal_rest/__init__.py,sha256=ziculmF_i6trw63LzZGFX-6W3Lwq9mCR8_UpkpvpaHI,152
 openapi_fal_rest/client.py,sha256=G6BpJg9j7-JsrAUGddYwkzeWRYickBjPdcVgXoPzxuE,2817
@@ -133,8 +133,8 @@ openapi_fal_rest/models/workflow_node_type.py,sha256=-FzyeY2bxcNmizKbJI8joG7byRi
 openapi_fal_rest/models/workflow_schema.py,sha256=4K5gsv9u9pxx2ItkffoyHeNjBBYf6ur5bN4m_zePZNY,2019
 openapi_fal_rest/models/workflow_schema_input.py,sha256=2OkOXWHTNsCXHWS6EGDFzcJKkW5FIap-2gfO233EvZQ,1191
 openapi_fal_rest/models/workflow_schema_output.py,sha256=EblwSPAGfWfYVWw_WSSaBzQVju296is9o28rMBAd0mc,1196
-fal-1.11.0.dist-info/METADATA,sha256=gyk3rZlftiD1qXNBRHk5Z_g-CC65Qi9OTS_zuKfQGHE,4043
-fal-1.11.0.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-fal-1.11.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.11.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.11.0.dist-info/RECORD,,
+fal-1.11.1.dist-info/METADATA,sha256=vqzL53kDWgyVECQHZe8riqszOL0GALTXa1ztbCSal-Q,4043
+fal-1.11.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+fal-1.11.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.11.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.11.1.dist-info/RECORD,,

{fal-1.11.0.dist-info → fal-1.11.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any