douban.py

from mitmproxy import command, flow, ctx
from http import client
import pyperclip
import tempfile
from urllib import request, parse
import typing
import json
import re
import os


def create_pastebin(code):
    pass


class DoubanDomain:

    def _save_on_clip(self, content):
        try:
            pyperclip.copy(content)
            ctx.log.info("Copyed %s" % content)
        except pyperclip.PyperclipException as e:
            ctx.log.error(str(e))

    @command.command("douban.save_resp_content")
    def save_resp_content(self, flow: flow.Flow) -> None:
        content = flow.response.content
        fd, name = tempfile.mkstemp(prefix="douban-mitmproxy")
        os.write(fd, content)
        os.close(fd)
        self._save_on_clip(name)

    @command.command("douban.share_resp_content")
    def share_resp_content(self, flow: flow.Flow) -> None:
        content = flow.response.content
        url = create_pastebin(content)
        self._save_on_clip(url)


addons = [
    DoubanDomain()
]