windows下为Python脚本生成单独可执行程序的脚本（不用为每个脚本写setup.py了）

我们经常要用py2exe将python脚本转换为单独的可执行文件，但是有一个麻烦就是要针对脚本写一个setup.py，在setup.py中指定一些信息，然后运行setup.py py2exe来生成执行文件。为避免每次都要写setup.py，本人编写了一个工具脚本pyfly.py,只要运行pyfly.py yourfile.py就能生成yourfile.exe。还可以为其指定其他参数，比如-i myicon.ico为生成的可执行文件添加图标；-p console指定生成控制台程序，-p windows指定生成windows程序；-c yes指定用upx压缩可执行文件以减小大小，-c no执行不压缩可执行文件；-d out_directory 指定生成的可执行文件的输出目录。（-c yes 选项 需要将upx.exe放置在系统的环境变量PATH所包含的目录中，或者将upx.exe的路径加入PATH中)

运行"pyfly.py pyfly.py", 就立即在当前目录下生成pyfly.exe,将pyfly.exe置于环境变量PATH中能找到的位置，然后在任意目录中就可为你的python脚本生成单文件的可执行程序了。

#-*- coding: utf8 -*-

import os,sys,re,time,uuid,subprocess,shutil,WConio

def printgoodnews(news):

oldcolor=WConio.gettextinfo()[4]

WConio.textcolor(WConio.GREEN)

print news

WConio.textcolor(oldcolor)

def printbadnews(news):

oldcolor=WConio.gettextinfo()[4]

WConio.textcolor(WConio.RED)

print news

WConio.textcolor(oldcolor)

def delete_directory(path):

for root, subdirs,files in os.walk(path,topdown=False):

for file in files:

os.unlink(os.path.join(root,file))

for subdir in subdirs:

os.rmdir(os.path.join(root,subdir))

os.rmdir(path)

usage=r"""usage: pyfly script.py [-d out_directory] [-i iconfile] [-p console|windows] [-c yes|no]

('-c yes': call upx to compress exe)

e.g. : pyfly script.py -d c:\temp -i myicon.ico -p console

: pyfly script.py -d "c:\my directory"

"""

setuppy_console="""\

from distutils.core import setup

import py2exe

setup(

version = "",

description = "",

name = r"%s",

options = {"py2exe": {"compressed": 1,

"optimize": 2,

"ascii": 0,

"bundle_files": 1}},

zipfile = None,

# targets to build

console = [{"script": r"%s", "icon_resources": [(1, r"%s")]} ],

)

"""

setuppy_noicon_console="""\

from distutils.core import setup

import py2exe

setup(

version = "",

description = "",

name = r"%s",

options = {"py2exe": {"compressed": 1,

"optimize": 2,

"ascii": 0,

"bundle_files": 1}},

zipfile = None,

# targets to build

console = [{"script": r"%s", } ],

)

"""

setuppy_windows="""\

from distutils.core import setup

import py2exe

setup(

version = "",

description = "",

name = r"%s",

options = {"py2exe": {"compressed": 1,

"optimize": 2,

"ascii": 0,

"bundle_files": 1}},

zipfile = None,

# targets to build

windows = [{"script": r"%s", "icon_resources": [(1, r"%s")]} ],

)

"""

setuppy_noicon_windows="""\

from distutils.core import setup

import py2exe

setup(

version = "",

description = "",

name = r"%s",

options = {"py2exe": {"compressed": 1,

"optimize": 2,

"ascii": 0,

"bundle_files": 1}},

zipfile = None,

# targets to build

windows = [{"script": r"%s", } ],

)

"""

if __name__=="__main__":

if len(sys.argv)==1:

printgoodnews(usage)

sys.exit(-1)

script=sys.argv[1]

platform='console'

out_directory=os.getcwd()

iconfile=""

upx="no"

if script[-3:].lower()!=".py" or not os.path.exists(script):

printbadnews("%s is not a valid python source file or not exists!")

printgoodnews(usage)

sys.exit(-1)

try:

if len(sys.argv)>2 and len(sys.argv)<=8 and len(sys.argv) % 2==0:

options=[item.lower() for item in sys.argv[2::2]]

for option in options:

if option not in ['-p','-d','-i','-c']:

raise Exception(r'invalid option: %s!' % option)

if '-p' in options:

platform=sys.argv[sys.argv.index('-p')+1].lower()

if platform not in ['console','windows']:

raise Exception(r"%s is not a valid platform, you should use 'console' or 'windows'!" % platform)

if '-d' in options:

out_directory=sys.argv[sys.argv.index('-d')+1]

if not os.path.exists(os.path.abspath(out_directory)):

raise Exception(r"%s is not a valid directory!" % out_directory)

if '-i' in options:

iconfile=sys.argv[sys.argv.index('-i')+1]

if not os.path.exists(os.path.abspath(iconfile)) or iconfile[-4:].lower()!='.ico':

raise Exception(r"%s is not a valid icon file name or not exists!" % iconfile)

if '-c' in options:

upx=sys.argv[sys.argv.index('-c')+1].lower()

if upx not in ['yes','no']:

raise Exception(r"%s option must be set to 'yes' or 'no'!" % "-c")

elif len(sys.argv)==2:

pass

else:

raise Exception('Invalid counts of arguments!')

except Exception,e:

printbadnews(str(e))

printgoodnews(usage)

sys.exit(-1)

try:

tempdir=os.environ['temp']

#create setup.py in temp directory

setupfile=os.path.join(tempdir,str(uuid.uuid1())+".py")

while os.path.exists(setupfile):

setupfile=os.path.join(tempdir,str(uuid.uuid1())+".py")

g=open(setupfile,'wb')

if iconfile!="":

g.write(eval('setuppy_%s' % platform) % (os.path.basename(script),script,iconfile))

else:

g.write(eval('setuppy_noicon_%s' % platform) % (os.path.basename(script),script))

g.close()

del g

if " " in out_directory:

out_directory='"'+out_directory+'"'

cmd=r'python "%s" py2exe -O2 -d %s' %(setupfile,out_directory)

#print cmd

if subprocess.call(cmd)==0:

#delete temporary file and 'build' directory

os.unlink(setupfile)

cwd=os.getcwd()

build=os.path.join(cwd,'build')

bdist_win32=os.path.join(build,'bdist.win32')

delete_directory(bdist_win32)

#if build is a blank directory, delete it; otherwise keep it.

i=0

for root,subdirs,files in os.walk(build,topdown=True):

if subdirs!=[] or files!=[]:

i+=1

break

if i==0:

delete_directory(build)

# compress exe

if upx=='yes':

subprocess.call('upx -9 -v "%s.exe"' % os.path.join(out_directory,os.path.basename(script)[0:-3]))

except Exception,e:

printbadnews(str(e))