Python/pyshell_w.pyw at main · fufvocdi/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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import sys,os, traceback,marshal
import runpy
try: # 用于打开pyc文件
    from importlib._bootstrap_external import MAGIC_NUMBER
except ImportError:
    from importlib._bootstrap import MAGIC_NUMBER # Python 3.4等较低版本

# 默认的环境变量
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():
            try:co=compile(code,'<PyShell #{}>'.format(line),'single')
            except SyntaxError:
                co=compile(code,'<PyShell #{}>'.format(line),'exec')
            exec(co, vars_)
    except BaseException:handle()

def run_file(filename):
    code=open(filename,'rb').read()
    if filename.lower().endswith(".pyc"):
        if code[:4] != MAGIC_NUMBER:
            warnings.warn("警告: 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:
        if isinstance(code,str):
            co=compile(code,filename,"exec")
        else:
            co=code
        exec(co,vars_)
    except BaseException:
        # 避免打包为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:
            if os.path.isfile(sys.argv[0]):
                app_path=sys.argv[0]
            else:
                app_path=sys.executable
            autoexecfile=os.path.split(app_path)[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:# 检查有无提供代码
                code=sys.argv[2]
                sys.argv=[sys.argv[1]]+sys.argv[3:] # 原版python解释器中，第一项为“-c”
                try:exec_code(code)
                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()