recoll/src/filters/rclrtf.py
Jean-Francois Dockes 3956d0a867 rcltext.py
2015-09-13 10:34:40 +02:00

56 lines
1.5 KiB
Python
Executable file

#!/usr/bin/env python
import rclexecm
import rclexec1
import re
import sys
import os
# Processing the output from unrtf
class RTFProcessData:
def __init__(self, em):
self.em = em
self.out = ""
self.gothead = 0
self.patendhead = re.compile('''</head>''')
self.patcharset = re.compile('''^<meta http-equiv=''')
# Some versions of unrtf put out a garbled charset line.
# Apart from this, we pass the data untouched.
def takeLine(self, line):
if not self.gothead:
if self.patendhead.search(line):
self.out += '<meta http-equiv="Content-Type" ' + \
'content="text/html;charset=UTF-8">' + "\n"
self.out += line + "\n"
self.gothead = 1
elif not self.patcharset.search(line):
self.out += line + "\n"
else:
self.out += line + "\n"
def wrapData(self):
return self.out
class RTFFilter:
def __init__(self, em):
self.em = em
def reset(self):
pass
def getCmd(self, fn):
cmd = rclexecm.which("unrtf")
if cmd:
return ([cmd, "--nopict", "--html"], RTFProcessData(self.em))
else:
return ([],None)
if __name__ == '__main__':
if not rclexecm.which("unrtf"):
print("RECFILTERROR HELPERNOTFOUND antiword")
sys.exit(1)
proto = rclexecm.RclExecM()
filter = RTFFilter(proto)
extract = rclexec1.Executor(proto, filter)
rclexecm.main(proto, extract)