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
|
# -*- coding: utf-8 -*-
#
# PartMgr GUI - Storage widgets
#
# Copyright 2014 Michael Buesch <m@bues.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
from partmgr.gui.itemselect import *
from partmgr.gui.groupselect import *
from partmgr.gui.quantity import *
from partmgr.gui.util import *
from partmgr.core.storage import *
class OneStorageSelectWidget(ItemSelectWidget):
quantityChanged = Signal(int)
def __init__(self, storage, parent=None):
self.quantity = QuantityWidget(storage)
ItemSelectWidget.__init__(self, parent,
actionButtonLabel = "Remove",
itemLabel = "Location:",
itemSpecificWidget = self.quantity)
self.storage = storage
self.updateData(storage.db.getLocations(),
storage.getLocation())
self.selectionChanged.connect(self.__locationSelChanged)
self.quantity.quantityChanged.connect(self.__quantityChanged)
def __locationSelChanged(self, location):
self.storage.setLocation(location)
def __quantityChanged(self, newQuantity):
self.storage.setQuantity(newQuantity)
self.quantityChanged.emit(newQuantity)
def actionButtonPressed(self):
qty = str(self.storage.getQuantity())
loc = self.storage.getLocation()
if loc:
loc = loc.getName()
else:
loc = "<none>"
res = QMessageBox.question(self,
"Delete storage?",
"Delete the storage '%s' / '%s'?" %\
(loc, qty),
QMessageBox.Yes | QMessageBox.No)
if res != QMessageBox.Yes:
return
self.storage.delete()
self.quantityChanged.emit(0)
self.parent().updateData()
class StoragesSelectWidget(GroupSelectWidget):
quantityChanged = Signal()
contentChanged = Signal()
def __init__(self, parent=None):
GroupSelectWidget.__init__(self, parent,
"Add new storage")
self.currentStockItem = None
def updateData(self, stockItem=None):
if not stockItem:
stockItem = self.currentStockItem
self.clear()
self.currentStockItem = stockItem
storages = stockItem.getStorages()
storages.sort(key=lambda s: s.getId())
for storage in storages:
widget = OneStorageSelectWidget(storage, self)
widget.selectionChanged.connect(self.contentChanged)
widget.quantityChanged.connect(self.quantityChanged)
self.addItemSelectWidget(widget)
self.finishUpdate()
def createNewGroup(self):
stockItem = self.currentStockItem
if not stockItem:
return
storage = Storage("", stockItem = stockItem)
stockItem.db.modifyStorage(storage)
self.updateData(stockItem)
self.contentChanged.emit()
|
