python 脚本 实现 luajit 批量加密

# -*- coding: utf-8 -*-

import os
import subprocess

defaultSrcFolder = '../../XXX/src'
defaultTargetFolder = '../../XXX/src_jit_android'

filenums = 0

# luaJit encrypt src files to target folder
def encryptFile(src, out):
    global filenums
    for i in os.listdir(src):
        path_file = os.path.join(src, i)
        out_file = os.path.join(out, i)
        if os.path.isfile(path_file):
            
            if not os.path.exists(out):
                os.makedirs(out)

            if not os.path.exists(out_file):
                cmdStr = 'luajit -b' + ' ' + path_file + ' ' + out_file
                filenums += 1
                print 'encrypting ', out_file
                subprocess.call(cmdStr, shell=True)

        if os.path.isdir(path_file):
            encryptFile(path_file, out_file)


def del_file(path):
    if not os.path.exists(path):
        return
    for i in os.listdir(path):
        path_file = os.path.join(path, i)
        if os.path.isfile(path_file):
            os.remove(path_file)
        else:
            del_file(path_file)
            os.rmdir(path_file)



if __name__ == '__main__':
    # drag src folder into
    folder = raw_input('drag src folder:')
    if len(folder) == 0:
        folder = defaultSrcFolder

    outfolder = raw_input('input out folder(or use default): ')
    if len(outfolder) == 0:
        outfolder = defaultTargetFolder

    del_file(outfolder)

    encryptFile(folder, outfolder)
    
    print 'encrypt successful, %s files in total ' % filenums