Наполнение базы данных
Теперь можно наполнить таблицы значениями. Следует начать с расшифровки числовых значений для дней недели:
weekdays = ["Воскресенье", "Понедельник", "Вторник", "Среда", "Четверг", "Пятница", "Суббота", "Воскресенье"]
import sqlite as db
c = db.connect(database="tvprogram") cu = c.cursor() cu.execute("""DELETE FROM wd;""") cu.executemany("""INSERT INTO wd VALUES (%s, %s);""", enumerate(weekdays)) c.commit() c.close()
Стоит напомнить, что встроенная функция enumerate() создает список пар номер-значение, например:
>>> print [i for i in enumerate(['a', 'b', 'c'])] [(0, 'a'), (1, 'b'), (2, 'c')]
Из приведенного примера ясно, что метод executemany() объекта-курсора использует второй параметр - последовательность - для массового ввода данных с помощью SQL-инструкции INSERT.
Предположим, что телепрограмма задана в файле tv.csv в формате CSV (он уже обсуждался):
10.02.2003 9.00|ОРТ|Новости|Новости|9.15 10.02.2003 9.15|ОРТ|"НЕЖНЫЙ ЯД"|Сериал|10.15 10.02.2003 10.15|ОРТ|"Маски-шоу"|Юмористическая программа|10.45 10.02.2003 10.45|ОРТ|"Человек и закон"||11.30 10.02.2003 11.30|ОРТ|"НОВЫЕ ПРИКЛЮЧЕНИЯ СИНДБАДА"|Сериал|12.00
Следующая программа разбирает CSV-файл и записывает данные в таблицу tv:
import calendar, csv import sqlite as db from sqlite.main import Time, Date ## Только для db.Date, db.Time = Date, Time ## sqlite
c = db.connect(database="tvprogram") cu = c.cursor()
input_file = open("tv.csv", "rb") rdr = csv.DictReader(input_file, fieldnames=['begt', 'channel', 'prname', 'prgenre', 'endt']) for rec in rdr: bd, bt = rec['begt'].split() bdd, bdm, bdy = map(int, bd.split('.')) bth, btm = map(int, bt.split('.')) eth, etm = map(int, rec['endt'].split('.')) rec['wd'] = calendar.weekday(bdy, bdm, bdd) rec['begd'] = db.Date(bdy, bdm, bdd) rec['begt'] = db.Time(bth, btm, 0) rec['endt'] = db.Time(eth, etm, 0)
cu.execute("""INSERT INTO tv (tvdate, tvweekday, tvchannel, tvtime1, tvtime2, prname, prgenre) VALUES ( %(begd)s, %(wd)s, %(channel)s, %(begt)s, %(endt)s, %(prname)s, %(prgenre)s);""", rec) input_file.close() c.commit()
Большая часть преобразований связана с получением дат и времен ( приходится разбивать строки на части в соответствии с форматом даты и времени). День недели получен с помощью функции из модуля calendar.
Примечание:
Из-за небольшой ошибки в пакете sqlite конструкторы Date, Time и т.д. не попадают из модуля sqlite.main при импорте из sqlite, поэтому пришлось добавить две строки, специфичные для sqlite, в универсальный "модуль" с именем db.
В этом же примере было продемонстрировано использование словаря для вставки значений в таблицу базы данных. Следует заметить, что подстановка выполняется внутри вызова execute() в соответствии с типами переданных значений. SQL-инструкция INSERT была бы некорректной при попытке выполнить подстановку самостоятельно, например, операцией форматирования %.
