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)