鏍囩锛?a href='http://www.mamicode.com/so/1/rand' title='rand'>rand   str   鐢熸垚   with open   enumerate   range   鏂囦欢   鏂瑰紡   new   import csv

import random

class XmlFormat(object):

def __init__(self):

self.wf = open(鈥榙ata/res_location.xml鈥? 鈥榳鈥? encoding=鈥榰tf-8鈥?

self.add_header()

def add_header(self):

header = """<?xml version="1.0" encoding="utf-8"?>

"""

# print(header)

self.wf.write(header)

def add_tail(self):

tail = """

"""

self.wf.write(tail)

def get_random(self):

return random.randrange(1000, 9999, 1)

def readfile(self):

with open(鈥榙ata/location.csv鈥? 鈥榬鈥? encoding=鈥榰tf-8鈥? as f:

reder = csv.reader(f, delimiter=鈥?鈥?

for i, line in enumerate(reder):

if not line:

continue

# todo: 鑷姩鎶撳彇琛ㄥご淇℃伅,骞惰缃敭鍊煎:

if i == 0:

# 瀛樺偍琛ㄥご瀛楁

if not line:

raise

header_line = line

continue

data = {}

# print(line)

# 澶勭悊姣忎竴琛岀殑鏁版嵁

for e, var in enumerate(line):

# print(e,var,header_line[e])

if header_line[e] == 鈥榥ew-code鈥?

data[header_line[e]] = var or self.get_random()

elif header_line[e] == 鈥榠s_warehouse鈥?

data[header_line[e]] = var or 0

else:

data[header_line[e]] = var

if (i) % 1000 == 0:

# 鍏抽棴涔嬪墠鐨勬枃浠? self.add_tail()

self.wf.close()

self.wf = open(鈥榙ata/res_location.xml鈥?format(i), 鈥榳鈥? encoding=鈥榰tf-8鈥?

print(鈥榎鈥榙ata/res_location{}.xml鈥?鈥?format(i))

self.add_header()

# 杩欓噷浼犻€掑弬鏁扮殑鏂瑰紡鍙互浣跨敤**kw

self.generate_xml_code(**data)

def generate_xml_code(self, **kwargs):

"""

璁剧疆xml 妯℃澘,骞剁敓鎴愮浉搴旂殑xml鏂囨湰,

澶ф嫭鍙蜂腑鐨勫弬鏁伴渶瑕佸拰csv 鏂囦欢鐨勫瓧娈典俊鎭繚鎸佷竴鑷?濡倇extra_id} 瀵瑰簲csv 鏂囦欢涓殑extra_id瀛楁

xml 妯℃澘鎸夌収鍙寜鐓ц嚜宸辩殑闇€姹傝繘琛屾洿鎹?

娉? csv 鏂囦欢鐨勫瓧娈典笉鑳介噸澶?

:param kwargs:

:return:

"""

str_format = """

{name}

internal

{is_warehouse}

1

{new-code}

"""

res = str_format.format(**kwargs)

self.wf.write(res)

def __del__(self):

self.add_tail()

self.wf.close()

if __name__ == 鈥榑_main__鈥?

a = XmlFormat()

a.readfile()

璇存槑:

csv鏂囦欢闇€瑕佹斁鍦╠ata鏂囦欢澶逛腑,鍏跺疄鍙互淇敼璇诲彇鏂囦欢鐨勪綅缃?/li>