- 01
- 02
- 03
- 04
- 05
- 06
- 07
- 08
- 09
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
#!/usr/bin/env python
from sys import argv
from pymongo import MongoClient
conn = MongoClient('127.0.0.1', 27017)
base = conn.xmldata
try:
from lxml import etree
except ImportError:
import xml.etree.ElementTree as etree
def parse_xml(filename):
tree = etree.parse(filename)
root = tree.getroot()
cntr = root.getchildren()
for country in cntr:
data_struct = {}
data_struct['name'] = country.attrib['name']
for key in ['rank', 'year', 'gdppc']:
value = country.find(key).text
data_struct[key] = value
neighbors = []
for row in country.iter('neighbor'):
neighbor = {}
neighbor['name'] = row.attrib['name']
neighbor['direction'] = row.attrib['direction']
neighbors.append(neighbor)
data_struct['neighbors'] = neighbors
insert_data("country", data_struct)
def insert_data(collection, data_struct):
coll = base[collection]
try:
coll.insert_one(data_struct)
except Exception as e:
raise
filename = argv[1]
parse_xml(filename)
Python учЮ... говно?