pgmini-migrate 0.1.1__tar.gz → 0.1.2__tar.gz

pgmini_migrate-0.1.2/PKG-INFO

@@ -0,0 +1,40 @@
+Metadata-Version: 2.4
+Name: pgmini-migrate
+Version: 0.1.2
+Summary: A minimal PostgreSQL migration tool for Python
+Author-email: Nguyen Vu Duy Luan <nvdluan@gmail.com>
+License: Copyright (c) 2025 Nguyen Vu Duy Luan
+Project-URL: Homepage, https://github.com/NVDLuan/pgmini-migrate
+Project-URL: Issues, https://github.com/NVDLuan/pgmini-migrate/issues
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: psycopg[binary]>=3.1.8
+Requires-Dist: dotenv<0.10.0,>=0.9.9
+Requires-Dist: click<9.0.0,>=8.2.1
+Dynamic: license-file
+
+# pgmini-migrate
+
+`pgmini-migrate` is a **lightweight SQL migration tool** for PostgreSQL.  
+It allows you to easily **create migration files** and run **upgrades/downgrades** using simple CLI commands.  
+
+No ORM required — just pure SQL.
+
+---
+
+## ✨ Features
+- Auto-generate migration files (`upgrade` + `downgrade`).
+- Run **upgrade** or **downgrade** up to a specific version.
+- Two connection options:
+  - via `DATABASE_URL`
+  - via `host`, `port`, `user`, `password`, `dbname`.
+- Configurable migrations directory.
+- Minimal, simple, and PostgreSQL-focused.
+
+---
+
+## 📦 Installation
+
+```bash
+pip install pgmini-migrate

pgmini_migrate-0.1.2/README.md

@@ -0,0 +1,24 @@
+# pgmini-migrate
+
+`pgmini-migrate` is a **lightweight SQL migration tool** for PostgreSQL.  
+It allows you to easily **create migration files** and run **upgrades/downgrades** using simple CLI commands.  
+
+No ORM required — just pure SQL.
+
+---
+
+## ✨ Features
+- Auto-generate migration files (`upgrade` + `downgrade`).
+- Run **upgrade** or **downgrade** up to a specific version.
+- Two connection options:
+  - via `DATABASE_URL`
+  - via `host`, `port`, `user`, `password`, `dbname`.
+- Configurable migrations directory.
+- Minimal, simple, and PostgreSQL-focused.
+
+---
+
+## 📦 Installation
+
+```bash
+pip install pgmini-migrate

pgmini_migrate-0.1.2/pgmini_migrate/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

pgmini_migrate-0.1.2/pgmini_migrate/__main__.py

@@ -0,0 +1,4 @@
+from .cli import main
+
+if __name__ == "__main__":
+    main()

pgmini_migrate-0.1.2/pgmini_migrate/cli.py

