nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

save read housings, and display new ones with a blue background

Browse source
This commit is contained in:
Romain Bignon 2012-02-18 11:58:28 +01:00
commit 379083d2d7
2 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
weboob/applications/qflatboob/main_window.py
View file

@ -17,7 +17,7 @@
# You should have received a copy of the GNU Affero General Public License # You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
from PyQt4.QtGui import QListWidgetItem, QImage, QPixmap, QLabel, QIcon from PyQt4.QtGui import QListWidgetItem, QImage, QPixmap, QLabel, QIcon, QBrush, QColor
from PyQt4.QtCore import SIGNAL, Qt from PyQt4.QtCore import SIGNAL, Qt
from weboob.tools.application.qt import QtMainWindow, QtDo, HTMLDelegate from weboob.tools.application.qt import QtMainWindow, QtDo, HTMLDelegate
@ -29,12 +29,13 @@ from .ui.main_window_ui import Ui_MainWindow
from .query import QueryDialog from .query import QueryDialog
class MainWindow(QtMainWindow): class MainWindow(QtMainWindow):
def __init__(self, config, weboob, parent=None): def __init__(self, config, storage, weboob, parent=None):
QtMainWindow.__init__(self, parent) QtMainWindow.__init__(self, parent)
self.ui = Ui_MainWindow() self.ui = Ui_MainWindow()
self.ui.setupUi(self) self.ui.setupUi(self)
self.config = config self.config = config
self.storage = storage
self.weboob = weboob self.weboob = weboob
self.process = None self.process = None
self.housing = None self.housing = None
@ -135,6 +136,9 @@ class MainWindow(QtMainWindow):
housing.area, housing.cost, housing.currency, housing.backend, housing.text)) housing.area, housing.cost, housing.currency, housing.backend, housing.text))
item.setData(Qt.UserRole, housing) item.setData(Qt.UserRole, housing)
if not '%s@%s' % (housing.id, housing.backend) in self.storage.get('read'):
item.setBackground(QBrush(QColor(219, 224, 255)))
if housing.photos is NotLoaded: if housing.photos is NotLoaded:
process = QtDo(self.weboob, lambda b, c: self.setPhoto(c, item)) process = QtDo(self.weboob, lambda b, c: self.setPhoto(c, item))
process.do('fillobj', housing, ['photos'], backends=housing.backend) process.do('fillobj', housing, ['photos'], backends=housing.backend)
@ -152,6 +156,12 @@ class MainWindow(QtMainWindow):
housing = item.data(Qt.UserRole).toPyObject() housing = item.data(Qt.UserRole).toPyObject()
self.ui.queriesFrame.setEnabled(False) self.ui.queriesFrame.setEnabled(False)
item.setBackground(QBrush())
read = set(self.storage.get('read'))
read.add('%s@%s' % (housing.id, housing.backend))
self.storage.set('read', list(read))
self.storage.save()
self.setHousing(housing) self.setHousing(housing)
self.process = QtDo(self.weboob, self.gotHousing) self.process = QtDo(self.weboob, self.gotHousing)

4
weboob/applications/qflatboob/qflatboob.py
View file

@ -31,11 +31,13 @@ class QFlatBoob(QtApplication):
DESCRIPTION = 'Qt application to find housings.' DESCRIPTION = 'Qt application to find housings.'
CAPS = ICapHousing CAPS = ICapHousing
CONFIG = {'queries': {}} CONFIG = {'queries': {}}
STORAGE = {'bookmarks': [], 'read': []}
def main(self, argv): def main(self, argv):
self.load_backends(ICapHousing) self.load_backends(ICapHousing)
self.create_storage()
self.load_config(klass=YamlConfig) self.load_config(klass=YamlConfig)
self.main_window = MainWindow(self.config, self.weboob) self.main_window = MainWindow(self.config, self.storage, self.weboob)
self.main_window.show() self.main_window.show()
return self.weboob.loop() return self.weboob.loop()