File: fake_downloader.py

package info (click to toggle)
chromium-browser 57.0.2987.98-1~deb8u1
  • links: PTS, VCS
  • area: main
  • in suites: jessie
  • size: 2,637,852 kB
  • ctags: 2,544,394
  • sloc: cpp: 12,815,961; ansic: 3,676,222; python: 1,147,112; asm: 526,608; java: 523,212; xml: 286,794; perl: 92,654; sh: 86,408; objc: 73,271; makefile: 27,698; cs: 18,487; yacc: 13,031; tcl: 12,957; pascal: 4,875; ml: 4,716; lex: 3,904; sql: 3,862; ruby: 1,982; lisp: 1,508; php: 1,368; exp: 404; awk: 325; csh: 117; jsp: 39; sed: 37
file content (37 lines) | stat: -rw-r--r-- 1,149 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
#!/usr/bin/python
# Copyright (c) 2014 The Native Client Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Fake downloader which can be used in place of http_download."""

import os
import shutil
import urllib2

class FakeDownloader(object):
  """Testing replacement for http_download that copies files."""

  def __init__(self, copy_func=None):
    self._urls = {}
    if copy_func is None:
      self._copy_func = shutil.copyfile
    else:
      self._copy_func = copy_func
    self._download_count = 0

  def StoreURL(self, url, filepath):
    """Stores a known url into the downloader to download."""
    self._urls[url] = filepath

  def Download(self, url, target, username=None, verbose=True, password=None,
               logger=None):
    if url not in self._urls:
      raise urllib2.HTTPError(url, 404,
                              'Fake Downloader retrieved invalid URL',
                              [], None)
    self._copy_func(self._urls[url], target)
    self._download_count += 1

  def GetDownloadCount(self):
    return self._download_count