Using Django post_migrate signal

blog_django_code.py

######################################
############# models.py ##############
######################################

from architect.commands import partition
from django.db import ProgrammingError

def create_partitions(sender, **kwargs):
    """
    After running migrations, go through each of the models
    in the app and ensure the partitions have been setup
    """
    paths = {model.__module__ for model in sender.get_models()}
    for path in paths:
        try:
            partition.run(dict(module=path))
        except ProgrammingError:
            # Possibly because models were just un-migrated or
            # fields have been changed that effect Architect
            print(f"Unable to apply partitions for module '{path}'")
        else:
            print(f"Applied partitions for module '{path}'")

######################################
############# apps.py ################
######################################
        
from django.apps import AppConfig
from django.db.models.signals import post_migrate
from .models import create_partitions

class CoreConfig(AppConfig):
    name = 'core'

    def ready(self):
        # Hook up Architect to the post migrations signal
        post_migrate.connect(create_partitions, sender=self)