fairypp / Python_Learning

14 stars 8 forks source link

looking for an answer to a traceback #27

Closed karishma-gangwani closed 3 years ago

karishma-gangwani commented 3 years ago

Hi I wrote the code for this problem as below but it shows me a traceback Traceback (most recent call last): File "15tracks.py", line 39, in fname = Library.xml NameError: name 'Library' is not defined

import xml.etree.ElementTree as ET import sqlite3

conn = sqlite3.connect('trackdb.sqlite') cur = conn.cursor()

cur.executescript(''' DROP TABLE IF EXISTS Artist; DROP TABLE IF EXISTS Album; DROP TABLE IF EXISTS Genre; DROP TABLE IF EXISTS Track;

CREATE TABLE Artist ( id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE, name TEXT UNIQUE );

CREATE TABLE Album ( id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE, artist_id INTEGER, title TEXT UNIQUE );

CREATE TABLE Genre ( id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE, name TEXT UNIQUE );

CREATE TABLE Track ( id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE, title TEXT UNIQUE, album_id INTEGER, genre_id INTEGER, len INTEGER, rating INTEGER, count INTEGER ); ''')

fname = input ('Enter filename: ') if(len(fname)<1):fname='Library.xml'

def lookup(d, key): found = False for child in d: if found : return child.text if child.tag == 'key' and child.text == key : found = True return None

stuff = ET.parse(fname) all = stuff.findall('dict/dict/dict') print('Dict count:'), len(all) for entry in all: if ( lookup(entry, 'Track ID') is None ) : continue

name = lookup(entry, 'Name')
genre = lookup(entry, 'Genre')
artist = lookup(entry, 'Artist')
album = lookup(entry, 'Album')
count = lookup(entry, 'Play Count')
rating = lookup(entry, 'Rating')
length = lookup(entry, 'Total Time')

if name is None or artist is None or album is None or genre is None:
    continue
print (name,artist, album, count, rating, length)

cur.execute('''INSERT OR IGNORE INTO Artist (name)
VALUES ( ? )''', ( artist, ) )
cur.execute('SELECT id FROM Artist WHERE name = ? ', (artist, ))
artist_id = cur.fetchone()[0]

cur.execute('''INSERT OR IGNORE INTO Album (title, artist_id)
VALUES( ? )''', ( genre, ) )
cur.execute('SELECT id FROM Album WHERE title = ? ', (album, ))
artist_id= cur.fetchone()[0]

cur.execute('''INSERT OR IGNORE INTO Genre (name)
VALUES ( ? )''', ( genre, ) )
cur.execute('SELECT id FROM Genre WHERE name = ? ', (genre, ))
genre_id = cur.fetchone()[0]

cur.execute('''INSERT OR REPLACE INTO Track
(title, album_id,genre_id, len, rating, count)
VALUES ( ?, ?, ?, ?, ?,? )''',
( name, album_id,genre_id, length, rating, count ) )

conn.commit()