jetbase 0.7.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

jetbase/core/upgrade.py

@@ -1,75 +0,0 @@
-import os
-
-from jetbase.core.dry_run import process_dry_run
-from jetbase.core.file_parser import parse_upgrade_statements
-from jetbase.core.repository import (
-    create_migrations_table_if_not_exists,
-    get_last_updated_version,
-    run_migration,
-)
-from jetbase.core.version import get_migration_filepaths_by_version
-from jetbase.enums import MigrationOperationType
-
-
-def upgrade_cmd(
-    count: int | None = None, to_version: str | None = None, dry_run: bool = False
-) -> None:
-    """
-    Run database migrations by applying all pending SQL migration files.
-    Executes migration files in order starting from the last applied version,
-    updating the migrations tracking table after each successful migration.
-
-    Returns:
-        None
-    """
-
-    if count is not None and to_version is not None:
-        raise ValueError(
-            "Cannot specify both 'count' and 'to_version' for upgrade. "
-            "Select only one, or do not specify either to run all pending migrations."
-        )
-
-    create_migrations_table_if_not_exists()
-    latest_version: str | None = get_last_updated_version()
-
-    all_versions: dict[str, str] = get_migration_filepaths_by_version(
-        directory=os.path.join(os.getcwd(), "migrations"),
-        version_to_start_from=latest_version,
-    )
-
-    if latest_version:
-        all_versions = dict(list(all_versions.items())[1:])
-
-    if count:
-        all_versions = dict(list(all_versions.items())[:count])
-    elif to_version:
-        if all_versions.get(to_version) is None:
-            raise ValueError(
-                f"The specified to_version '{to_version}' does not exist among pending migrations."
-            )
-        all_versions_list = []
-        for file_version, file_path in all_versions.items():
-            all_versions_list.append((file_version, file_path))
-            if file_version == to_version:
-                break
-        all_versions = dict(all_versions_list)
-
-    if not dry_run:
-        for version, file_path in all_versions.items():
-            sql_statements: list[str] = parse_upgrade_statements(file_path=file_path)
-            filename: str = os.path.basename(file_path)
-
-            run_migration(
-                sql_statements=sql_statements,
-                version=version,
-                migration_operation=MigrationOperationType.UPGRADE,
-                filename=filename,
-            )
-
-            print(f"Migration applied successfully: {filename}")
-
-    else:
-        process_dry_run(
-            version_to_filepath=all_versions,
-            migration_operation=MigrationOperationType.UPGRADE,
-        )

jetbase/core/version.py

