68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
"""
|
||
|
Unit Name: dupeguru.export
|
||
|
Created By: Virgil Dupras
|
||
|
Created On: 2006/09/16
|
||
|
Last modified by:$Author: virgil $
|
||
|
Last modified on:$Date: 2009-05-28 15:22:39 +0200 (Thu, 28 May 2009) $
|
||
|
$Revision: 4385 $
|
||
|
Copyright 2004-2006 Hardcoded Software (http://www.hardcoded.net)
|
||
|
"""
|
||
|
from xml.dom import minidom
|
||
|
import tempfile
|
||
|
import os.path as op
|
||
|
import os
|
||
|
from StringIO import StringIO
|
||
|
|
||
|
from hsutil.files import FileOrPath
|
||
|
|
||
|
def output_column_xml(outfile, columns):
|
||
|
"""Creates a xml file outfile with the supplied columns.
|
||
|
|
||
|
outfile can be a filename or a file object.
|
||
|
columns is a list of 2 sized tuples (display,enabled)
|
||
|
"""
|
||
|
doc = minidom.Document()
|
||
|
root = doc.appendChild(doc.createElement('columns'))
|
||
|
for display,enabled in columns:
|
||
|
col_node = root.appendChild(doc.createElement('column'))
|
||
|
col_node.setAttribute('display', display)
|
||
|
col_node.setAttribute('enabled', {True:'y',False:'n'}[enabled])
|
||
|
with FileOrPath(outfile, 'wb') as fp:
|
||
|
doc.writexml(fp, '\t','\t','\n', encoding='utf-8')
|
||
|
|
||
|
def merge_css_into_xhtml(xhtml, css):
|
||
|
with FileOrPath(xhtml, 'r+') as xhtml:
|
||
|
with FileOrPath(css) as css:
|
||
|
try:
|
||
|
doc = minidom.parse(xhtml)
|
||
|
except Exception:
|
||
|
return False
|
||
|
head = doc.getElementsByTagName('head')[0]
|
||
|
links = head.getElementsByTagName('link')
|
||
|
for link in links:
|
||
|
if link.getAttribute('rel') == 'stylesheet':
|
||
|
head.removeChild(link)
|
||
|
style = head.appendChild(doc.createElement('style'))
|
||
|
style.setAttribute('type','text/css')
|
||
|
style.appendChild(doc.createTextNode(css.read()))
|
||
|
xhtml.truncate(0)
|
||
|
doc.writexml(xhtml, '\t','\t','\n', encoding='utf-8')
|
||
|
xhtml.seek(0)
|
||
|
return True
|
||
|
|
||
|
def export_to_xhtml(xml, xslt, css, columns, cmd='xsltproc --path "%(folder)s" "%(xslt)s" "%(xml)s"'):
|
||
|
folder = op.split(xml)[0]
|
||
|
output_column_xml(op.join(folder,'columns.xml'),columns)
|
||
|
html = StringIO()
|
||
|
cmd = cmd % {'folder': folder, 'xslt': xslt, 'xml': xml}
|
||
|
html.write(os.popen(cmd).read())
|
||
|
html.seek(0)
|
||
|
merge_css_into_xhtml(html,css)
|
||
|
html.seek(0)
|
||
|
html_path = op.join(folder,'export.htm')
|
||
|
html_file = open(html_path,'w')
|
||
|
html_file.write(html.read().encode('utf-8'))
|
||
|
html_file.close()
|
||
|
return html_path
|