python3inizing scripts
This commit is contained in:
Ville Rantanen
@@ -1,6 +1,7 @@
|
||||
# Python library for ansi colorization
|
||||
import re, sys
|
||||
|
||||
|
||||
class code:
|
||||
K = "\033[1;30m"
|
||||
R = "\033[1;31m"
|
||||
@@ -29,27 +30,68 @@ class code:
|
||||
bc = "\033[46m"
|
||||
bw = "\033[47m"
|
||||
|
||||
S = '\033[1m'#strong
|
||||
s = '\033[2m'#strong off
|
||||
U = '\033[4m'#underline
|
||||
u = '\033[24m'#underline off
|
||||
Z = '\033[0m'#zero colors
|
||||
ic = '\033[7m'#inverse colors
|
||||
io = '\033[27m'#inverse off
|
||||
st = '\033[9m'#strike on
|
||||
so = '\033[29m'#strike off
|
||||
CLR = '\033[2J'
|
||||
CLREND = '\033[K'
|
||||
CLRBEG = '\033[1K'
|
||||
S = "\033[1m" # strong
|
||||
s = "\033[2m" # strong off
|
||||
U = "\033[4m" # underline
|
||||
u = "\033[24m" # underline off
|
||||
Z = "\033[0m" # zero colors
|
||||
ic = "\033[7m" # inverse colors
|
||||
io = "\033[27m" # inverse off
|
||||
st = "\033[9m" # strike on
|
||||
so = "\033[29m" # strike off
|
||||
CLR = "\033[2J"
|
||||
CLREND = "\033[K"
|
||||
CLRBEG = "\033[1K"
|
||||
CLRSCR = CLR + "\033[0;0H"
|
||||
|
||||
color_keys="K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR".split(",")
|
||||
color_list=[K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR]
|
||||
custom_match=re.compile(r'(\${)([0-9;]*[ABCDEFGHJKSTfminsu]+)(})')
|
||||
color_keys = "K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR".split(
|
||||
","
|
||||
)
|
||||
color_list = [
|
||||
K,
|
||||
R,
|
||||
G,
|
||||
B,
|
||||
Y,
|
||||
M,
|
||||
C,
|
||||
W,
|
||||
k,
|
||||
r,
|
||||
g,
|
||||
b,
|
||||
y,
|
||||
m,
|
||||
c,
|
||||
w,
|
||||
S,
|
||||
s,
|
||||
U,
|
||||
u,
|
||||
Z,
|
||||
ic,
|
||||
io,
|
||||
st,
|
||||
so,
|
||||
bk,
|
||||
br,
|
||||
bg,
|
||||
by,
|
||||
bb,
|
||||
bm,
|
||||
bc,
|
||||
bw,
|
||||
CLR,
|
||||
CLREND,
|
||||
CLRBEG,
|
||||
CLRSCR,
|
||||
]
|
||||
custom_match = re.compile(r"(\${)([0-9;]*[ABCDEFGHJKSTfminsu]+)(})")
|
||||
|
||||
def pos(self, y, x):
|
||||
""" Go to absolute position """
|
||||
return "\033[" + str(y) + ";" + str(x) + "H"
|
||||
|
||||
def column(self, x):
|
||||
""" Go to absolute column """
|
||||
return "\033[" + str(x) + "G"
|
||||
@@ -61,27 +103,35 @@ class code:
|
||||
|
||||
def clear(self):
|
||||
sys.stdout.write(self.CLRSCR + self.pos(0, 0))
|
||||
|
||||
def clear_to_end(self):
|
||||
sys.stdout.write(self.CLREND)
|
||||
|
||||
def clear_to_beginning(self):
|
||||
sys.stdout.write(self.CLRBEG)
|
||||
|
||||
def up(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "A")
|
||||
|
||||
def down(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "B")
|
||||
|
||||
def right(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "C")
|
||||
|
||||
def left(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "D")
|
||||
|
||||
def up_line(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "F")
|
||||
|
||||
def down_line(self, n=1):
|
||||
sys.stdout.write("\033[" + str(n) + "E")
|
||||
|
||||
def save(self):
|
||||
""" Save cursor position """
|
||||
sys.stdout.write("\033[s")
|
||||
|
||||
def restore(self):
|
||||
""" Restore cursor position """
|
||||
sys.stdout.write("\033[u")
|
||||
@@ -90,15 +140,17 @@ class code:
|
||||
for i, c in enumerate(self.color_keys):
|
||||
s = s.replace("${" + c + "}", self.color_list[i])
|
||||
return self.custom_color(s)
|
||||
|
||||
def nocolor_string(self, s):
|
||||
for i, c in enumerate(self.color_keys):
|
||||
s = s.replace("${" + c + "}", "")
|
||||
return self.custom_nocolor(s)
|
||||
|
||||
def custom_color(self, s):
|
||||
return self.custom_match.sub('\033[\\2',s)
|
||||
return self.custom_match.sub("\033[\\2", s)
|
||||
|
||||
def custom_nocolor(self, s):
|
||||
return self.custom_match.sub('',s)
|
||||
return self.custom_match.sub("", s)
|
||||
|
||||
def get_keys(self):
|
||||
return self.color_keys
|
||||
@@ -142,4 +194,4 @@ ${S}=====================${Z}
|
||||
C .right() Right y;xH .pos() Absolute Position
|
||||
D .left() Left """
|
||||
|
||||
return(c.color_string(unformatted))
|
||||
return c.color_string(unformatted)
|
||||
|
||||
@@ -1,36 +1,31 @@
|
||||
#!/usr/bin/env python
|
||||
#!/usr/bin/env python3
|
||||
import sys, os
|
||||
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
import ansicodes
|
||||
from argparse import ArgumentParser
|
||||
|
||||
|
||||
def setup_options():
|
||||
parser = ArgumentParser()
|
||||
parser.add_argument(
|
||||
"-v",
|
||||
default = False,
|
||||
action = "store_true"
|
||||
)
|
||||
parser.add_argument("-v", default=False, action="store_true")
|
||||
parser.add_argument(
|
||||
"cs",
|
||||
type=str,
|
||||
default="cp437",
|
||||
nargs = '?',
|
||||
help = "Character set to show, ex. ascii, cp437, cp850, cp1250, latin1 ..."
|
||||
nargs="?",
|
||||
help="Character set to show, ex. ascii, cp437, cp850, cp1250, latin1 ...",
|
||||
)
|
||||
opts = parser.parse_args()
|
||||
return opts
|
||||
|
||||
|
||||
def long_format(table):
|
||||
bc = ansicodes.code()
|
||||
for x, col in enumerate(table):
|
||||
for y, row in enumerate(col):
|
||||
bc.posprint(
|
||||
y + 2,
|
||||
1 + x*7,
|
||||
bc.color_string(
|
||||
"${c}%03d: ${G}%s${Z}"%( row )
|
||||
)
|
||||
y + 2, 1 + x * 7, bc.color_string("${c}%03d: ${G}%s${Z}" % (row))
|
||||
)
|
||||
|
||||
|
||||
@@ -38,22 +33,18 @@ def short_format(table):
|
||||
bc = ansicodes.code()
|
||||
for x, col in enumerate(table):
|
||||
for y, row in enumerate(col):
|
||||
bc.posprint(
|
||||
x + 3,
|
||||
y * 2 + 1,
|
||||
bc.color_string(
|
||||
"${G}%s${Z}"%( row[1], )
|
||||
)
|
||||
)
|
||||
print('')
|
||||
bc.posprint(x + 3, y * 2 + 1, bc.color_string("${G}%s${Z}" % (row[1],)))
|
||||
print("")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
opts = setup_options()
|
||||
table = [[] for x in range(8)]
|
||||
for c in range(0, 256):
|
||||
col = c/32
|
||||
table[col].append((c,chr(c).decode(opts.cs,'ignore').encode('utf-8','ignore')))
|
||||
col = int(c / 32)
|
||||
table[col].append(
|
||||
(c, chr(c))
|
||||
)
|
||||
# sys.stdout.write(chr(c).decode('latin1'))
|
||||
# remove control chars
|
||||
table.pop(0)
|
||||
@@ -68,5 +59,4 @@ if __name__ == "__main__":
|
||||
long_format(table)
|
||||
else:
|
||||
short_format(table)
|
||||
print('')
|
||||
|
||||
print("")
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
#!/usr/bin/env python
|
||||
#!/usr/bin/env python3
|
||||
import sys
|
||||
import os
|
||||
import pwd
|
||||
@@ -6,13 +6,15 @@ from argparse import ArgumentParser
|
||||
|
||||
MINSIZE = 0
|
||||
|
||||
|
||||
def setup_options():
|
||||
parser = ArgumentParser(description="List file usage")
|
||||
parser.add_argument('startpath', action="append", nargs='+')
|
||||
parser.add_argument("startpath", action="append", nargs="+")
|
||||
|
||||
options = parser.parse_args()
|
||||
return options
|
||||
|
||||
|
||||
def add_recurse(db, options, start):
|
||||
for path, dirs, files in os.walk(start, followlinks=False):
|
||||
files = clean_syms(files, path)
|
||||
@@ -26,6 +28,7 @@ def add_recurse(db,options,start):
|
||||
db[owner] += size
|
||||
return db
|
||||
|
||||
|
||||
def clean_syms(files, path):
|
||||
nonsyms = []
|
||||
for f in files:
|
||||
@@ -33,6 +36,7 @@ def clean_syms(files,path):
|
||||
nonsyms.append(f)
|
||||
return nonsyms
|
||||
|
||||
|
||||
def print_db(db):
|
||||
namedb = dict()
|
||||
for u in db:
|
||||
@@ -40,15 +44,16 @@ def print_db(db):
|
||||
namedb[pwd.getpwuid(u).pw_name] = humanize_size(db[u])
|
||||
except:
|
||||
namedb[str(u)] = humanize_size(db[u])
|
||||
names=namedb.keys()
|
||||
names = list(namedb.keys())
|
||||
names.sort()
|
||||
for n in names:
|
||||
print(n + ": " + namedb[n])
|
||||
|
||||
|
||||
def humanize_size(size, precision=1):
|
||||
if size == None:
|
||||
return 'nan'
|
||||
suffixes=['B','KB','MB','GB','TB']
|
||||
return "nan"
|
||||
suffixes = ["B", "KB", "MB", "GB", "TB"]
|
||||
suffixIndex = 0
|
||||
defPrecision = 0
|
||||
while size > 1024:
|
||||
@@ -57,13 +62,14 @@ def humanize_size(size,precision=1):
|
||||
defPrecision = precision
|
||||
return "%.*f%s" % (defPrecision, size, suffixes[suffixIndex])
|
||||
|
||||
|
||||
def main():
|
||||
options=setup_options();
|
||||
options = setup_options()
|
||||
db = dict()
|
||||
for f in options.startpath[0]:
|
||||
db = add_recurse(db, options, f)
|
||||
print_db(db)
|
||||
sys.exit(0)
|
||||
|
||||
main()
|
||||
|
||||
main()
|
||||
|
||||
@@ -17,7 +17,7 @@
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
|
||||
'''Markslider: a slideshow engine based on markdown.'''
|
||||
"""Markslider: a slideshow engine based on markdown."""
|
||||
|
||||
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
||||
__version__ = "1.3"
|
||||
@@ -25,12 +25,14 @@ __version__ = "1.3"
|
||||
import sys, os, argparse, re, datetime
|
||||
from argparse import ArgumentParser
|
||||
import traceback, tty, termios, subprocess, signal
|
||||
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
import ansicodes, md_color
|
||||
|
||||
HL = ">"
|
||||
EOS = "# End of Slides"
|
||||
|
||||
|
||||
class getch:
|
||||
def get(self):
|
||||
fd = sys.stdin.fileno()
|
||||
@@ -44,11 +46,14 @@ class getch:
|
||||
|
||||
|
||||
class EndProgram(Exception):
|
||||
''' Nice exit '''
|
||||
""" Nice exit """
|
||||
|
||||
pass
|
||||
|
||||
|
||||
class slide_reader:
|
||||
""" Class for reading files. """
|
||||
|
||||
def __init__(self, files, opts):
|
||||
self.filename = files[0]
|
||||
self.files = files
|
||||
@@ -71,16 +76,12 @@ class slide_reader:
|
||||
self.pygments = False
|
||||
if opts.syntax_pygments:
|
||||
try:
|
||||
self.pygments = Pygmentizer(
|
||||
opts.syntax_formatter,
|
||||
opts.syntax_style
|
||||
)
|
||||
self.pygments = Pygmentizer(opts.syntax_formatter, opts.syntax_style)
|
||||
except ImportError as e:
|
||||
self.pygments = False
|
||||
|
||||
|
||||
def read(self):
|
||||
''' Read a file, set pages and data '''
|
||||
""" Read a file, set pages and data """
|
||||
|
||||
self.pages = 0
|
||||
self.background = []
|
||||
@@ -89,7 +90,7 @@ class slide_reader:
|
||||
first_slide_found = False
|
||||
for fname in self.files:
|
||||
first_slide_found = False
|
||||
f=open(fname,'r')
|
||||
f = open(fname, "r")
|
||||
new_page = []
|
||||
in_code = False
|
||||
for row in f:
|
||||
@@ -131,11 +132,13 @@ class slide_reader:
|
||||
|
||||
def get_data(self):
|
||||
return self.data
|
||||
|
||||
def get_current_filename(self):
|
||||
for i, offset in enumerate(self.file_start_page):
|
||||
if offset > self.page:
|
||||
return self.files[i]
|
||||
return "NA"
|
||||
|
||||
def get_filename(self):
|
||||
return self.filename
|
||||
|
||||
@@ -153,6 +156,7 @@ class slide_reader:
|
||||
|
||||
def get_page_no(self):
|
||||
return self.page
|
||||
|
||||
def inc_page_no(self, inc=1, loop=False):
|
||||
self.page += inc
|
||||
if self.page < 0:
|
||||
@@ -165,16 +169,22 @@ class slide_reader:
|
||||
self.page = 0
|
||||
self.width = max([len(x) for x in self.data[self.page]])
|
||||
self.height = len(self.data[self.page])
|
||||
|
||||
def last_page(self):
|
||||
self.page = self.pages - 1
|
||||
|
||||
def first_page(self):
|
||||
self.page = 0
|
||||
|
||||
def get_page_height(self):
|
||||
return self.height
|
||||
|
||||
def get_page_width(self):
|
||||
return self.width
|
||||
|
||||
def get_max_height(self):
|
||||
return self.max_height
|
||||
|
||||
def get_max_width(self):
|
||||
return self.max_width
|
||||
|
||||
@@ -194,14 +204,21 @@ class slide_reader:
|
||||
if re.search("^##[^#]", line):
|
||||
subh = [subh[0] + 1, 0, 0]
|
||||
TOC.append(" %d.%d. %s" % (h1 + 1, subh[0], title))
|
||||
if self.opts.toc_depth==2: continue
|
||||
if self.opts.toc_depth == 2:
|
||||
continue
|
||||
if re.search("^###[^#]", line):
|
||||
subh = [subh[0], subh[1] + 1, 0]
|
||||
TOC.append(" %d.%d.%d. %s"%(h1+1,subh[0],subh[1],title))
|
||||
if self.opts.toc_depth==3: continue
|
||||
TOC.append(
|
||||
" %d.%d.%d. %s" % (h1 + 1, subh[0], subh[1], title)
|
||||
)
|
||||
if self.opts.toc_depth == 3:
|
||||
continue
|
||||
if re.search("^####[^#]", line):
|
||||
subh = [subh[0], subh[1], subh[2] + 1]
|
||||
TOC.append(" %d.%d.%d.%d. %s"%(h1+1,subh[0],subh[1],subh[2],title))
|
||||
TOC.append(
|
||||
" %d.%d.%d.%d. %s"
|
||||
% (h1 + 1, subh[0], subh[1], subh[2], title)
|
||||
)
|
||||
return TOC
|
||||
|
||||
def toc(self):
|
||||
@@ -227,18 +244,17 @@ class slide_reader:
|
||||
"""Launch in a string using tags `command`>
|
||||
Remove empty lines from beginning and end of stdout.
|
||||
"""
|
||||
output = subprocess.check_output(
|
||||
command.group(2).strip(),
|
||||
shell = True
|
||||
)
|
||||
output = subprocess.check_output(command.group(2).strip(), shell=True)
|
||||
if type(output) == bytes:
|
||||
output = output.decode('utf-8')
|
||||
output = output.decode("utf-8")
|
||||
output = output.split("\n")
|
||||
while len(output[0].strip()) == 0:
|
||||
if len(output)==1: return [""]
|
||||
if len(output) == 1:
|
||||
return [""]
|
||||
del output[0]
|
||||
while len(output[-1].strip()) == 0:
|
||||
if len(output)==1: return [""]
|
||||
if len(output) == 1:
|
||||
return [""]
|
||||
del output[-1]
|
||||
return_value = [command.group(1)]
|
||||
return_value.extend(output)
|
||||
@@ -254,16 +270,18 @@ class slide_reader:
|
||||
# ~ 3=image command
|
||||
output = subprocess.check_output(
|
||||
"convert %s JPEG:- | jp2a --colors --width=70 -" % (image.group(3),),
|
||||
shell=True
|
||||
shell=True,
|
||||
)
|
||||
if type(output) == bytes:
|
||||
output = output.decode('utf-8')
|
||||
output = output.decode("utf-8")
|
||||
output = output.split("\n")
|
||||
while len(output[0].strip()) == 0:
|
||||
if len(output)==1: return [""]
|
||||
if len(output) == 1:
|
||||
return [""]
|
||||
del output[0]
|
||||
while len(output[-1].strip()) == 0:
|
||||
if len(output)==1: return [""]
|
||||
if len(output) == 1:
|
||||
return [""]
|
||||
del output[-1]
|
||||
return_value = [image.group(1)]
|
||||
return_value.extend(output)
|
||||
@@ -288,12 +306,14 @@ class slide_reader:
|
||||
if titles[page[0]] > 1:
|
||||
page[0] += " [%d/%d]" % (page_no, titles[page[0]])
|
||||
|
||||
class Pygmentizer():
|
||||
|
||||
class Pygmentizer:
|
||||
def __init__(self, formatter, style):
|
||||
import pygments
|
||||
import pygments.lexers
|
||||
import pygments.formatters
|
||||
import pygments.styles
|
||||
|
||||
self.pygments = pygments
|
||||
self.lexers = pygments.lexers
|
||||
self.formatters = pygments.formatters
|
||||
@@ -301,12 +321,10 @@ class Pygmentizer():
|
||||
|
||||
self.style = self.styles.get_style_by_name(style)
|
||||
self.formatter = self.formatters.get_formatter_by_name(
|
||||
formatter,
|
||||
style = self.style
|
||||
formatter, style=self.style
|
||||
)
|
||||
self.lexer = None
|
||||
|
||||
|
||||
def format(self, code):
|
||||
in_block = False
|
||||
blocks = []
|
||||
@@ -318,11 +336,7 @@ class Pygmentizer():
|
||||
if len(lexer_name) == 0:
|
||||
in_block = False
|
||||
if in_block:
|
||||
blocks.append({
|
||||
"lexer": lexer_name,
|
||||
"code": [],
|
||||
"rows": []
|
||||
})
|
||||
blocks.append({"lexer": lexer_name, "code": [], "rows": []})
|
||||
current_block += 1
|
||||
else:
|
||||
if in_block:
|
||||
@@ -343,7 +357,7 @@ class Pygmentizer():
|
||||
|
||||
|
||||
def get_interactive_help_text():
|
||||
return ''' left/right,page up/down,home,end
|
||||
return """ left/right,page up/down,home,end
|
||||
change page
|
||||
c list contents (toc)
|
||||
M modify file with VIM
|
||||
@@ -354,11 +368,13 @@ def get_interactive_help_text():
|
||||
,/. scroll page
|
||||
up/down move highlight
|
||||
enter execute highlighted line
|
||||
h help'''
|
||||
h help"""
|
||||
|
||||
|
||||
def setup_options():
|
||||
''' Create command line options '''
|
||||
usage='''
|
||||
""" Create command line options """
|
||||
usage = (
|
||||
"""
|
||||
MarkSlider: a markdown based slideshow engine
|
||||
Special syntaxes:
|
||||
* Colors: insert string ${C}, where C is one of KRGBYMCWkrgbymcwSUZ
|
||||
@@ -369,67 +385,171 @@ Special syntaxes:
|
||||
* Convert images to ASCII, with jp2a: >
|
||||
|
||||
Keyboard shortcuts:
|
||||
'''+get_interactive_help_text()
|
||||
"""
|
||||
+ get_interactive_help_text()
|
||||
)
|
||||
|
||||
parser=ArgumentParser(description=usage,
|
||||
parser = ArgumentParser(
|
||||
description=usage,
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
epilog=__author__)
|
||||
epilog=__author__,
|
||||
)
|
||||
parser.add_argument("-v", "--version", action="version", version=__version__)
|
||||
parser.add_argument("--export",action="store",dest="screenshots",default=False,
|
||||
type=str, help="Take screenshots of the slideshow in the given folder.")
|
||||
parser.add_argument(
|
||||
"--export",
|
||||
action="store",
|
||||
dest="screenshots",
|
||||
default=False,
|
||||
type=str,
|
||||
help="Take screenshots of the slideshow in the given folder.",
|
||||
)
|
||||
|
||||
content = parser.add_argument_group('content')
|
||||
execution = parser.add_argument_group('execution')
|
||||
control = parser.add_argument_group('controls')
|
||||
content = parser.add_argument_group("content")
|
||||
execution = parser.add_argument_group("execution")
|
||||
control = parser.add_argument_group("controls")
|
||||
|
||||
content.add_argument("--background",action="store_true",dest="background",default = False,
|
||||
help="Use the rows before the first # header as slide background")
|
||||
content.add_argument("--center",action="store_true",dest="center",default=False,
|
||||
help="Center slides on screen.")
|
||||
content.add_argument("--dc",action="store_true",dest="dark_colors",default=False,
|
||||
help="Use dark colorscheme, better for white background terminals.")
|
||||
content.add_argument("-m",action="store_false",dest="autocolor",default=True,
|
||||
help="Disable color by markdown structure.")
|
||||
content.add_argument("--no-color","-n",action="store_false",dest="color",default=True,
|
||||
help="Disable color.")
|
||||
content.add_argument("--no-rename","--nr",action="store_false",dest="rename_title", default=True,
|
||||
help="Disable automatic renaming of duplicate titles.")
|
||||
content.add_argument("--toc",action="store",dest="toc",default=False,
|
||||
const="Table of Contents", type=str, nargs='?',
|
||||
help="Insert table of contents. Define the header, or use default: %(const)s")
|
||||
content.add_argument("--toc-page",action="store",dest="toc_page",default=2, type=int,
|
||||
help="Insert table of contents on a chosen page. default: %(const)s")
|
||||
content.add_argument("--toc-depth",action="store",dest="toc_depth",default=2, type=int,
|
||||
content.add_argument(
|
||||
"--background",
|
||||
action="store_true",
|
||||
dest="background",
|
||||
default=False,
|
||||
help="Use the rows before the first # header as slide background",
|
||||
)
|
||||
content.add_argument(
|
||||
"--center",
|
||||
action="store_true",
|
||||
dest="center",
|
||||
default=False,
|
||||
help="Center slides on screen.",
|
||||
)
|
||||
content.add_argument(
|
||||
"--dc",
|
||||
action="store_true",
|
||||
dest="dark_colors",
|
||||
default=False,
|
||||
help="Use dark colorscheme, better for white background terminals.",
|
||||
)
|
||||
content.add_argument(
|
||||
"-m",
|
||||
action="store_false",
|
||||
dest="autocolor",
|
||||
default=True,
|
||||
help="Disable color by markdown structure.",
|
||||
)
|
||||
content.add_argument(
|
||||
"--no-color",
|
||||
"-n",
|
||||
action="store_false",
|
||||
dest="color",
|
||||
default=True,
|
||||
help="Disable color.",
|
||||
)
|
||||
content.add_argument(
|
||||
"--no-rename",
|
||||
"--nr",
|
||||
action="store_false",
|
||||
dest="rename_title",
|
||||
default=True,
|
||||
help="Disable automatic renaming of duplicate titles.",
|
||||
)
|
||||
content.add_argument(
|
||||
"--toc",
|
||||
action="store",
|
||||
dest="toc",
|
||||
default=False,
|
||||
const="Table of Contents",
|
||||
type=str,
|
||||
nargs="?",
|
||||
help="Insert table of contents. Define the header, or use default: %(const)s",
|
||||
)
|
||||
content.add_argument(
|
||||
"--toc-page",
|
||||
action="store",
|
||||
dest="toc_page",
|
||||
default=2,
|
||||
type=int,
|
||||
help="Insert table of contents on a chosen page. default: %(const)s",
|
||||
)
|
||||
content.add_argument(
|
||||
"--toc-depth",
|
||||
action="store",
|
||||
dest="toc_depth",
|
||||
default=2,
|
||||
type=int,
|
||||
choices=list(range(1, 5)),
|
||||
help="Table of contents display depth. default: %(const)s")
|
||||
help="Table of contents display depth. default: %(const)s",
|
||||
)
|
||||
|
||||
content.add_argument("--no-pygments",action="store_false",dest="syntax_pygments",default = True,
|
||||
help="Disable autocoloring syntax with Pygments. Used only for ``` code blocks, if language is mentioned, ex.: ```python . Note! Do not have empty lines after ```")
|
||||
content.add_argument("--syntax-formatter",action="store",dest="syntax_formatter",default = "terminal256",
|
||||
choices=['terminal', 'terminal256', 'terminal16m'],
|
||||
help="Syntax highlighter formatter. default: %(default)s")
|
||||
content.add_argument("--syntax-style",action="store",dest="syntax_style",default = "monokai",
|
||||
help="Syntax highlighter style, see Pygments styles. default: %(default)s")
|
||||
content.add_argument(
|
||||
"--no-pygments",
|
||||
action="store_false",
|
||||
dest="syntax_pygments",
|
||||
default=True,
|
||||
help="Disable autocoloring syntax with Pygments. Used only for ``` code blocks, if language is mentioned, ex.: ```python . Note! Do not have empty lines after ```",
|
||||
)
|
||||
content.add_argument(
|
||||
"--syntax-formatter",
|
||||
action="store",
|
||||
dest="syntax_formatter",
|
||||
default="terminal256",
|
||||
choices=["terminal", "terminal256", "terminal16m"],
|
||||
help="Syntax highlighter formatter. default: %(default)s",
|
||||
)
|
||||
content.add_argument(
|
||||
"--syntax-style",
|
||||
action="store",
|
||||
dest="syntax_style",
|
||||
default="monokai",
|
||||
help="Syntax highlighter style, see Pygments styles. default: %(default)s",
|
||||
)
|
||||
|
||||
execution.add_argument(
|
||||
"-e",
|
||||
action="store_true",
|
||||
dest="execute",
|
||||
default=False,
|
||||
help="Execute commands in `! or `> tags at show time with Enter key. WARNING: Potentially very dangerous to run others' slides with this switch!",
|
||||
)
|
||||
execution.add_argument(
|
||||
"-E",
|
||||
action="store_true",
|
||||
dest="execute_read",
|
||||
default=False,
|
||||
help="Execute commands in ``> tags at file read time. WARNING: Potentially very dangerous to run others' slides with this switch!",
|
||||
)
|
||||
|
||||
execution.add_argument("-e",action="store_true",dest="execute",default=False,
|
||||
help="Execute commands in `! or `> tags at show time with Enter key. WARNING: Potentially very dangerous to run others' slides with this switch!")
|
||||
execution.add_argument("-E",action="store_true",dest="execute_read",default=False,
|
||||
help="Execute commands in ``> tags at file read time. WARNING: Potentially very dangerous to run others' slides with this switch!")
|
||||
control.add_argument(
|
||||
"--exit",
|
||||
action="store_true",
|
||||
dest="exit_last",
|
||||
default=False,
|
||||
help="Exit after last slide.",
|
||||
)
|
||||
|
||||
control.add_argument("--exit",action="store_true",dest="exit_last",default=False,
|
||||
help="Exit after last slide.")
|
||||
control.add_argument(
|
||||
"-s",
|
||||
action="store_false",
|
||||
dest="menu",
|
||||
default=True,
|
||||
help="Disable status bar.",
|
||||
)
|
||||
control.add_argument(
|
||||
"--timer",
|
||||
action="store",
|
||||
dest="slideTimer",
|
||||
default=False,
|
||||
type=int,
|
||||
help="Timer for slideshow. If set, starts automatic slide changing.",
|
||||
)
|
||||
content.add_argument(
|
||||
"-w",
|
||||
action="store_false",
|
||||
dest="wrap",
|
||||
default=True,
|
||||
help="Disable line wrapping. Cuts long lines.",
|
||||
)
|
||||
|
||||
control.add_argument("-s",action="store_false",dest="menu",default=True,
|
||||
help="Disable status bar.")
|
||||
control.add_argument("--timer",action="store",dest="slideTimer",default=False, type=int,
|
||||
help="Timer for slideshow. If set, starts automatic slide changing.")
|
||||
content.add_argument("-w",action="store_false",dest="wrap",default=True,
|
||||
help="Disable line wrapping. Cuts long lines.")
|
||||
|
||||
|
||||
parser.add_argument("files",type=str, nargs='+',
|
||||
help="File(s) to show")
|
||||
parser.add_argument("files", type=str, nargs="+", help="File(s) to show")
|
||||
opts = parser.parse_args()
|
||||
opts.slideShow = not not opts.slideTimer
|
||||
if opts.screenshots:
|
||||
@@ -440,7 +560,7 @@ Keyboard shortcuts:
|
||||
|
||||
|
||||
def page_print(reader, opts, offset):
|
||||
''' Print a page '''
|
||||
""" Print a page """
|
||||
|
||||
page = reader.get_current_page()
|
||||
scrsize = opts.size
|
||||
@@ -459,23 +579,21 @@ def page_print(reader,opts,offset):
|
||||
coloring = "${b}${U}"
|
||||
else:
|
||||
coloring = "${U}${Y}"
|
||||
print(
|
||||
align_pad +
|
||||
colorify(
|
||||
coloring + page[0],
|
||||
opts) +
|
||||
bc.Z
|
||||
)
|
||||
print(align_pad + colorify(coloring + page[0], opts) + bc.Z)
|
||||
if opts.background:
|
||||
bc.save()
|
||||
if opts.color:
|
||||
sys.stdout.write("\n".join([align_pad + bc.color_string(x) for x in reader.background]))
|
||||
sys.stdout.write(
|
||||
"\n".join([align_pad + bc.color_string(x) for x in reader.background])
|
||||
)
|
||||
else:
|
||||
sys.stdout.write("\n".join([align_pad + bc.nocolor_string(x) for x in reader.background]))
|
||||
sys.stdout.write(
|
||||
"\n".join([align_pad + bc.nocolor_string(x) for x in reader.background])
|
||||
)
|
||||
bc.restore()
|
||||
if (sys.version_info < (3, 0)):
|
||||
if sys.version_info < (3, 0):
|
||||
# python2 magic
|
||||
page = [row.decode('utf-8') for row in page]
|
||||
page = [row.decode("utf-8") for row in page]
|
||||
# Print page rows
|
||||
if not opts.wrap:
|
||||
page = [cut_line(row, scrsize[1] - 1) for row in page]
|
||||
@@ -495,13 +613,9 @@ def page_print(reader,opts,offset):
|
||||
)
|
||||
|
||||
for row_i in preformatted_rows:
|
||||
parsed[row_i][0] = 'preformatted'
|
||||
parsed[row_i][0] = "preformatted"
|
||||
parsed[row_i][1] = preformatted[row_i]
|
||||
colored = md_color.colorize(
|
||||
parsed,
|
||||
not opts.color,
|
||||
opts.dark_colors
|
||||
)
|
||||
colored = md_color.colorize(parsed, not opts.color, opts.dark_colors)
|
||||
else:
|
||||
if opts.color:
|
||||
colored = [bc.color_string(row[1]) for row in parsed]
|
||||
@@ -509,8 +623,10 @@ def page_print(reader,opts,offset):
|
||||
colored = [bc.nocolor_string(row[1]) for row in parsed]
|
||||
r = 0
|
||||
for row_i in range(len(page)):
|
||||
if row_i == 0: continue
|
||||
if row_i < offset[0]: continue
|
||||
if row_i == 0:
|
||||
continue
|
||||
if row_i < offset[0]:
|
||||
continue
|
||||
row = page[row_i]
|
||||
# page[1+offset[0]:]:
|
||||
if opts.wrap:
|
||||
@@ -533,13 +649,20 @@ def page_print(reader,opts,offset):
|
||||
|
||||
def print_menu(reader, opts):
|
||||
|
||||
bc.posprint( opts.size[0], 0,
|
||||
colorify("${y}%d${Z}/%d %s|%s"%(
|
||||
bc.posprint(
|
||||
opts.size[0],
|
||||
0,
|
||||
colorify(
|
||||
"${y}%d${Z}/%d %s|%s"
|
||||
% (
|
||||
1 + reader.get_page_no(),
|
||||
reader.get_pages(),
|
||||
os.path.basename(reader.get_current_filename()),
|
||||
"slideshow" if opts.slideShow else ""),
|
||||
opts))
|
||||
"slideshow" if opts.slideShow else "",
|
||||
),
|
||||
opts,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
def print_time(opts):
|
||||
@@ -547,22 +670,22 @@ def print_time(opts):
|
||||
bc.posprint(
|
||||
opts.size[0],
|
||||
opts.size[1] - 5,
|
||||
colorify(
|
||||
"%02d:%02d"%(
|
||||
now.hour,
|
||||
now.minute
|
||||
),
|
||||
opts
|
||||
)
|
||||
colorify("%02d:%02d" % (now.hour, now.minute), opts),
|
||||
)
|
||||
|
||||
|
||||
def print_help(reader, opts):
|
||||
''' Create a window with help message '''
|
||||
helptext=get_interactive_help_text().split('\n')
|
||||
""" Create a window with help message """
|
||||
helptext = get_interactive_help_text().split("\n")
|
||||
maxlen = max([len(x) for x in helptext])
|
||||
bc.posprint(3, 5, " +" + "-" * maxlen + "+ ")
|
||||
bc.posprint(4,5,colorify(" |${U}${Y}"+("{:^"+str(maxlen)+"}").format("Help")+"${Z}| ",opts))
|
||||
bc.posprint(
|
||||
4,
|
||||
5,
|
||||
colorify(
|
||||
" |${U}${Y}" + ("{:^" + str(maxlen) + "}").format("Help") + "${Z}| ", opts
|
||||
),
|
||||
)
|
||||
for y, row in enumerate(helptext):
|
||||
bc.posprint(5 + y, 5, (" |{:<" + str(maxlen) + "}| ").format(row))
|
||||
bc.posprint(6 + y, 5, " +" + "-" * maxlen + "+ ")
|
||||
@@ -571,7 +694,7 @@ def print_help(reader,opts):
|
||||
|
||||
|
||||
def print_toc(reader, opts):
|
||||
''' Create a window with TOC '''
|
||||
""" Create a window with TOC """
|
||||
text = reader.get_toc(display_position=True)
|
||||
title = opts.toc if opts.toc else "Table of Contents"
|
||||
maxlen = max([len(x) for x in text])
|
||||
@@ -594,7 +717,7 @@ def print_toc(reader,opts):
|
||||
|
||||
|
||||
def offset_change(opts, reader, offset, new_offset):
|
||||
''' Change the display position of page '''
|
||||
""" Change the display position of page """
|
||||
new_offset = (offset[0] + new_offset[0], offset[1] + new_offset[1])
|
||||
offsety = min(reader.get_page_height() - 1, new_offset[0])
|
||||
offseth = min(reader.get_page_height(), new_offset[1])
|
||||
@@ -614,7 +737,7 @@ def getkeypress():
|
||||
|
||||
|
||||
def browser(opts, files):
|
||||
''' Main function for printing '''
|
||||
""" Main function for printing """
|
||||
|
||||
try:
|
||||
reader = slide_reader(files, opts)
|
||||
@@ -675,24 +798,24 @@ def browser(opts,files):
|
||||
if inkey == 70 or inkey == 52: # END
|
||||
reader.last_page()
|
||||
offset = (0, 0)
|
||||
if inkey==ord('c'):
|
||||
if inkey == ord("c"):
|
||||
print_toc(reader, opts)
|
||||
if inkey==ord('h'):
|
||||
if inkey == ord("h"):
|
||||
print_help(reader, opts)
|
||||
if inkey==ord('s'):
|
||||
if inkey == ord("s"):
|
||||
opts.menu = not opts.menu
|
||||
if inkey==ord('t'):
|
||||
if inkey == ord("t"):
|
||||
opts.slideShow = not opts.slideShow
|
||||
if inkey==ord('r'):
|
||||
if inkey == ord("r"):
|
||||
reader.read()
|
||||
offset = offset_change(opts, reader, offset, (0, 0))
|
||||
if inkey==ord('M'):
|
||||
if inkey == ord("M"):
|
||||
modify_file(reader, offset)
|
||||
reader.read()
|
||||
offset = offset_change(opts, reader, offset, (0, 0))
|
||||
if inkey==ord(','):
|
||||
if inkey == ord(","):
|
||||
offset = offset_change(opts, reader, offset, (-1, 0))
|
||||
if inkey==ord('.'):
|
||||
if inkey == ord("."):
|
||||
offset = offset_change(opts, reader, offset, (1, 0))
|
||||
if inkey == 65: # up
|
||||
offset = offset_change(opts, reader, offset, (0, -1))
|
||||
@@ -702,7 +825,6 @@ def browser(opts,files):
|
||||
launch(reader, opts, offset)
|
||||
break
|
||||
|
||||
|
||||
except IOError:
|
||||
pass
|
||||
except KeyboardInterrupt:
|
||||
@@ -716,7 +838,7 @@ def browser(opts,files):
|
||||
|
||||
|
||||
def get_console_size():
|
||||
rows, columns = os.popen('stty size', 'r').read().split()
|
||||
rows, columns = os.popen("stty size", "r").read().split()
|
||||
return (int(rows), int(columns))
|
||||
|
||||
|
||||
@@ -731,10 +853,7 @@ def colorify(s,opts):
|
||||
def cut_line(s, i):
|
||||
""" cut a color tagged string, and remove control chars """
|
||||
s = s[:i]
|
||||
s=re.sub("\$$","",
|
||||
re.sub("\$\{$","",
|
||||
re.sub("\$\{.$","",
|
||||
s)))
|
||||
s = re.sub("\$$", "", re.sub("\$\{$", "", re.sub("\$\{.$", "", s)))
|
||||
return s
|
||||
|
||||
|
||||
@@ -754,16 +873,15 @@ def launch(reader,opts,offset):
|
||||
Detects URLS and markdown images 
|
||||
"""
|
||||
if not opts.execute:
|
||||
bc.posprint(
|
||||
offset[1]-offset[0]+1,
|
||||
0,
|
||||
"Execution not enabled!"
|
||||
)
|
||||
bc.posprint(offset[1] - offset[0] + 1, 0, "Execution not enabled!")
|
||||
inkey = getch.get()
|
||||
return
|
||||
s = reader.get_current_page()[offset[1]]
|
||||
urls = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', s)
|
||||
images = re.findall('!\[[^]]+\]\([^\)]+\)', s)
|
||||
urls = re.findall(
|
||||
"http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+",
|
||||
s,
|
||||
)
|
||||
images = re.findall("!\[[^]]+\]\([^\)]+\)", s)
|
||||
# sanity check
|
||||
if s.find("`") == -1 and len(urls) == 0 and len(images) == 0:
|
||||
return
|
||||
@@ -771,56 +889,53 @@ def launch(reader,opts,offset):
|
||||
run_command = re.match("(.*)`(.*)`!(.*)", s)
|
||||
show_command = re.match("(.*)`(.*)`>(.*)", s)
|
||||
if show_command != None:
|
||||
output = subprocess.check_output(
|
||||
show_command.group(2).strip(),
|
||||
shell = True
|
||||
)
|
||||
output = subprocess.check_output(show_command.group(2).strip(), shell=True)
|
||||
if type(output) == bytes:
|
||||
output = output.decode('utf-8')
|
||||
output = output.decode("utf-8")
|
||||
output = output.split("\n")
|
||||
while len(output[0].strip()) == 0:
|
||||
if len(output)==1: return
|
||||
if len(output) == 1:
|
||||
return
|
||||
del output[0]
|
||||
while len(output[-1].strip()) == 0:
|
||||
if len(output)==1: return
|
||||
if len(output) == 1:
|
||||
return
|
||||
del output[-1]
|
||||
for y, l in enumerate(output):
|
||||
bc.posprint(y+offset[1]-offset[0]+2,0,' '*len(l))
|
||||
bc.posprint(y + offset[1] - offset[0] + 2, 0, " " * len(l))
|
||||
bc.clear_to_end()
|
||||
bc.posprint(y + offset[1] - offset[0] + 2, 0, l)
|
||||
inkey = getch.get()
|
||||
return
|
||||
if run_command != None:
|
||||
subprocess.call(
|
||||
run_command.group(2),
|
||||
shell = True,
|
||||
executable = "/bin/bash"
|
||||
)
|
||||
subprocess.call(run_command.group(2), shell=True, executable="/bin/bash")
|
||||
inkey = getch.get()
|
||||
return
|
||||
# Open URLS last
|
||||
if len(urls) > 0:
|
||||
# Remove ) at the end of url: [name](link) markdown syntax
|
||||
subprocess.call(
|
||||
"%s '%s' &"%(
|
||||
"%s '%s' &"
|
||||
% (
|
||||
get_open_command(),
|
||||
urls[0].rstrip(")"),
|
||||
),
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
shell = True
|
||||
shell=True,
|
||||
)
|
||||
return
|
||||
if len(images) > 0:
|
||||
image = re.sub('.*\(([^\)]+)\).*', "\\1",images[0])
|
||||
image = re.sub(".*\(([^\)]+)\).*", "\\1", images[0])
|
||||
subprocess.call(
|
||||
"%s '%s' &"%(
|
||||
"%s '%s' &"
|
||||
% (
|
||||
get_open_command(),
|
||||
image,
|
||||
),
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
shell = True
|
||||
shell=True,
|
||||
)
|
||||
return
|
||||
return
|
||||
@@ -836,30 +951,21 @@ def modify_file(reader,offset):
|
||||
if (page + 1) in row_restarts:
|
||||
row = 1
|
||||
subprocess.call(
|
||||
"vim +%d -c 'exe \"normal! zt\"' -c %d %s"%(
|
||||
row,
|
||||
row + offset[1],
|
||||
reader.get_current_filename()
|
||||
),
|
||||
shell = True
|
||||
"vim +%d -c 'exe \"normal! zt\"' -c %d %s"
|
||||
% (row, row + offset[1], reader.get_current_filename()),
|
||||
shell=True,
|
||||
)
|
||||
|
||||
|
||||
def take_screenshot(reader, opts):
|
||||
out_file = os.path.join(
|
||||
opts.screenshots,
|
||||
"slide%03d.png"%(reader.page + 1,)
|
||||
)
|
||||
out_file = os.path.join(opts.screenshots, "slide%03d.png" % (reader.page + 1,))
|
||||
if not os.path.exists(opts.screenshots):
|
||||
os.mkdir(opts.screenshots)
|
||||
subprocess.call(
|
||||
"sleep 0.5; import -window %s '%s'"%(
|
||||
opts.xwinid,
|
||||
out_file
|
||||
),
|
||||
"sleep 0.5; import -window %s '%s'" % (opts.xwinid, out_file),
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
shell = True
|
||||
shell=True,
|
||||
)
|
||||
|
||||
|
||||
@@ -879,29 +985,26 @@ def main():
|
||||
opts = setup_options()
|
||||
if opts.screenshots:
|
||||
print("Click the terminal window to read window ID")
|
||||
xwininfo = subprocess.check_output(
|
||||
"xwininfo",
|
||||
shell = True
|
||||
)
|
||||
xwininfo = subprocess.check_output("xwininfo", shell=True)
|
||||
if type(xwininfo) == bytes:
|
||||
xwininfo = xwininfo.decode('utf-8')
|
||||
xwininfo = xwininfo.decode("utf-8")
|
||||
xwininfo = re.search(r"Window id: (0x[0-9]+)", xwininfo)
|
||||
if xwininfo:
|
||||
opts.xwinid = xwininfo.group(1)
|
||||
else:
|
||||
print("Cannot parse window ID")
|
||||
sys.exit(1)
|
||||
browser(
|
||||
opts,
|
||||
opts.files
|
||||
)
|
||||
browser(opts, opts.files)
|
||||
print("\n\n")
|
||||
if opts.screenshots:
|
||||
print("Crop the images for terminal tabs, and PDFize e.g.:\n- mogrify -chop 0x50 %s/*png\n- convert %s/*png %s.pdf"%(
|
||||
print(
|
||||
"Crop the images for terminal tabs, and PDFize e.g.:\n- mogrify -chop 0x50 %s/*png\n- convert %s/*png %s.pdf"
|
||||
% (
|
||||
opts.screenshots,
|
||||
opts.screenshots,
|
||||
os.path.basename(opts.screenshots),
|
||||
))
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Binary file not shown.
@@ -1,56 +1,96 @@
|
||||
#!/usr/bin/env python
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import sys, os, re
|
||||
from argparse import ArgumentParser, RawDescriptionHelpFormatter
|
||||
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
import ansicodes as ansi
|
||||
|
||||
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
||||
__version__ = "0.3"
|
||||
|
||||
''' Rules modified from mistune project '''
|
||||
""" Rules modified from mistune project """
|
||||
|
||||
|
||||
def setup_options():
|
||||
bc = ansi.code()
|
||||
''' Create command line options '''
|
||||
usage = '''
|
||||
""" Create command line options """
|
||||
usage = (
|
||||
"""
|
||||
Markdown syntax color in ansi codes.
|
||||
Special syntaxes:
|
||||
- Colors: insert string e.g. ${C}.
|
||||
- Any ANSI control code: ${3A}, ${1;34;42m}, see the table..
|
||||
|
||||
''' + ansi.demo()
|
||||
"""
|
||||
+ ansi.demo()
|
||||
)
|
||||
|
||||
parser = ArgumentParser(
|
||||
formatter_class=RawDescriptionHelpFormatter,
|
||||
description=usage,
|
||||
epilog = __author__
|
||||
epilog=__author__,
|
||||
)
|
||||
|
||||
parser.add_argument("-v", "--version", action="version", version=__version__)
|
||||
parser.add_argument("-D",action="store_true",dest="debug",default=False,
|
||||
help="Debug mode")
|
||||
parser.add_argument("--dc",action="store_true",dest="dark_colors",default=False,
|
||||
help="Use dark colorscheme, better for white background terminals.")
|
||||
parser.add_argument("--no-color","-n",action="store_false",dest="color",default=True,
|
||||
help="Disable color.")
|
||||
parser.add_argument("--print-template",action="store_true",dest="print_template",default=False,
|
||||
help="Print customizable color template.")
|
||||
parser.add_argument("--template",action="store",type=str,dest="template",default=None,
|
||||
help="Use a custom template file for colorization.")
|
||||
parser.add_argument("-z",action="store_true",dest="zero",default=False,
|
||||
help="Reset coloring at the end of each line.")
|
||||
parser.add_argument("-Z",action="store_false",dest="zero_final",default=True,
|
||||
help="Disable reset of colors at the end of file.")
|
||||
parser.add_argument("filename",type=str, nargs='?',
|
||||
help="File to show, - for stdin")
|
||||
parser.add_argument(
|
||||
"-D", action="store_true", dest="debug", default=False, help="Debug mode"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--dc",
|
||||
action="store_true",
|
||||
dest="dark_colors",
|
||||
default=False,
|
||||
help="Use dark colorscheme, better for white background terminals.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--no-color",
|
||||
"-n",
|
||||
action="store_false",
|
||||
dest="color",
|
||||
default=True,
|
||||
help="Disable color.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--print-template",
|
||||
action="store_true",
|
||||
dest="print_template",
|
||||
default=False,
|
||||
help="Print customizable color template.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--template",
|
||||
action="store",
|
||||
type=str,
|
||||
dest="template",
|
||||
default=None,
|
||||
help="Use a custom template file for colorization.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-z",
|
||||
action="store_true",
|
||||
dest="zero",
|
||||
default=False,
|
||||
help="Reset coloring at the end of each line.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-Z",
|
||||
action="store_false",
|
||||
dest="zero_final",
|
||||
default=True,
|
||||
help="Disable reset of colors at the end of file.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"filename", type=str, nargs="?", help="File to show, - for stdin"
|
||||
)
|
||||
opts = parser.parse_args()
|
||||
return opts
|
||||
|
||||
|
||||
def parse(data):
|
||||
data = [[None, row] for row in data]
|
||||
block='text'
|
||||
new_block='text'
|
||||
block = "text"
|
||||
new_block = "text"
|
||||
multiline_block = False
|
||||
# Parse styles
|
||||
for i, line in enumerate(data):
|
||||
@@ -59,9 +99,9 @@ def parse(data):
|
||||
# Previous lines have set the style already
|
||||
continue
|
||||
for match in blocks:
|
||||
if block_match[match]['re'].match(row):
|
||||
if block_match[match]["re"].match(row):
|
||||
new_block = match
|
||||
if match.startswith('multiline'):
|
||||
if match.startswith("multiline"):
|
||||
if multiline_block:
|
||||
multiline_block = False
|
||||
else:
|
||||
@@ -70,12 +110,14 @@ def parse(data):
|
||||
if multiline_block:
|
||||
new_block = multiline_block
|
||||
# Lists must end with empty line
|
||||
if new_block not in ('empty','list_bullet') and block.startswith('list_'):
|
||||
new_block='list_loose'
|
||||
if new_block not in ("empty", "list_bullet") and block.startswith("list_"):
|
||||
new_block = "list_loose"
|
||||
|
||||
if 'mod' in block_match[match]:
|
||||
if "mod" in block_match[match]:
|
||||
# Style sets block in previous or next lines
|
||||
data[i+block_match[match]['mod']['pos']][0]=block_match[match]['mod']['name']
|
||||
data[i + block_match[match]["mod"]["pos"]][0] = block_match[match]["mod"][
|
||||
"name"
|
||||
]
|
||||
data[i][0] = new_block
|
||||
if new_block != block:
|
||||
block = new_block
|
||||
@@ -86,23 +128,22 @@ def colorize(data, remove_colors = False, dark_colors = False, debug = False):
|
||||
# Start inserting colors, and printing
|
||||
bc = ansi.code()
|
||||
colorized = []
|
||||
cs = 'dc' if dark_colors else 'bc'
|
||||
csc = cs + 'c'
|
||||
cs = "dc" if dark_colors else "bc"
|
||||
csc = cs + "c"
|
||||
for i, line in enumerate(data):
|
||||
row = line[1]
|
||||
block = line[0]
|
||||
multiline_block = block.startswith('multiline')
|
||||
multiline_block = block.startswith("multiline")
|
||||
if multiline_block:
|
||||
row = block_match[block][csc] + row
|
||||
if block_match[block][cs]:
|
||||
row = block_match[block]['re'].sub(block_match[block][cs], row)
|
||||
row = block_match[block]["re"].sub(block_match[block][cs], row)
|
||||
# No coloring inside block_code, nor multiline
|
||||
if not (multiline_block or block == 'block_code'):
|
||||
if not (multiline_block or block == "block_code"):
|
||||
for match in inlines:
|
||||
if inline_match[match]['re'].search(row):
|
||||
row = inline_match[match]['re'].sub(
|
||||
inline_match[match][cs] + block_match[block][csc],
|
||||
row
|
||||
if inline_match[match]["re"].search(row):
|
||||
row = inline_match[match]["re"].sub(
|
||||
inline_match[match][cs] + block_match[block][csc], row
|
||||
)
|
||||
if remove_colors:
|
||||
colored = bc.nocolor_string(row)
|
||||
@@ -114,31 +155,34 @@ def colorize(data, remove_colors = False, dark_colors = False, debug = False):
|
||||
colorized.append(colored)
|
||||
return colorized
|
||||
|
||||
|
||||
def md_re_compile(d):
|
||||
''' Returns a re.compiled dict '''
|
||||
""" Returns a re.compiled dict """
|
||||
n = {}
|
||||
for t in d:
|
||||
n[t] = {}
|
||||
for i in d[t]:
|
||||
n[t][i] = d[t][i]
|
||||
try:
|
||||
if n[t]['re']:
|
||||
n[t]['re']=re.compile(n[t]['re'])
|
||||
if n[t]["re"]:
|
||||
n[t]["re"] = re.compile(n[t]["re"])
|
||||
except err:
|
||||
print("Error compiling: %s"%(n[t]['re']))
|
||||
print("Error compiling: %s" % (n[t]["re"]))
|
||||
sys.exit(1)
|
||||
return n
|
||||
|
||||
|
||||
def read_data2(fp):
|
||||
data = []
|
||||
# Read data
|
||||
for row in f:
|
||||
if not row:
|
||||
continue
|
||||
row = row.decode('utf-8').rstrip("\n\r ")
|
||||
row = row.decode("utf-8").rstrip("\n\r ")
|
||||
data.append(row)
|
||||
return data
|
||||
|
||||
|
||||
def read_data3(fp):
|
||||
data = []
|
||||
# Read data
|
||||
@@ -149,14 +193,16 @@ def read_data3(fp):
|
||||
data.append(row)
|
||||
return data
|
||||
|
||||
|
||||
def write_colored2(colored, opts):
|
||||
for c in colored:
|
||||
sys.stdout.write(c.encode('utf-8'))
|
||||
sys.stdout.write(c.encode("utf-8"))
|
||||
if opts.zero:
|
||||
sys.stdout.write(bc.Z)
|
||||
sys.stdout.write("\n")
|
||||
if opts.zero_final:
|
||||
sys.stdout.write(bc.Z.encode('utf-8'))
|
||||
sys.stdout.write(bc.Z.encode("utf-8"))
|
||||
|
||||
|
||||
def write_colored3(colored, opts):
|
||||
for c in colored:
|
||||
@@ -168,112 +214,207 @@ def write_colored3(colored, opts):
|
||||
sys.stdout.write(bc.Z)
|
||||
|
||||
|
||||
|
||||
# re: regular expression, bc: bright colors, bcc: continue with this color after inline
|
||||
# dc: dark colors, dcc: continued color after inline
|
||||
block_match_str = {
|
||||
'block_code': {'re':'^( {4}[^*])(.*)$',
|
||||
'bc' :'${Z}${c}\\1\\2', 'bcc':'${Z}${c}',
|
||||
'dc' :'${Z}${m}\\1\\2', 'dcc':'${Z}${m}'}, # code
|
||||
'multiline_code' : {'re':'^ *(`{3,}|~{3,}) *(\S*)',
|
||||
'bc' :'${Z}${c}\\1\\2', 'bcc':'${Z}${c}',
|
||||
'dc' :'${Z}${m}\\1\\2', 'dcc':'${Z}${m}'}, # ```lang
|
||||
'block_quote': {'re':'^(>[ >]* )',
|
||||
'bc':'${K}\\1${Z}','bcc':'${Z}',
|
||||
'dc':'${Y}\\1${Z}','dcc':'${Z}'}, # > > quote
|
||||
'hrule': {'re':'^ {0,3}[-*_]([-*_]){2,}$',
|
||||
'bc':'False','bcc':'${Z}',
|
||||
'dc':'False','dcc':'${Z}'}, # ----
|
||||
'heading' : {'re':'^ *(#{1,6}) *([^\n]+?) *#* *(?:\n+|$)',
|
||||
'bc':'${W}\\1 ${U}\\2${Z}','bcc':'${W}${U}',
|
||||
'dc':'${B}\\1 ${U}\\2${Z}','dcc':'${B}${U}'}, # # heading
|
||||
'lheading' : {'re':'^(=+|-+)$',
|
||||
'bc':'${W}\\1', 'bcc':'${W}',
|
||||
'dc':'${B}\\1', 'dcc':'${B}',
|
||||
'mod':{'pos':-1,'name':'lheading.mod'}}, # ======= under header
|
||||
'lheading.mod' : {'re':'^([^\n]+)',
|
||||
'bc':'${W}\\1', 'bcc':'${W}',
|
||||
'dc':'${B}\\1', 'dcc':'${B}'}, # over the ======= under header
|
||||
'list_bullet': {'re':'^( *)([*+-]|[\d\.]+)( +)',
|
||||
'bc':'\\1${y}\\2${Z}\\3','bcc':'${Z}',
|
||||
'dc':'\\1${r}\\2${Z}\\3','dcc':'${Z}'}, # * or 1.
|
||||
'list_loose': {'re':'None',
|
||||
'bc':'False','bcc':'${Z}',
|
||||
'dc':'False','dcc':'${Z}'},
|
||||
'text': {'re':'^([^\n]+)',
|
||||
'bc':'${Z}\\1','bcc':'${Z}',
|
||||
'dc':'${Z}\\1','dcc':'${Z}'},
|
||||
'empty': {'re':'(^$)',
|
||||
'bc':'${Z}\\1','bcc':'${Z}',
|
||||
'dc':'${Z}\\1','dcc':'${Z}'},
|
||||
'preformatted': {'re': 'a^', # Never matches anything
|
||||
'bc':'','bcc':'',
|
||||
'dc':'','dcc':''},
|
||||
"block_code": {
|
||||
"re": "^( {4}[^*])(.*)$",
|
||||
"bc": "${Z}${c}\\1\\2",
|
||||
"bcc": "${Z}${c}",
|
||||
"dc": "${Z}${m}\\1\\2",
|
||||
"dcc": "${Z}${m}",
|
||||
}, # code
|
||||
"multiline_code": {
|
||||
"re": "^ *(`{3,}|~{3,}) *(\S*)",
|
||||
"bc": "${Z}${c}\\1\\2",
|
||||
"bcc": "${Z}${c}",
|
||||
"dc": "${Z}${m}\\1\\2",
|
||||
"dcc": "${Z}${m}",
|
||||
}, # ```lang
|
||||
"block_quote": {
|
||||
"re": "^(>[ >]* )",
|
||||
"bc": "${K}\\1${Z}",
|
||||
"bcc": "${Z}",
|
||||
"dc": "${Y}\\1${Z}",
|
||||
"dcc": "${Z}",
|
||||
}, # > > quote
|
||||
"hrule": {
|
||||
"re": "^ {0,3}[-*_]([-*_]){2,}$",
|
||||
"bc": "False",
|
||||
"bcc": "${Z}",
|
||||
"dc": "False",
|
||||
"dcc": "${Z}",
|
||||
}, # ----
|
||||
"heading": {
|
||||
"re": "^ *(#{1,6}) *([^\n]+?) *#* *(?:\n+|$)",
|
||||
"bc": "${W}\\1 ${U}\\2${Z}",
|
||||
"bcc": "${W}${U}",
|
||||
"dc": "${B}\\1 ${U}\\2${Z}",
|
||||
"dcc": "${B}${U}",
|
||||
}, # # heading
|
||||
"lheading": {
|
||||
"re": "^(=+|-+)$",
|
||||
"bc": "${W}\\1",
|
||||
"bcc": "${W}",
|
||||
"dc": "${B}\\1",
|
||||
"dcc": "${B}",
|
||||
"mod": {"pos": -1, "name": "lheading.mod"},
|
||||
}, # ======= under header
|
||||
"lheading.mod": {
|
||||
"re": "^([^\n]+)",
|
||||
"bc": "${W}\\1",
|
||||
"bcc": "${W}",
|
||||
"dc": "${B}\\1",
|
||||
"dcc": "${B}",
|
||||
}, # over the ======= under header
|
||||
"list_bullet": {
|
||||
"re": "^( *)([*+-]|[\d\.]+)( +)",
|
||||
"bc": "\\1${y}\\2${Z}\\3",
|
||||
"bcc": "${Z}",
|
||||
"dc": "\\1${r}\\2${Z}\\3",
|
||||
"dcc": "${Z}",
|
||||
}, # * or 1.
|
||||
"list_loose": {
|
||||
"re": "None",
|
||||
"bc": "False",
|
||||
"bcc": "${Z}",
|
||||
"dc": "False",
|
||||
"dcc": "${Z}",
|
||||
},
|
||||
"text": {
|
||||
"re": "^([^\n]+)",
|
||||
"bc": "${Z}\\1",
|
||||
"bcc": "${Z}",
|
||||
"dc": "${Z}\\1",
|
||||
"dcc": "${Z}",
|
||||
},
|
||||
"empty": {
|
||||
"re": "(^$)",
|
||||
"bc": "${Z}\\1",
|
||||
"bcc": "${Z}",
|
||||
"dc": "${Z}\\1",
|
||||
"dcc": "${Z}",
|
||||
},
|
||||
"preformatted": {
|
||||
"re": "a^", # Never matches anything
|
||||
"bc": "",
|
||||
"bcc": "",
|
||||
"dc": "",
|
||||
"dcc": "",
|
||||
},
|
||||
}
|
||||
block_match = md_re_compile(block_match_str)
|
||||
blocks=['block_quote', 'multiline_code','hrule', 'heading','lheading','list_bullet', 'block_code', 'text', 'empty']
|
||||
blocks = [
|
||||
"block_quote",
|
||||
"multiline_code",
|
||||
"hrule",
|
||||
"heading",
|
||||
"lheading",
|
||||
"list_bullet",
|
||||
"block_code",
|
||||
"text",
|
||||
"empty",
|
||||
]
|
||||
|
||||
inline_match_str = {
|
||||
'bold1': {'re':r'(^| |})(_[^_]+_)',
|
||||
'bc':'\\1${W}\\2','dc':'\\1${W}\\2'}, # _bold_
|
||||
'bold2': {'re':r'(^| |})(\*{1,2}[^\*]+\*{1,2})',
|
||||
'bc':'\\1${W}\\2','dc':'\\1${W}\\2'}, # **bold**
|
||||
'code': {'re':r'([`]+[^`]+[`]+)',
|
||||
'bc':'${c}\\1','dc':'${m}\\1'}, # `code`
|
||||
'code_special': {'re':r'([`]+[^`]+[`]+)([!>])',
|
||||
'bc':'${c}\\1${g}\\2','dc':'${m}\\1${r}\\2'}, # `code`! or `code`> for markslider
|
||||
'link': {'re':r'(\[)([^\]]+)(\])\(([^\)]+)\)',
|
||||
'bc':'${B}\\1${Z}\\2${B}\\3(${U}\\4${u})',
|
||||
'dc':'${b}\\1${Z}\\2${b}\\3(${U}\\4${u})'}, # [text](link)
|
||||
'image': {'re':r'(!\[[^\]]+\]\([^\)]+\))',
|
||||
'bc':'${r}\\1','dc':'${g}\\1'}, # 
|
||||
'underline': {'re':r'(^|\W)(__)([^_]+)(__)',
|
||||
'bc':'\\1\\2${U}\\3${Z}\\4','dc':'\\1\\2${U}\\3${Z}\\4'}, # __underline__
|
||||
'strikethrough': {'re':r'(~~)([^~]+)(~~)',
|
||||
'bc':'\\1${st}\\2${so}\\3','dc':'\\1${st}\\2${so}\\3'}, # ~~strike~~
|
||||
'checkbox': {'re':r'(\[[x ]\])',
|
||||
'bc':'${y}\\1','dc':'${r}\\1'}, # [x] [ ]
|
||||
"bold1": {
|
||||
"re": r"(^| |})(_[^_]+_)",
|
||||
"bc": "\\1${W}\\2",
|
||||
"dc": "\\1${W}\\2",
|
||||
}, # _bold_
|
||||
"bold2": {
|
||||
"re": r"(^| |})(\*{1,2}[^\*]+\*{1,2})",
|
||||
"bc": "\\1${W}\\2",
|
||||
"dc": "\\1${W}\\2",
|
||||
}, # **bold**
|
||||
"code": {"re": r"([`]+[^`]+[`]+)", "bc": "${c}\\1", "dc": "${m}\\1"}, # `code`
|
||||
"code_special": {
|
||||
"re": r"([`]+[^`]+[`]+)([!>])",
|
||||
"bc": "${c}\\1${g}\\2",
|
||||
"dc": "${m}\\1${r}\\2",
|
||||
}, # `code`! or `code`> for markslider
|
||||
"link": {
|
||||
"re": r"(\[)([^\]]+)(\])\(([^\)]+)\)",
|
||||
"bc": "${B}\\1${Z}\\2${B}\\3(${U}\\4${u})",
|
||||
"dc": "${b}\\1${Z}\\2${b}\\3(${U}\\4${u})",
|
||||
}, # [text](link)
|
||||
"image": {
|
||||
"re": r"(!\[[^\]]+\]\([^\)]+\))",
|
||||
"bc": "${r}\\1",
|
||||
"dc": "${g}\\1",
|
||||
}, # 
|
||||
"underline": {
|
||||
"re": r"(^|\W)(__)([^_]+)(__)",
|
||||
"bc": "\\1\\2${U}\\3${Z}\\4",
|
||||
"dc": "\\1\\2${U}\\3${Z}\\4",
|
||||
}, # __underline__
|
||||
"strikethrough": {
|
||||
"re": r"(~~)([^~]+)(~~)",
|
||||
"bc": "\\1${st}\\2${so}\\3",
|
||||
"dc": "\\1${st}\\2${so}\\3",
|
||||
}, # ~~strike~~
|
||||
"checkbox": {"re": r"(\[[x ]\])", "bc": "${y}\\1", "dc": "${r}\\1"}, # [x] [ ]
|
||||
}
|
||||
inline_match = md_re_compile(inline_match_str)
|
||||
inlines = ['bold1','bold2','code_special','code','image','link','underline','strikethrough', 'checkbox']
|
||||
inlines = [
|
||||
"bold1",
|
||||
"bold2",
|
||||
"code_special",
|
||||
"code",
|
||||
"image",
|
||||
"link",
|
||||
"underline",
|
||||
"strikethrough",
|
||||
"checkbox",
|
||||
]
|
||||
|
||||
if __name__ == "__main__":
|
||||
opts = setup_options()
|
||||
if opts.print_template:
|
||||
import json
|
||||
print(json.dumps({'about':'re: regular expression, bc: bright colors, bcc: continue with this color after inline, dc: dark colors, dcc: continued color after inline. "blocks" and "inlines" list keys of matchers. ',
|
||||
'blocks':blocks,
|
||||
'block_match':block_match_str,
|
||||
'inline_match':inline_match_str,
|
||||
'inlines':inlines},
|
||||
indent=2,sort_keys=True))
|
||||
|
||||
print(
|
||||
json.dumps(
|
||||
{
|
||||
"about": 're: regular expression, bc: bright colors, bcc: continue with this color after inline, dc: dark colors, dcc: continued color after inline. "blocks" and "inlines" list keys of matchers. ',
|
||||
"blocks": blocks,
|
||||
"block_match": block_match_str,
|
||||
"inline_match": inline_match_str,
|
||||
"inlines": inlines,
|
||||
},
|
||||
indent=2,
|
||||
sort_keys=True,
|
||||
)
|
||||
)
|
||||
sys.exit(0)
|
||||
if opts.template:
|
||||
import json
|
||||
template=json.load(open(opts.template,'r'))
|
||||
if 'inlines' in template: inlines=template['inlines']
|
||||
if 'blocks' in template: blocks=template['blocks']
|
||||
if 'block_match' in template:
|
||||
block_match_str=template['block_match']
|
||||
|
||||
template = json.load(open(opts.template, "r"))
|
||||
if "inlines" in template:
|
||||
inlines = template["inlines"]
|
||||
if "blocks" in template:
|
||||
blocks = template["blocks"]
|
||||
if "block_match" in template:
|
||||
block_match_str = template["block_match"]
|
||||
block_match = md_re_compile(block_match_str)
|
||||
if 'inline_match' in template:
|
||||
inline_match_str=template['inline_match']
|
||||
if "inline_match" in template:
|
||||
inline_match_str = template["inline_match"]
|
||||
inline_match = md_re_compile(inline_match_str)
|
||||
|
||||
bc = ansi.code()
|
||||
if opts.filename == "-" or opts.filename == None:
|
||||
f = sys.stdin
|
||||
else:
|
||||
f = open(opts.filename, 'r')
|
||||
if (sys.version_info > (3, 0)):
|
||||
f = open(opts.filename, "r")
|
||||
if sys.version_info > (3, 0):
|
||||
data = read_data3(f)
|
||||
else:
|
||||
data = read_data2(f)
|
||||
|
||||
data = parse(data)
|
||||
colored = colorize(data, not opts.color, opts.dark_colors, opts.debug)
|
||||
if (sys.version_info > (3, 0)):
|
||||
if sys.version_info > (3, 0):
|
||||
write_colored3(colored, opts)
|
||||
else:
|
||||
write_colored2(colored, opts)
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
#!/usr/bin/env python
|
||||
#!/usr/bin/env python3
|
||||
# coding=utf-8
|
||||
#
|
||||
# Copyright 2016 Ville Rantanen
|
||||
@@ -17,52 +17,75 @@
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
|
||||
'''Color by ${X} notation.'''
|
||||
"""Color by ${X} notation."""
|
||||
|
||||
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
||||
__version__ = "0.1"
|
||||
|
||||
import sys, os, argparse, re
|
||||
from argparse import ArgumentParser
|
||||
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
import ansicodes as ansi
|
||||
|
||||
|
||||
def setup_options():
|
||||
''' Create command line options '''
|
||||
usage='''
|
||||
""" Create command line options """
|
||||
usage = (
|
||||
"""
|
||||
Color notation renderer in ANSI codes
|
||||
Special syntaxes:
|
||||
* Colors: insert string ${X}, where X values in the table below.
|
||||
|
||||
''' + ansi.demo()
|
||||
"""
|
||||
+ ansi.demo()
|
||||
)
|
||||
|
||||
parser=ArgumentParser(description=usage,
|
||||
parser = ArgumentParser(
|
||||
description=usage,
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
epilog=__author__)
|
||||
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("-Z",action="store_false",dest="zero_final",default=True,
|
||||
help="Disable reset of colors at the end of file.")
|
||||
parser.add_argument("filename",type=str,
|
||||
help="File to show, - for stdin")
|
||||
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(
|
||||
"-Z",
|
||||
action="store_false",
|
||||
dest="zero_final",
|
||||
default=True,
|
||||
help="Disable reset of colors at the end of file.",
|
||||
)
|
||||
parser.add_argument("filename", type=str, help="File to show, - for stdin")
|
||||
opts = parser.parse_args()
|
||||
return opts
|
||||
|
||||
|
||||
bc = ansi.code()
|
||||
opts = setup_options()
|
||||
if opts.filename == "-":
|
||||
f = sys.stdin
|
||||
else:
|
||||
f=open(opts.filename,'r')
|
||||
f = open(opts.filename, "r")
|
||||
for row in f:
|
||||
if not row:
|
||||
continue
|
||||
if type(row) == bytes:
|
||||
row = row.decode('utf-8')
|
||||
row = row.decode("utf-8")
|
||||
row = row.rstrip("\n\r ")
|
||||
if opts.color:
|
||||
colored = bc.color_string(row)
|
||||
@@ -74,4 +97,3 @@ for row in f:
|
||||
sys.stdout.write("\n")
|
||||
if opts.zero_final and opts.color:
|
||||
sys.stdout.write(bc.Z)
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
#!/usr/bin/env python
|
||||
#!/usr/bin/env python3
|
||||
# coding=utf-8
|
||||
#
|
||||
# Copyright 2016 Ville Rantanen
|
||||
@@ -17,7 +17,7 @@
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
|
||||
'''Color by notation.'''
|
||||
"""Color by notation."""
|
||||
|
||||
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
||||
__version__ = "0.2"
|
||||
@@ -30,6 +30,8 @@ PAUSE="====PAUSE"
|
||||
|
||||
import sys, os, argparse, re, time
|
||||
from argparse import ArgumentParser
|
||||
|
||||
|
||||
class bc:
|
||||
K = "\033[1;30m"
|
||||
R = "\033[1;31m"
|
||||
@@ -67,35 +69,77 @@ class bc:
|
||||
bc = "\033[46m"
|
||||
bw = "\033[47m"
|
||||
|
||||
S = '\033[1m'
|
||||
s = '\033[2m'#strong off
|
||||
U = '\033[4m'#underline
|
||||
u = '\033[24m'#underline off
|
||||
ic = '\033[7m'#inverse colors
|
||||
io = '\033[27m'#inverse off
|
||||
st = '\033[9m'#strike on
|
||||
so = '\033[29m'#strike off
|
||||
S = "\033[1m"
|
||||
s = "\033[2m" # strong off
|
||||
U = "\033[4m" # underline
|
||||
u = "\033[24m" # underline off
|
||||
ic = "\033[7m" # inverse colors
|
||||
io = "\033[27m" # inverse off
|
||||
st = "\033[9m" # strike on
|
||||
so = "\033[29m" # strike off
|
||||
|
||||
Z = '\033[0m'
|
||||
CLR = '\033[2J'
|
||||
CLREND = '\033[K'
|
||||
Z = "\033[0m"
|
||||
CLR = "\033[2J"
|
||||
CLREND = "\033[K"
|
||||
CLRSCR = CLR + "\033[0;0H"
|
||||
|
||||
color_keys = "pqwertyu01234567;asdfghjPQWERTYUxXczvVbBnN"
|
||||
color_list=[K, R, G, Y, B, M, C, W, k, r, g, y, b, m, c, w,
|
||||
bk,br,bg,by,bb,bm,bc,bw,nk,nr,ng,ny,nb,nm,nc,nw,
|
||||
S,s,CLRSCR,Z,U,u,st,so,ic,io]
|
||||
color_list = [
|
||||
K,
|
||||
R,
|
||||
G,
|
||||
Y,
|
||||
B,
|
||||
M,
|
||||
C,
|
||||
W,
|
||||
k,
|
||||
r,
|
||||
g,
|
||||
y,
|
||||
b,
|
||||
m,
|
||||
c,
|
||||
w,
|
||||
bk,
|
||||
br,
|
||||
bg,
|
||||
by,
|
||||
bb,
|
||||
bm,
|
||||
bc,
|
||||
bw,
|
||||
nk,
|
||||
nr,
|
||||
ng,
|
||||
ny,
|
||||
nb,
|
||||
nm,
|
||||
nc,
|
||||
nw,
|
||||
S,
|
||||
s,
|
||||
CLRSCR,
|
||||
Z,
|
||||
U,
|
||||
u,
|
||||
st,
|
||||
so,
|
||||
ic,
|
||||
io,
|
||||
]
|
||||
|
||||
def pos(self, y, x):
|
||||
return "\033[" + str(y) + ";" + str(x) + "H"
|
||||
|
||||
def posprint(self, y, x, s):
|
||||
sys.stdout.write( (self.pos(y,x) + str(s)).encode('utf-8') )
|
||||
sys.stdout.write((self.pos(y, x) + str(s)).encode("utf-8"))
|
||||
|
||||
def clear(self):
|
||||
sys.stdout.write( (self.CLR+self.pos(0,0)).encode('utf-8') )
|
||||
sys.stdout.write((self.CLR + self.pos(0, 0)).encode("utf-8"))
|
||||
|
||||
def clear_to_end(self):
|
||||
sys.stdout.write( self.CLREND.encode('utf-8') )
|
||||
sys.stdout.write(self.CLREND.encode("utf-8"))
|
||||
|
||||
def color_char(self, s):
|
||||
for i, c in enumerate(self.color_keys):
|
||||
@@ -103,24 +147,28 @@ class bc:
|
||||
s = s.replace(c, self.color_list[i])
|
||||
return s
|
||||
return ""
|
||||
|
||||
def nocolor_string(self, s):
|
||||
return ""
|
||||
|
||||
|
||||
def end_drawing(opts):
|
||||
if opts.zero_final:
|
||||
sys.stdout.write(bc.Z.encode('utf-8'))
|
||||
sys.stdout.write(bc.Z.encode("utf-8"))
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
def open_data_file(opts):
|
||||
if opts.test_image:
|
||||
return test_data()
|
||||
if opts.filename=='-':
|
||||
if opts.filename == "-":
|
||||
return sys.stdin
|
||||
return open(opts.filename,'r')
|
||||
return open(opts.filename, "r")
|
||||
|
||||
|
||||
def setup_options():
|
||||
''' Create command line options '''
|
||||
usage='''
|
||||
""" Create command line options """
|
||||
usage = """
|
||||
Color image renderer in ANSI codes. Example file:
|
||||
|
||||
first layer of characters
|
||||
@@ -145,31 +193,61 @@ Color image renderer in ANSI codes. Example file:
|
||||
====EOF
|
||||
end of file
|
||||
|
||||
'''
|
||||
"""
|
||||
|
||||
parser=ArgumentParser(description=usage,
|
||||
parser = ArgumentParser(
|
||||
description=usage,
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
epilog=__author__)
|
||||
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("--test",action="store_true",dest="test_image",default=False,
|
||||
help="Show test image showing features.")
|
||||
parser.add_argument("-z",action="store_true",dest="zero",default=False,
|
||||
help="Reset color codes at the end of each line.")
|
||||
parser.add_argument("-Z",action="store_false",dest="zero_final",default=True,
|
||||
help="Disable reset of colors at the end of file.")
|
||||
parser.add_argument("-c",action="store_true",dest="clear",default=False,
|
||||
help="Clear screen first")
|
||||
parser.add_argument("filename",type=str, nargs='?', default=None,
|
||||
help="File to show, - for stdin")
|
||||
parser.add_argument(
|
||||
"--no-color",
|
||||
"-n",
|
||||
action="store_false",
|
||||
dest="color",
|
||||
default=True,
|
||||
help="Disable color.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--test",
|
||||
action="store_true",
|
||||
dest="test_image",
|
||||
default=False,
|
||||
help="Show test image showing features.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-z",
|
||||
action="store_true",
|
||||
dest="zero",
|
||||
default=False,
|
||||
help="Reset color codes at the end of each line.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-Z",
|
||||
action="store_false",
|
||||
dest="zero_final",
|
||||
default=True,
|
||||
help="Disable reset of colors at the end of file.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-c",
|
||||
action="store_true",
|
||||
dest="clear",
|
||||
default=False,
|
||||
help="Clear screen first",
|
||||
)
|
||||
parser.add_argument(
|
||||
"filename", type=str, nargs="?", default=None, help="File to show, - for stdin"
|
||||
)
|
||||
opts = parser.parse_args()
|
||||
if not opts.test_image and opts.filename == None:
|
||||
parser.error("Need a file or - as argument")
|
||||
|
||||
return opts
|
||||
|
||||
|
||||
def test_data():
|
||||
testimage = ''' NOT VISIBLE
|
||||
_ _ _ ____ ___
|
||||
@@ -221,8 +299,11 @@ u f u
|
||||
; f U
|
||||
; f U
|
||||
====EOI
|
||||
Outputs:'''.encode('utf-8')
|
||||
return testimage.split('\n')
|
||||
Outputs:'''.encode(
|
||||
"utf-8"
|
||||
)
|
||||
return testimage.split("\n")
|
||||
|
||||
|
||||
bc = bc()
|
||||
opts = setup_options()
|
||||
@@ -244,7 +325,7 @@ while True:
|
||||
for row in f:
|
||||
if not row:
|
||||
end_drawing(opts)
|
||||
row=row.decode('utf-8').rstrip("\n\r")
|
||||
row = row.decode("utf-8").rstrip("\n\r")
|
||||
if row.startswith(NEWIMAGE):
|
||||
break
|
||||
if row.startswith(ENDOFFILE):
|
||||
@@ -282,37 +363,36 @@ while True:
|
||||
for i in range(len(gray[0])):
|
||||
for frame in colors:
|
||||
if len(frame) < i + 1:
|
||||
frame.append('')
|
||||
frame.append("")
|
||||
while len(gray[0][i]) > len(frame[i]):
|
||||
frame[i] = frame[i] + " "
|
||||
if len(gray) > 1:
|
||||
for layer in gray[1:]:
|
||||
if len(layer) < i + 1:
|
||||
layer.append('')
|
||||
layer.append("")
|
||||
while len(gray[0][i]) > len(layer[i]):
|
||||
layer[i] = layer[i] + " "
|
||||
|
||||
|
||||
for i in range(len(gray[0])):
|
||||
for c in range(len(gray[0][i])):
|
||||
if opts.color:
|
||||
for frame in colors:
|
||||
try:
|
||||
if frame[i][c] != " ":
|
||||
sys.stdout.write(bc.color_char(frame[i][c]).encode('utf-8'))
|
||||
sys.stdout.write(bc.color_char(frame[i][c]).encode("utf-8"))
|
||||
except IndexError:
|
||||
pass
|
||||
char = " "
|
||||
for layer in gray:
|
||||
if layer[i][c]==" ": continue
|
||||
if layer[i][c] == " ":
|
||||
continue
|
||||
char = layer[i][c]
|
||||
|
||||
sys.stdout.write(char.encode('utf-8'))
|
||||
sys.stdout.write(char.encode("utf-8"))
|
||||
if opts.color and opts.zero:
|
||||
sys.stdout.write(bc.Z.encode('utf-8'))
|
||||
sys.stdout.write(bc.Z.encode("utf-8"))
|
||||
sys.stdout.write("\n")
|
||||
if pause > 0:
|
||||
time.sleep(pause)
|
||||
if opts.test_image:
|
||||
end_drawing(opts)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user