def test_pop(self):
visitor = mnfy.SourceCode()
visitor._write('X')
with self.assertRaises(ValueError):
visitor._pop('sadf')
visitor._write('X')
self.assertEqual(visitor._pop(), 'X')
def test_roundtrip(cls, source=None):
if source is None: # pragma: no cover
try:
source = inspect.getsource(mnfy)
except IOError:
pass
original_ast = ast.parse(source)
minifier = mnfy.SourceCode()
minifier.visit(original_ast)
minified_source = str(minifier)
minified_ast = ast.parse(minified_source)
node_pairs = zip(ast.walk(minified_ast), ast.walk(original_ast))
for minified, original in node_pairs:
cls.compare_ast_nodes(minified, original)
return minified_source
def miniy_py(content):
minifier = mnfy.SourceCode()
minifier.visit(ast.parse(content))
return str(minifier)
def verify(self, given_ast, expect):
visitor = mnfy.SourceCode()
visitor.visit(given_ast)
self.assertEqual(str(visitor), expect)
def write_minified(infile, outfile, minify=True, readable=False):
print("Turning {0} into {1}...".format(infile, outfile))
try:
if not minify:
print("Not minifying.")
with open(infile, 'r') as source_file:
source = source_file.read()
with open(outfile, 'w') as fw:
fw.write(source)
elif minify and readable:
print("Using simple readable minification")
with open(infile, 'r') as fr, open(outfile, 'w') as fw:
for line in fr:
# Check for empty lines or comment lines, and skip them
if not line.lstrip() or line.lstrip()[0] == '#':
continue
fw.write(line)
# Use mnfy lib to minify
elif HAVE_PYMINIFIER:
print("Using pyminifier library!")
# Adapted from mnfy.py
subprocess.call("pyminifier -o {0} {1}".format(outfile,
infile).split())
# Use mnfy lib to minify
elif HAVE_MNFY:
print("Using mnfy library!")
# Adapted from mnfy.py
with open(infile, 'r') as source_file:
source = source_file.read()
import ast
source_ast = ast.parse(source)
# Use safe transforms
for transform in mnfy.safe_transforms:
transformer = transform()
source_ast = transformer.visit(source_ast)
minifier = mnfy.SourceCode()
minifier.visit(source_ast)
with open(outfile, 'w') as fw:
fw.write(str(minifier))
# More aggressive simple removal of blank lines and comments; combines some continued lines
else:
print("Using simple minification")
with open(infile, 'r') as fr, open(outfile, 'w') as fw:
commabuffer = ""
for line in fr:
# Check for empty lines or comment lines, and skip them
if not line.lstrip() or line.lstrip()[0] == '#':
continue
if commabuffer:
commabuffer += line.strip()
#if commabuffer[-1] != ",":
if "#" in line or commabuffer[-1] not in "[(,":
fw.write(commabuffer + "\n")
commabuffer = ""
#elif line.rstrip()[-1] == ",":
elif "#" in line:
fw.write(line)
commabuffer = ""
elif line.rstrip()[-1] in "[(,":
commabuffer = line.rstrip()
else:
fw.write(line)
commabuffer = ""
if commabuffer:
fw.write(commabuffer)
size_bytes = os.stat(outfile).st_size
size_kb = math.ceil(size_bytes / 1024.)
bytes_diff = 1024 - size_bytes % 1024
print("\t{0:20}\t{1}kb\t({2} free bytes in last block)".format(
os.path.basename(outfile) + ":", size_kb, bytes_diff))
return size_kb
except SyntaxError as e:
print("ERROR: Syntax error when minifying {0}".format(infile))
return 0