latch 2.51.2__tar.gz → 2.51.2.dev2__tar.gz

{latch-2.51.2/latch.egg-info → latch-2.51.2.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: latch
-Version: 2.51.2
+Version: 2.51.2.dev2
 Summary: The Latch SDK
 Author-email: kenny@latch.bio
 Classifier: Programming Language :: Python :: 3.8

{latch-2.51.2 → latch-2.51.2.dev2/latch.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: latch
-Version: 2.51.2
+Version: 2.51.2.dev2
 Summary: The Latch SDK
 Author-email: kenny@latch.bio
 Classifier: Programming Language :: Python :: 3.8

{latch-2.51.2 → latch-2.51.2.dev2}/latch.egg-info/SOURCES.txt

@@ -111,7 +111,13 @@ latch_cli/services/execute/utils.py
 latch_cli/services/init/__init__.py
 latch_cli/services/init/init.py
 latch_cli/services/init/__pycache__/__init__.cpython-310.pyc
+latch_cli/services/init/__pycache__/__init__.cpython-311.pyc
+latch_cli/services/init/__pycache__/__init__.cpython-38.pyc
+latch_cli/services/init/__pycache__/__init__.cpython-39.pyc
 latch_cli/services/init/__pycache__/init.cpython-310.pyc
+latch_cli/services/init/__pycache__/init.cpython-311.pyc
+latch_cli/services/init/__pycache__/init.cpython-38.pyc
+latch_cli/services/init/__pycache__/init.cpython-39.pyc
 latch_cli/services/init/assemble_and_sort/.env
 latch_cli/services/init/assemble_and_sort/LICENSE
 latch_cli/services/init/assemble_and_sort/README.md
@@ -119,18 +125,22 @@ latch_cli/services/init/assemble_and_sort/__init__.py
 latch_cli/services/init/assemble_and_sort/assemble.py
 latch_cli/services/init/assemble_and_sort/sort.py
 latch_cli/services/init/assemble_and_sort/system-requirements.txt
+latch_cli/services/init/assemble_and_sort/__pycache__/__init__.cpython-310.pyc
 latch_cli/services/init/common/.dockerignore
 latch_cli/services/init/example_conda/__init__.py
 latch_cli/services/init/example_conda/conda_task.py
 latch_cli/services/init/example_conda/environment.yaml
+latch_cli/services/init/example_conda/__pycache__/__init__.cpython-310.pyc
 latch_cli/services/init/example_docker/__init__.py
 latch_cli/services/init/example_docker/task.py
+latch_cli/services/init/example_nf_integration/latch_metadata/__pycache__/__init__.cpython-311.pyc
 latch_cli/services/init/example_nfcore/Dockerfile
 latch_cli/services/init/example_nfcore/__init__.py
 latch_cli/services/init/example_nfcore/task.py
 latch_cli/services/init/example_r/__init__.py
 latch_cli/services/init/example_r/environment.R
 latch_cli/services/init/example_r/r_task.py
+latch_cli/services/init/example_r/__pycache__/__init__.cpython-310.pyc
 latch_cli/services/init/example_snakemake/Dockerfile
 latch_cli/services/init/example_snakemake/Snakefile
 latch_cli/services/init/example_snakemake/config.yaml
@@ -143,6 +153,7 @@ latch_cli/services/init/template/LICENSE
 latch_cli/services/init/template/README.md
 latch_cli/services/init/template/__init__.py
 latch_cli/services/init/template/task.py
+latch_cli/services/init/template/__pycache__/__init__.cpython-310.pyc
 latch_cli/services/register/__init__.py
 latch_cli/services/register/constants.py
 latch_cli/services/register/register.py

{latch-2.51.2 → latch-2.51.2.dev2}/latch_cli/menus.py

@@ -1,13 +1,66 @@
 import os
 import sys
-import termios
-import tty
+from functools import wraps
 from typing import Any, Callable, Generic, List, Optional, Tuple, TypeVar
 
-from typing_extensions import TypedDict
+from typing_extensions import ParamSpec, TypedDict
 
 from latch_cli.click_utils import AnsiCodes
 
+P = ParamSpec("P")
+T = TypeVar("T")
+
+
+def raw_input_if_available(f: Callable[P, T]) -> Callable[P, T]:
+    if sys.platform == "win32":
+
+        @wraps(f)
+        def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+            try:
+                from ctypes import pointer, windll
+                from ctypes.wintypes import DWORD, HANDLE
+            except ImportError:
+                return f(*args, **kwargs)
+
+            original_mode = DWORD()
+            handle = HANDLE(windll.kernel32.GetStdHandle(STD_INPUT_HANDLE))
+            windll.kernel32.GetConsoleMode(handle, pointer(original_mode))
+
+            ENABLE_ECHO_INPUT = 0x0004
+            ENABLE_LINE_INPUT = 0x0002
+            ENABLE_PROCESSED_INPUT = 0x0001
+
+            windll.kernel32.SetConsoleMode(
+                handle,
+                original_mode.value
+                & ~(ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT | ENABLE_PROCESSED_INPUT),
+            )
+
+            try:
+                return f(*args, **kwargs)
+            finally:
+                windll.kernel32.SetConsoleMode(handle, original_mode)
+
+    else:
+
+        @wraps(f)
+        def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+            try:
+                import termios
+                import tty
+            except ImportError:
+                return f(*args, **kwargs)
+
+            old_settings = termios.tcgetattr(sys.stdin.fileno())
+            tty.setraw(sys.stdin.fileno())
+
+            try:
+                return f(*args, **kwargs)
+            finally:
+                termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings)
+
+    return wrapper
+
 
 def buffered_print() -> Tuple[Callable, Callable]:
     buffer = []
@@ -196,14 +249,12 @@ def read_bytes(num_bytes: int) -> bytes:
     return result
 
 
-T = TypeVar("T")
-
-
 class SelectOption(TypedDict, Generic[T]):
     display_name: str
     value: T
 
 
+@raw_input_if_available
 def select_tui(
     title: str, options: List[SelectOption[T]], clear_terminal: bool = True
 ) -> Optional[T]:
@@ -261,9 +312,6 @@ def select_tui(
 
         return num_lines_rendered
 
-    old_settings = termios.tcgetattr(sys.stdin.fileno())
-    tty.setraw(sys.stdin.fileno())
-
     curr_selected = 0
     start_index = 0
     _, term_height = os.get_terminal_size()
@@ -319,4 +367,3 @@ def select_tui(
         clear(num_lines_rendered)
         reveal_cursor()
         _show()
-        termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings)

{latch-2.51.2 → latch-2.51.2.dev2}/latch_cli/services/execute/main.py

@@ -4,15 +4,14 @@ import json
 import os
 import signal
 import sys
-import termios
-import tty
-from typing import Generic, Literal, Optional, Tuple, TypedDict, TypeVar, Union
+from typing import Literal, Optional, TypedDict, Union
 from urllib.parse import urljoin, urlparse
 
 import websockets.client as websockets
 from latch_sdk_config.latch import NUCLEUS_URL
 from typing_extensions import TypeAlias
 
+from latch_cli.menus import raw_input_if_available
 from latch_cli.services.execute.utils import (
     ContainerNode,
     EGNNode,
@@ -174,6 +173,7 @@ async def connect(egn_info: EGNNode, container_info: Optional[ContainerNode]):
             pass
 
 
+@raw_input_if_available
 def exec(
     execution_id: Optional[str] = None,
     egn_id: Optional[str] = None,
@@ -186,10 +186,4 @@ def exec(
     egn_info = get_egn_info(execution_info, egn_id)
     container_info = get_container_info(egn_info, container_index)
 
-    old_settings_stdin = termios.tcgetattr(sys.stdin.fileno())
-    tty.setraw(sys.stdin)
-
-    try:
-        asyncio.run(connect(egn_info, container_info))
-    finally:
-        termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings_stdin)
+    asyncio.run(connect(egn_info, container_info))

{latch-2.51.2 → latch-2.51.2.dev2}/latch_cli/services/get_executions.py

@@ -1,8 +1,5 @@
 import os
-import sys
-import termios
 import textwrap
-import tty
 from pathlib import Path
 from typing import Dict, List
 
@@ -50,12 +47,10 @@ def get_executions():
     ]
 
     for execution_data in sorted(data.values(), key=lambda x: -int(x["id"])):
-        options.append(
-            {
-                **execution_data,
-                "workflow_tagged": f'{execution_data["workflow_name"]}/{execution_data["workflow_version"]}',
-            }
-        )
+        options.append({
+            **execution_data,
+            "workflow_tagged": f'{execution_data["workflow_name"]}/{execution_data["workflow_version"]}',
+        })
 
     _all_executions(
         title="All Executions",
@@ -64,6 +59,7 @@ def get_executions():
     )
 
 
+@menus.raw_input_if_available
 def _all_executions(
     title: str,
     column_names: List[str],
@@ -142,9 +138,6 @@ def _all_executions(
         menus._show()
         return max_row_len
 
-    old_settings = termios.tcgetattr(sys.stdin.fileno())
-    tty.setraw(sys.stdin.fileno())
-
     curr_selected = hor_index = 0
 
     menus.remove_cursor()
@@ -223,13 +216,13 @@ def _all_executions(
                 prev = (curr_selected, hor_index, term_width, term_height)
                 menus.clear_screen()
                 max_row_len = render(curr_selected, hor_index, term_width, term_height)
-    except KeyboardInterrupt: ...
+    except KeyboardInterrupt:
+        ...
     finally:
         menus.clear_screen()
         menus.reveal_cursor()
         menus.move_cursor((0, 0))
         menus._show()
-        termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings)
 
 
 def _execution_dashboard(execution_data: Dict[str, str], workflow_graph: Dict):
@@ -316,7 +309,8 @@ def _execution_dashboard(execution_data: Dict[str, str], workflow_graph: Dict):
                 menus.clear_screen()
                 prev = (curr_selected, term_width, term_height)
                 render(curr_selected, term_width, term_height)
-    except KeyboardInterrupt: ...
+    except KeyboardInterrupt:
+        ...
     finally:
         menus.clear_screen()
         menus.move_cursor((0, 0))
@@ -452,7 +446,8 @@ def _log_window(execution_data, fixed_workflow_graph: list, selected: int):
                 menus.clear_screen()
                 prev_term_dims = (vert_index, hor_index, term_width, term_height)
                 render(vert_index, hor_index, term_width, term_height)
-    except KeyboardInterrupt: ...
+    except KeyboardInterrupt:
+        ...
     finally:
         log_sched.shutdown()
         log_file.unlink(missing_ok=True)
@@ -513,7 +508,8 @@ def _abort_modal(execution_data):
             if prev_term_dims != (term_width, term_height):
                 prev_term_dims = (term_width, term_height)
                 render(term_width, term_height)
-    except KeyboardInterrupt: ...
+    except KeyboardInterrupt:
+        ...
     finally:
         menus.clear_screen()
         menus.move_cursor((0, 0))

{latch-2.51.2 → latch-2.51.2.dev2}/latch_cli/services/local_dev_old.py

@@ -4,10 +4,8 @@ import os
 import signal
 import subprocess
 import sys
-import termios
 import time
 import traceback
-import tty
 from enum import Enum
 from http.client import HTTPException
 from pathlib import Path
@@ -25,6 +23,7 @@ from watchfiles import awatch
 
 from latch.utils import current_workspace, retrieve_or_login
 from latch_cli.constants import docker_image_name_illegal_pat
+from latch_cli.menus import raw_input_if_available
 from latch_cli.tinyrequests import post
 from latch_cli.utils import TemporarySSHCredentials, identifier_suffix_from_str
 
@@ -253,11 +252,10 @@ async def _send_resize_message(
     )
 
 
+@raw_input_if_available
 async def _shell_session(
     ws: client.WebSocketClientProtocol,
 ):
-    old_settings_stdin = termios.tcgetattr(sys.stdin.fileno())
-    tty.setraw(sys.stdin)
 
     loop = asyncio.get_event_loop()
     reader = asyncio.StreamReader(loop=loop)
@@ -332,7 +330,6 @@ async def _shell_session(
     except asyncio.CancelledError:
         ...
     finally:
-        termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings_stdin)
         signal.signal(signal.SIGWINCH, old_sigwinch_handler)
 
 

{latch-2.51.2 → latch-2.51.2.dev2}/latch_cli/services/preview.py

@@ -1,7 +1,4 @@
 import os
-import sys
-import termios
-import tty
 import webbrowser
 from pathlib import Path
 from typing import List
@@ -72,7 +69,8 @@ def preview(pkg_root: Path):
     webbrowser.open(url)
 
 
-# TODO(ayush): abstract this logic in a unified interface that all tui commands use
+# todo(ayush): abstract this logic in a unified interface that all tui commands use
+@menus.raw_input_if_available
 def _select_workflow_tui(title: str, options: List[str], clear_terminal: bool = True):
     """
     Renders a terminal UI that allows users to select one of the options
@@ -125,9 +123,6 @@ def _select_workflow_tui(title: str, options: List[str], clear_terminal: bool =
 
         return num_lines_rendered
 
-    old_settings = termios.tcgetattr(sys.stdin.fileno())
-    tty.setraw(sys.stdin.fileno())
-
     curr_selected = 0
     start_index = 0
     _, term_height = os.get_terminal_size()
@@ -176,9 +171,9 @@ def _select_workflow_tui(title: str, options: List[str], clear_terminal: bool =
                 start_index=start_index,
                 max_per_page=max_per_page,
             )
-    except KeyboardInterrupt: ...
+    except KeyboardInterrupt:
+        ...
     finally:
         menus.clear(num_lines_rendered)
         menus.reveal_cursor()
         menus._show()
-        termios.tcsetattr(sys.stdin.fileno(), termios.TCSANOW, old_settings)

{latch-2.51.2 → latch-2.51.2.dev2}/setup.py

@@ -13,7 +13,7 @@ if cur_ver < (3, 8) or cur_ver > (3, 11):
 
 setup(
     name="latch",
-    version="v2.51.2",
+    version="v2.51.2.dev2",
     author_email="kenny@latch.bio",
     description="The Latch SDK",
     packages=find_packages(),