diff --git a/apiserver/mongo/initialize/migration.py b/apiserver/mongo/initialize/migration.py index 04f6da3..8837604 100644 --- a/apiserver/mongo/initialize/migration.py +++ b/apiserver/mongo/initialize/migration.py @@ -18,9 +18,15 @@ _migration_dir = _parent_dir / _migrations def check_mongo_empty() -> bool: - return not all( - get_db(alias).list_collection_names() for alias in utils.get_options(Database) - ) + for alias in utils.get_options(Database): + collection_names = get_db(alias).list_collection_names() + if collection_names and any( + name in collection_names + for name in ["company", "user", "versions"] + ): + return False + + return True def get_last_server_version() -> Version: