From 2e9b024390fc8b7309a4497280cedc5b0cbe8b81 Mon Sep 17 00:00:00 2001
From: DarkPhoenix <phoenix@mail.ru>
Date: Thu, 1 Aug 2019 11:54:33 +0300
Subject: [PATCH] Add browser lite handlers to graphs window

---
 gui/builtinContextMenus/fitAddBrowse.py |  3 ++-
 gui/graphFrame/lists.py                 | 11 ++++++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/gui/builtinContextMenus/fitAddBrowse.py b/gui/builtinContextMenus/fitAddBrowse.py
index 836121f40..0f6eae676 100644
--- a/gui/builtinContextMenus/fitAddBrowse.py
+++ b/gui/builtinContextMenus/fitAddBrowse.py
@@ -28,7 +28,8 @@ class AddBrowsedFits(ContextMenuUnconditional):
         excludedFitIDs = callingWindow.getExistingFitIDs()
         dlg = FitBrowserLiteDialog(self.mainFrame, title=titles[fullContext[0]], excludedFitIDs=excludedFitIDs)
         if dlg.ShowModal() == wx.ID_OK:
-            pass
+            fitIDs = dlg.getFitIDsToAdd()
+            callingWindow.addFitsByIDs(fitIDs)
 
 
 AddBrowsedFits.register()
diff --git a/gui/graphFrame/lists.py b/gui/graphFrame/lists.py
index 9888bb812..bdf3d2b52 100644
--- a/gui/graphFrame/lists.py
+++ b/gui/graphFrame/lists.py
@@ -22,7 +22,6 @@
 import wx
 
 import gui.display
-import gui.globalEvents as GE
 from eos.saveddata.targetProfile import TargetProfile
 from gui.contextMenu import ContextMenu
 from service.const import GraphCacheCleanupReason
@@ -169,6 +168,16 @@ class BaseList(gui.display.Display):
     def getExistingFitIDs(self):
         return [f.ID for f in self.fits]
 
+    def addFitsByIDs(self, fitIDs):
+        sFit = Fit.getInstance()
+        for fitID in fitIDs:
+            fit = sFit.getFit(fitID)
+            if fit is not None:
+                self.fits.append(fit)
+        self.updateView()
+        self.graphFrame.draw()
+        
+
 class FitList(BaseList):
 
     def __init__(self, graphFrame, parent):