pomf.py 文件源码

def upload(url):
    cclive = subprocess.Popen("cclive --support | xargs | tr ' ' '|'", stdout=subprocess.PIPE, shell=True)
    (cclive_formats, err) = cclive.communicate()

    re_youtube = "youtube|youtu\.be|yooouuutuuube"
    search = ".*(?:{}|{}).*".format(re_youtube, cclive_formats)
    try:
        if re.match(search, url, re.I):
            if re.match(".*(?:{}).*".format(re_youtube), url, re.I):
                cmd = "youtube-dl --quiet --recode-video webm --format webm/mp4 --output /tmp/%\(id\)s.webm {}".format(url)
                p = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
                yt = ".*(?:youtube.*?(?:v=|/v/)|youtu\.be/|yooouuutuuube.*?id=)([-_a-zA-Z0-9]+).*"
                file = "/tmp/{}.webm".format(re.match(yt, url, re.I).group(1))
            else:
                cmd = "cclive --quiet -f fmt43_360p {} --O /tmp/pomf.webm --exec 'echo -n %f'".format(url, "/tmp")
                p = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
                (file, err) = p.communicate()
        else:
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:37.0) Gecko/20100101 Firefox/37.0',
                'Referer': 'http://www.amazon.com/'
            }

            extension = guess_extension(guess_type(url)[0]).replace('jpe','jpg')
            temp = tempfile.NamedTemporaryFile(suffix=extension)
            content = requests.get(url).content
            temp.write(content)
            file = temp.name

        fh = open(file, "rb")
        fh.seek(0)

        content = requests.post(url="http://pomf.se/upload.php", files={"files[]":fh})
        if not content.status_code // 100 == 2:
            raise Exception("Unexpected response {}".format(content))
        return "http://a.pomf.se/{}".format(content.json()["files"][0]["url"])
    except Exception as e:
        return "Error: {}".format(e)