From 269f2b48505ae4ff67bb656effb03b9c80f6a537 Mon Sep 17 00:00:00 2001
From: Yuri Kuznetsov <yurikuzn@users.noreply.github.com>
Date: Sun, 12 Jul 2020 14:27:07 +0300
Subject: [PATCH] extension and upgrade scripts fix

---
 extension.php | 4 +---
 upgrade.php   | 4 +---
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/extension.php b/extension.php
index 2ba444a4b6..dc967fe139 100644
--- a/extension.php
+++ b/extension.php
@@ -47,13 +47,11 @@ if (!isset($pathInfo['extension']) || $pathInfo['extension'] !== 'zip' || !is_fi
 }
 
 $app = new \Espo\Core\Application();
+$app->setupSystemUser();
 
 $config = $app->getContainer()->get('config');
 $entityManager = $app->getContainer()->get('entityManager');
 
-$user = $entityManager->getEntity('User', 'system');
-$app->getContainer()->setUser($user);
-
 $upgradeManager = new \Espo\Core\ExtensionManager($app->getContainer());
 
 echo "Starting installation process...\n";
diff --git a/upgrade.php b/upgrade.php
index 4f44ce9407..4ac07f3f02 100644
--- a/upgrade.php
+++ b/upgrade.php
@@ -52,13 +52,11 @@ if (!isset($pathInfo['extension']) || $pathInfo['extension'] !== 'zip' || !is_fi
 }
 
 $app = new \Espo\Core\Application();
+$app->setupSystemUser();
 
 $config = $app->getContainer()->get('config');
 $entityManager = $app->getContainer()->get('entityManager');
 
-$user = $entityManager->getEntity('User', 'system');
-$app->getContainer()->setUser($user);
-
 $upgradeManager = new \Espo\Core\UpgradeManager($app->getContainer());
 
 echo "Current version is " . $config->get('version') . "\n";