@@ -1,163 +0,0 @@
-import os
-
-from jetbase.core.file_parser import is_filename_format_valid, is_filename_length_valid
-
-
-def _get_version_key_from_filename(filename: str) -> str:
-    """
-    Extract and normalize version key from a filename.
-
-    The function extracts the version part from a filename that follows the format:
-    'V{version}__{description}.sql' where version can be like '1', '1_1', or '1.1'.
-
-    Args:
-        filename (str): The filename to extract version from.
-            Must follow pattern like 'V1__description.sql' or 'V1_1__description.sql'
-
-    Returns:
-        str: Normalized version string where underscores are replaced with periods.
-
-    Raises:
-        ValueError: If the filename doesn't follow the expected format.
-
-    Examples:
-        >>> _get_version_key_from_filename("V1__my_description.sql")
-        '1'
-        >>> _get_version_key_from_filename("V1_1__my_description.sql")
-        '1.1'
-        >>> _get_version_key_from_filename("V1.1__my_description.sql")
-        '1.1'
-    """
-    try:
-        version = filename.split("__")[0][1:]
-    except Exception:
-        raise (
-            ValueError(
-                "Filename must be in the following format: V1__my_description.sql, V1_1__my_description.sql, V1.1__my_description.sql"
-            )
-        )
-    return version.replace("_", ".")
-
-
-def _convert_version_tuple_to_str(version_tuple: tuple[str, ...]) -> str:
-    """
-    Convert a version tuple to a string representation.
-
-    Args:
-        version_tuple (tuple[str, ...]): A tuple containing version components as strings.
-
-    Returns:
-        str: A string representation of the version, with components joined by periods.
-
-    Example:
-        >>> _convert_version_tuple_to_str(('1', '2', '3'))
-        '1.2.3'
-    """
-    return ".".join(version_tuple)
-
-
-def convert_version_to_tuple(version: str) -> tuple[str, ...]:
-    """
-    Convert a version string to a tuple of version components.
-
-    Args:
-        version_str (str): A version string with components separated by periods.
-
-    Returns:
-        tuple[str, ...]: A tuple containing the version components as strings.
-
-    Example:
-        >>> convert_version_to_tuple("1.2.3")
-        ('1', '2', '3')
-    """
-    return tuple(version.split("."))
-
-
-def get_migration_filepaths_by_version(
-    directory: str,
-    version_to_start_from: str | None = None,
-    end_version: str | None = None,
-) -> dict[str, str]:
-    """
-    Retrieve migration file paths organized by version number.
-
-    Walks through the specified directory to find SQL migration files and creates
-    a dictionary mapping version strings to their file paths. Files are validated
-    for proper naming format and length. Results can be filtered by version range.
-
-    Args:
-        directory (str): The directory path to search for SQL migration files.
-        version_to_start_from (str | None): Optional minimum version (inclusive).
-            Only files with versions >= this value are included. Defaults to None.
-        end_version (str | None): Optional maximum version (exclusive).
-            Only files with versions < this value are included. Defaults to None.
-
-    Returns:
-        dict[str, str]: Dictionary mapping version strings to file paths,
-            sorted in ascending order by version number.
-
-    Raises:
-        ValueError: If a filename doesn't match the required format or exceeds
-            the maximum length of 512 characters.
-
-    Example:
-        >>> get_migration_filepaths_by_version('/path/to/migrations')
-        {'1.0.0': '/path/to/migrations/V1_0_0__init.sql',
-         '1.2.0': '/path/to/migrations/V1_2_0__add_users.sql'}
-        >>> get_migration_filepaths_by_version('/path/to/migrations', version_to_start_from='1.1.0')
-        {'1.2.0': '/path/to/migrations/V1_2_0__add_users.sql'}
-    """
-    version_to_filepath_dict: dict[str, str] = {}
-
-    for root, _, files in os.walk(directory):
-        for filename in files:
-            if filename.endswith(".sql") and not is_filename_format_valid(
-                filename=filename
-            ):
-                raise ValueError(
-                    f"Invalid migration filename format: {filename}. "
-                    "Filenames must start with 'V', followed by the version number, "
-                    "two underscores '__', a description, and end with '.sql'. "
-                    "V<version_number>__<my_description>.sql. "
-                    "Examples: 'V1_2_0__add_new_table.sql' or 'V1.2.0__add_new_table.sql'"
-                )
-
-            if filename.endswith(".sql") and not is_filename_length_valid(
-                filename=filename
-            ):
-                raise ValueError(
-                    f"Migration filename too long: {filename}. "
-                    f"Filename is currently {len(filename)} characters. "
-                    "Filenames must not exceed 512 characters."
-                )
-
-            if is_filename_format_valid(filename=filename):
-                file_path: str = os.path.join(root, filename)
-                version: str = _get_version_key_from_filename(filename=filename)
-                version_tuple: tuple[str, ...] = convert_version_to_tuple(
-                    version=version
-                )
-
-                if end_version:
-                    if version_tuple > convert_version_to_tuple(version=end_version):
-                        continue
-
-                if version_to_start_from:
-                    if version_tuple >= convert_version_to_tuple(
-                        version=version_to_start_from
-                    ):
-                        version_to_filepath_dict[
-                            _convert_version_tuple_to_str(version_tuple=version_tuple)
-                        ] = file_path
-
-                else:
-                    version_to_filepath_dict[
-                        _convert_version_tuple_to_str(version_tuple=version_tuple)
-                    ] = file_path
-
-    ordered_version_to_filepath_dict: dict[str, str] = {
-        version: version_to_filepath_dict[version]
-        for version in sorted(version_to_filepath_dict.keys())
-    }
-
-    return ordered_version_to_filepath_dict

jetbase/queries.py

