diff --git a/cdek/api.py b/cdek/api.py index 04d73b6..1eb1e17 100644 --- a/cdek/api.py +++ b/cdek/api.py @@ -9,6 +9,8 @@ from django.conf import settings from django.core.files.base import ContentFile from requests import Request +from store.utils import is_migration_running + os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'poizonstore.settings') diff --git a/store/utils.py b/store/utils.py index 54ce737..2ab9c86 100644 --- a/store/utils.py +++ b/store/utils.py @@ -1,4 +1,5 @@ import os +import sys import textwrap from typing import Tuple @@ -153,3 +154,7 @@ def get_primary_key_related_model(model_class, **kwargs): return model_class.to_representation(self, data) return PrimaryKeyNestedMixin(**kwargs) + + +def is_migration_running(): + return 'makemigrations' in sys.argv or 'migrate' in sys.argv