[FIXED] Re: urllib2.urlopen Progress bar

[Ritesh Raj Sarraf]

-----BEGIN PGP SIGNED MESSAGE-----
Hash: SHA1

Calling urllib2.urlopen the below mentioned way fixed my problem.

try:
        os.chdir (sSourceDir)
        block_size = 4096
        i = 0
        count = 0
        temp = urllib2.urlopen(sUrl)
        headers = temp.info()
        size = int(headers['Content-Length'])
        data = open (sFile, 'wb')
        while i < size:
                data.write (temp.read(block_size))
                i += block_size
                counter += 1
                progressbar.myReportHook(counter, block_size, size)
        print "\n"
        data.close()
        temp.close()


Thanks,
Ritesh

Ritesh Raj Sarraf on Sunday 15 Jan 2006 12:55 wrote:

> -----BEGIN PGP SIGNED MESSAGE-----
> Hash: SHA1
> 
> Hi,
> 
> In urllib.urlretrieve I can use reporthook to implement a progress bar.
> 
> But in urllib2.urlopen I can't.
> I have to use urllib2.urlopen because it throws exceptions which are
> important to be handled where as urlretrieve is dumb.
> 
> Is there a way to implement urllib2.urlopen to enable reporthook ?
> 
> progress.py code:
> 
> #!/usr/bin/env python
> 
> import time, urllib2, urllib
> 
> class progressBar:
>     def __init__(self, minValue = 0, maxValue = 10, totalWidth=12):
>         self.progBar = "[]"   # This holds the progress bar string
>         self.min = minValue
>         self.max = maxValue
>         self.span = maxValue - minValue
>         self.width = totalWidth
>         self.amount = 0       # When amount == max, we are 100% done
>         self.updateAmount(0)  # Build progress bar string
> 
>     def updateAmount(self, newAmount = 0):
>         if newAmount < self.min: newAmount = self.min
>         if newAmount > self.max: newAmount = self.max
>         self.amount = newAmount
> 
>         # Figure out the new percent done, round to an integer
>         diffFromMin = float(self.amount - self.min)
>         percentDone = (diffFromMin / float(self.span)) * 100.0
>         percentDone = round(percentDone)
>         percentDone = int(percentDone)
> 
>         # Figure out how many hash bars the percentage should be
>         allFull = self.width - 2
>         numHashes = (percentDone / 100.0) * allFull
>         numHashes = int(round(numHashes))
> 
>         # build a progress bar with hashes and spaces
>         self.progBar = "[" + '#'*numHashes + ' '*(allFull-numHashes) + "]"
> 
>         # figure out where to put the percentage, roughly centered
>         percentPlace = (len(self.progBar) / 2) - len(str(percentDone))
>         percentString = str(percentDone) + "%"
> 
>         # slice the percentage into the bar
>         #self.progBar = self.progBar[0:percentPlace] + percentString +
> self.progBar[percentPlace+len(percentString):]
>         self.progBar = self.progBar[0:percentPlace] +
> self.progBar[percentPlace+len(percentString):] + percentString
> 
>     def __str__(self):
>         return str(self.progBar)
>         
> def myReportHook(count, blockSize, totalSize):
>     import sys
>     global prog
> 
>     if prog == "":
>             prog = progressBar(0,totalSize,50)
>     #print count, blockSize, totalSize
>     #prog = progressBar(0, totalSize, 77)
>     prog.updateAmount(count*blockSize)
>     sys.stdout.write (str(prog))
>     sys.stdout.write ("\r")
>     #print count * (blockSize/1024) , "kb of " , (totalSize/1024) , "kb
> downloaded.\n"
> prog = ""
> sFile = "new.rpm"
> sUrl="http://localhost/new.rpm"
> urllib.urlretrieve(sUrl, sFile, reporthook=myReportHook)
> print "\n\n"
> #temp = urllib2.urlopen(sUrl)
> #lastval = int(temp.headers['Content-Length'])
> #prog = progressBar(0, lastval, 77)
> #
> #for x in range(101):
> #    prog.updateAmount(x)
> #    print prog, "\r", time.sleep(0.5)
> #
> #data = open(sFile,'wb')
> #data.write(temp.read())
> #data.close()
> #temp.close()
> 
> 
> Regards,
> Ritesh
> - --
> Ritesh Raj Sarraf
> RESEARCHUT -- http://www.researchut.com
> Gnupg Key ID: 04F130BC
> "Stealing logic from one person is plagiarism, stealing from many is
> research."
> "Necessity is the mother of invention."
> -----BEGIN PGP SIGNATURE-----
> Version: GnuPG v1.4.2 (GNU/Linux)
> 
> iD8DBQFDyfkM4Rhi6gTxMLwRAlmMAJ9yePv6DvXCcnUb0DYhWSuIonciSACgrxDj
> aKSGkfg8I1T/q3r6zWLTXmw=
> =uomB
> -----END PGP SIGNATURE-----
> 

- -- 
Ritesh Raj Sarraf
RESEARCHUT -- http://www.researchut.com
"Stealing logic from one person is plagiarism, stealing from many is
research."
"Necessity is the mother of invention."
-----BEGIN PGP SIGNATURE-----
Version: GnuPG v1.4.2 (GNU/Linux)

iD8DBQFDyicF4Rhi6gTxMLwRAldhAJ9cdnma+FerMVxe/EAG+eokL1POdQCgr5A3
ZyMQFN96Yrkp5frgHPYUnfU=
=Y9wS
-----END PGP SIGNATURE-----