dayhoff-tools 1.0.18__py3-none-any.whl → 1.0.19__py3-none-any.whl

dayhoff_tools/cli/utility_commands.py

@@ -7,6 +7,7 @@ import subprocess
 import sys
 from pathlib import Path
 
+import toml
 import typer
 import yaml
 
@@ -201,6 +202,10 @@ def build_and_upload_wheel(bump_part: str = "patch"):
         )
         return
 
+    # ANSI color codes
+    BLUE = "\033[94m"
+    RESET = "\033[0m"
+
     # --- Authentication Setup ---
     token = os.environ.get("UV_PUBLISH_TOKEN")
 
@@ -283,13 +288,13 @@ def build_and_upload_wheel(bump_part: str = "patch"):
 
         # Build wheel and sdist
         build_cmd = ["uv", "build"]
-        print(f"Running command: {' '.join(build_cmd)}")
+        # Print command in blue
+        print(f"Running command: {BLUE}{' '.join(build_cmd)}{RESET}")
         subprocess.run(build_cmd, check=True)
 
         # Upload using uv publish with explicit arguments
-        print(
-            f"Running command: {' '.join(publish_cmd_safe_print)}"
-        )  # Print masked command
+        # Print masked command in blue
+        print(f"Running command: {BLUE}{' '.join(publish_cmd_safe_print)}{RESET}")
         subprocess.run(
             publish_cmd,  # Use the actual command with token
             check=True,
@@ -361,24 +366,108 @@ def build_and_upload_wheel(bump_part: str = "patch"):
 
 
 def update_dayhoff_tools():
-    """Update the dayhoff-tools package to the latest version using uv.
+    """Update dayhoff-tools to latest, update pyproject.toml, and sync.
 
-    Runs `uv add dayhoff-tools@latest`. Note: This command might not correctly
-    update dayhoff-tools if it's only listed in a top-level [dependency-groups]
-    table in the target project's pyproject.toml. It works best for dependencies
-    in [project.dependencies] or [project.optional-dependencies].
+    1. Runs `uv lock --upgrade-package dayhoff-tools`
+    2. Parses uv.lock to find the new version.
+    3. Updates the version constraint in pyproject.toml to `>=<new_version>`.
+    4. Runs `uv sync --no-install-project`.
     """
+    # ANSI color codes
+    BLUE = "\033[94m"
+    RESET = "\033[0m"
+
+    lock_file_path = Path("uv.lock")
+    pyproject_path = Path("pyproject.toml")
+
     try:
-        print("Attempting to update dayhoff-tools to the latest version using uv...")
-        update_cmd = ["uv", "add", "dayhoff-tools@latest"]
-        print(f"Running command: {' '.join(update_cmd)}")
-        subprocess.run(update_cmd, check=True)
-        print("Update attempt complete! Check pyproject.toml and run sync if needed.")
+        print("Updating dayhoff-tools lock...")
+
+        # Step 1: Update lock file for dayhoff-tools
+        lock_cmd = ["uv", "lock", "--upgrade-package", "dayhoff-tools"]
+        print(f"Running command: {BLUE}{' '.join(lock_cmd)}{RESET}")
+        subprocess.run(lock_cmd, check=True)
+
+        # Step 2: Parse uv.lock to find the new version
+        print(f"Reading {lock_file_path} to find new version...")
+        if not lock_file_path.exists():
+            print(f"Error: {lock_file_path} not found after lock command.")
+            return
+
+        locked_version = None
+        try:
+            lock_data = toml.load(lock_file_path)
+            # Find dayhoff-tools in the lock file packages
+            for package in lock_data.get("package", []):
+                if package.get("name") == "dayhoff-tools":
+                    locked_version = package.get("version")
+                    break
+        except toml.TomlDecodeError as e:
+            print(f"Error parsing {lock_file_path}: {e}")
+            return
+        except Exception as e:
+            print(f"Error reading lock file: {e}")
+            return
+
+        if not locked_version:
+            print(f"Error: Could not find dayhoff-tools version in {lock_file_path}.")
+            return
+
+        print(f"Found dayhoff-tools version {locked_version} in lock file.")
+
+        # Step 3: Update pyproject.toml
+        print(f"Updating {pyproject_path} version constraint...")
+        try:
+            content = pyproject_path.read_text()
+            # Regex to find the dayhoff-tools dependency line, capturing name+extra
+            # Matches 'dayhoff-tools' or \"dayhoff-tools\", optional [extra], >= constraints etc.
+            pattern = re.compile(
+                r"(^[\'\"](dayhoff-tools(?:\[[^\\\]]+\\])?)[\'\"]\s*)[\\>\\<\\=\\~\\^\\(\\)\\s,0-9\\.\\*\'\"]+",
+                re.MULTILINE,
+            )
+            new_constraint = f">= {locked_version}"
+            new_content, num_replacements = pattern.subn(
+                rf"\1{new_constraint}", content
+            )
+
+            if num_replacements > 0:
+                pyproject_path.write_text(new_content)
+                print(
+                    f"Updated dayhoff-tools constraint in {pyproject_path} to '{new_constraint}'"
+                )
+            else:
+                print(
+                    f"Warning: Could not find dayhoff-tools dependency line in {pyproject_path} to update constraint."
+                )
+
+        except FileNotFoundError:
+            print(f"Error: {pyproject_path} not found.")
+            return
+        except Exception as e:
+            print(f"Error updating {pyproject_path}: {e}")
+            # Continue to sync step even if pyproject update fails?
+            # For now, let's proceed to sync.
+            print("Proceeding with sync despite pyproject.toml update error.")
+
+        # Step 4: Sync environment based on updated lock file
+        print("Syncing environment...")
+        sync_cmd = ["uv", "sync", "--no-install-project"]
+        print(f"Running command: {BLUE}{' '.join(sync_cmd)}{RESET}")
+        subprocess.run(sync_cmd, check=True)
+
+        print(
+            "dayhoff-tools updated, pyproject.toml modified, and environment synced successfully."
+        )
+
     except subprocess.CalledProcessError as e:
-        print(f"Error occurred while attempting update: {e}")
+        print(f"Error occurred during update/sync process: {e}")
     except FileNotFoundError:
+        # This catches if uv command itself is not found
         print("Error: 'uv' command not found. Is uv installed and in PATH?")
         sys.exit(1)
+    except Exception as e:
+        # Catch any other unexpected errors
+        print(f"An unexpected error occurred: {e}")
 
 
 def sync_dependencies(
@@ -394,16 +483,22 @@ def sync_dependencies(
     the local project itself (--no-install-project). Use --install-project to
     include the local project.
     """
+    # ANSI color codes
+    BLUE = "\033[94m"
+    RESET = "\033[0m"
+
     try:
         lock_cmd = ["uv", "lock"]
-        print(f"Running command: {' '.join(lock_cmd)}")
+        # Print command in blue
+        print(f"Running command: {BLUE}{' '.join(lock_cmd)}{RESET}")
         subprocess.run(lock_cmd, check=True)
 
         sync_cmd = ["uv", "sync", "--all-groups"]
         if not install_project:
             sync_cmd.append("--no-install-project")
 
-        print(f"Running command: {' '.join(sync_cmd)}")
+        # Print command in blue
+        print(f"Running command: {BLUE}{' '.join(sync_cmd)}{RESET}")
         subprocess.run(sync_cmd, check=True)
 
         print("Dependencies synced successfully.")

{dayhoff_tools-1.0.18.dist-info → dayhoff_tools-1.0.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.0.18
+Version: 1.0.19
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com
@@ -10,21 +10,26 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Provides-Extra: base
+Provides-Extra: lite
 Requires-Dist: biopython (>=1.84)
 Requires-Dist: boto3 (>=1.36.8)
 Requires-Dist: docker (>=7.1.0)
 Requires-Dist: fair-esm (>=2.0.0)
-Requires-Dist: firebase-admin (>=6.5.0) ; extra == "base"
+Requires-Dist: firebase-admin (>=6.5.0)
+Requires-Dist: firebase-admin (>=6.5.0) ; extra == "lite"
 Requires-Dist: h5py (>=3.11.0)
 Requires-Dist: pandas (>=2.2.3)
-Requires-Dist: pyyaml (>=6.0) ; extra == "base"
+Requires-Dist: pyyaml (>=6.0)
+Requires-Dist: pyyaml (>=6.0) ; extra == "lite"
 Requires-Dist: questionary (>=2.0.1)
 Requires-Dist: rdkit-pypi (>=2022.9.5)
-Requires-Dist: requests (>=2.31.0) ; extra == "base"
+Requires-Dist: requests (>=2.31.0)
+Requires-Dist: requests (>=2.31.0) ; extra == "lite"
 Requires-Dist: sqlalchemy (>=2.0.40,<3.0.0)
+Requires-Dist: toml (>=0.10)
 Requires-Dist: transformers (==4.36.2)
-Requires-Dist: typer (>=0.9.0) ; extra == "base"
+Requires-Dist: typer (>=0.9.0)
+Requires-Dist: typer (>=0.9.0) ; extra == "lite"
 Description-Content-Type: text/markdown
 
 # dayhoff-tools

{dayhoff_tools-1.0.18.dist-info → dayhoff_tools-1.0.19.dist-info}/RECORD

@@ -5,7 +5,7 @@ dayhoff_tools/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 dayhoff_tools/cli/cloud_commands.py,sha256=KiYEuD3nSg8QPWBYfrhdze2La_CJe4iqK-8uOAHyS8U,35827
 dayhoff_tools/cli/main.py,sha256=yorL117hknzhNybt7CTz4oKXVuQRIDXPl7huIEWga-M,3666
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
-dayhoff_tools/cli/utility_commands.py,sha256=217qbmt4qVoaPwStvwONoAj9NP9k9UkExs5WyKNCQt8,15586
+dayhoff_tools/cli/utility_commands.py,sha256=rbcrTsvdKivBEuVDhOFz3SoPJeLx9pCSj3XUmc6Q0lk,19044
 dayhoff_tools/deployment/base.py,sha256=u-AjbtHnFLoLt33dhYXHIpV-6jcieMEHHGBGN_U9Hm0,15626
 dayhoff_tools/deployment/deploy_aws.py,sha256=O0gQxHioSU_sNU8T8MD4wSOPvWc--V8eRRZzlRu035I,16446
 dayhoff_tools/deployment/deploy_gcp.py,sha256=DxBM4sUzwPK9RWLP9bSfr38n1HHl-TVrp4TsbdN8pUA,5795
@@ -25,7 +25,7 @@ dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/structure.py,sha256=ufN3gAodQxhnt7psK1VTQeu9rKERmo_PhoxIbB4QKMw,27660
 dayhoff_tools/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJqE4,16456
 dayhoff_tools/warehouse.py,sha256=TqV8nex1AluNaL4JuXH5zuu9P7qmE89lSo6f_oViy6U,14965
-dayhoff_tools-1.0.18.dist-info/METADATA,sha256=lbn5KAcT-XrkbrPfcPL6mQQrK9PoP0BUyV2Kjv4SIDc,2049
-dayhoff_tools-1.0.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.0.18.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.0.18.dist-info/RECORD,,
+dayhoff_tools-1.0.19.dist-info/METADATA,sha256=okG1WGmODVeiTDHJtFIiuoJCMT1aCdnKZMiK2yLAB9o,2214
+dayhoff_tools-1.0.19.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.0.19.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.0.19.dist-info/RECORD,,