XML文件为空时如何退出Python代码(XML到CSV转换)

2024-09-27 21:33:58 发布

您现在位置:Python中文网/ 问答频道 /正文

如果XML文件是空的,即没有根元素,如何在python中处理?当有XML记录时,下面的Python代码可以正常工作,但每当XML文件为空时,转换进程文件。我正在寻找一个退出,而不是失败的过程。你知道吗

XML格式:

<?xml version = '1.0' encoding = 'UTF-8'?>
<ns2:exportEmpData xmlns:ns2="http://webservice.example.com/"/>

Python代码:

import xml
import csv
import xml.etree.ElementTree as ET

tree = ET.parse('C:/emp/emplist.xml')
root = tree.getroot()

# open a file for writing

Emp_data = open('C:/emp/emplist.csv', 'wb')

# create the csv writer object

csvwriter = csv.writer(Emp_data)
emp_head = []

count = 0
for member in root.findall('emplist'):
emp_nodes = []
if count == 0:
  empId = member.find('empId').tag
  emp_head.append(empId)
  fullName = member.find('fullName').tag
  emp_head.append(fullName)
  currentAddress = member.find('currentAddress').tag
  emp_head.append(currentAddress)
  csvwriter.writerow(emp_head)
  count = count + 1

  empId = member.find('empId').text
  emp_nodes.append(empId)
  fullName = member.find('fullName').text
  emp_nodes.append(fullName)
  currentAddress = member.find('currentAddress').attrib.get('city')
  emp_nodes.append(currentAddress)
  csvwriter.writerow(emp_nodes)
Emp_data.close()

Tags: 文件csvimportcountxmlfindheadmember

热门问题