manage.py

#!/usr/bin/env python
"""Django's command-line utility for administrative tasks."""
import multiprocessing
import os
import sys


def apply_test_settings():
    sys.argv.append("--settings=django_smg.settings.test")
    print("Running tests with django_smg.test_settings to supress log output")


def patch_mac_fork_bug():
    # Workaround for https://code.djangoproject.com/ticket/31169
    if os.environ.get("OBJC_DISABLE_INITIALIZE_FORK_SAFETY", "") != "YES":
        sys.exit(
            "Set OBJC_DISABLE_INITIALIZE_FORK_SAFETY=YES in your "
            "environment to work around use of forking in Django's "
            "test runner."
        )
    multiprocessing.set_start_method("fork")


def main():
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "django_smg.settings")
    try:
        from django.core.management import execute_from_command_line
    except ImportError as exc:
        raise ImportError(
            "Couldn't import Django. Are you sure it's installed and "
            "available on your PYTHONPATH environment variable? Did you "
            "forget to activate a virtual environment?"
        ) from exc
    if "test" in sys.argv:
        apply_test_settings()
        if sys.platform == "darwin":
            patch_mac_fork_bug()
    execute_from_command_line(sys.argv)


if __name__ == "__main__":
    main()