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

引言

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

#!/usr/bin/env python3
# -*-coding: utf-8-*-
# author: Chris
# blog: http://www.cnblogs.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.cnblogs.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 sys
import time
from functools import partial
import os

def 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.cnblogs.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()

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

引言