Python/pyshell_w.pyw at main · YuZhouRen86/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import sys,os,traceback,marshal
import runpy

vars_={"__annotations__":{},"__builtins__":__builtins__,"__doc__":None,
      "__loader__":__loader__,"__name__":"__main__","__package__":None,
      "__spec__":None,"sys":sys,"os":os}
_TITLE="PyShell"
__version__="1.3.3"

def exec_code(code,line=0):
    # 执行代码
    try:
        if code.strip():
            exec(compile(code,'<PyShell_w>'.format(line),'single'),vars_)
    except BaseException:handle()

def run_file(filename):
    code=open(filename,'rb').read()
    if filename.endswith(".pyc"):
        code=code[16:] if code[16]==227 else code[12:]
        code=marshal.loads(code)
    else:
        code=code.decode(encoding='utf-8').strip('\ufeff')
    backup=sys.argv
    sys.argv=sys.argv[1:]
    sys.path.append(os.path.split(filename)[0])
    vars_["__file__"]=filename
    try:
        exec(code,vars_)
    except:
        # 避免打包为pyinstaller出现错误failed to execute script ...
        try:handle()
        except:pass
    sys.argv=backup

def in_pythonw(): # 判断是否为pythonw(无控制台窗口)的环境
    return "pythonw.exe" in sys.executable or "pyshell_w.exe" in sys.executable

def handle():
    traceback.print_exc()

def help():
    print("""用法 Usage:\n
    pyshell_w 某py,pyw或pyc文件: 运行脚本。
    pyshell_w -c <Python代码> : 运行指定的Python指令。
    pyshell_w -m <模块名称> : 运行一个模块。
    pyshell_w <其他参数> --no-autoexec : 不自动运行autoexec.py。
    pyshell_w -h (或 /?): 显示此帮助信息。
""")

def main():
    # 执行autoexec.py
    if '--no-autoexec' not in sys.argv:
        try:
            autoexecfile=os.path.split(sys.argv[0])[0]+"\\autoexec.py"
            run_file(autoexecfile)
        except BaseException:handle()
    else:sys.argv.remove('--no-autoexec') # 便于处理其他参数


    # 处理命令行参数
    if len(sys.argv)>1:
        if sys.argv[1]=="-c":
            if len(sys.argv)>=3:# 检查有无提供代码
                try:exec_code(sys.argv[2])
                except BaseException:handle()
            else:help()
        elif sys.argv[1]=="-m":
            if len(sys.argv)>=3:
                global vars_
                modname=sys.argv[2]
                del sys.argv[1:3]
                try:vars_ = runpy.run_module(modname,init_globals=vars_,
                                             run_name='__main__')
                except BaseException:handle()
            else:help()
        elif sys.argv[1] in ('/?','-?','-h','--help'):
            help()
        else:
            filename=os.path.realpath(sys.argv[1])
            run_file(filename)

    # pyshell_w中移除了交互功能

if __name__=="__main__":main()