fr.close()

关于 yml 文件的操作,有一个专门的库 yaml ,yaml 直接给出了一个把 yml 文件读取到字典里的方法 load,所以要读取 yml 文件非常简单,只需要引入 yaml 库,用 python 的 open 方法打开文件,用 load 方法读取文件,就可以轻而易举的读取 yml 文件到字典里了。当然其中还是有点坑的,那就是解码的问题,因为默认读取的文件不是字符编码格式,所以要把它转换为字符编码,再传给 load。

文件写入方法示例:

    import yaml
    fr = open('yml_file_address', 'w',encoding='utf-8')
    temp = {"fruits":"banana"}
    data = yaml.dump(temp,fr, default_flow_style=False)
    print(data)
    fr.close()

yaml 库提供了一个 dump 方法,只需要打开文件,给出要写入的参数(字典形式等),调用 dump 方法就可以很快捷的写入信息到 yml 文件啦

这两个方法在项目中的运用:

修改配置文件昵称
def changeUserName(user):
    fr = open('yml_file_address', 'r',encoding='utf-8')
    data = yaml.load(fr)
    fr.close()
    data['robot_name_cn'] = user
    fr = open('yml_file_address', 'w', encoding='utf-8')
    yaml.dump(data, fr, allow_unicode=True, default_flow_style=False)
    fr.close()
    return data['robot_name_cn']
修改配置文件密码
def changePassword(password):
    fr = open('yml_file_address', 'r',encoding='utf-8')
    data = yaml.load(fr)
    fr.close()
    data['password'] = password
    if 'password' in data.keys():
        fr = open('yml_file_address', 'w', encoding='utf-8')
        yaml.dump(data, fr, allow_unicode=True, default_flow_style=False)
        fr.close()
        return data['password']
    else:
        fr = open('yml_file_address', 'a', encoding='utf-8')
        yaml.dump(data, fr, allow_unicode=True, default_flow_style=False)
        fr.close()
        return data['password']

这是两个修改配置信息的函数 changeUserName() 和 changePassword(),故名思议,就是修改密码和昵称,两个函数的思路都是一样的,先读取配置文件,把相应的信息替换进配置文件里,然后通过写入方法,把这两个信息写入文件中。

其中的难点可能就是解码和生成 yaml 文件的样式问题了。一开始各种报解码错误,在打开文件的时候加一个 encoding,在写入函数中加一个 allow_unicode=True 搞定了。

解决替换信息在文件中的样式问题, 是在写入函数中加入 default_flow_style=False 搞定的。

非淡泊无以明志