@@ -0,0 +1,340 @@
+import os
+import sys
+import psycopg
+from contextlib import contextmanager
+from datetime import datetime
+import argparse
+
+# Default layout
+DEFAULT_MIGRATIONS_DIR = "migrations"
+UPGRADE_SUBDIR = "upgrade"
+DOWNGRADE_SUBDIR = "downgrade"
+MIGRATION_TABLE = "schema_migrations"
+
+
+def load_env_file(env_path=".env"):
+    """Load simple env file KEY=VALUE into os.environ (if not already set)."""
+    if not os.path.exists(env_path):
+        return
+    with open(env_path, "r") as f:
+        for ln in f:
+            ln = ln.strip()
+            if not ln or ln.startswith("#"):
+                continue
+            if "=" not in ln:
+                continue
+            k, v = ln.split("=", 1)
+            k = k.strip()
+            v = v.strip().strip('"').strip("'")
+            # do not overwrite existing env set by host
+            if k not in os.environ:
+                os.environ[k] = v
+
+
+def build_conninfo_from_args(args):
+    """Return a psycopg-compatible connection string or dict kwargs"""
+    if args.database_url:
+        return args.database_url
+    # build a dict for psycopg.connect(...)
+    kw = {}
+    if args.db_host:
+        kw["host"] = args.db_host
+    if args.db_port:
+        kw["port"] = args.db_port
+    if args.db_name:
+        kw["dbname"] = args.db_name
+    if args.db_user:
+        kw["user"] = args.db_user
+    if args.db_password:
+        kw["password"] = args.db_password
+    return kw
+
+
+@contextmanager
+def get_connection(conninfo):
+    """Context manager returning a connected psycopg connection (sync)."""
+    if isinstance(conninfo, str):
+        conn = psycopg.connect(conninfo)
+    else:
+        conn = psycopg.connect(**conninfo)
+    try:
+        yield conn
+    finally:
+        conn.close()
+
+
+def ensure_migration_table(conn):
+    with conn.cursor() as cur:
+        cur.execute(
+            f"""
+            CREATE TABLE IF NOT EXISTS {MIGRATION_TABLE} (
+                version VARCHAR(255) PRIMARY KEY,
+                applied_at TIMESTAMP DEFAULT NOW()
+            )
+            """
+        )
+    conn.commit()
+
+
+def get_applied_migrations(conn):
+    with conn.cursor() as cur:
+        cur.execute(f"SELECT version FROM {MIGRATION_TABLE} ORDER BY version")
+        rows = cur.fetchall()
+        return [row[0] for row in rows]
+
+
+def apply_migration(conn, version, sql_path):
+    with open(sql_path, "r", encoding="utf-8") as f:
+        sql = f.read()
+    try:
+        with conn.cursor() as cur:
+            print(f"⬆️  Applying {version} -> {os.path.basename(sql_path)}")
+            cur.execute(sql)
+            cur.execute(
+                f"INSERT INTO {MIGRATION_TABLE} (version) VALUES (%s)", (version,)
+            )
+        conn.commit()
+        print(f"✅ {version} applied.")
+    except Exception as e:
+        conn.rollback()
+        print(f"❌ Failed to apply {version}: {e}")
+        raise
+
+
+def rollback_migration_from_file(conn, version, sql_path):
+    with open(sql_path, "r", encoding="utf-8") as f:
+        sql = f.read()
+    try:
+        with conn.cursor() as cur:
+            print(f"⬇️  Rolling back {version} -> {os.path.basename(sql_path)}")
+            cur.execute(sql)
+            cur.execute(f"DELETE FROM {MIGRATION_TABLE} WHERE version = %s", (version,))
+        conn.commit()
+        print(f"✅ {version} rolled back.")
+    except Exception as e:
+        conn.rollback()
+        print(f"❌ Failed to rollback {version}: {e}")
+        raise
+
+
+def find_migration_files(migrations_root):
+    """Return sorted list of upgrade filenames in migrations/upgrade"""
+    upgrade_dir = os.path.join(migrations_root, UPGRADE_SUBDIR)
+    if not os.path.isdir(upgrade_dir):
+        return []
+    files = [f for f in os.listdir(upgrade_dir) if f.endswith(".sql")]
+    files = sorted(files)
+    return files
+
+
+def parse_version_from_filename(filename):
+    """Expect filenames like 001.description.upgrade.sql or 001.description.sql"""
+    base = os.path.basename(filename)
+    parts = base.split(".", 1)
+    return parts[0]
+
+
+def full_upgrade_path(migrations_root, filename):
+    return os.path.join(migrations_root, UPGRADE_SUBDIR, filename)
+
+
+def full_downgrade_path(migrations_root, filename):
+    return os.path.join(migrations_root, DOWNGRADE_SUBDIR, filename)
+
+
+def get_next_version_str(migrations_root):
+    upgrade_dir = os.path.join(migrations_root, UPGRADE_SUBDIR)
+    os.makedirs(upgrade_dir, exist_ok=True)
+    files = [f for f in os.listdir(upgrade_dir) if f[0:3].isdigit()]
+    versions = [int(f[0:3]) for f in files if f[0:3].isdigit()]
+    next_v = max(versions) + 1 if versions else 1
+    return f"{next_v:03d}"
+
+
+def create_migration(migrations_root, description):
+    desc = description.strip()
+    # sanitize description
+    desc_clean = "_".join(
+        c for c in desc.lower().replace(" ", "_") if (c.isalnum() or c == "_")
+    )
+    version = get_next_version_str(migrations_root)
+    upgrade_name = f"{version}.{desc_clean}.upgrade.sql"
+    downgrade_name = f"{version}.{desc_clean}.downgrade.sql"
+
+    upgrade_path = full_upgrade_path(migrations_root, upgrade_name)
+    downgrade_path = full_downgrade_path(migrations_root, downgrade_name)
+
+    # ensure dirs
+    os.makedirs(os.path.dirname(upgrade_path), exist_ok=True)
+    os.makedirs(os.path.dirname(downgrade_path), exist_ok=True)
+
+    ts = datetime.utcnow().isoformat() + "Z"
+    with open(upgrade_path, "w", encoding="utf-8") as f:
+        f.write(f"-- +upgrade {version} {desc}\n-- created at {ts}\n\n")
+    with open(downgrade_path, "w", encoding="utf-8") as f:
+        f.write(f"-- +downgrade {version} {desc}\n-- created at {ts}\n\n")
+
+    print(f"Created:\n  {upgrade_path}\n  {downgrade_path}")
+    return upgrade_path, downgrade_path
+
+
+def cmd_upgrade(args):
+    conninfo = build_conninfo_from_args(args)
+    migrations_root = args.migrations
+    files = find_migration_files(migrations_root)
+    if not files:
+        print("No migration files found.")
+        return
+
+    with get_connection(conninfo) as conn:
+        ensure_migration_table(conn)
+        applied = set(get_applied_migrations(conn))
+        for fname in files:
+            version = parse_version_from_filename(fname)
+            if version in applied:
+                print(f"Skipping {fname} (already applied).")
+                continue
+            sql_path = full_upgrade_path(migrations_root, fname)
+            apply_migration(conn, version, sql_path)
+
+
+def cmd_history(args):
+    conninfo = build_conninfo_from_args(args)
+    with get_connection(conninfo) as conn:
+        ensure_migration_table(conn)
+        applied = get_applied_migrations(conn)
+        print("Applied migrations (ordered):")
+        for v in applied:
+            print("  -", v)
+
+
+def cmd_show(args):
+    conninfo = build_conninfo_from_args(args)
+    with get_connection(conninfo) as conn:
+        ensure_migration_table(conn)
+        applied = get_applied_migrations(conn)
+        if not applied:
+            print("Database has no applied migrations (base).")
+        else:
+            print("Current version:", applied[-1])
+
+
+def cmd_downgrade(args):
+    conninfo = build_conninfo_from_args(args)
+    migrations_root = args.migrations
+
+    with get_connection(conninfo) as conn:
+        ensure_migration_table(conn)
+        applied = get_applied_migrations(conn)
+
+        if not applied:
+            print("No migrations applied.")
+            return
+
+        target = args.target  # None => step=1, "base" => all, specific version => to that
+        if target is None:
+            # default: rollback 1 step
+            target_index = len(applied) - 2
+        elif target == "base":
+            target_index = -1
+        else:
+            if target not in applied:
+                print(f"Target {target} not found in applied migrations.")
+                return
+            target_index = applied.index(target)
+
+        to_rollback = applied[target_index + 1 :]
+        if not to_rollback:
+            print("Nothing to rollback.")
+            return
+
+        # rollback in reverse order
+        for version in reversed(to_rollback):
+            # find corresponding file in downgrade dir
+            # find filename that starts with version in upgrade dir to get remainder name
+            upgrade_dir = os.path.join(migrations_root, UPGRADE_SUBDIR)
+            candidates = [f for f in os.listdir(upgrade_dir) if f.startswith(version)]
+            if not candidates:
+                print(f"⚠️  No upgrade file found for version {version} (can't map to downgrade).")
+                continue
+            # infer name: replace 'upgrade' -> 'downgrade'
+            up_fname = candidates[0]
+            down_fname = up_fname.replace(".upgrade.", ".downgrade.")
+            down_path = full_downgrade_path(migrations_root, down_fname)
+            if not os.path.exists(down_path):
+                print(f"⚠️  No rollback file for {version} -> expected {down_path}")
+                continue
+            rollback_migration_from_file(conn, version, down_path)
+
+
+def build_arg_parser():
+    p = argparse.ArgumentParser(prog="pgmigrate", description="Simple PostgreSQL migration tool")
+    p.add_argument("--env-file", default=".env", help="Path to .env file (default: .env)")
+    p.add_argument(
+        "--migrations",
+        default=DEFAULT_MIGRATIONS_DIR,
+        help=f"Root migrations dir (default: {DEFAULT_MIGRATIONS_DIR})",
+    )
+
+    # DB connection options (mutually override database_url)
+    p.add_argument("--database-url", help="Full DATABASE_URL (postgresql://...)", default=None)
+    p.add_argument("--db-host", help="DB host", default=os.getenv("POSTGRES_HOST"))
+    p.add_argument("--db-port", help="DB port", default=os.getenv("POSTGRES_PORT"))
+    p.add_argument("--db-name", help="DB name", default=os.getenv("POSTGRES_DB"))
+    p.add_argument("--db-user", help="DB user", default=os.getenv("POSTGRES_USER"))
+    p.add_argument("--db-password", help="DB password", default=os.getenv("POSTGRES_PASSWORD"))
+
+    sub = p.add_subparsers(dest="cmd", required=True)
+
+    sub.add_parser("upgrade", help="Apply all pending migrations")
+
+    down_p = sub.add_parser("downgrade", help="Rollback migrations to target version (or 1 step if no target)")
+    down_p.add_argument("target", nargs="?", help="target version (e.g. 002) or 'base'")
+
+    sub.add_parser("history", help="Show applied migrations (history)")
+
+    sub.add_parser("show", help="Show current applied version")
+
+    create_p = sub.add_parser("create", help="Create new migration skeleton")
+    create_p.add_argument("description", nargs="+", help="Migration description (quoted)")
+
+    return p
+
+
+def main(argv=None):
+    argv = argv if argv is not None else sys.argv[1:]
+    parser = build_arg_parser()
+    args = parser.parse_args(argv)
+
+    # Load env file first (do not overwrite existing env vars)
+    if args.env_file:
+        load_env_file(args.env_file)
+
+    # prefer CLI args, else env
+    # Re-populate db fields from env if not provided explicitly
+    # (build_conninfo handles database_url precedence)
+    # Dispatch commands:
+    if args.cmd == "create":
+        desc = " ".join(args.description)
+        create_migration(args.migrations, desc)
+        return
+
+    # For commands that touch DB, ensure connection info exists
+    # Build conninfo now (will use args or environment)
+    # If neither database_url nor db_name provided, abort
+    conninfo = build_conninfo_from_args(args)
+    if not conninfo:
+        print("No database connection info provided. Provide --database-url or host/name/user/password or set env vars.")
+        sys.exit(2)
+
+    if args.cmd == "upgrade":
+        cmd_upgrade(args)
+    elif args.cmd == "downgrade":
+        cmd_downgrade(args)
+    elif args.cmd == "history":
+        cmd_history(args)
+    elif args.cmd == "show":
+        cmd_show(args)
+    else:
+        print("Unknown command:", args.cmd)
+        sys.exit(2)

