axmol/tools/framework-compile/bin/utils_cocos.py

#!/usr/bin/python
#-*- coding: UTF-8 -*-

import os
import sys
import shutil

if sys.platform == 'win32':
	import _winreg

def execute_command(cmdstring, cwd=None, timeout=None, shell=True):
	""" 执行一个SHELL命令
		封装了subprocess的Popen方法, 支持超时判断，支持读取stdout和stderr
		参数:
		cwd: 运行命令时更改路径，如果被设定，子进程会直接先更改当前路径到cwd
		timeout: 超时时间，秒，支持小数，精度0.1秒
		shell: 是否通过shell运行
		Returns: return_code
		Raises:  Exception: 执行超时
	"""

	import shlex
	import datetime
	import subprocess
	import time

	#if os_is_win32():
	#	cmdstring = convert_to_python_path(cmdstring)

	print("")
	print("Execute command:")
	print(cmdstring)
	print("")

	if shell:
		cmdstring_list = cmdstring
	else:
		cmdstring_list = shlex.split(cmdstring)
	if timeout:
		end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout)

	# 没有指定标准输出和错误输出的管道,因此会打印到屏幕上
	sub = None
	try:
		sub = subprocess.Popen(cmdstring_list, cwd=cwd, stdin=subprocess.PIPE, shell=shell, bufsize=4096)
	except Exception, e:
		print "execute command fail:%s" % cmdstring
		raise e

	# subprocess.poll()方法：检查子进程是否结束了，如果结束了，设定并返回码，放在subprocess.returncode变量中
	while sub.poll() is None:
		time.sleep(0.1)
		if timeout:
			if end_time <= datetime.datetime.now():
				raise Exception("Timeout：%s"%cmdstring)

	if 0 != sub.returncode :
		errStr = "[ERROR] execute command fail:%s" % cmdstring
		print errStr
		raise Exception(errStr)

	return sub.returncode

def get_required_vs_version(proj_file):
		# Now VS2012 is the mini version required
		return "11.0"

def get_vs_cmd_path(vs_reg, proj_path, vs_version):
	# get required vs version
	required_vs_version = get_required_vs_version(proj_path)
	if required_vs_version is None:
		raise Exception("Can't parse the sln file to find required VS version")

	# get the correct available VS path
	needUpgrade = False
	vsPath = None

	if vs_version is None:
		i = 0
		try:
			while True:
				version = _winreg.EnumKey(vs_reg, i)
				try:
					if float(version) >= float(required_vs_version):
						key = _winreg.OpenKey(vs_reg, r"SxS\VS7")
						vsPath, type = _winreg.QueryValueEx(key, version)

						if float(version) > float(required_vs_version):
							needUpgrade = True

						key.close()
						break
				except:
					pass
				i += 1
		except WindowsError:
			pass
	else:
		if vs_version == 2012:
			vs_ver = "11.0"
			needUpgrade = False
		else:
			vs_ver = "12.0"
			needUpgrade = True

		try:
			key = _winreg.OpenKey(vs_reg, r"SxS\VS7")
			vsPath, type = _winreg.QueryValueEx(key, vs_ver)
		except:
			raise Exception("Can't find VS%d" % vs_version)

	if vsPath is None:
		message = "Can't find correct Visual Studio's path in the regedit"
		raise Exception(message)

	commandPath = os.path.join(vsPath, "Common7", "IDE", "devenv")
	return (commandPath, needUpgrade)

def rmdir(folder):
	if os.path.exists(folder):
		if sys.platform == 'win32':
			execute_command("rd /s/q %s" % folder)
		else:
			shutil.rmtree(folder)

def mkdir(folder):
	if not os.path.exists(folder):
		os.makedirs(folder)

def cpdir(source, dest):
	source_dir = source
	dest_dir = dest

	if not os.path.exists(source_dir):
		raise Exception("cpdir source_dir (%s) not exists" % source_dir)

	mkdir(dest_dir)

	for f in os.listdir(source_dir):
		path = os.path.join(source_dir, f)
		if os.path.isfile(path):
			shutil.copy(path, dest_dir)
		elif os.path.isdir(path):
			new_dest = os.path.join(dest_dir, f)
			cpdir(path, new_dest)

def win2unix(filePath):
	try:
		oldfile = open(filePath, "rb+")
		path, name = os.path.split(filePath)
		newfile = open(path + '$' + name, "wb+")

		old = b'\r'
		new = b''

		data = b''
		while (True):
			data = oldfile.read(200)
			newData = data.replace(old, new)
			newfile.write(newData)
			if len(data) < 200:
				break
		newfile.close()
		oldfile.close()

		os.remove(filePath)
		os.rename(path + '$' + name, filePath)
	except IOError as e:
		print(e)

def copy_files_with_cb(src, dst, cb):
	for item in os.listdir(src):
		path = os.path.join(src, item)
		if os.path.isfile(path):
			if cb is None:
				shutil.copy(path, dst)
			else:
				cb(path, dst)
		if os.path.isdir(path):
			new_dst = os.path.join(dst, item)
			if not os.path.isdir(new_dst):
				os.makedirs(new_dst)
			copy_files_with_cb(path, new_dst, cb)