mitmproxy script for injecting script tag for loading weinre

weinre.py

import argparse

class Weinrer:
    def __init__(self, host, port):
        self.host = host
        self.port = port

    def response(self, flow):

        text = flow.response.text

        if len(text) > 0 and hasattr(text, 'find') and text.find('</head>') > -1:
            flow.response.set_text(text.replace(
                '</head>',
                '<script src="http://{self.host}:{self.port}/target/target-script-min.js#anonymous"></script></head>'.format(self=self)))

def start():
    parser = argparse.ArgumentParser()
    parser.add_argument('host', type=str, default='localhost')
    parser.add_argument('port', type=str, default='8080')
    args = parser.parse_args()

    return Weinrer(args.host, args.port)

"""
Usage:
$ mitmproxy -s 'weinre.py 192.168.0.5 8080'
"""