SwissFEL/grum
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use DESC_TYPES and stored type to choose DescType on file load

Browse Source
This commit is contained in:
augustin_s
2023-04-15 13:45:35 +02:00
parent 4bb052d665
commit 839b1a1e21
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
grum/mainwin.py
View File

@ -2,7 +2,7 @@ from PyQt5.QtCore import Qt, pyqtSignal
from PyQt5.QtWidgets import QMainWindow, QSplitter from PyQt5.QtWidgets import QMainWindow, QSplitter
from . import assets from . import assets
from .descs import PlotDescription, ImageDescription from .descs import DESC_TYPES, PlotDescription, ImageDescription
from .dictlist import DictList from .dictlist import DictList
from .exampledata import exampledata from .exampledata import exampledata
from .h5filedlg import open_h5_files_dialog, save_h5_file_dialog from .h5filedlg import open_h5_files_dialog, save_h5_file_dialog
@ -208,7 +208,8 @@ class MainWindow(QMainWindow):
for fn in fns: for fn in fns:
data = read_dict(fn) data = read_dict(fn)
for k, v in data.items(): for k, v in data.items():
DescType = ImageDescription if "image" in v else PlotDescription #TODO tn = v.pop("type")
DescType = DESC_TYPES[tn]
self.add_new_desc_to_list(DescType, k, v) self.add_new_desc_to_list(DescType, k, v)