mirror of
https://github.com/tdeerenberg/Musort.git
synced 2025-07-16 00:54:19 +00:00
Python Source Code
This commit is contained in:
tdeerenberg
211
src/musort.py
Executable file
211
src/musort.py
Executable file
@ -0,0 +1,211 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
from tinytag import TinyTag
|
||||
import os
|
||||
import sys, getopt
|
||||
|
||||
help=\
|
||||
"""Musort © 2023 tdeerenberg (github.com/tdeerenberg)
|
||||
|
||||
DESCRIPTION:
|
||||
A Python3 program that renames all selected music/audio files in a folder with a specified naming convention
|
||||
|
||||
USAGE:
|
||||
musort [DIRECTORY] [NAMING_CONVENTION] [OPTIONAL_OPTIONS]...
|
||||
|
||||
USAGE EXAMPLES:
|
||||
musort ~/music track.title.year -s _ -r
|
||||
musort /local/music disc.artist.title.album -r
|
||||
musort ~/my_music track.title
|
||||
|
||||
OPTIONAL OPTIONS:
|
||||
-h, --help Show the help menu
|
||||
-s, --separator Set the separator for the filename (ex. '-s .' -> 01.track.flac and '-s -' -> 01-track.mp3)
|
||||
Default separator ( . ) will be used if none is given
|
||||
-r, --recursive Rename files in subdirectories as well
|
||||
|
||||
NAMING CONVENTION:
|
||||
FORMAT_OPTION.FORMAT_OPTION... The amount of format options does not matter.
|
||||
It can be one, two, three, even all of them.
|
||||
(See FORMAT OPTIONS below for all options)
|
||||
|
||||
FORMAT OPTIONS:
|
||||
album album as string
|
||||
albumartist album artist as string
|
||||
artist artist name as string
|
||||
audio_offset number of bytes before audio data begins
|
||||
bitdepth bit depth for lossless audio
|
||||
bitrate bitrate in kBits/s
|
||||
comment file comment as string
|
||||
composer composer as string
|
||||
disc disc number
|
||||
disc_total the total number of discs
|
||||
duration duration of the song in seconds
|
||||
filesize file size in bytes
|
||||
genre genre as string
|
||||
samplerate samples per second
|
||||
title title of the song
|
||||
track track number as string
|
||||
track_total total number of tracks as string
|
||||
year year or date as string"""
|
||||
|
||||
class Music:
|
||||
def get_compatible(self, directory):
|
||||
"""Scans the set directory for compatible audio files"""
|
||||
music = []
|
||||
all_files = list(map(lambda x: os.path.join(os.path.abspath(directory), x),os.listdir(directory)))
|
||||
for file in all_files:
|
||||
match file.split("."):
|
||||
case [*_, "flac"]:
|
||||
music.append(file)
|
||||
case [*_, "mp3"]:
|
||||
music.append(file)
|
||||
case [*_, "mp1"]:
|
||||
music.append(file)
|
||||
case [*_, "mp2"]:
|
||||
music.append(file)
|
||||
case [*_, "opus"]:
|
||||
music.append(file)
|
||||
case [*_, "ogg"]:
|
||||
music.append(file)
|
||||
case [*_, "wma"]:
|
||||
music.append(file)
|
||||
self.directory = directory
|
||||
self.compatible = music
|
||||
|
||||
def get_compatible_recursive(self, directory):
|
||||
"""Scans the set directory with subdirectory for compatible audio files"""
|
||||
files = []
|
||||
music = []
|
||||
for a, b, c in os.walk(directory):
|
||||
for d in c:
|
||||
files.append(os.path.join(a,d))
|
||||
for file in files:
|
||||
match file.split("."):
|
||||
case [*_, "flac"]:
|
||||
music.append(file)
|
||||
case [*_, "mp3"]:
|
||||
music.append(file)
|
||||
case [*_, "mp1"]:
|
||||
music.append(file)
|
||||
case [*_, "mp2"]:
|
||||
music.append(file)
|
||||
case [*_, "opus"]:
|
||||
music.append(file)
|
||||
case [*_, "ogg"]:
|
||||
music.append(file)
|
||||
case [*_, "wma"]:
|
||||
music.append(file)
|
||||
self.directory = directory
|
||||
self.compatible = music
|
||||
|
||||
def set_separator(self, sep):
|
||||
"""Sets the separator for naming the audio files
|
||||
(ex. 01-songname.mp3 or 01.songname.flac)"""
|
||||
self.separator = sep
|
||||
self.separator_status = True
|
||||
def set_format(self, val):
|
||||
"""Sets the naming convention of the audio files
|
||||
(ex. title-artist or artist-track-title)"""
|
||||
self.format = val.split(".")
|
||||
|
||||
# Rename files
|
||||
def rename_music(self):
|
||||
"""Rename all compatible music files"""
|
||||
|
||||
"""Get the file extension (ex. .flac, .mp3, etc)"""
|
||||
for file in self.compatible:
|
||||
ext = file.split(".")
|
||||
ext = "." + ext[-1]
|
||||
|
||||
"""Let TinyTag module read the audio file"""
|
||||
track = TinyTag.get(file)
|
||||
|
||||
"""Print the progress (Current track)"""
|
||||
print(f"\nCurrent track: '{track.title}' by '{track.artist}'")
|
||||
rename = []
|
||||
|
||||
"""Uses the given format to set new filename"""
|
||||
for f in self.format:
|
||||
match f:
|
||||
case "track":
|
||||
rename.append(f"{int(track.track):02}")
|
||||
case "album":
|
||||
rename.append(track.album)
|
||||
case "albumartist":
|
||||
rename.append(track.albumartist)
|
||||
case "artist":
|
||||
rename.append(track.artist)
|
||||
case "audio_offset":
|
||||
rename.append(track.audio_offset)
|
||||
case "bitdepth":
|
||||
rename.append(track.bitdepth)
|
||||
case "bitrate":
|
||||
rename.append(track.bitrate)
|
||||
case "comment":
|
||||
rename.append(track.commment)
|
||||
case "composer":
|
||||
rename.append(track.composer)
|
||||
case "disc":
|
||||
rename.append(track.disc)
|
||||
case "disc_total":
|
||||
rename.append(track.disc_total)
|
||||
case "duration":
|
||||
rename.append(track.duration)
|
||||
case "filesize":
|
||||
rename.append(track.filesize)
|
||||
case "genre":
|
||||
rename.append(track.genre)
|
||||
case "samplerate":
|
||||
rename.append(track.samplerate)
|
||||
case "title":
|
||||
rename.append(track.title)
|
||||
case "track_total":
|
||||
rename.append(track.track_total)
|
||||
case "year":
|
||||
rename.append(track.year)
|
||||
rename.append(self.separator)
|
||||
rename.pop()
|
||||
rename = ''.join(rename)+ext
|
||||
|
||||
"""Get the absolute path and rename the audio file"""
|
||||
dst = os.path.join(os.path.abspath(os.path.dirname(file)), rename)
|
||||
os.rename(file, dst)
|
||||
print("Done")
|
||||
|
||||
|
||||
|
||||
def main():
|
||||
"""Runs the whole program"""
|
||||
argv = sys.argv[3:]
|
||||
try:
|
||||
opts, args = getopt.getopt(argv, "s:rh", ["sep=", "recursive=", "help="])
|
||||
except getopt.GetoptError as err:
|
||||
print(err)
|
||||
exit()
|
||||
|
||||
music = Music()
|
||||
for opt, arg in opts:
|
||||
if opt in ['-h']:
|
||||
print(help)
|
||||
exit()
|
||||
if opt in ['-s']:
|
||||
music.set_separator(arg)
|
||||
if opt in ['-r']:
|
||||
music.get_compatible_recursive(sys.argv[1])
|
||||
|
||||
try:
|
||||
music.compatible
|
||||
except:
|
||||
print("Running not recursively")
|
||||
music.get_compatible(sys.argv[1])
|
||||
try:
|
||||
music.separator
|
||||
except:
|
||||
print("Using default separator: .")
|
||||
music.set_separator(".")
|
||||
|
||||
music.set_format(sys.argv[2])
|
||||
music.rename_music()
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user