pyqt4 sample
This commit is contained in:
Jean-Francois Dockes
parent
9f7d32992c
commit
12c60d39d9
2 changed files with 272 additions and 0 deletions
208
src/python/samples/recollgui/qrecoll.py
Executable file
208
src/python/samples/recollgui/qrecoll.py
Executable file
|
|
@ -0,0 +1,208 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import sys
|
||||
import datetime
|
||||
try:
|
||||
from recoll import recoll
|
||||
from recoll import rclextract
|
||||
hasextract = True
|
||||
except:
|
||||
import recoll
|
||||
hasextract = False
|
||||
|
||||
import queryarea
|
||||
from getopt import getopt
|
||||
|
||||
from PyQt4 import QtCore, QtGui
|
||||
|
||||
class RecollQuery(QtCore.QAbstractTableModel):
|
||||
def __init__(self):
|
||||
QtCore.QAbstractTableModel.__init__(self)
|
||||
self.totres = -1
|
||||
self.query = None
|
||||
self.docs = []
|
||||
self.pagelen = 10
|
||||
self.attrs = ("filename", "title", "mtime", "url", "ipath")
|
||||
def rowCount(self, parent):
|
||||
ret = len(self.docs)
|
||||
#print "RecollQuery.rowCount(): ", ret
|
||||
return ret
|
||||
def columnCount(self, parent):
|
||||
#print "RecollQuery.columnCount()"
|
||||
if parent.isValid():
|
||||
return 0
|
||||
else:
|
||||
return len(self.attrs)
|
||||
def setquery(self, db, q):
|
||||
"""Parse and execute query on open db"""
|
||||
print "RecollQuery.setquery():"
|
||||
# Get query object
|
||||
self.query = db.query()
|
||||
# Parse/run input query string
|
||||
self.totres = self.query.execute(q)
|
||||
self.docs = []
|
||||
self.fetchMore(None)
|
||||
def getdoc(self, index):
|
||||
if index.row() < len(self.docs):
|
||||
return self.docs[index.row()]
|
||||
else:
|
||||
return None
|
||||
def headerData(self, idx, orient, role):
|
||||
if orient == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
|
||||
return self.attrs[idx]
|
||||
return None
|
||||
def data(self, index, role):
|
||||
#print "RecollQuery.data: row %d, role: " % (index.row(),), role
|
||||
if not index.isValid():
|
||||
return QtCore.QVariant()
|
||||
|
||||
if index.row() >= len(self.docs):
|
||||
return QtCore.QVariant()
|
||||
|
||||
if role == QtCore.Qt.DisplayRole:
|
||||
#print "RecollQuery.data: row %d, col %d role: " % \
|
||||
# (index.row(), index.column()), role
|
||||
attr = self.attrs[index.column()]
|
||||
value = getattr(self.docs[index.row()], attr)
|
||||
if attr == "mtime":
|
||||
dte = datetime.datetime.fromtimestamp(int(value))
|
||||
value = str(dte)
|
||||
return value
|
||||
else:
|
||||
return QtCore.QVariant()
|
||||
def canFetchMore(self, parent):
|
||||
#print "RecollQuery.canFetchMore:"
|
||||
if len(self.docs) < self.totres:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
def fetchMore(self, parent):
|
||||
#print "RecollQuery.fetchMore:"
|
||||
self.beginInsertRows(QtCore.QModelIndex(), len(self.docs), \
|
||||
len(self.docs) + self.pagelen)
|
||||
count = 0
|
||||
while self.query.next >= 0 and self.query.next < self.totres \
|
||||
and count < self.pagelen:
|
||||
#print "Got: ", title.encode("utf-8")
|
||||
self.docs.append(self.query.fetchone())
|
||||
count += 1
|
||||
self.endInsertRows()
|
||||
|
||||
class HlMeths:
|
||||
def __init__(self, groups):
|
||||
self.groups = groups
|
||||
def startMatch(self, idx):
|
||||
ugroup = " ".join(self.groups[idx][1])
|
||||
return '<font color="blue">'
|
||||
def endMatch(self):
|
||||
return '</font>'
|
||||
|
||||
def extract(doc):
|
||||
extractor = rclextract.Extractor(doc)
|
||||
newdoc = extractor.textextract(doc.ipath)
|
||||
return newdoc
|
||||
|
||||
def extractofile(doc, outfilename=""):
|
||||
extractor = rclextract.Extractor(doc)
|
||||
outfilename = extractor.idoctofile(doc.ipath, doc.mimetype, \
|
||||
ofilename=outfilename)
|
||||
return outfilename
|
||||
|
||||
class RclGui_Main(QtGui.QWidget):
|
||||
def __init__(self, db, parent=None):
|
||||
QtGui.QWidget.__init__(self, parent)
|
||||
self.ui = queryarea.Ui_queryArea()
|
||||
self.ui.setupUi(self)
|
||||
self.db = db
|
||||
self.qmodel = RecollQuery()
|
||||
self.connect(self.ui.exitPB, QtCore.SIGNAL("clicked()"),
|
||||
self.onexit)
|
||||
def on_searchEntry_returnPressed(self):
|
||||
self.startQuery()
|
||||
def on_resTable_clicked(self, index):
|
||||
doc = self.qmodel.getdoc(index)
|
||||
query = self.qmodel.query;
|
||||
groups = self.qmodel.query.getgroups()
|
||||
meths = HlMeths(groups)
|
||||
if doc is not None:
|
||||
ipath = doc.get('ipath')
|
||||
print "ipath[", ipath, "]"
|
||||
if index.column() == 1:
|
||||
newdoc = extract(doc)
|
||||
print "newdoc.mimetype:", newdoc.mimetype
|
||||
if newdoc.mimetype == 'text/html':
|
||||
ishtml = True
|
||||
else:
|
||||
ishtml = False
|
||||
text = query.highlight(newdoc.text,
|
||||
methods=meths,
|
||||
ishtml=ishtml,
|
||||
eolbr=True)
|
||||
print text
|
||||
|
||||
text = '<qt><head></head><body>' + text + '</body></qt>'
|
||||
self.ui.resDetail.setText(text)
|
||||
elif index.column() == 3 and ipath:
|
||||
fn = QtGui.QFileDialog.getSaveFileName(self)
|
||||
if fn:
|
||||
docitems = doc.items()
|
||||
fn = extractofile(doc, str(fn.toLocal8Bit()))
|
||||
print "Saved as", fn
|
||||
else:
|
||||
print >> sys.stderr, "Canceled"
|
||||
else:
|
||||
abs = query.makedocabstract(doc, methods=meths)
|
||||
self.ui.resDetail.setText(abs)
|
||||
|
||||
def startQuery(self):
|
||||
self.qmodel.setquery(self.db, self.ui.searchEntry.text())
|
||||
self.ui.resTable.setModel(self.qmodel)
|
||||
def onexit(self):
|
||||
sys.exit(0)
|
||||
|
||||
def Usage():
|
||||
print >> sys.stderr, '''Usage: qt.py [<qword1> [<qword2> ...]]'''
|
||||
sys.exit(1)
|
||||
|
||||
def main(args):
|
||||
|
||||
app = QtGui.QApplication(args)
|
||||
|
||||
confdir=""
|
||||
extra_dbs = []
|
||||
# Snippet params
|
||||
maxchars = 300
|
||||
contextwords = 6
|
||||
|
||||
# Process options: [-c confdir] [-i extra_db [-i extra_db] ...]
|
||||
options, args = getopt(args[1:], "c:i:")
|
||||
for opt,val in options:
|
||||
if opt == "-c":
|
||||
confdir = val
|
||||
elif opt == "-i":
|
||||
extra_dbs.append(val)
|
||||
else:
|
||||
print >> sys.stderr, "Bad opt: ", opt
|
||||
Usage()
|
||||
|
||||
# The query should be in the remaining arg(s)
|
||||
q = None
|
||||
if len(args) > 0:
|
||||
q = ""
|
||||
for word in args:
|
||||
q += word + " "
|
||||
|
||||
db = recoll.connect(confdir=confdir, extra_dbs=extra_dbs)
|
||||
db.setAbstractParams(maxchars=maxchars, contextwords=contextwords)
|
||||
|
||||
topwindow = RclGui_Main(db)
|
||||
topwindow.show()
|
||||
if q is not None:
|
||||
topwindow.ui.searchEntry.setText(q)
|
||||
topwindow.startQuery()
|
||||
|
||||
sys.exit(app.exec_())
|
||||
|
||||
|
||||
if __name__=="__main__":
|
||||
main(sys.argv)
|
||||
64
src/python/samples/recollgui/queryarea.ui
Normal file
64
src/python/samples/recollgui/queryarea.ui
Normal file
|
|
@ -0,0 +1,64 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<ui version="4.0">
|
||||
<class>queryArea</class>
|
||||
<widget class="QWidget" name="queryArea">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>0</x>
|
||||
<y>0</y>
|
||||
<width>614</width>
|
||||
<height>759</height>
|
||||
</rect>
|
||||
</property>
|
||||
<property name="windowTitle">
|
||||
<string>Recoll query</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_2">
|
||||
<item>
|
||||
<layout class="QVBoxLayout" name="verticalLayout">
|
||||
<item>
|
||||
<widget class="QPushButton" name="exitPB">
|
||||
<property name="text">
|
||||
<string>Exit</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QLineEdit" name="searchEntry"/>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QSplitter" name="splitter">
|
||||
<property name="orientation">
|
||||
<enum>Qt::Vertical</enum>
|
||||
</property>
|
||||
<widget class="QTableView" name="resTable">
|
||||
<property name="sizePolicy">
|
||||
<sizepolicy hsizetype="Expanding" vsizetype="Expanding">
|
||||
<horstretch>0</horstretch>
|
||||
<verstretch>4</verstretch>
|
||||
</sizepolicy>
|
||||
</property>
|
||||
<property name="editTriggers">
|
||||
<set>QAbstractItemView::NoEditTriggers</set>
|
||||
</property>
|
||||
<property name="cornerButtonEnabled">
|
||||
<bool>false</bool>
|
||||
</property>
|
||||
</widget>
|
||||
<widget class="QTextBrowser" name="resDetail">
|
||||
<property name="sizePolicy">
|
||||
<sizepolicy hsizetype="Expanding" vsizetype="Expanding">
|
||||
<horstretch>0</horstretch>
|
||||
<verstretch>0</verstretch>
|
||||
</sizepolicy>
|
||||
</property>
|
||||
</widget>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
<resources/>
|
||||
<connections/>
|
||||
</ui>
|
||||
Loading…
Add table
Add a link
Reference in a new issue