Jetbase is a simple, lightweight database migration tool for Python projects.
Jetbase helps you manage database migrations in a simple, version-controlled way. Whether you're adding a new table, modifying columns, or need to undo a change, Jetbase makes it super easy!
Created and maintained by Jaz π οΈ
- π¦ Simple Setup β Get started with just one command
- β¬οΈ Easy Upgrades β Apply pending migrations with confidence
- β¬οΈ Safe Rollbacks β Made a mistake? No problem, roll it back!
- π Clear Status β Always know which migrations have been applied and which are pending
- π Migration Locking β Prevents conflicts when multiple processes try to migrate
- β Checksum Validation β Detects if migration files have been modified
- π Repeatable Migrations β Support for migrations that run on every upgrade
Using pip:
pip install jetbase
Using uv:
uv add jetbase
Note for Snowflake and Databricks Users:
To use Jetbase with Snowflake or Databricks, install the appropriate extras:pip install "jetbase[snowflake]" pip install "jetbase[databricks]"
jetbase init
cd jetbaseThis creates a jetbase/ directory with:
- A
migrations/folder for your SQL files - An
env.pyconfiguration file
Edit jetbase/env.py with your database connection string (currently support for postgres, sqlite snowflake, databricks):
PostgreSQL example:
sqlalchemy_url = "postgresql+psycopg2://user:password@localhost:5432/mydb"
SQLite example:
sqlalchemy_url = "sqlite:///mydb.db"
jetbase new "create users table" -v 1This creates a new SQL file called V1__create_users_table.sql.
Tip:
You can also create migrations manually by adding SQL files in thejetbase/migrationsdirectory, using theV<version>__<description>.sqlnaming convention (e.g.,V1__add_users_table.sql,V2.4__add_users_table.sql).
Open the newly created file and add your SQL:
-- upgrade CREATE TABLE users ( id SERIAL PRIMARY KEY, name VARCHAR(100) NOT NULL, email VARCHAR(255) UNIQUE NOT NULL ); CREATE TABLE items ( id SERIAL PRIMARY KEY, name VARCHAR(100) NOT NULL ); -- rollback DROP TABLE items; DROP TABLE users;
jetbase upgrade
That's it! Your database is now up to date. π
Note:
Jetbase uses SQLAlchemy under the hood to manage database connections.
For any database other than SQLite, you must install the appropriate Python database driver.
For example, to use Jetbase with PostgreSQL:
pip install psycopg2
You can also use another compatible driver if you prefer (such as asyncpg, pg8000, etc.).
Jetbase currently supports:
- β PostgreSQL
- β SQLite
- β Snowflake
- β Databricks
- β MySQL
Open an issue on GitHub!