#!/usr/local/bin/python3

import os

import sys

import re

import shutil

# trawl through b2 files and find ones name something disk x

# move them all into one album - but with tracks prepended with

# disk x

print(sys.argv[1])

for dirname,dirnames,filenames in os.walk(sys.argv[1]):

  for subdirname in dirnames:

    print(os.path.join(dirname, subdirname))

    m=re.search("\s+[Dd]isk\s*\d+$",subdirname)

    if m:

      #print("Multi-disk: "+subdirname.replace(m.group(0),""))

      singlename=os.path.join(dirname,subdirname.replace(m.group(0),""))

      if not os.path.isdir(singlename):

        os.mkdir(singlename)

  for filename in filenames:

    print(os.path.join(dirname, filename))

    f=re.search("\.(flac|mp3|jpg|wav)$",filename)

    if f is None and (filename!="discid" and filename!="toc"):

        print("skipsies")

        continue

    m=re.search("\s+[Dd]isk\s*\d+$",dirname)

    if m:

      singlename=dirname.replace(m.group(0),"")

      n=m.group(0).replace(" ","")

      print("mv "+os.path.join(dirname, filename)+" "+os.path.join(singlename, n+" - "+filename))

      shutil.move(os.path.join(dirname, filename),os.path.join(singlename, n+" - "+filename))

  if len(os.listdir(dirname) ) == 0:

    print("Cleaning Empty Dir: "+dirname)

    os.rmdir(dirname)