retag-mp3.py

#!/usr/bin/python3

import os
import sys
import re
import csv
import eyed3
from datetime import datetime

def main():
  for arg in sys.argv[1:]:
    basepath = arg
    logfilename = 'log_retag_' + datetime.now().strftime("%Y_%m_%d_%H%M%S")
    if not os.path.exists('logs'):
      os.makedirs('logs')
    with open('logs/{}.csv'.format(logfilename), 'w', newline='') as logfile:
      writer = csv.writer(logfile)
      writer.writerow(["oldArtist", "newArtist", "oldTitle", "newTitle", "dirty", "file"])
      print("Processing files in directory: \"%s\"" % (basepath))
      print("================================")
      for subdir, dirs, files in os.walk(basepath):
        for file in files:
          fpath = subdir + os.sep + file
          # ITERATING OVER DIRECTORY CONTENT
          if not file.startswith('.') and file.endswith('.mp3'):
            # ITERATING OVER MP3s
            audiofile = eyed3.load(fpath)
            artist, title = getFormatedTags(audiofile)
            isDirty = (audiofile.tag.artist != artist)|(audiofile.tag.title != title)
            writer.writerow([audiofile.tag.artist, artist, audiofile.tag.title, title, isDirty, fpath])
            audiofile.tag.artist = artist
            audiofile.tag.title = title
            audiofile.tag.save()
            print("Successfully tagged: {}".format(file))

def getFormatedTags(af):
  if hasattr(af, 'tag'):
    # check existence tags
    if all(hasattr(af.tag, attr)\
    for attr in ["artist", "title"]):
      artist = af.tag.artist
      title = af.tag.title
      newTitle = title.replace('[', '(').replace(']', ')')
      newArtist = artist.replace('[', '(').replace(']', ')')

      # remove unnecessary title infos
      newTitle = re.sub(r'(?: \(Remix\)| \(Remixes\)| \(New Remixes\)| \(Original Mix\)| \(Extended Mix\)| \(Club Mix\)| \(Extended Explicit Mix\) | \(Extended Explicit Mix\)| \(Deluxe Mix\))', '', newTitle)
      # replace few title formatings
      newTitle = re.sub(r'(?:Extended Remix\)|Club Remix\))', 'Remix)', newTitle)
      newTitle = re.sub(r'(?:Original Mix\)|Club Mix\)|Extended Mix\))', 'Mix)', newTitle)

      # replace artist featuring formating
      newArtist = re.sub(r'(?:feat\.|FEAT\.|featuring|Featuring)', 'ft.', newArtist)

      # move featuring artists from title to artist tag
      ftMatchRegex = r'\((?:feat\.|FEAT\.|featuring|Featuring|ft.) (.*?)\)'
      titleFeatStrs = re.findall(ftMatchRegex, newTitle)
      if len(titleFeatStrs) > 0:
        newTitle = re.sub(ftMatchRegex, '', newTitle)
        newArtist += ' ft. ' + ' & '.join(titleFeatStrs)

      return newArtist, newTitle

if __name__ == "__main__":
  main()