diff --git a/test/check_database_constraints.py b/test/check_database_constraints.py index 06a81fe81..7002dcf4a 100644 --- a/test/check_database_constraints.py +++ b/test/check_database_constraints.py @@ -30,9 +30,9 @@ # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -from ietf import settings -from django.core import management -management.setup_environ(settings) +import os + +os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ietf.settings") from django.db.models.fields.related import ForeignKey from django import db diff --git a/test/find_non_ascii.py b/test/find_non_ascii.py index d7b162bdd..afd302ba6 100644 --- a/test/find_non_ascii.py +++ b/test/find_non_ascii.py @@ -30,9 +30,9 @@ # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -from ietf import settings -from django.core import management -management.setup_environ(settings) +import os + +os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ietf.settings") from django.db.models.fields import CharField, TextField from django import db