add new application qflatboob
This commit is contained in:
Romain Bignon
parent
a93daefb4a
commit
07f105910e
11 changed files with 1068 additions and 1 deletions
219
weboob/applications/qflatboob/main_window.py
Normal file
219
weboob/applications/qflatboob/main_window.py
Normal file
|
|
@ -0,0 +1,219 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright(C) 2010-2012 Romain Bignon
|
||||
#
|
||||
# This file is part of weboob.
|
||||
#
|
||||
# weboob is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Affero General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# weboob 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 Affero General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Affero General Public License
|
||||
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
from PyQt4.QtGui import QListWidgetItem, QImage, QPixmap, QLabel
|
||||
from PyQt4.QtCore import SIGNAL, Qt
|
||||
|
||||
from weboob.tools.application.qt import QtMainWindow, QtDo, HTMLDelegate
|
||||
from weboob.tools.application.qt.backendcfg import BackendCfg
|
||||
from weboob.capabilities.housing import ICapHousing, Query, City
|
||||
from weboob.capabilities.base import NotLoaded
|
||||
|
||||
from .ui.main_window_ui import Ui_MainWindow
|
||||
from .query import QueryDialog
|
||||
|
||||
class MainWindow(QtMainWindow):
|
||||
def __init__(self, config, weboob, parent=None):
|
||||
QtMainWindow.__init__(self, parent)
|
||||
self.ui = Ui_MainWindow()
|
||||
self.ui.setupUi(self)
|
||||
|
||||
self.config = config
|
||||
self.weboob = weboob
|
||||
self.process = None
|
||||
self.housing = None
|
||||
self.displayed_photo_idx = 0
|
||||
self.process_photo = {}
|
||||
|
||||
self.ui.housingsList.setItemDelegate(HTMLDelegate())
|
||||
self.ui.housingFrame.hide()
|
||||
|
||||
self.connect(self.ui.actionBackends, SIGNAL("triggered()"), self.backendsConfig)
|
||||
self.connect(self.ui.queriesList, SIGNAL('currentIndexChanged(int)'), self.queryChanged)
|
||||
self.connect(self.ui.addQueryButton, SIGNAL('clicked()'), self.addQuery)
|
||||
self.connect(self.ui.housingsList, SIGNAL('itemClicked(QListWidgetItem*)'), self.housingSelected)
|
||||
self.connect(self.ui.previousButton, SIGNAL('clicked()'), self.previousClicked)
|
||||
self.connect(self.ui.nextButton, SIGNAL('clicked()'), self.nextClicked)
|
||||
|
||||
self.reloadQueriesList()
|
||||
self.refreshHousingsList()
|
||||
|
||||
if self.weboob.count_backends() == 0:
|
||||
self.backendsConfig()
|
||||
|
||||
def backendsConfig(self):
|
||||
bckndcfg = BackendCfg(self.weboob, (ICapHousing,), self)
|
||||
if bckndcfg.run():
|
||||
pass
|
||||
|
||||
def reloadQueriesList(self, select_name=None):
|
||||
self.disconnect(self.ui.queriesList, SIGNAL('currentIndexChanged(int)'), self.queryChanged)
|
||||
self.ui.queriesList.clear()
|
||||
for name in self.config.get('queries', default={}).iterkeys():
|
||||
self.ui.queriesList.addItem(name)
|
||||
if name == select_name:
|
||||
self.ui.queriesList.setCurrentIndex(len(self.ui.queriesList))
|
||||
self.connect(self.ui.queriesList, SIGNAL('currentIndexChanged(int)'), self.queryChanged)
|
||||
|
||||
def addQuery(self):
|
||||
querydlg = QueryDialog(self.weboob, self)
|
||||
if querydlg.exec_():
|
||||
name = unicode(querydlg.ui.nameEdit.text())
|
||||
query = {}
|
||||
query['cities'] = []
|
||||
for i in xrange(len(querydlg.ui.citiesList)):
|
||||
item = querydlg.ui.citiesList.item(i)
|
||||
city = item.data(Qt.UserRole).toPyObject()
|
||||
query['cities'].append({'id': city.id, 'backend': city.backend, 'name': city.name})
|
||||
query['area_min'] = querydlg.ui.areaMin.value()
|
||||
query['area_max'] = querydlg.ui.areaMax.value()
|
||||
query['cost_min'] = querydlg.ui.costMin.value()
|
||||
query['cost_max'] = querydlg.ui.costMax.value()
|
||||
self.config.set('queries', name, query)
|
||||
self.config.save()
|
||||
|
||||
self.reloadQueriesList(name)
|
||||
|
||||
def queryChanged(self, i):
|
||||
self.refreshHousingsList()
|
||||
|
||||
def refreshHousingsList(self):
|
||||
name = unicode(self.ui.queriesList.itemText(self.ui.queriesList.currentIndex()))
|
||||
q = self.config.get('queries', name)
|
||||
|
||||
self.ui.housingsList.clear()
|
||||
self.ui.queriesList.setEnabled(False)
|
||||
|
||||
query = Query()
|
||||
query.cities = []
|
||||
for c in q['cities']:
|
||||
city = City(c['id'])
|
||||
city.backend = c['backend']
|
||||
city.name = c['name']
|
||||
query.cities.append(city)
|
||||
|
||||
query.area_min = int(q['area_min']) or None
|
||||
query.area_max = int(q['area_max']) or None
|
||||
query.cost_min = int(q['cost_min']) or None
|
||||
query.cost_max = int(q['cost_max']) or None
|
||||
|
||||
self.process = QtDo(self.weboob, self.addHousing)
|
||||
self.process.do('search_housings', query)
|
||||
|
||||
def addHousing(self, backend, housing):
|
||||
if not backend:
|
||||
self.ui.queriesList.setEnabled(True)
|
||||
self.process = None
|
||||
return
|
||||
|
||||
item = QListWidgetItem()
|
||||
item.setText(u'<h2>%s</h2><i>%s — %s%s (%s)</i><br />%s' % (housing.title, housing.date.strftime('%Y-%m-%d') if housing.date else 'Unknown',
|
||||
housing.cost, housing.currency, housing.backend, housing.text))
|
||||
item.setData(Qt.UserRole, housing)
|
||||
self.ui.housingsList.addItem(item)
|
||||
|
||||
def housingSelected(self, item):
|
||||
housing = item.data(Qt.UserRole).toPyObject()
|
||||
self.ui.queriesFrame.setEnabled(False)
|
||||
|
||||
self.setHousing(housing)
|
||||
|
||||
self.process = QtDo(self.weboob, self.gotHousing)
|
||||
self.process.do('fillobj', housing, backends=housing.backend)
|
||||
|
||||
def setHousing(self, housing, nottext='Loading...'):
|
||||
self.housing = housing
|
||||
|
||||
self.ui.housingFrame.show()
|
||||
|
||||
self.display_photo()
|
||||
|
||||
self.ui.titleLabel.setText('<h1>%s</h1>' % housing.title)
|
||||
self.ui.areaLabel.setText(u'%s m²' % housing.area)
|
||||
self.ui.costLabel.setText(u'%s %s' % (housing.cost, housing.currency))
|
||||
self.ui.dateLabel.setText(housing.date.strftime('%Y-%m-%d') if housing.date else nottext)
|
||||
self.ui.phoneLabel.setText(housing.phone or nottext)
|
||||
self.ui.locationLabel.setText(housing.location or nottext)
|
||||
self.ui.stationLabel.setText(housing.station or nottext)
|
||||
|
||||
self.ui.descriptionEdit.setText(housing.text or nottext)
|
||||
|
||||
while self.ui.detailsFrame.layout().count() > 0:
|
||||
child = self.ui.detailsFrame.layout().takeAt(0)
|
||||
child.widget().hide()
|
||||
child.widget().deleteLater()
|
||||
|
||||
if housing.details:
|
||||
for key, value in housing.details.iteritems():
|
||||
label = QLabel(value)
|
||||
label.setTextInteractionFlags(Qt.TextSelectableByMouse|Qt.LinksAccessibleByMouse)
|
||||
self.ui.detailsFrame.layout().addRow('<b>%s:</b>' % key, label)
|
||||
|
||||
def gotHousing(self, backend, housing):
|
||||
if not backend:
|
||||
self.ui.queriesFrame.setEnabled(True)
|
||||
self.process = None
|
||||
return
|
||||
|
||||
self.setHousing(housing, nottext='')
|
||||
|
||||
def previousClicked(self):
|
||||
if len(self.housing.photos) == 0:
|
||||
return
|
||||
self.displayed_photo_idx = (self.displayed_photo_idx - 1) % len(self.housing.photos)
|
||||
self.display_photo()
|
||||
|
||||
def nextClicked(self):
|
||||
if len(self.housing.photos) == 0:
|
||||
return
|
||||
self.displayed_photo_idx = (self.displayed_photo_idx + 1) % len(self.housing.photos)
|
||||
self.display_photo()
|
||||
|
||||
def display_photo(self):
|
||||
if not self.housing.photos:
|
||||
self.ui.photoUrlLabel.setText('')
|
||||
return
|
||||
|
||||
if self.displayed_photo_idx >= len(self.housing.photos):
|
||||
self.displayed_photo_idx = len(self.housing.photos) - 1
|
||||
if self.displayed_photo_idx < 0:
|
||||
self.ui.photoUrlLabel.setText('')
|
||||
return
|
||||
|
||||
photo = self.housing.photos[self.displayed_photo_idx]
|
||||
if photo.data:
|
||||
data = photo.data
|
||||
if photo.id in self.process_photo:
|
||||
self.process_photo.pop(photo.id)
|
||||
else:
|
||||
self.process_photo[photo.id] = QtDo(self.weboob, lambda b,p: self.display_photo())
|
||||
self.process_photo[photo.id].do('fillobj', photo, ['data'], backends=self.housing.backend)
|
||||
|
||||
if photo.thumbnail_data:
|
||||
data = photo.thumbnail_data
|
||||
else:
|
||||
return
|
||||
|
||||
img = QImage.fromData(data)
|
||||
img = img.scaledToWidth(self.width()/3)
|
||||
|
||||
self.ui.photoLabel.setPixmap(QPixmap.fromImage(img))
|
||||
if photo.url is not NotLoaded:
|
||||
text = '<a href="%s">%s</a>' % (photo.url, photo.url)
|
||||
self.ui.photoUrlLabel.setText(text)
|
||||
Loading…
Add table
Add a link
Reference in a new issue