Exemple #1
0
import xmltodict
import sys
sys.path.append('../')
from db import Tag, Word

with open('wordbook.xml', encoding='utf8') as fd:
    doc = xmltodict.parse(fd.read())

for item in doc['wordbook']['item']:
    num = Tag.query_count(tag=item['tags'])
    if num == 0:
        tag = Tag(tag=item['tags'])
        Tag.insert(tag)
    else:
        tag = Tag.query_one(tag=item['tags'])
    wordNum = Word.query_count(word=item['word'])
    if wordNum == 0:
        word = Word(word=item['word'], explanation=item['trans'])
    else:
        continue
    word.tags.append(tag)
    Word.insert(word)