pgmini-migrate 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

pgmini_migrate/cli.py

@@ -0,0 +1,30 @@
+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/generator.py

@@ -0,0 +1,37 @@
+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/runner.py

@@ -0,0 +1,156 @@
+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.0.dist-info → pgmini_migrate-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgmini-migrate
-Version: 0.1.0
+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

pgmini_migrate-0.1.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+pgmini_migrate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgmini_migrate/cli.py,sha256=8bfm5BDWiTgiIk-RCySKHhtLoLRd7b12W40MMRbsPH0,1092
+pgmini_migrate/generator.py,sha256=-AO2AQ5rqBz6n1ydutJG4oYNJZGBRuBuklKI4u4QhR4,1215
+pgmini_migrate/runner.py,sha256=4LvJqwrT4p_i_z5LxPmFh6-OJw9xOE2gxGXcLnSSSbI,4859
+pgmini_migrate-0.1.1.dist-info/licenses/LICENSE,sha256=n8ph1nmbOC4LqVLKRE-5SrUxzlv71A4saf-2_iNWR2Y,37
+pgmini_migrate-0.1.1.dist-info/METADATA,sha256=dhj1ESyg7CxugVhMNQuuo-U-J1-QYX_t-clndcAbdWg,649
+pgmini_migrate-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pgmini_migrate-0.1.1.dist-info/entry_points.txt,sha256=Zy6EN7sJuzTPbMhyoFjri1OdQ2iX1HuPw4UcSi4G6OY,54
+pgmini_migrate-0.1.1.dist-info/top_level.txt,sha256=ok556JVjy6jjrWTHi9d_lE4TteBXjQIoLOoJFD7_yhg,15
+pgmini_migrate-0.1.1.dist-info/RECORD,,

pgmini_migrate-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+pgmini_migrate

pgmini_migrate-0.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-pgmini_migrate-0.1.0.dist-info/licenses/LICENSE,sha256=n8ph1nmbOC4LqVLKRE-5SrUxzlv71A4saf-2_iNWR2Y,37
-pgmini_migrate-0.1.0.dist-info/METADATA,sha256=WSE1F2hQk8NAT-l4K365L-r6tpCD4o1J3uG1-D9SGQ0,649
-pgmini_migrate-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pgmini_migrate-0.1.0.dist-info/entry_points.txt,sha256=Zy6EN7sJuzTPbMhyoFjri1OdQ2iX1HuPw4UcSi4G6OY,54
-pgmini_migrate-0.1.0.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-pgmini_migrate-0.1.0.dist-info/RECORD,,

pgmini_migrate-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-