upstream/IPod-Shuffle-4g
1
0
Fork 1
mirror of https://github.com/nims11/IPod-Shuffle-4g.git synced 2025-12-07 16:08:00 +09:00
Code Issues Projects Releases Packages Wiki Activity Actions

Made mutagen import optional

Browse source
This commit is contained in:
NicoHood 2016-08-27 21:02:59 +02:00
parent a21b0a2fe1
commit 3ff22a4950
1 changed files with 44 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

70
ipod-shuffle-4g.py
View file

@ -1,11 +1,11 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# Builtin libraries
import sys import sys
import struct import struct
import urllib.request, urllib.parse, urllib.error import urllib.request, urllib.parse, urllib.error
import os import os
import hashlib import hashlib
import mutagen
import binascii
import subprocess import subprocess
import collections import collections
import errno import errno
@ -15,6 +15,12 @@ import re
import tempfile import tempfile
import signal import signal
# External libraries
try:
import mutagen
except ImportError:
mutagen = None
audio_ext = (".mp3", ".m4a", ".m4b", ".m4p", ".aa", ".wav") audio_ext = (".mp3", ".m4a", ".m4b", ".m4p", ".aa", ".wav")
list_ext = (".pls", ".m3u") list_ext = (".pls", ".m3u")
def make_dir_if_absent(path): def make_dir_if_absent(path):
@ -359,31 +365,34 @@ class Track(Record):
self["filetype"] = 2 self["filetype"] = 2
text = os.path.splitext(os.path.basename(filename))[0] text = os.path.splitext(os.path.basename(filename))[0]
audio = None
try:
audio = mutagen.File(filename, easy = True)
except:
print("Error calling mutagen. Possible invalid filename/ID3Tags (hyphen in filename?)")
if audio:
# Note: Rythmbox IPod plugin sets this value always 0.
self["stop_at_pos_ms"] = int(audio.info.length * 1000)
artist = audio.get("artist", ["Unknown"])[0] # Try to get album and artist information with mutagen
if artist in self.artists: if mutagen:
self["artistid"] = self.artists.index(artist) audio = None
else: try:
self["artistid"] = len(self.artists) audio = mutagen.File(filename, easy = True)
self.artists.append(artist) except:
print("Error calling mutagen. Possible invalid filename/ID3Tags (hyphen in filename?)")
if audio:
# Note: Rythmbox IPod plugin sets this value always 0.
self["stop_at_pos_ms"] = int(audio.info.length * 1000)
album = audio.get("album", ["Unknown"])[0] artist = audio.get("artist", ["Unknown"])[0]
if album in self.albums: if artist in self.artists:
self["albumid"] = self.albums.index(album) self["artistid"] = self.artists.index(artist)
else: else:
self["albumid"] = len(self.albums) self["artistid"] = len(self.artists)
self.albums.append(album) self.artists.append(artist)
if audio.get("title", "") and audio.get("artist", ""): album = audio.get("album", ["Unknown"])[0]
text = " - ".join(audio.get("title", "") + audio.get("artist", "")) if album in self.albums:
self["albumid"] = self.albums.index(album)
else:
self["albumid"] = len(self.albums)
self.albums.append(album)
if audio.get("title", "") and audio.get("artist", ""):
text = " - ".join(audio.get("title", "") + audio.get("artist", ""))
# Handle the VoiceOverData # Handle the VoiceOverData
if isinstance(text, str): if isinstance(text, str):
@ -622,10 +631,15 @@ class Shuffler(object):
self.lists.append(os.path.abspath(dirpath)) self.lists.append(os.path.abspath(dirpath))
if self.auto_id3_playlists != None: if self.auto_id3_playlists != None:
for grouped_list in group_tracks_by_id3_template(self.tracks, self.auto_id3_playlists): if mutagen:
self.lists.append(grouped_list) for grouped_list in group_tracks_by_id3_template(self.tracks, self.auto_id3_playlists):
self.lists.append(grouped_list)
else:
print("Error: No mutagen found. Cannot generate auto-id3-playlists.")
sys.exit(1)
def write_database(self): def write_database(self):
print("Writing database. This may take a while...")
with open(os.path.join(self.path, "iPod_Control", "iTunes", "iTunesSD"), "wb") as f: with open(os.path.join(self.path, "iPod_Control", "iTunes", "iTunesSD"), "wb") as f:
try: try:
f.write(self.tunessd.construct()) f.write(self.tunessd.construct())
@ -633,6 +647,7 @@ class Shuffler(object):
print("I/O error({0}): {1}".format(e.errno, e.strerror)) print("I/O error({0}): {1}".format(e.errno, e.strerror))
print("Error: Writing iPod database failed.") print("Error: Writing iPod database failed.")
sys.exit(1) sys.exit(1)
print("Database written successfully:") print("Database written successfully:")
print("Tracks", len(self.tracks)) print("Tracks", len(self.tracks))
print("Albums", len(self.albums)) print("Albums", len(self.albums))
@ -740,6 +755,9 @@ if __name__ == '__main__':
if result.rename_unicode: if result.rename_unicode:
check_unicode(result.path) check_unicode(result.path)
if not mutagen:
print("Warning: No mutagen found. Database will not contain any album nor artist information.")
verboseprint("Playlist voiceover requested:", result.playlist_voiceover) verboseprint("Playlist voiceover requested:", result.playlist_voiceover)
verboseprint("Track voiceover requested:", result.track_voiceover) verboseprint("Track voiceover requested:", result.track_voiceover)
if (result.track_voiceover or result.playlist_voiceover): if (result.track_voiceover or result.playlist_voiceover):