Почему словарь не записывается в JSON-файл(Python)?

Question

[Maxim045]

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

import mutagen, datetime, pygame, os, json
from mutagen.id3 import ID3
from tkinter import *
from tkinter.filedialog import askdirectory

root = Tk()
root.minsize(300,300)
all_tags={}

def write_tags() -> None:
    '''Функция записывает словарь 'all_tags' в JSON-файл.'''
    with open("C:\\Users\\101ap\\Desktop\\Player\\tags.json", 'w') as t:       
        json.dump(all_tags, t, indent=4)


def extract_tags(audiofile: str) -> None:
    '''Функция получает путь к MP3-файлу, извлекает из него теги и наполняет ими
       словарь 'song_tags', который передает в словарь 'all_tags'.'''
    song_tags = {}
    song_tags['Path to the File']=audiofile
    song_title = audiofile.tags.getall('TIT2')
    song_tags['Song Title'] = str(song_title[0])
    '''Метод извлекает из MP3-файла название композиции.'''
    album_title = audiofile.tags.getall('TALB')
    song_tags['Album Title'] = str(album_title[0])
    '''Метод извлекает из MP3-файла название альбома.'''
    song_artist = audiofile.tags.getall('TPE1')
    song_tags['Song Artist'] = str(song_artist[0])
    '''Метод извлекает из MP3-файла имя исполнителя композиции.'''
    album_artist = audiofile.tags.getall('TPE2')
    song_tags['Album Artist'] = str(album_artist[0])
    '''Метод извлекает из MP3-файла имя исполнителя альбома.'''
    year_of_publishing = audiofile.tags.getall('TDRC')
    song_tags['Year of Publishing'] = str(year_of_publishing[0])
    '''Метод извлекает из MP3-файла имя исполнителя альбома.'''
    track_number = audiofile.tags.getall('TRCK')
    song_tags['Track Number'] = str(track_number[0])
    '''Метод извлекает из MP3-файла порядковый номер композиции.'''
    song_tags['Length'] = str(datetime.timedelta(seconds=audiofile.info.length))
    '''Метод извлекает из MP3-файла длину композиции.'''
    song_tags['Bitrate'] = audiofile.info.bitrate
    '''Метод извлекает из MP3-файла битрейт композиции.'''
    all_tags[audiofile]=song_tags
    write_tags()


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


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


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

root.mainloop()

Comments

[AWEme]

Почитай про функции в питоне, и особенно почитай, что делает инструкция return.

[Maxim045]

AWEme, хорошо

Answer 1

[Nikita Panuhin]

1) choose_files(directory) не выполняется из-за return строчкой выше.
2) extract_tags(audiofile) не выполняется из-за return строчкой выше.
4) Вы передаёте в json.dump словарь, ключами которого являются объекты класса mutagen.mp3.MP3, что делать нельзя (они не умеют сами преобразовываться в тип str).
5) Замечание: Если открываете файл с ключом 'w', то он будет очищаться при каждом открытии. Таким образом в нём окажется информация только о последнем файле, потому что вызываете вы write_tags() почему-то не в конце, а после каждого файла
6) Замечание: Вы передаёте в функцию extract_tags не строку, а объект mutagen.mp3.MP3 . Так что extract_tags(audiofile: str) это не верно.

Исправленный код

import mutagen
import datetime

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

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

root = Tk()
root.minsize(300, 300)
all_tags = {}


def write_tags() -> None:
    '''Функция записывает словарь 'all_tags' в JSON-файл.'''
    with open("C:\\Users\\101ap\\Desktop\\Player\\tags.json", 'w') as t:
        json.dump(all_tags, t, indent=4, ensure_ascii=False)


def extract_tags(path) -> None:
    '''Функция получает путь до MP3-файла, извлекает из него теги и наполняет ими
       словарь 'song_tags', который передает в словарь 'all_tags'.'''

    audiofile = mutagen.File(path)

    # print(path, audiofile)

    song_tags = {}
    song_tags['Path to the File'] = path

    if audiofile.tags:

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

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

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

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

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

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

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

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

    return song_tags


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


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

[Maxim045]

Я указываю точный путь, наверное, ошибки из-за этого возникнуть не должно. Думаю, что косячу где-то более масштабно. Но, в любом случае, спасибо!

[Nikita Panuhin]

Хорошо, сейчас доделаю так, чтобы работало и добавлю.

[Maxim045]

Никита Панюхин, да, положение немного изменилось. Ошибка, конечно, не ушла, но все равно спасибо:)

[Nikita Panuhin]

Сейчас напишу, чтобы работало, подождите пару минут.

[Maxim045]

Никита Панюхин, спасибо! Буду исправлять:)

[Nikita Panuhin]

Ну подождите вы :) там я ещё не написал некоторые маленькие вещи. Сейчас всё будет...

[Maxim045]

Никита Панюхин, надеюсь, это все:)
Огромное спасибо! Что-то я уже совсем потерялся

[Nikita Panuhin]

Выложил код, теперь нормально работает, дальше подстройте под себя.

[Maxim045]

Никита Панюхин, Огромное спасибо!