0

gevent(1.0b4) を使用して html ファイルをダウンロードする場合、プログレス バーを使用して進行状況を表示したいと考えています。
以下のコードを書きましたが、コードには常にいくつかのエラーが存在します。誰かが助けてくれることを願っています!

file_path='temp'
url_count=len(urls)

def progress_bar(file_path, file_count): #
    file_count = long(file_count)
    width = 32
    last_count = 0
    try:
        while True:
            if os.path.isdir(file_path): 
                current_count = len(glob.glob1(myPath,"*.html"))
                percentage = current_count*100/file_count
                current_width = width*percentage/100
                sys.stderr.write('% 3d%% [%s%s] %s/s \r' % (percentage, '#'*current_width, ' '*(width-current_width), current_count - last_count))
                last_count = current_count
            time.sleep(1)    
    except:
        sys.stderr.write('100%% [%s]\n' % ('#'*width))

def print_head(url):     
    data = urllib2.urlopen(url)
    htmlFile = open(file_path+'/'+url+'.html', 'w')
    htmlFile.write(data.read())
    htmlFile.close()    
    raise Exception("done!")     

jobs = [gevent.spawn(print_head, url) for url in urls]
x = [g.link_exception(progress_bar,file_path,url_count) for g in jobs] 
gevent.joinall(jobs)

トレースバック

 Traceback (most recent call last):
      File "E:\tt\test.py", line 39, in <module>
        x = [g.link_exception(progress_bar,file_path,url_count) for g in jobs] #
    TypeError: link_exception() takes at most 3 arguments (4 given)
4

1 に答える 1

1

答えはあなたの質問にあります!

link_exception() takes at most 3 arguments (4 given)

このメソッドの定義を見ると、次のことがわかります。

def link_exception(self, callback, SpawnedLink=FailureSpawnedLink):
        """Like :meth:`link` but *callback* is only notified when the greenlet dies because of unhandled exception"""
        self.link(callback, SpawnedLink=SpawnedLink)

したがって、2 つ以上のパラメーターを渡すことはできません。

更新:私があなたを正しく理解していれば、次のようなものを使用できます:

    file_path = os.path.abspath(os.path.dirname(__file__))
    url_count = len(urls)

    def progress_bar(green):
        width = 32
        current_count = getattr(progress_bar, 'current_count', 0) + 1
        percentage = current_count * 100 / url_count
        current_width = width * percentage/100
        print('% 3d%% [%s%s] %s/s \r' % (percentage, '#' * current_width, ' ' * (width - current_width), current_count))
        setattr(progress_bar, 'current_count', current_count)

        url, exc = green.value
        if exc:
            print 'Download {} failed with error {}'.format(url, exc)
        else:
            print 'Download {} success'.format(url)

    def print_head(url):
        exc = None
        try:
            data = urllib2.urlopen(url)
            htmlFile = open(''.join([file_path, '/', clearFileName(url), '.html']), 'wb+')
            htmlFile.write(data.read())
            htmlFile.close()
        except Exception, ex:
            exc = ex
        return url,exc

    def clearFileName(url):
        return url.replace('/', '_').replace(':', '_').replace('.', '_').replace('#', '_').replace('=', '_')

    jobs = [gevent.spawn(print_head, url) for url in urls]
    [g.link(progress_bar) for g in jobs]
    gevent.joinall(jobs)
于 2013-03-04T13:37:35.183 に答える