· 8 years ago · Feb 28, 2018, 09:36 PM
1import xml.etree.ElementTree as ET
2import sqlite3
3conn = sqlite3.connect ('dvds.sqlite')
4cur = conn.cursor()
5
6cur.executescript('''
7DROP TABLE IF EXISTS Films;
8
9CREATE TABLE Films (
10 id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE,
11 title TEXT UNIQUE,
12 genre TEXT,
13 cert INTEGER
14
15);
16
17''')
18
19fname = raw_input('Enter file name: ')
20if ( len(fname) < 1 ) : fname = 'dvds.xml'
21
22def lookup(d, key):
23 found = False
24 for child in d:
25 if found : return child.text
26 if child.tag == 'key' and child.text == key :
27 found = True
28 return None
29
30stuff = ET.parse(fname)
31for entry in stuff.findall('film'):
32 title = entry.find('title').text
33 genre = entry.find('genre').text
34 cert = entry.find('certificate').text
35
36 print title, genre, cert
37
38 cur.execute('''INSERT OR IGNORE INTO Films (title, genre, cert)
39 VALUES ( ?, ?, ? )''', ( title, genre, cert ) )
40
41conn.commit()