用Python写的学校图书馆视频下载工具

学校图书馆视频用webplayer建的，上面的电影更新得比较快，主要还是在局域网内，速度应该可以保证。

但是这个这能在线看，同学都希望能有一个可以下载视频的工具。

我用酷抓6获取了url和http请求头后，用python写了这样一个下载工具，还比较简陋。

刚学Python不久，写的时候都是一边翻着手册的，代码也很乱，凑合着。

import httplib
import string
import os
import os.path
import time

def getfilename(vod_num):
    vod_url="http://vod.ccsu.cn/webmedia/webmedia.das?cmd=1&clientver=4801&prog_id="+vod_num+"&customer_id=2&local=172.16.145.30&proxy=172.16.145.30&uuid=c4cb4544a8c54133-a7ab-0700&osver=Unknown&useragent=9.0.8112.16421&requesttype=1&Offset=0"
    vod_headers={"Host":"vod.ccsu.cn","Accept":"image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, application/vnd.ms-powerpoint, application/vnd.ms-excel, application/msword, application/vgplayer, application/x-shockwave-flash, */*","User-Agent":"Viewgood/1.0 (1; 1; 1)","Connection":"Keep-Alive","Pragma":"no-cache","Accept-Encoding":"gzip, deflate"}
    try:
        conn=httplib.HTTPConnection("vod.ccsu.cn")
        conn.request("GET",vod_url,"",vod_headers)
        res=conn.getresponse()
    finally:
        conn.close()
    tmp=res.read()
    print tmp
    tmp=string.split(tmp,chr(0))
    stype=string.split(tmp[3],"=")
    sname=string.split(tmp[5],"=")
    return sname[1]+"."+stype[1]

def getnetfilesize(firstblock):
    filesize=1000000
    headsize=1024
    headsize=string.find(firstblock,"\r\n\r\n")+4
    ss=string.split(firstblock,"\r\n",2)
    ss=string.split(ss[1],":")
    filesize=int(ss[1])
    return filesize,headsize

def getfullfilesize(vod_num):
    vod_url="/webmedia/webmedia.tfs?cmd=1&clientver=4801&uuid=347f67d\d153c4887-9cf9-08a8&prog_id="\
    +vod_num+\
    "&server_id=1&customer_id=7&proxy=&requesttype=1&Offset=0"
    vod_headers={"Host":"vod.ccsu.cn","Accept":"image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, application/vnd.ms-powerpoint, application/vnd.ms-excel, application/msword, application/vgplayer, application/x-shockwave-flash, */*","User-Agent":"Viewgood/1.0 (1; 1; 1)","Connection":"Keep-Alive","Pragma":"no-cache","Accept-Encoding":"gzip, deflate"}   
    nfsize=0
    hsize=0
    try:
        conn=httplib.HTTPConnection("vod.ccsu.cn")
        conn.request("GET",vod_url,"",vod_headers)
        res=conn.getresponse()
        nfsize,hsize=getnetfilesize(res.read(100))
    finally:
        conn.close()
    return nfsize

def downloadfile(vod_num,offset=0):
    vod_url="/webmedia/webmedia.tfs?cmd=1&clientver=4801&uuid=347f67dd153c4887-9cf9-08a8&prog_id="+vod_num+"&server_id=1&customer_id=7&local=172.16.145.30&proxy=&requesttype=1&Offset="+str(offset)
    vod_headers={"Host":"vod.ccsu.cn","Accept":"image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, application/vnd.ms-powerpoint, application/vnd.ms-excel, application/msword, application/vgplayer, application/x-shockwave-flash, */*","User-Agent":"Viewgood/1.0 (1; 1; 1)","Connection":"Keep-Alive","Pragma":"no-cache","Accept-Encoding":"gzip, deflate"}
    fp=file(savefilename,"ab+")
    try:
        conn=httplib.HTTPConnection("vod.ccsu.cn")
        conn.request("GET",vod_url,"",vod_headers)
        res=conn.getresponse()
        fp.seek(0,2)
        bs=1024*100 #block size
        readsize=offset #size of read
        filetoreadsize=0 #
        longyu=0
        baifenbi=""
        print "saved in:",savefilename
        print "downloading..."
        while 1:  
            block=res.read(bs)
            if block == "":
                break
            if readsize == offset: 
                filetoreadsize,headsize=getnetfilesize(block)
                block=block[headsize:]
                readsize += len(block)
                fp.write(block)
               
            else:
                readsize += len(block)
                fp.write(block)
            baifenbi="%%%.2f"%(readsize*100.0/netfilesize)
            os.system("title "+baifenbi+"  "+str(readsize/1000)+"kb") 

    finally:
        conn.close()
        fp.close()
    print "-"*30
    print "download_size:%db"%(readsize)
    if readsize<netfilesize:
        print "lost of size:%db"%(netfilesize-readsize)

def getsavepath():
    #
    #
    pfp=open("path.txt","r+")
    try:
        ppath=pfp.readline()
    except:
        print "read path.txt error"
        return False
    finally:
        pfp.close()
    if os.path.isdir(ppath) == False:
        print "path error"
        print "use cwd"
        ppath=os.getcwd()
        pfp=open("path.txt","w")
        try:
            pfp.write(ppath)
        except:
            print "crate path.txt error"
            return False
        finally:
            pfp.close()
    return ppath

def main():
    #34778 is small
    print "****vod download****".decode("gbk")
    print "--code by gk"
    vod_num=raw_input("please input an id:")
    print "connecting..."
    global savefilename
    global netfilesize
    savefilename=getsavepath()+"\\"+vod_num+"_"+getfilename(vod_num)
    offset=0
    if os.path.isfile(savefilename):
        offset=os.path.getsize(savefilename)
        print "exist file size:%db"%(offset)
    netfilesize=getfullfilesize(vod_num)
    print "net file size:%db"%(netfilesize)
    logfp=open("vodlog.log","a")
    logfp.writelines(time.strftime('%Y-%m-%d',time.localtime(time.time()))+"  "+vod_num+"  "+savefilename+"\n")
    logfp.close()
    if offset<netfilesize:
        downloadfile(vod_num,offset)
    print "ok"
    os.system("pause")
    
if __name__ == "__main__":
    main()

用GUI2EXE打包成exe文件就可以给同学用了。

以后打算用wxPython写个图形界面