|
"""Peewee migrations -- 009_add_models.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.""" |
|
|
|
@migrator.create_model |
|
class File(pw.Model): |
|
id = pw.TextField(unique=True) |
|
user_id = pw.TextField() |
|
filename = pw.TextField() |
|
meta = pw.TextField() |
|
created_at = pw.BigIntegerField(null=False) |
|
|
|
class Meta: |
|
table_name = "file" |
|
|
|
|
|
def rollback(migrator: Migrator, database: pw.Database, *, fake=False): |
|
"""Write your rollback migrations here.""" |
|
|
|
migrator.remove_model("file") |
|
|
