首页 > 学院 > 开发设计 > 正文

记录Python处理文件因文件流没有关闭而出现的诡异问题

2019-11-14 17:01:44
字体:
来源:转载
供稿:网友

引言

最近在学习Python,每次在新建Python文件时都会在文件的开头部分加入如下的信息,然后将文件添加可执行权限。自然,这个用程序自动生成最好不过了。所以,一开始就使用了Shell脚本实现一个版本,基本工作正常。今天就想着用Python实现一个对应的版本吧,于是乎就开始做了;虽然很简单的逻辑,很简单的代码,不过还是让我这个新手吃了一点亏。虽然比较曲折,但是还是可以工作了。以下把遇到的问题即解决办法记录下来。

#!/usr/bin/env python3# -*-coding: utf-8-*-# author: Chris# blog: http://www.VEVb.com/chriscabin/# create_time: 2015-09-05 10:01:42'''Add description for this module.'''__author__ = "Christopher Lee"

问题一 Python运行linux shell 命令

主要有以下几种(将在学习后期做一个系统性的总结):

  • os.system()
  • os.popen()
  • subPRocess.call()

问题二 为何使用vim打开“创建”后的文件却并没有内容写入呢

最初版本的实现代码示例如下:

def create_py_file():    file_name='demo.py'    # 判断参数个数    args = sys.argv        if len(args) == 2:        file_name = args[1] + '.py'    if len(args) <= 2:        try:            with open(file_name, mode='w') as out_file:                # 向文件中写入信息                custom_print = partial(print, file=out_file)                custom_print('#!/usr/bin/env python3')                custom_print('# -*-coding: utf-8-*-')                # 此处省略。。。                                # 打开文件                command = 'vim %s +11' % (file_name)                os.system(command)        except Exception as e:            print(e.message)    else:        print("Too many arguments.")if __name__ == '__main__':    create_py_file()

现象说明

  • 本以为在custom_print执行后,文件中已经写入相关信息了,但是在执行完vim file_name命令后发现并没有内容。非常纳闷

寻找原因

  • 仔细分析后,才发现实际上在执行打开文件命令时,之前的信息并没有写入到新的文件中。因为在with语句块结束前,文件流并没有被关闭,也就没有保存到磁盘中,所以才导致问题的发生。(注意上面的代码的18、19行,明显还在with块中)

解决办法

  • 其实解决办法超级简单,只需要将18、19行代码往前缩进即可!

最终版本的实现代码

  • 最终版本的代码可以完成预期的要求:可以创建一个新的文件(如果文件存在,则直接打开),向其中添加头部的信息;最后给文件添加可执行权限,并且使用vim打开。
  • 只需要把文件移动到/usr/local/bin/,并且命名为new即可。这样,只需要在终端中运行new demo就可以轻松创建一个新的python文件啦!
#!/usr/bin/env python3# -*-coding: utf-8-*-# author: Chris# blog: http://www.VEVb.com/chriscabin/# create_time: 2015-09-05 10:25:46'''This is a python script, which aims to create a new python file automaticly.'''__author__ = "Christopher Lee"import sysimport timefrom functools import partialimport osdef create_py_file():    file_name = 'demo.py'    # 判断参数个数    args = sys.argv    if len(args) == 2:        file_name = args[1]    if len(args) <= 2:        try:            if os.path.exists(file_name):                print("File exists.")                command = 'vi %s' % file_name                os.system(command)                return None            with open(file_name, mode='w') as out_file:                # 向文件中写入信息                custom_print = partial(print, file=out_file)                custom_print('#!/usr/bin/env python3')                custom_print('# -*-coding: utf-8-*-')                custom_print('# author: Chris')                custom_print('# blog: http://www.VEVb.com/chriscabin/')                custom_print('# create_time:', time.strftime('%Y-%m-%d %H:%M:%S'))                custom_print('')                custom_print("'''/nAdd description for this module./n'''")                custom_print('')                custom_print('__author__ = "Christopher Lee"')                custom_print('')            # 当文件流关闭后再进行操作            # 添加可执行权限            command = 'chmod a+x %s' % file_name            os.system(command)            command = 'vim %s +12' % (file_name)            os.system(command)        except Exception as e:            print(e.message)    else:        print("Too many arguments.")if __name__ == '__main__':    create_py_file()

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表