3

我正在为一个大学项目使用一段自修改代码。

这里是：

import datetime
import inspect
import re
import sys

def main():
    # print the time it is last run
    lastrun = 'Mon Jun  8 16:31:27 2009'

    print "This program was last run at ",
    print lastrun

    # read in the source code of itself
    srcfile = inspect.getsourcefile(sys.modules[__name__])
    f = open(srcfile, 'r')
    src = f.read()
    f.close()

    # modify the embedded timestamp
    timestamp = datetime.datetime.ctime(datetime.datetime.now())
    match = re.search("lastrun = '(.*)'", src)
    if match:
        src = src[:match.start(1)] + timestamp + src[match.end(1):]

    # write the source code back
    f = open(srcfile, 'w')
    f.write(src)
    f.close()

if __name__=='__main__':
    main()

不幸的是，它不起作用。返回错误：

# This is the script's output
This program is last run at  Mon Jun  8 16:31:27 2009
# This is the error message
Traceback (most recent call last):
  File "C:\Users\Rui Gomes\Desktop\teste.py", line 30, in <module>
    main()
  File "C:\Users\Rui Gomes\Desktop\teste.py", line 13, in main
    srcfile = inspect.getsourcefile(sys.modules[__name__])
  File "C:\Python31\lib\inspect.py", line 439, in getsourcefile
    filename = getfile(object)
  File "C:\Python31\lib\inspect.py", line 401, in getfile
    raise TypeError('{!r} is a built-in module'.format(object))
TypeError: <module '__main__' (built-in)> is a built-in module

我会感谢任何解决方案。

4

2 回答 2

4

它在 IDLE 之外运行时运行完美——因此问题不在于您的代码，而在于您执行它的环境。当您在 IDLE 中运行代码的错误部分时，您会得到以下输出：

>>> import inspect
>>> sys.modules[__name__]
<module '__main__' from 'C:\Python26\Lib\idlelib\idle.pyw'>
>>> inspect.getsourcefile(sys.modules[__name__])
'C:\\Python26\\Lib\\idlelib\\idle.pyw'

当你在IDLE中运行它时

# read in the source code of itself
srcfile = inspect.getsourcefile(sys.modules[__name__])
f = open(srcfile, 'r')
src = f.read()
f.close()

您实际上是在尝试修改'C:\\Python26\\Lib\\idlelib\\idle.pyw'... IDLE 不允许您这样做。

它的长短似乎是你写的东西确实有效：但它不能在 IDLE 中运行。

于 2010-06-02T17:17:05.320 回答
3

您可以使用__file__global 属性来获取当前模块的源路径。

从 2.6 文档：

__file__是从文件中加载模块的文件的路径名（如果它是从文件中加载的）。__file__静态链接到解释器的 C 模块不存在该 属性；对于从共享库动态加载的扩展模块，它是共享库文件的路径名。

编辑：

我假设 inspect.getsourcefile() 在检查__main__模块时总是会抛出 TypeError 。只有从交互式解释器运行时才会出现这种情况。我站得更正了。德普。

于 2010-06-02T17:22:39.677 回答