manage.py

#!/usr/bin/env python


import os
import sys
import signal


def patch_create_suffix(original):
    """
    Patch mysql creation policy to handle the "utf8mb4" facility. This appears tricky but it's necessary
    due to the conception of "extended utf-8" in mysql. If we do not patch, the mysql backend cannot index
    ``VARCHAR(255)`` fields !
    see <http://bd808.com/blog/2017/04/17/making-django-migrations-that-work-with-mysql-55-and-utf8mb4/>
    for explanations

    :param original: the original function we are patching
    :return: the patched function
    """
    def patch(self):
        return original(self) + ' ROW_FORMAT=DYNAMIC'
    return patch


def sighandler(signum, frame):
    sys.exit(1)


if __name__ == '__main__':
    # Monkey-patch Django's broken signal handling
    # http://blog.lotech.org/fix-djangos-runserver-when-run-under-docker-or-pycharm.html
    signal.signal(signal.SIGTERM, sighandler)

    try:
        import django
    except ImportError:
        if os.environ.get('VIRTUAL_ENV') is None:
            print('\033[36;1mINFO: No VIRTUAL_ENV detected, '
            'usually contributors use virtual environment `zdsenv`\033[00;0m')
        elif os.path.normpath(os.environ.get('VIRTUAL_ENV')) != os.path.normpath('./zdsenv'):
            print('\033[36;1mINFO: The current VIRTUAL_ENV is `%s`, '
            'usually contributors use `./zdsenv`\033[00;0m' % os.environ.get('VIRTUAL_ENV'))

        print("\033[31;1mYour environment does not have django, please run pip install -r requirements-dev.txt\033[00;0m")
        sys.exit(1)

    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'zds.settings.dev')

    if len(sys.argv) > 1 and sys.argv[1] in ['migrate', 'test']:

        from django.db.backends.mysql.creation import BaseDatabaseCreation

        BaseDatabaseCreation.sql_table_creation_suffix = \
            patch_create_suffix(BaseDatabaseCreation.sql_table_creation_suffix)
        from django.db.backends.mysql.schema import DatabaseSchemaEditor

        DatabaseSchemaEditor.sql_create_table += ' ROW_FORMAT=DYNAMIC'
    from django.core.management import execute_from_command_line

    execute_from_command_line(sys.argv)