convert: playlist: Linter and import fixes

in m3u module and testsuite.
This commit is contained in:
J0J0 T 2022-08-24 23:14:04 +02:00 committed by J0J0 Todos
parent c1908d551a
commit 2c1163cbc5
2 changed files with 21 additions and 18 deletions

View file

@ -12,25 +12,21 @@
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
"""Provides utilities to read, write an manipulate m3u playlist files.
"""
"""Provides utilities to read, write an manipulate m3u playlist files."""
from beets.util import syspath
class EmptyPlaylistError(Exception):
"""An error that should be raised when a playlist file without media files
is saved or loaded.
"""
"""Raised when a playlist file without media files is saved or loaded."""
pass
class M3UFile():
"""Reads and writes m3u or m3u8 playlist files."""
def __init__(self, path):
"""Reads and writes m3u or m3u8 playlist files.
``path`` is the full path to the playlist file.
"""``path`` is the absolute path to the playlist file.
The playlist file type, m3u or m3u8 is determined by 1) the ending
being m3u8 and 2) the file paths contained in the list being utf-8
@ -42,8 +38,7 @@ class M3UFile():
self.media_list = []
def load(self):
"""Reads the m3u file from disk and sets the object's attributes.
"""
"""Reads the m3u file from disk and sets the object's attributes."""
with open(syspath(self.path), "r") as playlist_file:
raw_contents = playlist_file.readlines()
self.extm3u = True if raw_contents[0] == "#EXTM3U\n" else False
@ -56,9 +51,9 @@ class M3UFile():
raise EmptyPlaylistError
def set_contents(self, media_list, extm3u=True):
"""Sets self.media_list to a list of media file paths,
"""Sets self.media_list to a list of media file paths.
and sets additional flags, changing the final m3u-file's format.
Also sets additional flags, changing the final m3u-file's format.
``media_list`` is a list of paths to media files that should be added
to the playlist (relative or absolute paths, that's the responsibility

View file

@ -1,5 +1,5 @@
# This file is part of beets.
# Copyright 2016, Johannes Tiefenbacher.
# Copyright 2016, J0J0 Todos.
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
@ -11,22 +11,23 @@
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
"""Testsuite for the M3UFile class."""
from os import path, remove
from os import path
from tempfile import mkdtemp
from shutil import rmtree
import unittest
# from unittest.mock import Mock, MagicMock
from beets.util import M3UFile, EmptyPlaylistError
from beets.util import syspath, bytestring_path, py3_path, CHAR_REPLACE
from beets.util import bytestring_path
from beets.util.m3u import M3UFile, EmptyPlaylistError
from test._common import RSRC
class M3UFileTest(unittest.TestCase):
"""Tests the M3UFile class."""
def test_playlist_write_empty(self):
"""Test whether saving an empty playlist file raises an error."""
tempdir = bytestring_path(mkdtemp())
the_playlist_file = path.join(tempdir, b'playlist.m3u8')
m3ufile = M3UFile(the_playlist_file)
@ -35,6 +36,7 @@ class M3UFileTest(unittest.TestCase):
rmtree(tempdir)
def test_playlist_write(self):
"""Test saving ascii paths to a playlist file."""
tempdir = bytestring_path(mkdtemp())
the_playlist_file = path.join(tempdir, b'playlist.m3u')
m3ufile = M3UFile(the_playlist_file)
@ -47,6 +49,7 @@ class M3UFileTest(unittest.TestCase):
rmtree(tempdir)
def test_playlist_write_unicode(self):
"""Test saving unicode paths to a playlist file."""
tempdir = bytestring_path(mkdtemp())
the_playlist_file = path.join(tempdir, b'playlist.m3u8')
m3ufile = M3UFile(the_playlist_file)
@ -59,6 +62,7 @@ class M3UFileTest(unittest.TestCase):
rmtree(tempdir)
def test_playlist_load_ascii(self):
"""Test loading ascii paths from a playlist file."""
the_playlist_file = path.join(RSRC, b'playlist.m3u')
m3ufile = M3UFile(the_playlist_file)
m3ufile.load()
@ -66,6 +70,7 @@ class M3UFileTest(unittest.TestCase):
'/This/is/a/path/to_a_file.mp3\n')
def test_playlist_load_unicode(self):
"""Test loading unicode paths from a playlist file."""
the_playlist_file = path.join(RSRC, b'playlist.m3u8')
m3ufile = M3UFile(the_playlist_file)
m3ufile.load()
@ -73,12 +78,14 @@ class M3UFileTest(unittest.TestCase):
'/This/is/å/path/to_a_file.mp3\n')
def test_playlist_load_extm3u(self):
"""Test loading a playlist with an #EXTM3U header."""
the_playlist_file = path.join(RSRC, b'playlist.m3u')
m3ufile = M3UFile(the_playlist_file)
m3ufile.load()
self.assertTrue(m3ufile.extm3u)
def test_playlist_load_non_extm3u(self):
"""Test loading a playlist without an #EXTM3U header."""
the_playlist_file = path.join(RSRC, b'playlist_non_ext.m3u')
m3ufile = M3UFile(the_playlist_file)
m3ufile.load()
@ -86,6 +93,7 @@ class M3UFileTest(unittest.TestCase):
def suite():
"""This testsuite's main function."""
return unittest.TestLoader().loadTestsFromName(__name__)