diff --git a/apps/jumpserver/rewriting/db.py b/apps/jumpserver/rewriting/db.py index 1ccd49d8f..ddb29e538 100644 --- a/apps/jumpserver/rewriting/db.py +++ b/apps/jumpserver/rewriting/db.py @@ -26,7 +26,7 @@ class OneToOneField(models.OneToOneField, ForeignKey): def set_db_constraint(): if os.getenv('DB_CONSTRAINT', '1') != '0': return - if len(sys.argv) == 2 and sys.argv[1] == 'makemigrations': + if len(sys.argv) == 2 and sys.argv[1] in ['makemigrations', 'check']: return print("Set foreignkey db constraint False") transaction.atomic = atomic