Python开发【第七篇】：文件及文件夹操作

def open(file, mode='r', buffering=None, encoding=None, errors=None, newline=None, closefd=True): 

#1. 打开文件，得到文件句柄并赋值给一个变量
#2. 通过句柄对文件进行操作
#3. 关闭文件

#1. 打开文件，得到文件句柄并赋值给一个变量
f=open('a.txt','r',encoding='utf-8') #默认打开模式就为r

#2. 通过句柄对文件进行操作
data=f.read()

#3. 关闭文件
f.close()

# -*- coding: utf-8 -*-

# 声明字符编码
# coding:utf-8
import os

# with open('../files/a.txt', 'r', encoding='utf-8') as a_f, open('../files/b.txt', 'w', encoding='utf-8') as b_f:
#     a_data = a_f.read()
#     a_data = a_data + "方式二：将硬盘存放的该文件的内容一行一行地读入内存，修改完毕就写入新文件，最后用新文件覆盖源文件"
#     b_f.write(a_data)
# os.remove('../files/a.txt')
# os.rename('../files/b.txt', '../files/a.txt')


with open('../files/a.txt', 'r', encoding='utf-8') as a_f, open('../files/b.txt', 'w', encoding='utf-8') as b_f:
    for line in a_f:
        b_f.write(line)
    b_f.write("方式二：将硬盘存放的该文件的内容一行一行地读入内存，修改完毕就写入新文件，最后用新文件覆盖源文件")
os.remove('../files/a.txt')
os.rename('../files/b.txt', '../files/a.txt')

# -*- coding: utf-8 -*-

# 声明字符编码
# coding:utf-8
# 文件全类容
file_list = []


# 定义文件生成器
def file_generator():
    with open('../files/haproxy.conf', 'r', encoding='utf-8') as f:
        for row in f:
            yield row


# 判断当前内容是否包含
def is_contain(str):
    str = "backend %s" % str
    flag = False
    # 获取读取文件生成器
    generator = file_generator()
    for row in generator:
        r = row.strip()
        file_list.append(r)
        if r == str:
            flag = True

    print(file_list)
    return flag


# 查询
def fetch():
    # 是否查到标识
    flag = False
    print('=========欢迎进入查询功能=========')
    content = input('请输入您要查询的内容：')
    # 获取读取文件生成器
    generator = file_generator()
    backend_data = "backend %s" % content
    for row in generator:
        r = row.strip()
        # 查到了，如果开头为"backend"说明已经结束了。
        if flag is True and r.startswith('backend'):
            break
        if r == backend_data or flag is True:
            flag = True
            # 打印查询到的信息
            print(r)
    else:
        if flag is False:
            print('未查到当前节点下的内容。。。。')


# 新增
def insert():
    # 直接在文件中新加内容
    print('=========欢迎进入新增功能=========')
    content = input('请输入您要新增的内容：')
    if content is None:
        print("新增内容不能为空")
    # 思路也是先查询出原内容，组成列表，通过索引添加内容，在重写到原来的文件中
    pass


# 修改
def update():
    print('=========欢迎进入修改功能=========')
    # data = [{'backend': 'www.oldboy1.org', 'record': {'server': '2.2.2.4', 'weight': 20, 'maxconn': 3000}},{'backend': 'www.oldboy1.org', 'record': {'server': '3.3.3.3', 'weight': 30, 'maxconn': 1000}}]
    content = input('请输入您要查询的内容：').strip()
    # eval 转为列表结构
    data = eval(content)

    if is_contain(data[0]['backend']) is True:
        # 获取要修改的源数据
        old_server_record = '%sserver %s %s weight %s maxconn %s\n' % (' ' * 8, data[0]['record']['server'],
                                                                       data[0]['record']['server'],
                                                                       data[0]['record']['weight'],
                                                                       data[0]['record']['maxconn'])
        # 要修改的数据
        new_server_record = '%sserver %s %s weight %s maxconn %s\n' % (' ' * 8, data[1]['record']['server'],
                                                                       data[1]['record']['server'],
                                                                       data[1]['record']['weight'],
                                                                       data[1]['record']['maxconn'])

        old_server_record = old_server_record.strip()
        new_server_record = new_server_record.strip()
        print(old_server_record)
        print(new_server_record)

        print('修改前列表：')
        print(file_list)
        if old_server_record not in file_list:
            print('你要修改的记录不存在')
        else:
            index = file_list.index(old_server_record)
            file_list[index] = new_server_record
            print('修改后列表：')
            print(file_list)
            # 重新覆盖文件
            with open('../files/haproxy.conf', 'w+', encoding='utf-8') as f:
                # 写入后就变成一行了???
                f.writelines(file_list)

    else:
        print('对不起你要修改的内容不存在。。。')


# 删除
def delete():
    # 思路也是先查询出原内容，组成列表，通过索引删除内容，在重写到原来的文件中
    pass


if __name__ == "__main__":
    msg = """
    1:查询
    2:新增
    3:修改
    4:删除
    5:退出
    """
    msg_dic = {
        '1': fetch,
        '2': insert,
        '3': update,
        '4': delete
    }
    while True:
        print(msg)
        # 接收用户输入并且去除空格
        choice = input("请输入功能选择：").strip()
        if choice == '5': break
        operation = msg_dic.get(choice)
        # print(operation) # 打印的是函数地址
        if operation is not None:
            # 调用函数
            operation()
        else:
            print("系统暂无此功能。。。")

import os

=====OS模块====

os.getcwd():获取当前路径
os.listdir():展示当前目录内容 返回一个列表
os.chdir():改变当前路径
os.mkdir():创建目录
os.rmdir():删除目录
os.remove():删除文件
os.rename():重命名

os.path.isdir():判断是否是一个目录
os.path.isfile():判断是否是一个文件
os.path.join():路径拼接
os.path.dirname():所在目录/父及目录
os.path.abspath():绝对路径
os.path.relpath():相对路径
os.path.normpath():规范化路径
os.path.getsize():资源大小
os.path.getctime/getatime/getmtime:资源时间
os.path.exists():路径是否存在
os.path.isabs():是否是绝对路径


'''