软测入门(七)python操作数据文件(Json、yaml、csv、excel、xml)

python操作文件

txt文件

• read() : 读取所有
• readline() : 读取一行
• readlines() : 读取所有，且以行为单位，放入list列表中

file = open(r"F:\abc.txt", "r", encoding="utf-8")  # 以utf-8格式读取文件
# 读取所有
# print(file.read())

# 读取一行
# print(file.readline())

# 读取所有，且以行为单位，放入list列表中
lines = file.readlines()
for l in lines:
    print(l)

csv文件

CSV:逗号分隔值文件，其文件以纯文本形式存储表格数据，是自动化测试中常用的一种文件类型。

编写一个csv文件，内容如上，然后打开文件

数据以表格的形式存储。

import csv
file = open("aaa.csv", "r", encoding="utf-8")
c = csv.reader(file)
for cs in c:
    print(cs)
    
    
######################################################################################
['姓名', '性别', '年龄']
['张三', '男', '18']
['李四', '女', '19']
['王五', '男', '20']

Excel文件

excel是以二进制形式存储文件。

有一个excel文件，内容如上。

• 安装库，xlrd1.2.0版本支持 xlsx的文件

  pip install xlrd==1.2.0
  

• 使用

  import xlrd
  
  lsx = xlrd.open_workbook("bbb.xlsx")
  # 获取到第一个 sheet 页
  sheet = lsx.sheet_by_index(0)
  
  # 获取行数  /  列数
  print(sheet.nrows, sheet.ncols)
  # 拿取第一行
  print(sheet.row_values(0))
  
  print("=========================================")
  # 拿取所有遍历
  for i in range(sheet.nrows):
      print(sheet.row_values(i))
  

Json文件

整理一个Json文件，然后代码读取这些数据

import json

file = open("ac.json", "r", encoding="utf-8")
# 注意：这里从文件中读取后，是字符串，不是json的对象
json_str = file.read()
json_obj = json.loads(json_str)
print(json_obj)

print("+++++++++转换为json字符串++++++++")
# 转换为json字符串，不使用转码
js_str = json.dumps(json_obj, ensure_ascii=False)
print(js_str)

[{
    
    'name': '张三', 'sex': '男', 'age': 20}, {
    
    'name': '李四', 'sex': '女', 'age': 19}, {
    
    'name': '王五', 'sex': '男', 'age': 21}]
+++++++++转换为json字符串++++++++
[{
    
    "name": "张三", "sex": "男", "age": 20}, {
    
    "name": "李四", "sex": "女", "age": 19}, {
    
    "name": "王五", "sex": "男", "age": 21}]

Xml文件

准备一个xml文件

<?xml version="1.0" encoding="utf-8" ?>
<bookstore>
    <book category="COOKING">
        <title lang="en">Everyday Italian</title>
        <author>Giada De Laurentiis</author>
        <year>2005</year>
        <price>30.00</price>
    </book>
    
    <book category="CHILDREN">
        <title lang="en">Harry Potter</title>
        <author>J K. Rowling</author>
        <year>2005</year>
        <price>29.95</price>
    </book>
</bookstore>

开始读取

# 导包
try:
    import xml.etree.cElementTree as ET
except:
    import xml.etree.ElementTree as ET

# 解析 xml文件
tree = ET.parse("test.xml")
# 获取根节点
root = tree.getroot()

print(root.tag)     # 获取标签名
print(root.attrib)  # 获取属性
print(root.text)    # 获取标签文本

for child in root:
    print(child.tag)  # 获取标签名
    print(child.attrib)  # 获取属性
    print(child.text)  # 获取标签文本

    for children in child:
        print(children.tag)  # 获取标签名
        print(children.attrib)  # 获取属性
        print(children.text)  # 获取标签文本

yaml文件

• 安装

  pip install pyyaml
  

• 使用

  准备一个yml文件

  

  转换为 python 对象

  import yaml
  
  file = open("test.yml", "r", encoding="utf-8")
  yaml_obj = yaml.load(file.read(), Loader=yaml.FullLoader)
  print(yaml_obj)
  
  ################################################################
  {
        
        'animal': 'pets', 'hash': {
        
        'name': 'steve', 'foo': 'bar'}, 'lists': ['Cat', 'Dog'], 'aris': [['Barry', 'Couch', 'Zara'], ['Anna', 'John']], 'dans': ['A', 'B', 'C']}