|
"""Peewee migrations -- 002_add_local_sharing.py. |
|
|
|
Some examples (model - class or model name):: |
|
|
|
> Model = migrator.orm['table_name'] # Return model in current state by name |
|
> Model = migrator.ModelClass # Return model in current state by name |
|
|
|
> migrator.sql(sql) # Run custom SQL |
|
> migrator.run(func, *args, **kwargs) # Run python function with the given args |
|
> migrator.create_model(Model) # Create a model (could be used as decorator) |
|
> migrator.remove_model(model, cascade=True) # Remove a model |
|
> migrator.add_fields(model, **fields) # Add fields to a model |
|
> migrator.change_fields(model, **fields) # Change fields |
|
> migrator.remove_fields(model, *field_names, cascade=True) |
|
> migrator.rename_field(model, old_field_name, new_field_name) |
|
> migrator.rename_table(model, new_table_name) |
|
> migrator.add_index(model, *col_names, unique=False) |
|
> migrator.add_not_null(model, *field_names) |
|
> migrator.add_default(model, field_name, default) |
|
> migrator.add_constraint(model, name, sql) |
|
> migrator.drop_index(model, *col_names) |
|
> migrator.drop_not_null(model, *field_names) |
|
> migrator.drop_constraints(model, *constraints) |
|
|
|
""" |
|
|
|
from contextlib import suppress |
|
|
|
import peewee as pw |
|
from peewee_migrate import Migrator |
|
|
|
|
|
with suppress(ImportError): |
|
import playhouse.postgres_ext as pw_pext |
|
|
|
|
|
def migrate(migrator: Migrator, database: pw.Database, *, fake=False): |
|
"""Write your migrations here.""" |
|
|
|
if isinstance(database, pw.SqliteDatabase): |
|
migrate_sqlite(migrator, database, fake=fake) |
|
else: |
|
migrate_external(migrator, database, fake=fake) |
|
|
|
|
|
def migrate_sqlite(migrator: Migrator, database: pw.Database, *, fake=False): |
|
|
|
migrator.add_fields( |
|
"chat", |
|
created_at=pw.DateTimeField(null=True), |
|
updated_at=pw.DateTimeField(null=True), |
|
) |
|
|
|
|
|
migrator.sql( |
|
"UPDATE chat SET created_at = timestamp, updated_at = timestamp WHERE timestamp IS NOT NULL" |
|
) |
|
|
|
|
|
migrator.remove_fields("chat", "timestamp") |
|
|
|
|
|
migrator.change_fields( |
|
"chat", |
|
created_at=pw.DateTimeField(null=False), |
|
updated_at=pw.DateTimeField(null=False), |
|
) |
|
|
|
|
|
def migrate_external(migrator: Migrator, database: pw.Database, *, fake=False): |
|
|
|
migrator.add_fields( |
|
"chat", |
|
created_at=pw.BigIntegerField(null=True), |
|
updated_at=pw.BigIntegerField(null=True), |
|
) |
|
|
|
|
|
migrator.sql( |
|
"UPDATE chat SET created_at = timestamp, updated_at = timestamp WHERE timestamp IS NOT NULL" |
|
) |
|
|
|
|
|
migrator.remove_fields("chat", "timestamp") |
|
|
|
|
|
migrator.change_fields( |
|
"chat", |
|
created_at=pw.BigIntegerField(null=False), |
|
updated_at=pw.BigIntegerField(null=False), |
|
) |
|
|
|
|
|
def rollback(migrator: Migrator, database: pw.Database, *, fake=False): |
|
"""Write your rollback migrations here.""" |
|
|
|
if isinstance(database, pw.SqliteDatabase): |
|
rollback_sqlite(migrator, database, fake=fake) |
|
else: |
|
rollback_external(migrator, database, fake=fake) |
|
|
|
|
|
def rollback_sqlite(migrator: Migrator, database: pw.Database, *, fake=False): |
|
|
|
migrator.add_fields("chat", timestamp=pw.DateTimeField(null=True)) |
|
|
|
|
|
|
|
migrator.sql("UPDATE chat SET timestamp = created_at") |
|
|
|
|
|
migrator.remove_fields("chat", "created_at", "updated_at") |
|
|
|
|
|
migrator.change_fields("chat", timestamp=pw.DateTimeField(null=False)) |
|
|
|
|
|
def rollback_external(migrator: Migrator, database: pw.Database, *, fake=False): |
|
|
|
migrator.add_fields("chat", timestamp=pw.BigIntegerField(null=True)) |
|
|
|
|
|
|
|
migrator.sql("UPDATE chat SET timestamp = created_at") |
|
|
|
|
|
migrator.remove_fields("chat", "created_at", "updated_at") |
|
|
|
|
|
migrator.change_fields("chat", timestamp=pw.BigIntegerField(null=False)) |
|
|