dumper.py 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. '''
  2. Created on 26/06/2012
  3. @author: konstaa
  4. '''
  5. import core.api
  6. import re
  7. class Plugin(core.api.Plugin, core.api.Child, core.api.IConfigurable):
  8. POST_NAME = '.ss.std.code.dumper.html'
  9. def declare_configuration(self, parser):
  10. parser.add_option("--std.code.dumper.on", action="store_true", default=False,
  11. help="If the option is set (True), HTML files are generated for every parsed file containing code (for troubleshooting purposes only) [default: %default]")
  12. def configure(self, options):
  13. self.dump_html = options.__dict__['std.code.dumper.on']
  14. def initialize(self):
  15. if self.dump_html == True:
  16. core.api.subscribe_by_parents_interface(core.api.ICode, self)
  17. # do not process files dumped by previous run of this module
  18. self.get_plugin_loader().get_plugin('core.dir').add_exclude_rule(re.compile(r'.*' + Plugin.POST_NAME + r'$'))
  19. def callback(self, parent, data):
  20. file_name = data.get_path()
  21. text = data.get_content()
  22. import cgi
  23. f = open(file_name + Plugin.POST_NAME, 'w')
  24. f.write('<html><body><table><tr><td><pre>')
  25. last_pos = 0
  26. for marker in data.iterate_markers():
  27. f.write(cgi.escape(text[last_pos:marker.begin]))
  28. if marker.get_type() == data.get_marker_types().STRING:
  29. f.write('<span style="color:#0000FF">')
  30. elif marker.get_type() == data.get_marker_types().COMMENT:
  31. f.write('<span style="color:#009900">')
  32. elif marker.get_type() == data.get_marker_types().PREPROCESSOR:
  33. f.write('<span style="color:#990000">')
  34. f.write(cgi.escape(text[marker.begin:marker.end]))
  35. f.write('</span>')
  36. last_pos = marker.end
  37. f.write(cgi.escape(text[last_pos:]))
  38. last_pos = 0
  39. f.write('</pre></td><td><pre>')
  40. styles = ['<span style="background-color:#ffff80">', '<span style="background-color:#ff80ff">']
  41. for item in enumerate(data.iterate_regions(filter_group=data.get_region_types().FUNCTION)):
  42. reg = item[1]
  43. f.write(cgi.escape(text[last_pos:reg.get_offset_begin()]))
  44. f.write(styles[item[0] % 2])
  45. f.write('<a href="#line' + str(reg.get_cursor()) + '" id=line"' + str(reg.get_cursor()) + '"></a>')
  46. f.write(cgi.escape(text[reg.get_offset_begin():reg.get_offset_end()]))
  47. f.write('</span>')
  48. last_pos = reg.get_offset_end()
  49. f.write(cgi.escape(text[last_pos:]))
  50. f.write('</pre></td></tr></table></body></html>')
  51. f.close()