feat: create Makefile for managing migrations
This commit is contained in:
3 files changed
+27
-6
No files matched your search
@@ -60,7 +60,7 @@ class MigrationRunner:
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Manage PostgreSQL schema migrations")
|
||||
parser.add_argument("action", choices=("rollback",))
|
||||
parser.add_argument("action", choices=("migrate", "rollback"))
|
||||
parser.add_argument("--to", type=int, default=0, dest="target_version")
|
||||
args = parser.parse_args()
|
||||
|
||||
@@ -71,7 +71,11 @@ def main() -> None:
|
||||
user=os.environ.get("POSTGRES_USER", "zyrarbot"),
|
||||
password=os.environ.get("POSTGRES_PASSWORD", "zyrarbot"),
|
||||
) as connection:
|
||||
MigrationRunner(MIGRATIONS).rollback_to(connection, args.target_version)
|
||||
runner = MigrationRunner(MIGRATIONS)
|
||||
if args.action == "migrate":
|
||||
runner.apply_pending(connection)
|
||||
else:
|
||||
runner.rollback_to(connection, args.target_version)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in new issue
Block a user