212 lines
8.2 KiB
Python
212 lines
8.2 KiB
Python
#####################################################################################################################################################
|
|
# Copyright (c) 2006 Activision, Inc
|
|
# Confidential Information of Activision.
|
|
# Not for disclosure or distribution without Activision's prior written consent.
|
|
#####################################################################################################################################################
|
|
|
|
|
|
import os
|
|
import sys
|
|
import shutil
|
|
import stat
|
|
import time
|
|
import re
|
|
import logging
|
|
|
|
g_base_install_dir = "" #PC install Dir
|
|
g_base_game_dir = "" #P4 COD Root, env var? cmdline?
|
|
|
|
copy_directories = []
|
|
|
|
#################################################################################################################################################
|
|
# Flush the standard output and error streams
|
|
def flushStdStreams():
|
|
if 'flush' in dir(sys.stdout):
|
|
sys.stdout.flush() # Serialize stdout and stderr
|
|
if 'flush' in dir(sys.stderr):
|
|
sys.stderr.flush()
|
|
|
|
def syncFile( srcFilename, destFilename ):
|
|
if not os.path.exists(srcFilename):
|
|
logging.warning("Src file does not exist!: " + (srcFilename) )
|
|
return False
|
|
|
|
if not os.path.exists(destFilename):
|
|
if not os.path.exists( os.path.dirname(destFilename) ):
|
|
os.makedirs(os.path.dirname(destFilename) )
|
|
|
|
shutil.copyfile( srcFilename, destFilename )
|
|
shutil.copystat( srcFilename, destFilename )
|
|
os.chmod(destFilename,stat.S_IREAD|stat.S_IWRITE)
|
|
logging.info('Getting: %s' %destFilename)
|
|
return True
|
|
else:
|
|
srcStats = os.stat(srcFilename)
|
|
dstStats = os.stat(destFilename)
|
|
srcTime = time.localtime(srcStats[stat.ST_MTIME])
|
|
dstTime = time.localtime(dstStats[stat.ST_MTIME])
|
|
if ( srcTime != dstTime ):
|
|
|
|
try:
|
|
shutil.copyfile( srcFilename, destFilename )
|
|
shutil.copystat( srcFilename, destFilename )
|
|
os.chmod(destFilename,stat.S_IREAD|stat.S_IWRITE)
|
|
logging.info( "Updating: %s" %destFilename )
|
|
return True
|
|
except:
|
|
if os.path.exists(srcFilename):
|
|
logging.warning( "Cannot Copy File '%s'!" %(srcFilename) )
|
|
else:
|
|
logging.warning( "File '%s' does not exist!" %(srcFilename) )
|
|
return False
|
|
else:
|
|
logging.debug( "Up to date: " + destFilename )
|
|
|
|
return False
|
|
|
|
|
|
#################################################################################################################################################
|
|
# Recursively copy an entire directory tree
|
|
# @param src src folder
|
|
# @param dst dst path
|
|
# @param recursive recursive copy or not (True by default)
|
|
# @param maskedStrings list of strings used to mask files (ie: files containing any of the maskedStrings won't be copied)
|
|
# @param filterFunctor optional function object for filtering out source files
|
|
def copytree(src, dst, recursive=True, regMatchExpression=None):
|
|
totalFilesCopied = 0
|
|
if os.path.isdir(src):
|
|
|
|
numFiles = 0
|
|
fileList = os.listdir(src)
|
|
|
|
if not os.path.isdir(dst):
|
|
os.makedirs(dst)
|
|
|
|
for name in fileList:
|
|
srcname = os.path.join(src, name)
|
|
dstname = os.path.join(dst, name)
|
|
|
|
try:
|
|
if os.path.isdir(srcname):
|
|
if recursive:
|
|
totalFilesCopied += copytree(srcname, dstname, recursive, regMatchExpression)
|
|
else:
|
|
doCopy = True
|
|
|
|
if regMatchExpression != None:
|
|
doCopy = False
|
|
m = regMatchExpression.match(srcname)
|
|
if m:
|
|
doCopy = True
|
|
|
|
if doCopy:
|
|
if syncFile(srcname,dstname) :
|
|
totalFilesCopied+=1
|
|
|
|
numFiles+=1
|
|
|
|
if numFiles>100:
|
|
numFiles = 0
|
|
print ".",
|
|
flushStdStreams()
|
|
except (IOError, os.error), why:
|
|
logging.error( "Can't copy %s to %s: %s" % (`srcname`, `dstname`, str(why)) )
|
|
|
|
return totalFilesCopied
|
|
|
|
def Usage():
|
|
print 'Usage: CopyAssets.py [srcDir destDir config]'
|
|
print 'Default: srcDir=IW_PROJECT_GAMEDIR, destDir=Current WOrking Directory, config=CopyAssets.config\n\n'
|
|
|
|
if __name__ == "__main__":
|
|
|
|
configFile = ""
|
|
|
|
logging.basicConfig(level=logging.DEBUG,
|
|
format='%(asctime)s %(levelname)s %(message)s',
|
|
filename='copyAssets.log',
|
|
filemode='w')
|
|
|
|
arg = sys.argv[1:]
|
|
if len(arg) == 0:
|
|
g_base_install_dir = os.getcwd()
|
|
if g_base_install_dir.endswith('bin'):
|
|
g_base_install_dir = os.path.abspath(os.path.join( g_base_install_dir, '../'))
|
|
|
|
if 'cod5modtools' in g_base_install_dir:
|
|
logging.error( 'Must run from PC Release directory, not the DEPOT!' )
|
|
print 'Must run from PC Release directory, not the DEPOT!'
|
|
sys.exit(1)
|
|
|
|
g_base_game_dir = os.getenv('IW_PROJECT_GAMEDIR')
|
|
if g_base_game_dir == "":
|
|
logging.error( 'Must define IW_PROJECT_GAMEDIR env var or specify directory' )
|
|
print 'Must define IW_PROJECT_GAMEDIR env var or specify directory'
|
|
Usage()
|
|
sys.exit(1)
|
|
|
|
configFile = 'CopyAssets.config'
|
|
if not os.path.isfile(configFile):
|
|
logging.error( 'Could not find config file!' )
|
|
print 'Could not find config file!'
|
|
Usage()
|
|
sys.exit(1)
|
|
else:
|
|
FILE = open(configFile,'r')
|
|
exec(FILE)
|
|
|
|
elif len(arg) == 3:
|
|
g_base_game_dir = os.path.expandvars(arg[0])
|
|
if not os.path.isdir(g_base_game_dir):
|
|
logging.error( 'Could not find source directory! ' + g_base_game_dir )
|
|
print 'Could not find source directory! ' + g_base_game_dir
|
|
Usage()
|
|
sys.exit(1)
|
|
|
|
g_base_install_dir = os.path.expandvars(arg[1])
|
|
|
|
configFile = os.path.expandvars(arg[2])
|
|
if not os.path.isfile(configFile):
|
|
logging.error( 'Could not find config file!' )
|
|
print 'Could not find config file!'
|
|
Usage()
|
|
sys.exit(1)
|
|
else:
|
|
FILE = open(configFile,'r')
|
|
exec(FILE)
|
|
else:
|
|
Usage()
|
|
sys.exit(1)
|
|
|
|
|
|
logging.info( 'Src Dir: ' + g_base_game_dir )
|
|
logging.info( 'Dst Dir: ' + g_base_install_dir )
|
|
logging.info( 'Config : ' + configFile )
|
|
totalFileCopied = 0
|
|
logging.info( 'Copying files...' )
|
|
print 'Copying files...'
|
|
for item in copy_directories:
|
|
baseName = os.path.basename(item[0])
|
|
srcPath = os.path.join( g_base_game_dir, item[0] )
|
|
dstPath = os.path.join( g_base_install_dir, item[1] )
|
|
recursiveCopy = True
|
|
|
|
if len(item) >= 3:
|
|
recursiveCopy = item[2]
|
|
|
|
if len(item) == 4:
|
|
reExp = re.compile(item[3])
|
|
totalFileCopied += copytree(srcPath,dstPath,recursiveCopy,regMatchExpression=reExp)
|
|
else:
|
|
|
|
if os.path.isfile(srcPath):
|
|
logging.info( 'Copying File' + srcPath + ' -> ' + dstPath )
|
|
if syncFile(srcPath,dstPath):
|
|
totalFileCopied += 1
|
|
elif os.path.isdir(srcPath):
|
|
logging.info( 'Copying Dir' + srcPath + ' ----> ' + dstPath )
|
|
totalFileCopied += copytree(srcPath,dstPath,recursiveCopy)
|
|
|
|
logging.info( 'Copied ' + str(totalFileCopied) + ' files!' )
|
|
|
|
|