diff -r 0bab4ef95bfe -r bfaab8740995 src/cm/converters/abi_converters.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/cm/converters/abi_converters.py Thu Jul 21 09:42:33 2011 +0200 @@ -0,0 +1,283 @@ +import os +import tempfile +import re + +import pexpect + +from abi_error import AbiConverterError, AbiCommandError + + +TYPES_IN = {'602': '602', 'abw': 'abw', 'aw': 'aw', + 'awt': 'awt', 'cwk': 'cwk', 'dbk': 'dbk', + 'doc': 'doc', 'docm': 'docm', 'docx': 'docx', + 'dot': 'dot', 'dotm': 'dotm', 'dotx': 'dotx', + 'fo': 'fo', 'htm': 'htm', 'html': 'html', + 'hwp': 'hwp', 'isc': 'isc', 'iscii': 'iscii', + 'kwd': 'kwd', 'mif': 'mif', 'odt': 'odt', + 'opml': 'opml', 'ott': 'ott', 'pdb': 'pdb', + 'pdf': 'pdf', 'rtf': 'rtf', 'sdw': 'sdw', + 'stw': 'stw', 'sxw': 'sxw', 'text': 'text', + 'txt': 'txt', 'wml': 'wml', 'wp': 'wp', + 'wpd': 'wpd', 'wri': 'wri', 'xhtml': 'xhtml', + 'xml': 'xml', 'zabw': 'zabw'} + +TYPES_OUT = {'abw': 'abw', 'aw': 'aw', 'awt': 'awt', + 'dbk': 'dbk', 'doc': 'doc', 'eml': 'eml', + 'fo': 'fo', 'html': 'html', 'isc': 'isc', + 'iscii': 'iscii', 'kwd': 'kwd', 'latex': 'latex', + 'mht': 'mht', 'mif': 'mif', 'nroff': 'nroff', + 'nws': 'nws', 'odt': 'odt', 'pdb': 'pdb', + 'pdf': 'pdf', 'ps': 'ps', 'rtf': 'rtf', + 'sxw': 'sxw', 'text': 'text', 'txt': 'txt', + 'wml': 'wml', 'xml': 'xml', 'xml2ps': 'xml2ps', + 'zabw': 'zabw'} + +class AbiFileConverter(object): + """This let's you convert between all filetypes supperted by the + AbiWord program. Import type isn't checked, as AbiWord doesn't check + on extension, but on metadata. + """ + + def __init__(self, timeout=60): + self.id = None + self.timeout = timeout + self._start_abiword() + + def _start_abiword(self): + """ + Start abiword with the AbiCommand plugin, if not already started + """ + + # find the abiword executable + abicommand = None + for dir in os.environ['PATH'].split(':'): + if os.path.isfile(os.path.join(dir, 'abiword')): + abicommand = os.path.join(dir, 'abiword') + if not abicommand: + raise AbiConverterError('Can not find abiword executable') + + # start the abiword executable + try: + self.child = pexpect.spawn(abicommand + ' --plugin AbiCommand') + self.child.expect( + 'AbiWord command line plugin: Type "quit" to exit', 10) + except: + raise AbiConverterError('Can not open abiword executable') + + def stop_abiword(self): + """ + Stop the running abiword, kill it if necessary + """ + self.child.sendline('quit') + if self._is_running(): + os.kill(self.child.pid, 9) + + def _is_running(self): + """ + Test to see if abiword is running + """ + try: + self.child.sendline('writepid /dev/null') + self.child.expect('OK', 1) + return True + except: + return False + + def convert_file(self, in_file, out_file=None, type=None): + """ + Convert a file. If out_file is not specified, a byte string is + returned. If type is not specified, the file extension from out_file is + used to determine the type. If this fails, the type 'text' is used. + Return value is -1 if an error occurred. + """ + # is the out_file specified? + return_bytes = False + if out_file is None: + out_file = tempfile.mktemp(prefix="abiconvert_") + return_bytes = True + + # is the type specified + type = TYPES_OUT.get( + type or os.path.splitext(out_file)[1][1:], 'txt') + + # do the coversion + self._perform_conversion(in_file, out_file, type) + + # return a byte string if no out_file is specified + if return_bytes: + fp = open(out_file, 'r') + bytes = fp.read() + fp.close() + os.remove(out_file) + return bytes + + def _perform_conversion(self, in_file, out_file, type): + """ + Do the actual conversion + """ + # make sure we are up and running + if not self._is_running: + self._start_abiword() + + # convert the file + cmd = 'convert %s %s %s' % (os.path.abspath(in_file), + os.path.abspath(out_file), type) + self.child.sendline(cmd) + + # Check for errors + i = self.child.expect(['OK', pexpect.TIMEOUT]) + if i != 0: + raise AbiCommandError('Error performing AbiCommand: %s' %cmd) + + def convert_to_html(self, input): + """ + Convert input file to HTML + """ + + from tempfile import mkstemp,mkdtemp + + THE_OUTDIR = "outdir" + THE_OUTFILE = "outfile" + THE_INDIR = "indir" + THE_INFILE = "infile" + + infile = None + outfile = None + out_f = None + try: + # create in/out files + temp_dir = mkdtemp(prefix="cm_") + + # in + indir_name = os.path.join(temp_dir, THE_INDIR) + os.mkdir(indir_name) + infile_name = os.path.join(indir_name, THE_INFILE) + + # out + outdir_name = os.path.join(temp_dir, THE_OUTDIR) + os.mkdir(outdir_name) + outfile_name = os.path.join(outdir_name, THE_OUTFILE) + + # write infile + infile = open(infile_name,'w') + if type(input) == unicode: + input = input.encode('utf8') + infile.write(input) + infile.close() + + # fix perms + # TODO group permission should suffice + os.chmod(temp_dir, 0755) # read + os.chmod(indir_name, 0755) # read + os.chmod(infile_name, 0755) # read + os.chmod(outdir_name, 0777) # read / write + + # Do the job + self.convert_file(infile_name, outfile_name, 'html') + + out_f = open(outfile_name,'r') + output = out_f.read() + + # load other files (useful only for html) + img_res = [] + if os.path.isdir(outdir_name + '/' + THE_OUTFILE + '_files'): + image_names = [name for name in os.listdir(outdir_name + '/' + THE_OUTFILE + '_files') if name != THE_OUTFILE] + for image_name in image_names: + img_res.append(os.path.join(outdir_name + '/' + THE_OUTFILE + '_files', image_name)) + + # clean images paths + output = re.sub(r' + + + + + %s + + +""" %body +