mirror of https://github.com/ksherlock/prez.git
77 lines
1.7 KiB
Python
77 lines
1.7 KiB
Python
|
|
|
|
|
|
# __all__ = ("str_to_bytes", "make_string", "format_rect", "format_point", "format_size")
|
|
|
|
# decorators
|
|
def export_enum(cls):
|
|
global __all__
|
|
|
|
members = cls.__members__
|
|
globals().update(members)
|
|
if __all__ != None: __all__.extend(list(members))
|
|
return cls
|
|
|
|
|
|
# helper functions
|
|
def str_to_bytes(text):
|
|
if isinstance(text, str): return text.encode("macroman")
|
|
if isinstance(text, bytes): return text
|
|
if isinstance(text, bytearray): return bytes(text)
|
|
raise TypeError("Bad text type: {}".format(type(text)))
|
|
|
|
|
|
|
|
def format_rect(x):
|
|
return "{{ {:d}, {:d}, {:d}, {:d} }}".format(*x)
|
|
|
|
def format_point(x):
|
|
return "{{ {:d}, {:d} }}".format(*x)
|
|
|
|
def format_size(x):
|
|
return "{{ {:d}, {:d} }}".format(*x)
|
|
|
|
|
|
def _generate_map():
|
|
map = { x: "\\${:02x}".format(x) for x in range(0, 256) if x < 32 or x > 126 }
|
|
map[0x0d] = "\\r" # intentionally backwards. -- no more
|
|
map[0x0a] = "\\n" # intentionally backwards. -- no more
|
|
map[0x09] = "\\t"
|
|
# \b \f \v \? also supported.
|
|
map[ord('"')] = '\\"'
|
|
map[ord("'")] = "\\'"
|
|
map[ord("\\")] = "\\\\"
|
|
# map[0x7f] = "\\?" # rubout
|
|
|
|
return map
|
|
|
|
_map = _generate_map()
|
|
|
|
def format_string(bstring, quote=True):
|
|
s = "".join([_map[x] if x in _map else chr(x) for x in bstring])
|
|
if quote: return '"' + s + '"'
|
|
return s
|
|
|
|
# rez "char" is a string of length 1.
|
|
# 'x' is character literal which is a number.
|
|
def format_char(x):
|
|
if x == 0: return '""'
|
|
c = _map.get(x, chr(x))
|
|
return '"' + c + '"'
|
|
|
|
def multi_format_string(bstring, indent=""):
|
|
q = '"'
|
|
|
|
rv = []
|
|
tmp = [indent + q]
|
|
for x in bstring:
|
|
if x in _map: tmp.append(_map[x])
|
|
else: tmp.append(chr(x))
|
|
if len(tmp) > 32 or x == 0x0a:
|
|
tmp.append(q)
|
|
rv.append("".join(tmp))
|
|
tmp = [indent + q]
|
|
if len(tmp)>1:
|
|
tmp.append(q)
|
|
rv.append("".join(tmp))
|
|
return ("\n").join(rv) |