#  -*- coding:utf-8 -*-
# =============================================================================================================
#
#    brief   :    制作viruswall 的离线升级数据
#
#    author  :    hechangmin
#
#    date    :    2008.6 
#
# =============================================================================================================

import  os
import  threading, zipfile

class  AsyncZip(threading.Thread):
    
     def   __init__ (self, infile, outfile):
        threading.Thread. __init__ (self)        
        self.infile  =  infile
        self.outfile  =  outfile
     def  run(self):
        state  =   ' w '
         if  os.path.isfile(self.outfile)  ==  True:
            state  =   ' a '
        f  =  zipfile.ZipFile(self.outfile, state, zipfile.ZIP_DEFLATED)
        f.write(self.infile)
        f.close()
         print   ' Finished background zip of:  ' , self.infile
        
def  zipDir(src,dst):
    
    initPath  =  os.getcwd()
    
    tempDST  =   os.path.join(os.getcwd(),dst)
    tempSRC  =   os.path.join(os.getcwd(),src)
    os.chdir( tempSRC )
    files  =  os.listdir(os.curdir)
     print  files
     for  file  in  files:
        background  =  AsyncZip(file,dst)
        background.start()
        background.join()     #  Wait for the background task to finish
        
    os.chdir( initPath )
     print  os.getcwd()
# test ok
if   __name__   ==   ' __main__ ' :
    
    zipDir( " D:\\AutoUpdate\\DataDist\\viruswall\\Data\\KSVW-VirusDB\\ " , " d:\\AutoUpdate\\DataDist\\viruswall\\Data\\update\\KSVW-VirusDB.tgz " )