Почему не заполняется база данных SQLite?

Question

[Maxim045]

У меня есть код, который заполняет базу данных SQLite тегами MP3-файла. После выбора директории база данных остается пустой. Пожалуйста, подскажите, где я допустил ошибку?

import mutagen
import datetime

import os
os.environ['PYGAME_HIDE_SUPPORT_PROMPT'] = "hide"
# import pygame

import sqlite3
from mutagen.id3 import ID3
from tkinter import *
from tkinter.filedialog import askdirectory

con = sqlite3.connect("tagsdatabase.db")
cur = con.cursor()
cur.execute('''CREATE TABLE IF NOT EXISTS tags_db(Path TEXT,
                                                  Song_Title TEXT,
                                                  Album_Title TEXT,
                                                  Song_Artist TEXT,
                                                  Album_Artist TEXT,
                                                  Year_of_Publishing INTEGER,
                                                  Track_Number INTEGER,
                                                  Length REAL,
                                                  Bitrate INTEGER)''')
con.commit()

root = Tk()
root.minsize(300, 300)


def extract_tags(path: str) -> list:
    '''Функция получает путь до MP3-файла, извлекает из него теги и заполняет ими базу данных.'''

    audiofile = mutagen.File(path)

    if audiofile.tags:

        song_title_s = audiofile.tags.getall('TIT2')
        song_title = str(song_title_s[0]) if song_title_s else None
        # Метод извлекает из MP3-файла название композиции

        album_title_s = audiofile.tags.getall('TALB')
        album_title = str(album_title_s[0]) if album_title_s else None
        # Метод извлекает из MP3-файла название альбома

        song_artist_s = audiofile.tags.getall('TPE1')
        song_artist = str(song_artist_s[0]) if song_artist_s else None
        # Метод извлекает из MP3-файла имя исполнителя композиции

        album_artist_s = audiofile.tags.getall('TPE2')
        album_artist = str(album_artist_s[0]) if album_artist_s else None
        # Метод извлекает из MP3-файла имя исполнителя альбома

        year_of_publishing_s = audiofile.tags.getall('TDRC')
        year_of_publishing = str(year_of_publishing_s[0]) if year_of_publishing_s else None
        # Метод извлекает из MP3-файла имя исполнителя альбома

        track_number_s = audiofile.tags.getall('TRCK')
        track_number = str(track_number_s[0]) if track_number_s else None
        # Метод извлекает из MP3-файла порядковый номер композиции

        length = str(datetime.timedelta(seconds = audiofile.info.length))
        # Метод извлекает из MP3-файла длину композиции

        bitrate = audiofile.info.bitrate
        # Метод извлекает из MP3-файла битрейт композиции

    song_tags = (path, song_title, album_title, song_artist, album_artist, year_of_publishing, track_number, length, bitrate)

    try:
        sql = cur.execute('SELECT * FROM tags_db WHERE path=?', path)
        cur.execute('UPDATE tags_db SET Song_Title=?, Album_Title=?, Song_Artist=?, Album_Artist=?, Year_of_Publishing=?, Track_Number=?, Length=?, Bitrate=?  WHERE path=?', (song_title, album_title, song_artist, album_artist, year_of_publishing, track_number, length, bitrate, sql))
    except sqlite3.ProgrammingError:
        cur.execute('INSERT INTO tags_db VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?)', song_tags)

    con.commit()


def choose_files(directory: str) -> None:
    '''Функция получает путь к директории, отбирает MP3-файлы, применяет extract_tags к каждому из них.'''
    for file in os.listdir(directory):
        if file.endswith(".mp3"):
            path = os.path.realpath(file)
            extract_tags(path)


def choose_directory(event) -> None:
    '''Функция открывает диалоговое окно выбора директории с MP3-файлами'''
    directory = askdirectory()
    os.chdir(directory)
    choose_files(directory)

    # exit()


choosedirectory = Button(root, text = 'Choose Directory')
choosedirectory.pack()
choosedirectory.bind("<Button-1>", choose_directory)

root.mainloop()

Comments

[javedimka]

Добавляй принты в функцию выбора файлов, в цикл фор, и проверяй, что файлы у тебя действительно выбираются, ещё один принт в функцию экстракт тэгс , чтобы убедиться, что и она вызывается

[Maxim045]

javedimka, спасибо

[o5a]

Maxim045,
1. Чтобы передать параметры в execute, нужно использовать кортежи. Если параметр один, то все равно его нужно оформить как кортеж. Т.е. в данном случае нужно было писать

sql = cur.execute('SELECT * FROM tags_db WHERE path=?', (path, ) )

Без этого такое использование как раз вызывает исключение.
2. Но в целом конструкция нелогичная. sql заполняется, но не используется.
Логика должна быть такая (судя по всему, цель обновить данные, а если их не было - занести): считываем как есть в sql, и если он пустой, то делаем INSERT, в противном случае UPDATE. А не как сейчас.
Альтернатива этому - делать INSERT ... ON CONFLICT UPDATE (почитай про upsert). Но для этого нужно указать первичный ключ.

Но даже с учетом этого навскидку данные должны были занестись.

[Maxim045]

o5a, большое спасибо

[Maxim045]

o5a, я исправил цикл, но проблема осталась. Если не сложно, пожалуйста, скажите, где ошибся

song_tags = (path, song_title, album_title, song_artist, album_artist, year_of_publishing, track_number, length, bitrate)

record = cur.execute('SELECT * FROM tags_db WHERE Path=?', (path, ))

if not record:
    cur.execute('INSERT INTO tags_db VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?)', song_tags)
else:
    cur.execute('UPDATE tags_db SET Song_Title=?, Album_Title=?, Song_Artist=?, Album_Artist=?, Year_of_Publishing=?, Track_Number=?, Length=?, Bitrate=?  WHERE Path=?', (song_title, album_title, song_artist, album_artist, year_of_publishing, track_number, length, bitrate, path))

con.commit()

[o5a]

Maxim045, в данном случае record это объект курсор, он существует, даже если данных нет, так что проверять так бессмысленно. Нужно проверять наличие данных по этому курсору, через fetchall или fetchone.

record = cur.execute('SELECT * FROM tags_db WHERE Path=?', (path, ))
result = record.fetchall()

if not result:
# и далее ...

Тогда будет работать как планировалось - добавлять, если еще не было записей по данному path.

Но вообще не совсем понятно, что именно не работает? Вставка данных в вашем варианте все равно должна быть.

[Maxim045]

o5a, вставка была, не работала проверка на дубликат. Теперь все работает.
Я сидел над этим вопросом второй день, уже не знал, что делать. Огромное вам спасибо

Answer 1

[shurshur]

Включить autocommit или в конце скрипта либо после каждого запроса делать con.commit()

Comments

[Maxim045]

Я прописал con.commit() после создания таблицы и в теле функции extract_tags. Я не совсем понял, этого не хватает? Если так, пожалуйста, скажите чуть конкретнее, где еще нужно прописать con.commit()?

[shurshur]

Maxim045, да, проглядел. Но вообще и правда я бы проверил, что код до запросов вообще добирается, может быть они и не выполняются?

[Maxim045]

shurshur, хорошо, спасибо, буду смотреть