1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
import os
import pickle
from typing import Any, Dict
from AnyQt.QtWidgets import QSizePolicy, QStyle, QFileDialog
from AnyQt.QtCore import QTimer, QUrl
from orangewidget.workflow.drophandler import SingleFileDropHandler
from Orange.base import Model
from Orange.widgets import widget, gui
from Orange.widgets.model import owsavemodel
from Orange.widgets.utils.filedialogs import RecentPathsWComboMixin, RecentPath, \
stored_recent_paths_prepend, OWUrlDropBase
from Orange.widgets.utils import stdpaths
from Orange.widgets.utils.widgetpreview import WidgetPreview
from Orange.widgets.widget import Msg, Output
class OWLoadModel(OWUrlDropBase, RecentPathsWComboMixin):
name = "Load Model"
description = "Load a model from an input file."
priority = 3050
replaces = ["Orange.widgets.classify.owloadclassifier.OWLoadClassifier"]
icon = "icons/LoadModel.svg"
keywords = "load model, file, open, model"
class Outputs:
model = Output("Model", Model)
class Error(widget.OWWidget.Error):
load_error = Msg("An error occured while reading '{}'")
FILTER = ";;".join(owsavemodel.OWSaveModel.filters)
want_main_area = False
buttons_area_orientation = None
resizing_enabled = False
def __init__(self):
super().__init__()
RecentPathsWComboMixin.__init__(self)
self.loaded_file = ""
vbox = gui.vBox(self.controlArea, "File")
box = gui.hBox(vbox)
self.file_combo.setMinimumWidth(300)
box.layout().addWidget(self.file_combo)
self.file_combo.activated[int].connect(self.select_file)
button = gui.button(box, self, '...', callback=self.browse_file)
button.setIcon(self.style().standardIcon(QStyle.SP_DirOpenIcon))
button.setSizePolicy(
QSizePolicy.Maximum, QSizePolicy.Fixed)
button = gui.button(
box, self, "Reload", callback=self.reload, default=True)
button.setIcon(self.style().standardIcon(QStyle.SP_BrowserReload))
button.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.set_file_list()
QTimer.singleShot(0, self.open_file)
def browse_file(self):
start_file = self.last_path() or stdpaths.Documents
filename, _ = QFileDialog.getOpenFileName(
self, 'Open Model File', start_file, self.FILTER)
if not filename:
return
self.add_path(filename)
self.open_file()
def select_file(self, n):
super().select_file(n)
self.open_file()
def reload(self):
self.open_file()
def open_file(self):
self.clear_messages()
fn = self.last_path()
if not fn:
return
try:
with open(fn, "rb") as f:
model = pickle.load(f)
except (pickle.UnpicklingError, OSError, EOFError):
self.Error.load_error(os.path.split(fn)[-1])
self.Outputs.model.send(None)
else:
self.Outputs.model.send(model)
def canDropUrl(self, url: QUrl) -> bool:
if url.isLocalFile():
return OWLoadModelDropHandler().canDropFile(url.toLocalFile())
else:
return False
def handleDroppedUrl(self, url: QUrl) -> None:
if url.isLocalFile():
self.add_path(url.toLocalFile())
self.open_file()
class OWLoadModelDropHandler(SingleFileDropHandler):
WIDGET = OWLoadModel
def canDropFile(self, path: str) -> bool:
return path.endswith(".pkcls")
def parametersFromFile(self, path: str) -> Dict[str, Any]:
r = RecentPath(os.path.abspath(path), None, None,
os.path.basename(path))
return {"recent_paths": stored_recent_paths_prepend(self.WIDGET, r)}
if __name__ == "__main__": # pragma: no cover
WidgetPreview(OWLoadModel).run()
|
