mirror of
https://github.com/kennethreitz/dive-into-python3.git
synced 2026-06-05 15:00:18 +00:00
Mark Pilgrim
60 lines
2.2 KiB
Python
Executable File
60 lines
2.2 KiB
Python
Executable File
#!/usr/bin/python3
|
|
"""Quick-and-dirty HTML minimizer"""
|
|
|
|
import sys
|
|
import re
|
|
import html.entities
|
|
import itertools
|
|
import os.path
|
|
|
|
_SERVERS = ['a.wearehugh.com',
|
|
'b.wearehugh.com',
|
|
'c.wearehugh.com',
|
|
'd.wearehugh.com']
|
|
available_server = itertools.cycle(_SERVERS)
|
|
input_file = sys.argv[1]
|
|
output_file = sys.argv[2]
|
|
in_pre = False
|
|
with open(output_file, 'w', encoding="utf-8") as _out, open(input_file, encoding="utf-8") as _in:
|
|
for line in _in:
|
|
# round-robin image servers
|
|
if "<img src=i/" in line:
|
|
line = line.replace("<img src=i/", "<img src=http://" + next(available_server) + "/dip3/")
|
|
if "url(i/" in line:
|
|
line = line.replace("url(i/", "url(http://" + next(available_server) + "/dip3/")
|
|
|
|
# remove selected comments (but not all comments, because some are conditional comments for IE compat)
|
|
line = line.replace('<!-- toc -->', '')
|
|
line = line.replace('<!-- /toc -->', '')
|
|
|
|
# replace entities with Unicode characters
|
|
for e in re.findall('&(.+?);', line):
|
|
if e in ('lt', 'amp', 'quot', 'apos', 'nbsp'):
|
|
continue
|
|
n = html.entities.name2codepoint.get(e)
|
|
if not n:
|
|
if e.count('#x'):
|
|
# it's late, forgive me
|
|
n = eval(e.replace('#', '0'))
|
|
elif e.count('#'):
|
|
n = int(e.replace('#', ''))
|
|
else:
|
|
continue
|
|
line = line.replace('&' + e + ';', chr(n))
|
|
|
|
# strip leading and trailing whitespace, except inside <pre> blocks
|
|
g = line.strip()
|
|
if g.count('<pre'):
|
|
in_pre = True
|
|
if g.count('</pre'):
|
|
# XXX this will break if you have </pre><pre> in one line
|
|
in_pre = False
|
|
g = line.rstrip()
|
|
if in_pre:
|
|
_out.write(line)
|
|
else:
|
|
_out.write(g)
|
|
if g.lower() == '<!doctype html>':
|
|
_out.write('\n<!-- readable source @ https://github.com/diveintomark/diveintopython3/blob/master/{0} -->\n'.format(os.path.basename(input_file)))
|
|
|