fal 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

fal/_fal_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.2.0'
-__version_tuple__ = version_tuple = (1, 2, 0)
+__version__ = version = '1.2.1'
+__version_tuple__ = version_tuple = (1, 2, 1)

fal/logging/isolate.py

@@ -17,11 +17,31 @@ class IsolateLogPrinter:
 
     def __init__(self, debug: bool = False) -> None:
         self.debug = debug
+        self._current_source: LogSource | None = None
+
+    def _maybe_print_header(self, source: LogSource):
+        from fal.console import console
+
+        if source == self._current_source:
+            return
+
+        msg = {
+            LogSource.BUILDER: "Building the environment",
+            LogSource.BRIDGE: "Unpacking user code",
+            LogSource.USER: "Running",
+        }.get(source)
+
+        if msg:
+            console.print(f"==> {msg}", style="bold green")
+
+        self._current_source = source
 
     def print(self, log: Log):
         if log.level < LogLevel.INFO and not self.debug:
             return
 
+        self._maybe_print_header(log.source)
+
         if log.source == LogSource.USER:
             stream = sys.stderr if log.level == LogLevel.STDERR else sys.stdout
             print(log.message, file=stream)

fal/toolkit/utils/download_utils.py

@@ -1,11 +1,12 @@
 from __future__ import annotations
 
 import hashlib
+import os
 import shutil
 import subprocess
 import sys
 from pathlib import Path, PurePath
-from tempfile import TemporaryDirectory
+from tempfile import NamedTemporaryFile, TemporaryDirectory
 from urllib.parse import urlparse
 from urllib.request import Request, urlopen
 
