migration.py

from django.conf import settings
from django.db import migrations


'''
Django-allauth requires a site to be configured within the django.contrib.sites 
application. This is used when constructing URLs for the password reset emails 
and email confirmations etc. Therefore, it is important that this value is 
customised so that we can redirect those requests to redirect within the given 
application we need them to be.

For example, the auth service is hosted at a.com and the website at b.com. We 
want any URLs included in emails etc sent from auth-service to have b.com 
embedded in them so that we portray a consistent URL to all external users. Our 
services will deal with redirection of those requests.
'''


def forward_func(apps, schema_editor):
    domain = settings.CUSTOM_SITE_DOMAIN
    name = settings.CUSTOM_SITE_NAME

    Site = apps.get_model("sites", "Site")
    db_alias = schema_editor.connection.alias
    Site.objects.using(db_alias).create(domain=domain, name=name)


def reverse_func(apps, schema_editor):
    domain = settings.CUSTOM_SITE_DOMAIN
    name = settings.CUSTOM_SITE_NAME

    Site = apps.get_model("sites", "Site")
    db_alias = schema_editor.connection.alias
    Site.objects.using(db_alias).filter(domain=domain, name=name).delete()


class Migration(migrations.Migration):

    dependencies = [
        ('django.contrib.sites', '__latest__')
    ]

    operations = [
        migrations.RunPython(forward_func, reverse_func)
    ]