Python学习之文件的创建与写入详解

在前面章节我们通过 os包学习了如何创建、读取一个文件夹，但是并没有学习如何创建、读写一个文件，接下来我们就学习关于文件的处理。当我们学习完文件处理之后，就可以随意读写文件。

内置函数 - open 获取文件对象

open() 函数

open()函数是是python的读写文件的基本函数，它可以生成文件对象可以创建，也可以操作文件的读写。

用法：

open(path, mode)

参数说明：

path：文件路径

mode：操作模式；比如读文件的模式，写文件的模式。下文有详细介绍

返回值：

文件对象

示例用法如下：

file = open('d://test.txt', 'w')

# >>> 代码释义：使用 open() 函数 对 D 磁盘下的 'test.txt' 文件执行写入的操作 ( w 为写入的操作 ) 并赋值给变量 file

利用文件对象进行创建与写入

文件操作的写入模式

文件对象的写入操作方法

注意：操作文件完成后，必须使用 close 方法！！！不然可能会造成内存占用，如果反复生成文件对象进行写入操作，可能会造成内存溢出，我们的程序就会出问题了。

使用 w 模式 执行 write 方法

import os
import os.path

current_path = os.getcwd()
# print(current_path)

file_test = open(current_path + '/' + 'test.txt', 'w', encoding='utf-8')    
# 注意如果是在WIN系统，在写入中文时，需要设置编码格式;如果不是WIN系统，则不需要设置编码格式

file_test.write('Python 是一门优雅的编程语言')
file_test.close()

上文脚本我们提及，如果是在WIN系统，在写入中文时，需要设置编码格式。如果不设置编码格式，虽然不会报错，也可以写入成功。但是打开写入的文件，就如同下图：

我们继上面演示的案例，在 text.txt 文件的基础上，继续演示 w 模式的功能。

import os
import os.path

current_path = os.getcwd()
# print(current_path)

file_test = open(current_path + '/' + 'test.txt', 'w', encoding='utf-8')
file_test.write('Python 是一门优雅的编程语言')
file_test.close()

file_path = os.path.join(current_path, 'test.txt')
file_test = open(file_path, 'w', encoding='utf-8')
file_test.write('Python 不仅是一门优雅的编程语言，还不掉头发')

可以看到我们针对 text.txt 文件的基础上，继续演示的 w 模式 新传入的 Python 不仅是一门优雅的编程语言，还不掉头发 已经覆盖了 text.txt 文件原本的内容。

使用 w+ 模式 执行 write 方法

这里我们先 利用 w+ 模式 写入内容

import os.path

current_path = os.getcwd()

file_path = os.path.join(current_path, 'test.txt')
file_test = open(file_path, 'w+', encoding='utf-8')
file_test.write('人生苦短 我用Python')
file_test.close()

区别于 w 模式，w+ 模式 不仅可以创建文件，还可以读取文件。下面我看看看 如何 读取文件。（需要注意的是，读取的操作需要在终端操作，Pycharm 不显示读取的结果）

import os.path

current_path = os.getcwd()

file_path = os.path.join(current_path, 'test.txt')
file_test = open(file_path, 'w+', encoding='utf-8')
file_test.write('人生苦短 我用Python')
file_test.read()
file_test.seek(0)
file_test.read()
file_test.close()

注意：只有 w+ 模式可以使用 read() 与 seek() 函数 读取文件内容 ，w 模式使用 read() 函数 读取文件内容会报错 ，如下图：

使用 ab 模式 执行 write 方法

我们先 使用 ab 模式 尝试将 字符串 通过 write 方法以追加的形式写入文件

import os.path

current_path = os.getcwd()

file_path = os.path.join(current_path, 'test.txt')
file_test = open(file_path, 'ab')
file_test.write('Python 是一种很有意思的编程语言')

终端执行效果如下：

由于我们传入的是 字符串类型 ，无法直接追加写入文件，那么我们 将字符串转为 byte 类型是否能成功呢？

这又是为何？难道我们使用方法的姿势不对？其实不然，这里写入的 byte 类型 我们需要先转为 byte 类型 才可以成功的写入。示例如下：

此时我们再次执行写入的动作，就可以将其追加写入文件。

file_test.write(_message)
file_test.close()

使用 a 模式 执行 writelines 方法

借着上面演示的案例，我们定义一个列表的字符串，然后 通过 open() 函数 的 a模式，使用 writelines 方法将内容写入文件

import os.path

current_path = os.getcwd()

file_path = os.path.join(current_path, 'test.txt')
file_list = ['\n今天天气不错\n', '挺风和日丽的\n', '我们下午没有课\n']
file_test = open(file_path, 'a')
file_test.writelines(file_list)
file_test.close()

执行效果如下图：

实战小案例

需求：实现一个可以自动创建 python 包 的函数

import os

def create_package(path):# 定义一个创建 包 的函数
   if os.path.exists(path):# 判断路径是否存在，若已经存在，则抛出异常
       raise Exception('％s 已经存在，不可创建' ％ path)
   else:
       os.mkdir(path)

init_path = os.path.join(path, '__init__.py')
   file_init = open(init_path, 'w', encoding='utf-8')
   file_init.write('# coding:utf-8\n')
   file_init.close()

if __name__ == '__main__':
   current_path = os.getcwd()
   path = os.path.join(current_path, 'test_package')
   create_package(path)

执行结果如下：

拓展：从上文脚本中，我们可以发现每次，执行写入的时候，都要写入 \n 换行符，很是麻烦，我们可以自定义一个 open 类，从而实现每次我们执行写入操作时的自动换行。

class Open(object):
   def __init__(self, path, mode='w', is_return=True):     # 这里的 is_return 我们定义的是换行的意思，结合下文的 message 理解
       self.path = path
       self.mode = mode
       self.is_return = is_return

def write(self, message):
       file_test = open(self.path, mode=self.mode)
       if self.is_return:      # 如果返回 Ture 则 在 message 后，增加换行符
           message = '％s\n' ％ message
       file_test.write(message)
       file_test.close()

if __name__ == '__main__':
   current_path = os.getcwd()
   # path = os.path.join(current_path, 'test_package')
   open_path = os.path.join(current_path, 'test_new_line.txt')
   # create_package(path)
   # create_package(open_path)
   open_test = Open(open_path)
   open_test.write('这是一段测试换行符的字符串信息')

执行结果如下：

来源：https://blog.csdn.net/weixin_42250835/article/details/123675274