- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 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 учЮ... говно?
а так нормально
все таки пайтон хороший скриптовый язык, небольшие скрипты на нем всегда чистые и аккуратные
- строки 44-45 выделите в def main()
- в конце скрипта
- на всякий пожарный, я б добавил проверку на аргумент, есть он или нет
- не работал с монгой из питона, но в 38й строке, если в base нет collection все ли будет нормально?
pycharm наше все
mongo поддерживает batch insert, лучше все документы сохранять одним риквестом
raise
Смысл?
В питоне то, где их бросают на каждый чих? "Ааааа, массив кончился, всё пиздец исключение"
З.Ы. Или ты предлагаешь по бектрейсу найти кто кинул и туда бряк поставить?
>Или ты предлагаешь по бектрейсу найти кто кинул и туда бряк поставить?
Такое кстати есть.