From 39ceb5ac5c6fe3de796d849fcf01b4084cc263e1 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 25 Jul 2021 14:26:31 +0300 Subject: [PATCH] Fix pre-populate logic to avoid overriding existing users --- apiserver/mongo/initialize/pre_populate.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apiserver/mongo/initialize/pre_populate.py b/apiserver/mongo/initialize/pre_populate.py index 5aea81c..c97202f 100644 --- a/apiserver/mongo/initialize/pre_populate.py +++ b/apiserver/mongo/initialize/pre_populate.py @@ -298,8 +298,9 @@ class PrePopulate: if company_id is None: company_id = "" - # Always use a public user for pre-populated data - cls.user_cls(id=user_id, name=user_name, company="").save() + existing_user = cls.user_cls.objects(id=user_id).only("id").first() + if not existing_user: + cls.user_cls(id=user_id, name=user_name, company=company_id).save() cls._import(zfile, company_id, user_id, metadata)