@@ -1,72 +0,0 @@
-from sqlalchemy import TextClause, text
-
-LATEST_VERSION_QUERY: TextClause = text("""
-    SELECT 
-        version 
-    FROM 
-        jetbase_migrations
-    ORDER BY 
-        applied_at DESC
-    LIMIT 1
-""")
-
-CREATE_MIGRATIONS_TABLE_STMT: TextClause = text("""
-CREATE TABLE IF NOT EXISTS jetbase_migrations (
-    version VARCHAR(255) PRIMARY KEY,
-    description VARCHAR(500),
-    filename VARCHAR(512),
-    order_executed INT GENERATED ALWAYS AS IDENTITY,
-    applied_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-)
-""")
-
-INSERT_VERSION_STMT: TextClause = text("""
-INSERT INTO jetbase_migrations (version, description, filename) 
-VALUES (:version, :description, :filename)
-""")
-
-DELETE_VERSION_STMT: TextClause = text("""
-DELETE FROM jetbase_migrations 
-WHERE version = :version
-""")
-
-LATEST_VERSIONS_QUERY: TextClause = text("""
-    SELECT 
-        version 
-    FROM 
-        jetbase_migrations
-    ORDER BY 
-        applied_at DESC
-    LIMIT :limit
-""")
-
-LATEST_VERSIONS_BY_STARTING_VERSION_QUERY: TextClause = text("""
-    SELECT
-        version
-    FROM
-        jetbase_migrations
-    WHERE applied_at > 
-        (select applied_at from jetbase_migrations 
-            where version = :starting_version)
-    ORDER BY 
-        applied_at DESC
-""")
-
-CHECK_IF_VERSION_EXISTS_QUERY: TextClause = text("""
-    SELECT 
-        COUNT(*)
-    FROM 
-        jetbase_migrations
-    WHERE 
-        version = :version
-""")
-
-
-CHECK_IF_MIGRATIONS_TABLE_EXISTS_QUERY: TextClause = text("""
-SELECT EXISTS (
-    SELECT 1
-    FROM information_schema.tables
-    WHERE table_schema = 'public'
-      AND table_name = 'jetbase_migrations'
-)
-""")

jetbase-0.7.0.dist-info/METADATA

@@ -1,12 +0,0 @@
-Metadata-Version: 2.4
-Name: jetbase
-Version: 0.7.0
-Summary: Jetbase is a Python database migration tool
-Author-email: jaz <jaz.allibhai@gmail.com>
-License-File: LICENSE
-Requires-Python: >=3.10
-Requires-Dist: sqlalchemy>=2.0.10
-Requires-Dist: typer>=0.12.3
-Description-Content-Type: text/markdown
-
-# jetbase

jetbase-0.7.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-jetbase/config.py,sha256=-YZgS-fVW8k53azx9Ur0GjJvYCZLwOHkguMr_Ujc-mo,1731
-jetbase/constants.py,sha256=X2SYgXRxj77zWB-6N764RQsJmKKUd5Wa7RaQGN6eTiQ,339
-jetbase/enums.py,sha256=LFNZnWcBxvF4kIdZYdGKn2Z65ADHJs38MnGAqeEbZ_U,110
-jetbase/queries.py,sha256=jZ9ei8V-dEb_8ZAwtSAFqLnIEI5rumm3KvQmw4CiZk4,1619
-jetbase/cli/main.py,sha256=I0HIRGC7dmTvEhgdEBksF-p7QPoY4-CfVcm-niVhzck,1504
-jetbase/core/dry_run.py,sha256=zIV4NM_OL-aR7UJlEXYysiGairMOgytnT5rg8baFSms,1469
-jetbase/core/file_parser.py,sha256=Aau7iFaBoZxOgoxB-4C5XAvE_a9142kv7sJHOAANe2A,6069
-jetbase/core/initialize.py,sha256=lm3FWhe1YNKCTmZix465tBGjJsl_YshhRo2ENm7iPFM,955
-jetbase/core/repository.py,sha256=JrmqFkIIbICPsxnY7O3ei9kLL-0mXWFefU0QPHppcgw,5586
-jetbase/core/rollback.py,sha256=e87xlI3WJaOAQD_4QdmeZIDj9a5CW7wKUxQkgIFXy3c,2385
-jetbase/core/upgrade.py,sha256=GowNGAxXRD9dks8FWqSMoQPCMnMh_yP01HNEYB5Py2Q,2603
-jetbase/core/version.py,sha256=mPeJEgcqA0hZMb6YMZQcfc-MGcNI2Ls_7hiH6vJnC1k,6136
-jetbase-0.7.0.dist-info/METADATA,sha256=6vue-I4pBPIQRmMHY3JrxjU7j81fgL6ITMiWcwnmbOs,306
-jetbase-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-jetbase-0.7.0.dist-info/entry_points.txt,sha256=-Pe7A0r8aGF-6C14hB0Ck8aEuszuH5pdadt5OWSiWwA,50
-jetbase-0.7.0.dist-info/licenses/LICENSE,sha256=hyssQNKtnK32aFCk8mVR406ysMtGNP6_dZCig8zBk4Q,1065
-jetbase-0.7.0.dist-info/RECORD,,