0

我在 stem 中将这个示例与pytube结合起来，以测量我通过 Tor 下载 youtube 视频所需的时间。

这是代码：

import io
import pycurl
import stem.process
from stem.util import term
import pickle
import socks  # SocksiPy module
import socket
import urllib
from pytube import YouTube
import pdb
import time
import string
import random

SOCKS_PORT = 9050
# Set socks proxy and wrap the urllib module

socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, '127.0.0.1', SOCKS_PORT)
socket.socket = socks.socksocket

# Perform DNS resolution through the socket

def getaddrinfo(*args):
  return [(socket.AF_INET, socket.SOCK_STREAM, 6, '', (args[0], args[1]))]

socket.getaddrinfo = getaddrinfo

def id_generator(size=6, chars=string.ascii_uppercase + string.digits):
    return ''.join(random.choice(chars) for _ in range(size))

def query(url):

    try:
        yt = YouTube('https://www.youtube.com/watch?v=5mkm22yO-bs')
        yt.set_filename(id_generator(6))
        video = yt.get('3gp', '144p')
        start = time.time()
        video.download('/tmp/') 
        end = time.time()
        time_ = end - start
        return time_

    except:
        return "Unable to reach %s" % url

def print_bootstrap_lines(line):
    if "Bootstrapped " in line:
        print(term.format(line, term.Color.BLUE))

def main():

    print(term.format("Starting Tor:\n", term.Attr.BOLD))

    tor_process = stem.process.launch_tor_with_config(
        config = {
            'SocksPort': str(SOCKS_PORT),

        },
        init_msg_handler = print_bootstrap_lines,
    )

    url = 'https://www.youtube.com/watch?v=5mkm22yO-bs'
    times = [0]*10

    for i in xrange(10):

        times[i] = query(url)

    tor_process.kill()  # stops tor

    with open('times_yout_jung.pickle', 'wb') as f:
        pickle.dump(times, f)

if __name__ == "__main__":
    main()

但是，它给出了“无法访问 url”错误。当我尝试通过此代码正常路由流量（以查看 pytube 是否正常工作）时，它可以工作：

from pytube import YouTube
import pdb
import time     
import string
import random

def id_generator(size=6, chars=string.ascii_uppercase + string.digits):
    return ''.join(random.choice(chars) for _ in range(size))

def main():


    pdb.set_trace()
    yt = YouTube('https://www.youtube.com/watch?v=5mkm22yO-bs')
    yt.set_filename(id_generator(6))
    video = yt.get('3gp', '144p')
    start = time.time()
    video.download('/tmp/') 
    end = time.time()

    print("Time ", end - start)

if __name__ == "__main__":
    main()

我试图用 pdb 调试代码，但我似乎无法弄清楚可能出了什么问题。有人可以帮忙吗？

4

0 回答 0