@@ -215,10 +216,14 @@ def _download_file_python(
     Returns:
         The path where the downloaded file has been saved.
     """
-    import shutil
-    import tempfile
-
-    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+    basename = os.path.basename(target_path)
+    # NOTE: using the same directory to avoid potential copies across temp fs and target
+    # fs, and also to be able to atomically rename a downloaded file into place.
+    with NamedTemporaryFile(
+        delete=False,
+        dir=os.path.dirname(target_path),
+        prefix=f"{basename}.tmp",
+    ) as temp_file:
         try:
             file_path = temp_file.name
 
@@ -232,13 +237,14 @@ def _download_file_python(
 
                 print(progress_msg, end="\r\n")
 
-            # Move the file when the file is downloaded completely. Since the
-            # file used is temporary, in a case of an interruption, the downloaded
-            # content will be lost. So, it is safe to redownload the file in such cases.
-            shutil.move(file_path, target_path)
+            # NOTE: Atomically renaming the file into place when the file is downloaded
+            # completely.
+            #
+            # Since the file used is temporary, in a case of an interruption, the
+            # downloaded content will be lost. So, it is safe to redownload the file in
+            # such cases.
+            os.rename(file_path, target_path)
 
-        except Exception as error:
-            raise error
         finally:
             Path(temp_file.name).unlink(missing_ok=True)
 
@@ -403,35 +409,38 @@ def clone_repository(
             print(f"Removing the existing repository: {local_repo_path} ")
             shutil.rmtree(local_repo_path)
 
-    # Temporary directory to download the repo into.
-    temp_dir = TemporaryDirectory()
-    temp_dir_path = temp_dir.name
-
-    try:
-        print(f"Cloning the repository '{https_url}' .")
-
-        # Clone with disabling the logs and advices for detached HEAD state.
-        clone_command = [
-            "git",
-            "clone",
-            "--recursive",
-            https_url,
-            temp_dir_path,
-        ]
-        subprocess.check_call(clone_command)
-
-        if commit_hash:
-            checkout_command = ["git", "checkout", commit_hash]
-            subprocess.check_call(checkout_command, cwd=temp_dir_path)
-
-        # Move the repository when the clone and checkout is finished.
-        shutil.move(temp_dir_path, local_repo_path)
-
-    except Exception as error:
-        print(f"{error}\nFailed to clone repository '{https_url}' .")
-        temp_dir.cleanup()
+    # NOTE: using the target_dir to be able to avoid potential copies across temp fs
+    # and target fs, and also to be able to atomically rename repo_name dir into place
+    # when we are done setting it up.
+    os.makedirs(target_dir, exist_ok=True)  # type: ignore[arg-type]
+    with TemporaryDirectory(
+        dir=target_dir,
+        suffix=f"{local_repo_path.name}.tmp",
+    ) as temp_dir:
+        try:
+            print(f"Cloning the repository '{https_url}' .")
+
+            # Clone with disabling the logs and advices for detached HEAD state.
+            clone_command = [
+                "git",
+                "clone",
+                "--recursive",
+                https_url,
+                temp_dir,
+            ]
+            subprocess.check_call(clone_command)
+
+            if commit_hash:
+                checkout_command = ["git", "checkout", commit_hash]
+                subprocess.check_call(checkout_command, cwd=temp_dir)
+
+            # NOTE: Atomically renaming the repository directory into place when the
+            # clone and checkout are done.
+            os.rename(temp_dir, local_repo_path)
 
-        raise error
+        except Exception as error:
+            print(f"{error}\nFailed to clone repository '{https_url}' .")
+            raise error
 
     if include_to_path:
         __add_local_path_to_sys_path(local_repo_path)

{fal-1.2.0.dist-info → fal-1.2.1.dist-info}/METADATA

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

{fal-1.2.0.dist-info → fal-1.2.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-fal/_fal_version.py,sha256=zMnMemknXglcJs59xkicNzeEJTVgYd1omSfLWj76yWw,411
+fal/_fal_version.py,sha256=2U0Gn26fYI3Vgj5hgkLM8I3wI6YEVdffJGllaVW-sSc,411
 fal/_serialization.py,sha256=rD2YiSa8iuzCaZohZwN_MPEB-PpSKbWRDeaIDpTEjyY,7653
 fal/_version.py,sha256=EBGqrknaf1WygENX-H4fBefLvHryvJBBGtVJetaB0NY,266
 fal/api.py,sha256=x60GlBWynDd1yhHsVWeqf07WVTzgbwNC6cqCjhlTiFQ,40556
@@ -35,7 +35,7 @@ fal/exceptions/__init__.py,sha256=x3fp97qMr5zCQJghMq6k2ESXWSrkWumO1BZebh3pWsI,92
 fal/exceptions/_base.py,sha256=oF2XfitbiDGObmSF1IX50uAdV8IUvOfR-YsGmMQSE0A,161
 fal/exceptions/auth.py,sha256=gxRago5coI__vSIcdcsqhhq1lRPkvCnwPAueIaXTAdw,329
 fal/logging/__init__.py,sha256=snqprf7-sKw6oAATS_Yxklf-a3XhLg0vIHICPwLp6TM,1583
-fal/logging/isolate.py,sha256=uKZpF5p02jqaiyXNYt7VUp43ShdKc_N4UUi7z65nmRc,1741
+fal/logging/isolate.py,sha256=jJSgDHkFg4sB0xElYSqCYF6IAxy6jEgSfjwFuKJIZbA,2305
 fal/logging/style.py,sha256=ckIgHzvF4DShM5kQh8F133X53z_vF46snuDHVmo_h9g,386
 fal/logging/trace.py,sha256=OhzB6d4rQZimBc18WFLqH_9BGfqFFumKKTAGSsmWRMg,1904
 fal/logging/user.py,sha256=0Xvb8n6tSb9l_V51VDzv6SOdYEFNouV_6nF_W9e7uNQ,642
@@ -51,7 +51,7 @@ fal/toolkit/file/providers/r2.py,sha256=WxmOHF5WxHt6tKMcFjWj7ZWO8a1EXysO9lfYv_tB
 fal/toolkit/image/__init__.py,sha256=qNLyXsBWysionUjbeWbohLqWlw3G_UpzunamkZd_JLQ,71
 fal/toolkit/image/image.py,sha256=UDIHgkxae8LzmCvWBM9GayMnK8c0JMMfsrVlLnW5rto,4234
 fal/toolkit/utils/__init__.py,sha256=CrmM9DyCz5-SmcTzRSm5RaLgxy3kf0ZsSEN9uhnX2Xo,97
-fal/toolkit/utils/download_utils.py,sha256=fDUITgdGW0wRXLE0NuQg29YJnJS3yr6l0zbRKqX6zMU,17006
+fal/toolkit/utils/download_utils.py,sha256=9WMpn0mFIhkFelQpPj5KG-pC7RMyyOzGHbNRDSyz07o,17664
 openapi_fal_rest/__init__.py,sha256=ziculmF_i6trw63LzZGFX-6W3Lwq9mCR8_UpkpvpaHI,152
 openapi_fal_rest/client.py,sha256=G6BpJg9j7-JsrAUGddYwkzeWRYickBjPdcVgXoPzxuE,2817
 openapi_fal_rest/errors.py,sha256=8mXSxdfSGzxT82srdhYbR0fHfgenxJXaUtMkaGgb6iU,470
@@ -115,8 +115,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.2.0.dist-info/METADATA,sha256=sfrRbze0jlBv3qqW-eG6sDFb6L6G7zHfJ9UpivA--uY,3777
-fal-1.2.0.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-fal-1.2.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.2.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.2.0.dist-info/RECORD,,
+fal-1.2.1.dist-info/METADATA,sha256=1ViXkmFXOXQqD52q0HPxP-BuzHL5RzQ6x6MC__GOXn4,3777
+fal-1.2.1.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+fal-1.2.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.2.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.2.1.dist-info/RECORD,,