mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-26 10:16:27 +00:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import string
|
|
import lxml.html
|
|
import lxml.etree
|
|
from mitmproxy import contentviews
|
|
from mitmproxy.utils import strutils
|
|
|
|
|
|
class ViewPigLatin(contentviews.View):
|
|
name = "pig_latin_HTML"
|
|
prompt = ("pig latin HTML", "l")
|
|
content_types = ["text/html"]
|
|
|
|
def __call__(self, data, **metadata):
|
|
if strutils.is_xml(data):
|
|
parser = lxml.etree.HTMLParser(
|
|
strip_cdata=True,
|
|
remove_blank_text=True
|
|
)
|
|
d = lxml.html.fromstring(data, parser=parser)
|
|
docinfo = d.getroottree().docinfo
|
|
|
|
def piglify(src):
|
|
words = src.split()
|
|
ret = ''
|
|
for word in words:
|
|
idx = -1
|
|
while word[idx] in string.punctuation and (idx * -1) != len(word):
|
|
idx -= 1
|
|
if word[0].lower() in 'aeiou':
|
|
if idx == -1:
|
|
ret += word[0:] + "hay"
|
|
else:
|
|
ret += word[0:len(word) + idx + 1] + "hay" + word[idx + 1:]
|
|
else:
|
|
if idx == -1:
|
|
ret += word[1:] + word[0] + "ay"
|
|
else:
|
|
ret += word[1:len(word) + idx + 1] + word[0] + "ay" + word[idx + 1:]
|
|
ret += ' '
|
|
return ret.strip()
|
|
|
|
def recurse(root):
|
|
if hasattr(root, 'text') and root.text:
|
|
root.text = piglify(root.text)
|
|
if hasattr(root, 'tail') and root.tail:
|
|
root.tail = piglify(root.tail)
|
|
|
|
if len(root):
|
|
for child in root:
|
|
recurse(child)
|
|
|
|
recurse(d)
|
|
|
|
s = lxml.etree.tostring(
|
|
d,
|
|
pretty_print=True,
|
|
doctype=docinfo.doctype
|
|
)
|
|
return "HTML", contentviews.format_text(s)
|
|
|
|
|
|
pig_view = ViewPigLatin()
|
|
|
|
|
|
def start():
|
|
contentviews.add(pig_view)
|
|
|
|
|
|
def done():
|
|
contentviews.remove(pig_view)
|