pgmini_migrate-0.1.2/pgmini_migrate.egg-info/PKG-INFO

@@ -0,0 +1,40 @@
+Metadata-Version: 2.4
+Name: pgmini-migrate
+Version: 0.1.2
+Summary: A minimal PostgreSQL migration tool for Python
+Author-email: Nguyen Vu Duy Luan <nvdluan@gmail.com>
+License: Copyright (c) 2025 Nguyen Vu Duy Luan
+Project-URL: Homepage, https://github.com/NVDLuan/pgmini-migrate
+Project-URL: Issues, https://github.com/NVDLuan/pgmini-migrate/issues
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: psycopg[binary]>=3.1.8
+Requires-Dist: dotenv<0.10.0,>=0.9.9
+Requires-Dist: click<9.0.0,>=8.2.1
+Dynamic: license-file
+
+# pgmini-migrate
+
+`pgmini-migrate` is a **lightweight SQL migration tool** for PostgreSQL.  
+It allows you to easily **create migration files** and run **upgrades/downgrades** using simple CLI commands.  
+
+No ORM required — just pure SQL.
+
+---
+
+## ✨ Features
+- Auto-generate migration files (`upgrade` + `downgrade`).
+- Run **upgrade** or **downgrade** up to a specific version.
+- Two connection options:
+  - via `DATABASE_URL`
+  - via `host`, `port`, `user`, `password`, `dbname`.
+- Configurable migrations directory.
+- Minimal, simple, and PostgreSQL-focused.
+
+---
+
+## 📦 Installation
+
+```bash
+pip install pgmini-migrate

{pgmini_migrate-0.1.1 → pgmini_migrate-0.1.2}/pgmini_migrate.egg-info/SOURCES.txt

@@ -3,9 +3,9 @@ README.md
 pyproject.toml
 setup.py
 pgmini_migrate/__init__.py
+pgmini_migrate/__main__.py
 pgmini_migrate/cli.py
-pgmini_migrate/generator.py
-pgmini_migrate/runner.py
+pgmini_migrate/utils.py
 pgmini_migrate.egg-info/PKG-INFO
 pgmini_migrate.egg-info/SOURCES.txt
 pgmini_migrate.egg-info/dependency_links.txt

{pgmini_migrate-0.1.1 → pgmini_migrate-0.1.2}/pgmini_migrate.egg-info/requires.txt

@@ -1,2 +1,3 @@
 psycopg[binary]>=3.1.8
 dotenv<0.10.0,>=0.9.9
+click<9.0.0,>=8.2.1

{pgmini_migrate-0.1.1 → pgmini_migrate-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pgmini-migrate"
-version = "0.1.1"
+version = "0.1.2"
 description = "A minimal PostgreSQL migration tool for Python"
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -13,9 +13,10 @@ authors = [
 ]
 dependencies = [
     "psycopg[binary]>=3.1.8",
-    "dotenv (>=0.9.9,<0.10.0)"
+    "dotenv (>=0.9.9,<0.10.0)",
+    "click (>=8.2.1,<9.0.0)"
 ]
-requires-python = ">=3.8"
+requires-python = ">=3.10"
 
 [project.scripts]
 pgmigrate = "pgmini_migrate.cli:main"

{pgmini_migrate-0.1.1 → pgmini_migrate-0.1.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pgmini-migrate",
-    version="0.1.0",
+    version="0.1.2",
     packages=find_packages(),
     install_requires=["psycopg2-binary", 'psycopg', 'python-dotenv'],
     entry_points={

pgmini_migrate-0.1.1/PKG-INFO

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: pgmini-migrate
-Version: 0.1.1
-Summary: A minimal PostgreSQL migration tool for Python
-Author-email: Nguyen Vu Duy Luan <nvdluan@gmail.com>
-License: Copyright (c) 2025 Nguyen Vu Duy Luan
-Project-URL: Homepage, https://github.com/NVDLuan/pgmini-migrate
-Project-URL: Issues, https://github.com/NVDLuan/pgmini-migrate/issues
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: psycopg[binary]>=3.1.8
-Requires-Dist: dotenv<0.10.0,>=0.9.9
-Dynamic: license-file
-
-# pgmini-migrate
-
-A minimal PostgreSQL migration tool using psycopg3.
-
-## Install
-```bash
-pip install pgmini-migrate

