From 97736d8dde1c7402714184c66b36b5a2961069f1 Mon Sep 17 00:00:00 2001
From: Danila_Mochalov <volzhankaonelove@yandex.ru>
Date: Sun, 9 Apr 2023 16:13:54 +0400
Subject: [PATCH] =?UTF-8?q?=D0=98=D1=81=D0=BF=D1=80=D0=B0=D0=B2=D0=BB?=
 =?UTF-8?q?=D0=B5=D0=BD=20=D0=B0=D0=BF=D0=B4=D0=B5=D0=B9=D1=82=20=D0=BC?=
 =?UTF-8?q?=D0=B0=D0=B3=D0=B0=D0=B7=D0=B8=D0=BD=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 LawFirm/LawFirmDatabaseImplement/Implements/ShopStorage.cs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/LawFirm/LawFirmDatabaseImplement/Implements/ShopStorage.cs b/LawFirm/LawFirmDatabaseImplement/Implements/ShopStorage.cs
index 7d1c1f5..ef78595 100644
--- a/LawFirm/LawFirmDatabaseImplement/Implements/ShopStorage.cs
+++ b/LawFirm/LawFirmDatabaseImplement/Implements/ShopStorage.cs
@@ -79,14 +79,17 @@ namespace LawFirmDatabaseImplement.Implements
             using var transaction = context.Database.BeginTransaction();
             try
             {
-                var shop = context.Shops.FirstOrDefault(x => x.Id == model.Id);
+                var shop = context.Shops.Include(x => x.Documents).FirstOrDefault(x => x.Id == model.Id);
                 if (shop == null)
                 {
                     return null;
                 }
                 shop.Update(model);
                 context.SaveChanges();
-                shop.UpdateDocuments(context, model);
+                if (model.ShopDocuments.Count > 0)
+                {
+					shop.UpdateDocuments(context, model);
+				}
                 transaction.Commit();
                 return shop.GetViewModel;
             }