common replace rule, new coloring library

2016-06-04 20:41:08 +03:00
parent 4b57c0f416
commit c2deffa88f
3 changed files with 162 additions and 0 deletions
--- a/reporting/md-color.py
+++ b/reporting/md-color.py
@@ -0,0 +1,97 @@
+#!/usr/bin/env python
+
+import sys,os,re
+from argparse import ArgumentParser
+sys.path.append(os.path.dirname(os.path.realpath(__file__)))
+import bc
+
+__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
+__version__ = "0.1"
+
+
+def setup_options():
+    ''' Create command line options '''
+    usage='''
+Color notation renderer in ANSI codes
+Special syntaxes:
+    * Colors: insert string ${C}, where C is one of KRGBYMCWkrgbymcwSUZ
+
+'''
+ 
+    parser=ArgumentParser(description=usage,
+                          epilog=__author__)
+
+    parser.add_argument("-v","--version",action="version",version=__version__)
+    parser.add_argument("--no-color","-n",action="store_false",dest="color",default=True,
+                     help="Disable color.")
+    parser.add_argument("-z",action="store_true",dest="zero",default=False,
+                     help="Reset coloring at the end of each line.")
+    parser.add_argument("filename",type=str, 
+                     help="File to show, - for stdin")
+    opts=parser.parse_args()
+    return opts
+
+block_match={
+    'block_code': (re.compile(r'^( {4}[^*])(.*)$'),'{}\\1\\2','${c}'),
+    'multiline_code' : (re.compile(r'^ *(`{3,}|~{3,}) *(\S*)?'),'${c}\\1\\2','${c}'),  # ```lang
+    'block_quote': (re.compile(r'^ *>'),False),
+
+    'hrule': (re.compile(r'^ {0,3}[-*_]([-*_]){2,}$'),False),
+    'heading' : (re.compile(r'^ *(#{1,6}) *([^\n]+?) *#* *(?:\n+|$)'),'${W}\\1 \\2','${W}'),
+    'lheading' : (re.compile(r'^(=+|-+)$'),'${W}\\1','${W}'),
+    'list_bullet':(re.compile(r'^( *)([*+-]|\d+\.)( +)'),'\\1${y}\\2${Z}\\3'),
+    'text':       (re.compile(r'^[^\n]+'),False),
+    'empty':      (re.compile(r'^$'),False)
+}
+blocks=['block_code', 'block_quote', 'multiline_code','hrule', 'heading','lheading','list_bullet', 'text', 'empty']
+
+inline_match={
+    'code': (re.compile(r'(`+)([^`]+)(`+)'),'${c}\\1\\2\\3')  # `code`
+}
+inlines=['code']
+
+
+bc=bc.ansi()
+opts=setup_options()
+if opts.filename=="-":
+    f=sys.stdin
+else:
+    f=open(opts.filename,'r')
+block='text'
+new_block='text'
+multiline_block=False
+for row in f:
+    if not row:
+        continue
+    row=row.decode('utf-8').rstrip("\n\r ")
+    for match in blocks:
+        if block_match[match][0].match(row):
+            new_block=match
+            if match.startswith('multiline'):
+                if multiline_block:
+                    multiline_block=False
+                else:
+                    multiline_block=match
+            break
+    if multiline_block: 
+        new_block=multiline_block
+        
+    if new_block!=block:
+        block=new_block
+        if not multiline_block:
+            sys.stdout.write(bc.Z)
+    #print(multiline_block)
+    #print(block)
+    if block_match[block][1]:
+        row=block_match[block][0].sub(block_match[block][1],row)
+    #print(row)
+    if opts.color:
+        colored=bc.color_string(row)
+    else:
+        colored=bc.nocolor_string(row)
+    sys.stdout.write(colored.encode('utf-8'))
+    if opts.zero:
+        sys.stdout.write(bc.Z)
+    sys.stdout.write("\n")
+
+