pgmini_migrate-0.1.1/README.md

@@ -1,7 +0,0 @@
-# pgmini-migrate
-
-A minimal PostgreSQL migration tool using psycopg3.
-
-## Install
-```bash
-pip install pgmini-migrate

pgmini_migrate-0.1.1/pgmini_migrate/cli.py

@@ -1,30 +0,0 @@
-import argparse
-from pgmini_migrate.generator import create_migration
-from pgmini_migrate.runner import upgrade, downgrade
-
-def main():
-    parser = argparse.ArgumentParser(prog="pgmigrate", description="Mini migration tool for PostgreSQL")
-    sub = parser.add_subparsers(dest="command")
-
-    # create
-    create_cmd = sub.add_parser("create", help="Create new migration")
-    create_cmd.add_argument("description", nargs="+", help="Migration description")
-
-    # upgrade
-    upgrade_cmd = sub.add_parser("upgrade", help="Apply migrations")
-    upgrade_cmd.add_argument("target", nargs="?", help="Target version (default: latest)")
-
-    # downgrade
-    downgrade_cmd = sub.add_parser("downgrade", help="Rollback migrations")
-    downgrade_cmd.add_argument("target", nargs="?", help="Target version")
-
-    args = parser.parse_args()
-
-    if args.command == "create":
-        create_migration(" ".join(args.description))
-    elif args.command == "upgrade":
-        upgrade(args.target)
-    elif args.command == "downgrade":
-        downgrade(args.target)
-    else:
-        parser.print_help()

pgmini_migrate-0.1.1/pgmini_migrate/generator.py

@@ -1,37 +0,0 @@
-import os
-
-
-MIGRATIONS_DIR = os.getenv("MIGRATIONS_DIR", "migrations")
-UPGRADE_DIR = os.path.join(MIGRATIONS_DIR, "upgrade")
-DOWNGRADE_DIR = os.path.join(MIGRATIONS_DIR, "downgrade")
-
-
-def ensure_dirs():
-    os.makedirs(UPGRADE_DIR, exist_ok=True)
-    os.makedirs(DOWNGRADE_DIR, exist_ok=True)
-
-def get_next_version():
-    ensure_dirs()
-    files = os.listdir(UPGRADE_DIR)
-    versions = []
-    for f in files:
-        if f[0:3].isdigit():
-            versions.append(int(f[0:3]))
-    return max(versions) + 1 if versions else 1
-
-
-def create_migration(description: str):
-    version = get_next_version()
-    version_str = f"{version:03d}"
-    # format description thành tên file đẹp
-    desc_clean = description.lower().replace(" ", "_")
-
-    upgrade_file = os.path.join(UPGRADE_DIR, f"{version_str}.{desc_clean}.upgrade.sql")
-    downgrade_file = os.path.join(DOWNGRADE_DIR, f"{version_str}.{desc_clean}.downgrade.sql")
-
-    with open(upgrade_file, "w") as f:
-        f.write(f"-- Migration {version_str}: {description}\n")
-    with open(downgrade_file, "w") as f:
-        f.write(f"-- Rollback {version_str}: {description}\n")
-
-    print(f"✅ Created migration files:\n  {upgrade_file}\n  {downgrade_file}")

pgmini_migrate-0.1.1/pgmini_migrate/runner.py

@@ -1,156 +0,0 @@
-import os
-from contextlib import contextmanager
-
-import psycopg
-from dotenv import load_dotenv
-
-load_dotenv('.env')
-
-DB_CONFIG = {
-    "dbname": os.getenv("POSTGRES_DB", "postgres"),
-    "user": os.getenv("POSTGRES_USER", "postgres"),
-    "password": os.getenv("POSTGRES_PASSWORD", "postgres"),
-    "host": os.getenv("POSTGRES_HOST", "localhost"),
-    "port": os.getenv("POSTGRES_PORT", "5432"),
-}
-
-MIGRATIONS_DIR = os.getenv("MIGRATIONS_DIR", "migrations")
-UPGRADE_DIR = os.path.join(MIGRATIONS_DIR, "upgrade")
-DOWNGRADE_DIR = os.path.join(MIGRATIONS_DIR, "downgrade")
-
-
-@contextmanager
-def get_connection():
-    conn = psycopg.connect(**DB_CONFIG)
-    try:
-        yield conn
-    finally:
-        conn.close()
-
-
-def ensure_migration_table(conn):
-    with conn.cursor() as cur:
-        cur.execute("""
-                    CREATE TABLE IF NOT EXISTS schema_migrations
-                    (
-                        version
-                        VARCHAR
-                    (
-                        255
-                    ) PRIMARY KEY,
-                        applied_at TIMESTAMP DEFAULT NOW
-                    (
-                    )
-                        )
-                    """)
-    conn.commit()
-
-
-def get_applied_migrations(conn):
-    with conn.cursor() as cur:
-        cur.execute("SELECT version FROM schema_migrations ORDER BY version")
-        rows = cur.fetchall()
-        return [row[0] for row in rows]
-
-
-def apply_migration(conn, version, sql_path):
-    with open(sql_path, "r") as f:
-        sql = f.read()
-
-    try:
-        with conn.cursor() as cur:
-            print(f"⬆️  Applying migration {version} ...")
-            cur.execute(sql)
-            cur.execute(
-                "INSERT INTO schema_migrations (version) VALUES (%s)", (version,)
-            )
-        conn.commit()
-        print(f"✅ Migration {version} applied.")
-    except Exception as e:
-        conn.rollback()
-        print(f"❌ Migration {version} failed: {e}")
-        raise
-
-
-def rollback_migration(conn, version, sql_file):
-    down_file = f"{version}.{sql_file.split('.', 1)[1].replace('upgrade', 'downgrade')}"
-    sql_path = os.path.join(DOWNGRADE_DIR, down_file)
-
-    if not os.path.exists(sql_path):
-        print(f"⚠️  No rollback file for {version}")
-        return
-
-    with open(sql_path, "r") as f:
-        sql = f.read()
-
-    try:
-        with conn.cursor() as cur:
-            print(f"⬇️  Rolling back migration {version} ...")
-            cur.execute(sql)
-            cur.execute("DELETE FROM schema_migrations WHERE version = %s", (version,))
-        conn.commit()
-        print(f"✅ Migration {version} rolled back.")
-    except Exception as e:
-        conn.rollback()
-        print(f"❌ Rollback {version} failed: {e}")
-        raise
-
-
-def upgrade():
-    with get_connection() as conn:
-        ensure_migration_table(conn)
-        applied = set(get_applied_migrations(conn))
-
-        migration_files = sorted(os.listdir(UPGRADE_DIR))
-        for filename in migration_files:
-            version = filename.split(".")[0]  # 001_init.sql -> 001
-            if version in applied or not filename.endswith(".sql"):
-                continue
-            sql_path = os.path.join(UPGRADE_DIR, filename)
-            apply_migration(conn, version, sql_path)
-
-
-def downgrade(target_version=None):
-    with get_connection() as conn:
-        ensure_migration_table(conn)
-        applied = get_applied_migrations(conn)
-
-        if not applied:
-            print("⚠️  No migrations to rollback.")
-            return
-
-        if target_version == "base":
-            target_index = -1
-        elif target_version:
-            if target_version not in applied:
-                print(f"⚠️  Target version {target_version} not found in applied migrations.")
-                return
-            target_index = applied.index(target_version)
-        else:
-            # Nếu không truyền version → rollback 1 bước
-            target_index = len(applied) - 2
-
-        to_rollback = applied[target_index + 1:]  # những migration mới hơn target
-        if not to_rollback:
-            print("⚠️  Nothing to rollback.")
-            return
-
-        # rollback ngược thứ tự
-        for version in reversed(to_rollback):
-            sql_file = next(
-                (f for f in os.listdir(UPGRADE_DIR) if f.startswith(version)), None
-            )
-            if not sql_file:
-                print(f"⚠️  No migration file found for {version}")
-                continue
-            sql_path = os.path.join(UPGRADE_DIR, sql_file)
-            rollback_migration(conn, version, sql_path)
-
-
-def history():
-    with get_connection() as conn:
-        ensure_migration_table(conn)
-        applied = get_applied_migrations(conn)
-        print("📜 Applied migrations:")
-        for v in applied:
-            print(f"  - {v}")

pgmini_migrate-0.1.1/pgmini_migrate.egg-info/PKG-INFO

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: pgmini-migrate
-Version: 0.1.1
-Summary: A minimal PostgreSQL migration tool for Python
-Author-email: Nguyen Vu Duy Luan <nvdluan@gmail.com>
-License: Copyright (c) 2025 Nguyen Vu Duy Luan
-Project-URL: Homepage, https://github.com/NVDLuan/pgmini-migrate
-Project-URL: Issues, https://github.com/NVDLuan/pgmini-migrate/issues
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: psycopg[binary]>=3.1.8
-Requires-Dist: dotenv<0.10.0,>=0.9.9
-Dynamic: license-file
-
-# pgmini-migrate
-
-A minimal PostgreSQL migration tool using psycopg3.
-
-## Install
-```bash
-pip install pgmini-migrate