'''
Created on Feb 22, 2013
@author: changxue
@summary: aim: generate jar file
path: 1. copy original jar to target dir
2. java to class, class to jar to target dir
'''
import os, shutil
from common_util import _printlog, _exclude, _add_files_to_list, _execute_cmd
def copy_original_jar(src_path, jar_dir):
_printlog('############## copy jar starting ##############\n')
jar_list = []
os.chdir(src_path)
file_list = os.listdir(src_path)
for f in file_list:
archive_file_path = os.path.join(src_path, f)
try:
if not os.path.isfile(archive_file_path):
raise Exception('%s is not a file.\n'%f)
else:
if f.endswith('.jar'):
jar_file = os.path.join(jar_dir, f)
if os.path.exists(jar_file):
raise Exception('WARNING: File[%s] already exists.\n' % jar_file)
else:
shutil.copy2(archive_file_path, jar_dir)
_printlog('Copy file[%s] to %s\n'% archive_file_path, jar_dir)
jar_list.append(f)
except Exception, e:
_printlog(e)
continue
_printlog('total_count=%s\n'%len(file_list))
_printlog('success_count=%s\n'%len(jar_list))
_printlog('success_list=\n%s'%'\n'.join(jar_list))
_printlog('############## jar_list end ##############\n')
def generate_class_all(src_path, target_dir):
''' make all java files in each sub-dir in src_path into .class files which is placed in target_dir.'''
_printlog('############## Generate class starting ##############\n')
success_list = []
fail_list = []
operate_list = _exclude(src_path)
for dirname in operate_list:
try:
full_path = os.path.join(src_path, dirname)
java_file_list = _add_files_to_list(full_path, '.*\.java$')
if not java_file_list:
_printlog("WARNING: no java file in %s\n"%(full_path))
else:
dest_full_path = os.path.join(target_dir, dirname)
if os.path.exists(dest_full_path):
_printlog('WARNING: directory already exist: %s\n'%dest_full_path)
else:
os.makedirs(dest_full_path)
cmd = 'javac -d %s %s' % (dest_full_path, ' '.join(java_file_list)) # would fail if the list is too large, could instead of *.java
_execute_cmd(cmd)
success_list.append(dirname)
except Exception, e:
_printlog("WARNING ON: make class file: %s\n%s\n"%(full_path, e))
fail_list.append(dirname)
continue
_printlog('total_count=%s\n'%len(operate_list))
_printlog('success_count=%s\n'%len(success_list))
_printlog('success_list=\n%s'%'\n'.join(success_list))
_printlog('fail_count=%s\n'%len(fail_list))
_printlog('fail_list=\n%s'%'\n'.join(fail_list))
_printlog('############## Generate class end ##############\n')
def class_to_jar(dirname, full_path, target_dir):
''' generate .jar file from .class files and place it to target_dir.
notice: full path include dirname and do nothing if .jar file is already existed.'''
f = os.path.join(target_dir, '%s.jar'%dirname)
if os.path.exists(f):
raise Exception('WARNING: file already exists: %s\n'%f)
# os.remove(f)
# _printlog('WARNING: replace file %s' % f)
class_file_list = _add_files_to_list(full_path, '.*\.class$')
if not class_file_list:
_printlog('ERROR: no .class file in %s' % (full_path))
else:
os.chdir(full_path)
command = 'jar cf %s.jar' % (dirname)
start = len(full_path)+1
for f in class_file_list:
command = "%s %s" % (command, f[start:])
_execute_cmd(command)
shutil.move('%s.jar'%dirname, target_dir)
os.remove('%s.jar'%dirname)
def generate_jar_all(src_path, target_dir):
''' make all .class files in each sub-dir in src_path into .jar file which is placed in target_dir.'''
_printlog('############## Make jar starting ##############\n')
operate_list = _exclude(src_path)
success_list = []
fail_list = []
for dirname in operate_list:
full_path = os.path.join(src_path, dirname)
try:
class_to_jar(dirname, full_path, target_dir)
success_list.append(dirname)
except Exception, e:
_printlog("WARNING ON generate_jar_all: %s\n%s\n"%(full_path, e))
fail_list.append(dirname)
continue
_printlog('total_count=%s\n'%len(operate_list))
_printlog('success_count=%s\n'%len(success_list))
_printlog('success_list=\n%s'%'\n'.join(success_list))
_printlog('fail_count=%s\n'%len(fail_list))
_printlog('fail_list=\n%s'%'\n'.join(fail_list))
_printlog('############## Make jar end ##############\n')
def toJar(archive_dir, extract_dir, jar_dir):
''' Main Entry '''
copy_original_jar(archive_dir, jar_dir)
bin_path = os.path.join(extract_dir, 'bin')
generate_class_all(os.path.join(extract_dir, 'src'), bin_path)
generate_jar_all(bin_path, jar_dir)
if __name__ == '__main__':
usage = '''Usage:
\tgenerateJar.py copy_jar
\tgenerateJar.py to_class
\tgenerateJar.py archive_path extract_dir jar_dir
'''
import sys
argv_len = len(sys.argv)
src_path = 'C:\\works\\workload_src'
target_dir = "C:\\works\\workload_out"
jar_dir = "C:\\ProGuard\\proguard4.8\\proguard4.8\\lib"
if argv_len == 1:
toJar(src_path, target_dir, jar_dir)
elif argv_len == 2:
option = sys.argv[1]
if option == 'copy_jar':
copy_original_jar(src_path, jar_dir)
elif option == 'to_class':
generate_class_all(os.path.join(target_dir, 'src'), os.path.join(target_dir, 'bin'))
else:
print usage
sys.exit()
elif argv_len == 4:
toJar(sys.argv[1], sys.argv[2], sys.argv[3])
else:
print usage
sys.exit()
