'''

开发一个给大百度的接口，各种要求，写一个xml文件，倒是不是很难

'''

import xml,datetime,codecs

import xml.dom.minidom as minidom

def covert_to_unicode(msg):

'''''将转入的编码转换为unicode，只接受utf-8和unicode编码'''

__re_str = None

if isinstance(msg, unicode):

__re_str = msg

elif isinstance(msg, str):

try:

__re_str = msg.decode('utf-8')

except Exception, errinfo:

raise Exception, '%s,%s' % (errinfo, str(msg))

else:

raise Exception, '%s 必须为str或unicode类型' % msg

return __re_str

class CreateCovXml():

def __init__(self, cov_xml_path):

self.__cov_path = cov_xml_path

self.__dom = None

self.__root = None

def _covert_code(self, msg):

return covert_to_unicode(msg)

def __create_new_node(self, node_name, node_text = None):

'''''为xml文件添加一个结点,其中node_text可以为空'''

if self.__dom == None:

raise Exception, '创建结点时，document对象还不存在'

# 创建的是结点只有结点名

if None == node_text:

return self.__dom.createElement(self._covert_code(node_name))

else:

newNode = self.__dom.createElement(self._covert_code(node_name))

newText = self.__dom.createTextNode(self._covert_code(node_text))

newNode.appendChild(newText)

return newNode

def begin_cov(self):

'''''begin_cov(self) --> None 创建coverage.xml文档和根结点'''

try:

impl = minidom.getDOMImplementation()

self.__dom = impl.createDocument(None, u'urlset', None)

self.__root = self.__dom.documentElement

except:

traceback.print_exc()

raise Exception, '创建coverage xml根结点失败'

def add_cov(self, url, title, priority, Horizontalpicture1,dayRetrievalstatus,channel,Largeblock,smallblock,blocknumber):

if self.__root == None:

raise Exception, '创建结点时，root结点不存在对象不存不存在'

urlstart_node = self.__create_new_node("url")

loc_node = self.__create_new_node("loc")

lastmod_node = self.__create_new_node("lastmod")

changefreq_node = self.__create_new_node("changefreq")

priority_node = self.__create_new_node("priority",priority)

data_node = self.__create_new_node("data")

display_node = self.__create_new_node("display")

channel_node = self.__create_new_node("channel",channel)

Largeblock_node = self.__create_new_node("Largeblock",Largeblock)

smallblock_node = self.__create_new_node("smallblock",smallblock)

blocknumber_node = self.__create_new_node("blocknumber",blocknumber)

title_node = self.__create_new_node("title",title)

subtitle_node = self.__create_new_node("subtitle")

url_node = self.__create_new_node("url",url)

Horizontalpicture1_node = self.__create_new_node("Horizontalpicture1",Horizontalpicture1)

Horizontalpicture2_node = self.__create_new_node("Horizontalpicture2")

Horizontalpicture3_node = self.__create_new_node("Horizontalpicture3")

singer_node = self.__create_new_node("singer")

area_node = self.__create_new_node("area")

timelength_node = self.__create_new_node("timelength")

episode_node = self.__create_new_node("episode")

updatetime_node = self.__create_new_node("updatetime")

briefintroduction_node = self.__create_new_node("briefintroduction")

daySearchvolume_node = self.__create_new_node("daySearchvolume")

dayRetrievalstatus_node = self.__create_new_node("dayRetrievalstatus",dayRetrievalstatus)

sitename_node = self.__create_new_node("sitename")

self.__root.appendChild(urlstart_node)

urlstart_node.appendChild(loc_node)

urlstart_node.appendChild(lastmod_node)

urlstart_node.appendChild(changefreq_node)

urlstart_node.appendChild(priority_node)

urlstart_node.appendChild(data_node)

data_node.appendChild(display_node)

display_node.appendChild(channel_node)

display_node.appendChild(Largeblock_node)

display_node.appendChild(smallblock_node)

display_node.appendChild(blocknumber_node)

display_node.appendChild(title_node)

display_node.appendChild(subtitle_node)

display_node.appendChild(url_node)

display_node.appendChild(Horizontalpicture1_node)

display_node.appendChild(Horizontalpicture2_node)

display_node.appendChild(Horizontalpicture3_node)

display_node.appendChild(singer_node)

display_node.appendChild(area_node)

display_node.appendChild(timelength_node)

display_node.appendChild(episode_node)

display_node.appendChild(updatetime_node)

display_node.appendChild(briefintroduction_node)

display_node.appendChild(daySearchvolume_node)

display_node.appendChild(dayRetrievalstatus_node)

display_node.appendChild(sitename_node)

def end_cov(self):

try:

f = open(self.__cov_path, 'wb')

writer = codecs.lookup('utf-8')[3](f)

self.__dom.writexml(writer, encoding='utf-8')

writer.close()

f.close()

return True

except Exception, err:

traceback.print_exc()

raise Exception, '写coverage.xml文件出错'#测试用例

test = CreateCovXml(result_filename+'.xml')

test.begin_cov()

priority_count = 0

for linkInfo in result_linkinfos:

priority_count = priority_count + 1

test.add_cov(

url = linkInfo.getTarget(),

title = linkInfo.getTitle(),

priority = str(priority_count),

Horizontalpicture1 = linkInfo.getDetailPicUrl(),

dayRetrievalstatus = "1",

channel = "资讯",

Largeblock = "带图区",

smallblock = "焦点小图",

blocknumber = "3-16"

)

test.end_cov()

os.chdir(basePath)