PageRenderTime 82ms CodeModel.GetById 31ms RepoModel.GetById 1ms app.codeStats 0ms

/pypy/tool/rest/rest.py

https://bitbucket.org/evelyn559/pypy
Python | 79 lines | 57 code | 7 blank | 15 comment | 11 complexity | 8182dce07f3be1584308619ab9480cd8 MD5 | raw file
    

  1. import py
    2. 

  2. import sys, os, traceback
    3. 

  3. import re
    4. 

  4. 

  5. if hasattr(sys.stdout, 'fileno') and os.isatty(sys.stdout.fileno()):
    6. 

  6.     def log(msg):
    7. 

  7.         print msg 
    8. 

  8. else:
    9. 

  9.     def log(msg):
    10. 

  10.         pass
    11. 

  11. 

  12. def convert_rest_html(source, source_path, stylesheet=None, encoding='latin1'):
    13. 

  13.     """ return html latin1-encoded document for the given input. 
    14. 

  14.         source  a ReST-string
    15. 

  15.         sourcepath where to look for includes (basically)
    16. 

  16.         stylesheet path (to be used if any)
    17. 

  17.     """
    18. 

  18.     from docutils.core import publish_string
    19. 

  19.     kwargs = {
    20. 

  20.         'stylesheet' : stylesheet, 
    21. 

  21.         'stylesheet_path': None,
    22. 

  22.         'traceback' : 1, 
    23. 

  23.         'embed_stylesheet': 0,
    24. 

  24.         'output_encoding' : encoding, 
    25. 

  25.         #'halt' : 0, # 'info',
    26. 

  26.         'halt_level' : 2, 
    27. 

  27.     }
    28. 

  28.     # docutils uses os.getcwd() :-(
    29. 

  29.     source_path = os.path.abspath(str(source_path))
    30. 

  30.     prevdir = os.getcwd()
    31. 

  31.     try:
    32. 

  32.         #os.chdir(os.path.dirname(source_path))
    33. 

  33.         return publish_string(source, source_path, writer_name='html',
    34. 

  34.                               settings_overrides=kwargs)
    35. 

  35.     finally:
    36. 

  36.         os.chdir(prevdir)
    37. 

  37. 

  38. def process(txtpath, encoding='latin1'):
    39. 

  39.     """ process a textfile """
    40. 

  40.     log("processing %s" % txtpath)
    41. 

  41.     assert txtpath.check(ext='.txt')
    42. 

  42.     if isinstance(txtpath, py.path.svnwc):
    43. 

  43.         txtpath = txtpath.localpath
    44. 

  44.     htmlpath = txtpath.new(ext='.html')
    45. 

  45.     #svninfopath = txtpath.localpath.new(ext='.svninfo')
    46. 

  46. 

  47.     style = txtpath.dirpath('style.css')
    48. 

  48.     if style.check():
    49. 

  49.         stylesheet = style.basename
    50. 

  50.     else:
    51. 

  51.         stylesheet = None
    52. 

  52.     content = unicode(txtpath.read(), encoding)
    53. 

  53.     doc = convert_rest_html(content, txtpath, stylesheet=stylesheet, encoding=encoding)
    54. 

  54.     htmlpath.write(doc)
    55. 

  55.     #log("wrote %r" % htmlpath)
    56. 

  56.     #if txtpath.check(svnwc=1, versioned=1): 
    57. 

  57.     #    info = txtpath.info()
    58. 

  58.     #    svninfopath.dump(info) 
    59. 

  59. 

  60. rex1 = re.compile(ur'.*<body>(.*)</body>.*', re.MULTILINE | re.DOTALL)
    61. 

  61. rex2 = re.compile(ur'.*<div class="document">(.*)</div>.*', re.MULTILINE | re.DOTALL)
    62. 

  62. 

  63. def strip_html_header(string, encoding='utf8'):
    64. 

  64.     """ return the content of the body-tag """ 
    65. 

  65.     uni = unicode(string, encoding)
    66. 

  66.     for rex in rex1,rex2: 
    67. 

  67.         match = rex.search(uni) 
    68. 

  68.         if not match: 
    69. 

  69.             break 
    70. 

  70.         uni = match.group(1) 
    71. 

  71.     return uni 
    72. 

  72. 

  73. class Project: # used for confrest.py files 
    74. 

  74.     def __init__(self, sourcepath):
    75. 

  75.         self.sourcepath = sourcepath
    76. 

  76.     def process(self, path):
    77. 

  77.         return process(path)
    78. 

  78.     def get_htmloutputpath(self, path):
    79. 

  79.         return path.new(ext='html')
    80. 

  80.