Refactor complete for test
This commit is contained in:
parent
af974cf0e4
commit
c863b7bb2d
2
.gitignore
vendored
2
.gitignore
vendored
@ -123,3 +123,5 @@ tags
|
|||||||
|
|
||||||
|
|
||||||
# End of https://www.gitignore.io/api/vim,python,visualstudiocode
|
# End of https://www.gitignore.io/api/vim,python,visualstudiocode
|
||||||
|
|
||||||
|
test/test/*
|
||||||
|
211
CommandExecutioner.py
Normal file
211
CommandExecutioner.py
Normal file
@ -0,0 +1,211 @@
|
|||||||
|
from tinytag import TinyTag
|
||||||
|
import os
|
||||||
|
# from functools import reduce
|
||||||
|
import pprint
|
||||||
|
from binascii import b2a_hex
|
||||||
|
|
||||||
|
|
||||||
|
class CommandExecutioner:
|
||||||
|
def __init__(self, kwargs):
|
||||||
|
self.options = kwargs
|
||||||
|
|
||||||
|
def work(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def scan(self):
|
||||||
|
"""
|
||||||
|
set self.files to be a list of files
|
||||||
|
element in list are in full path
|
||||||
|
"""
|
||||||
|
self.files = []
|
||||||
|
|
||||||
|
for root, dirs, files in os.walk(self.source):
|
||||||
|
print("[+] Scan {}".format(root))
|
||||||
|
dirs[:] = [d for d in dirs if not d.startswith('.')]
|
||||||
|
for folder in dirs:
|
||||||
|
folder = os.path.join(root, folder)
|
||||||
|
for f in files:
|
||||||
|
self.files.append(os.path.join(root, f))
|
||||||
|
|
||||||
|
def _scan(self):
|
||||||
|
"""
|
||||||
|
deprecated
|
||||||
|
"""
|
||||||
|
# files to work on
|
||||||
|
# scan all files and store whole path on here
|
||||||
|
self.files = []
|
||||||
|
|
||||||
|
folder_queue = [self.source]
|
||||||
|
home_path_len = len(folder_queue[0])
|
||||||
|
|
||||||
|
while (len(folder_queue) > 0):
|
||||||
|
current_folder = folder_queue[0]
|
||||||
|
folder_queue = folder_queue[1:]
|
||||||
|
|
||||||
|
if len(current_folder[home_path_len:]) == 0:
|
||||||
|
print("[+] Scan /")
|
||||||
|
else:
|
||||||
|
print("[+] Scan {}".format(current_folder[home_path_len:]))
|
||||||
|
|
||||||
|
self.files += list(
|
||||||
|
filter(
|
||||||
|
lambda x: os.path.isfile(x),
|
||||||
|
os.listdir(current_folder))
|
||||||
|
)
|
||||||
|
for f in self.files:
|
||||||
|
f = current_folder + '/' + f
|
||||||
|
|
||||||
|
folders = list(
|
||||||
|
filter(
|
||||||
|
lambda x:
|
||||||
|
os.path.isdir(x) and x[0] != '.',
|
||||||
|
os.listdir(current_folder)))
|
||||||
|
|
||||||
|
for folder in folders:
|
||||||
|
folder_queue.append(current_folder + '/' + folder)
|
||||||
|
|
||||||
|
def makedict(self):
|
||||||
|
"""
|
||||||
|
from list of files
|
||||||
|
to dictionary of file by attribute specified
|
||||||
|
"""
|
||||||
|
self.data = {}
|
||||||
|
for f in self.files:
|
||||||
|
try:
|
||||||
|
tag = TinyTag.get(f)
|
||||||
|
except LookupError:
|
||||||
|
# not a music file
|
||||||
|
continue
|
||||||
|
except BaseException:
|
||||||
|
print("Cannot get tag from file --> Skip\n\t{}".format(f))
|
||||||
|
continue
|
||||||
|
tag = getattr(tag, self.attribute)
|
||||||
|
if tag in self.data:
|
||||||
|
self.data[tag].append(f)
|
||||||
|
else:
|
||||||
|
self.data[tag] = []
|
||||||
|
self.data[tag].append(f)
|
||||||
|
|
||||||
|
|
||||||
|
class CommandSort(CommandExecutioner):
|
||||||
|
def __init__(self, kwargs):
|
||||||
|
super().__init__(kwargs)
|
||||||
|
self.source = kwargs['source']
|
||||||
|
self.destination = kwargs['destination']
|
||||||
|
self.attribute = kwargs['attribute']
|
||||||
|
|
||||||
|
def work(self):
|
||||||
|
self.scan()
|
||||||
|
self.makedict()
|
||||||
|
pprint.pprint(self.data)
|
||||||
|
self.sort()
|
||||||
|
|
||||||
|
def sort(self):
|
||||||
|
# key is now the name of the folder
|
||||||
|
for folder_name, tracks in self.data.items():
|
||||||
|
# print(folder_name)
|
||||||
|
# continue
|
||||||
|
|
||||||
|
if folder_name is None:
|
||||||
|
# file attribute is None
|
||||||
|
folder = "Undefined"
|
||||||
|
elif folder_name == '':
|
||||||
|
# file attribute is empty string
|
||||||
|
folder = "Undefined"
|
||||||
|
else:
|
||||||
|
folder = folder_name
|
||||||
|
|
||||||
|
# because folder is taken from file tags
|
||||||
|
# some tags could have '/' in it
|
||||||
|
# which is not acceptable as a file name in linux
|
||||||
|
# folder = folder.replace('/', '-')
|
||||||
|
|
||||||
|
# print(self.destination)
|
||||||
|
folder = self.destination + '/' + folder
|
||||||
|
if not os.path.exists(folder):
|
||||||
|
os.makedirs(folder)
|
||||||
|
|
||||||
|
print("========================================")
|
||||||
|
print("Album: {}".format(folder_name))
|
||||||
|
print("Folder: {}".format(folder))
|
||||||
|
# continue
|
||||||
|
for track in tracks:
|
||||||
|
new_file = folder + '/' + os.path.basename(track)
|
||||||
|
if track == new_file:
|
||||||
|
# after sort, stay the same
|
||||||
|
continue
|
||||||
|
if os.path.exists(new_file):
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
print("[+] Replace\n\t{}\n\t{}".format(track, new_file))
|
||||||
|
# uncomment when you are ready
|
||||||
|
# os.rename(track, new_file)
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
class CommandPlaylist(CommandExecutioner):
|
||||||
|
def __init__(self, kwargs):
|
||||||
|
super().__init__(kwargs)
|
||||||
|
self.source = kwargs['source']
|
||||||
|
self.destination = kwargs['destination']
|
||||||
|
self.attribute = kwargs['attribute']
|
||||||
|
self.playlist = kwargs['playlist']
|
||||||
|
|
||||||
|
def work(self):
|
||||||
|
self.scan()
|
||||||
|
self.makedict()
|
||||||
|
print("Create playlist {}".format(self.playlist))
|
||||||
|
|
||||||
|
playlist = open(self.playlist, 'w')
|
||||||
|
playlist.write('#EXTM3U\n')
|
||||||
|
|
||||||
|
for key, musics in self.data.items():
|
||||||
|
for music in musics:
|
||||||
|
tag = TinyTag.get(music)
|
||||||
|
if tag.artist is None:
|
||||||
|
tag.artist = ''
|
||||||
|
if tag.title is None:
|
||||||
|
tag.title = ''
|
||||||
|
tag.duration = int(tag.duration)
|
||||||
|
|
||||||
|
# write comment
|
||||||
|
playlist.write('#EXTINF:{},{} - {}\n'
|
||||||
|
.format(
|
||||||
|
tag.duration,
|
||||||
|
tag.artist,
|
||||||
|
tag.title))
|
||||||
|
|
||||||
|
# write file direction
|
||||||
|
music = encode_to_hex(music)
|
||||||
|
playlist.write('file://{}\n'.format(music))
|
||||||
|
|
||||||
|
print("{} created".format(self.playlist))
|
||||||
|
|
||||||
|
|
||||||
|
def encode_to_hex(string):
|
||||||
|
"""
|
||||||
|
change special char to hex
|
||||||
|
"""
|
||||||
|
chars = list(string)
|
||||||
|
for i in range(len(string)):
|
||||||
|
hex_c = ord(chars[i])
|
||||||
|
if hex_c >= ord('!') and hex_c <= ord('~'):
|
||||||
|
# skip ascii characters
|
||||||
|
# be aware of special ascii!!!
|
||||||
|
continue
|
||||||
|
elif hex_c == ord(' '):
|
||||||
|
chars[i] = '%20'
|
||||||
|
else:
|
||||||
|
# not ascii change to hex
|
||||||
|
u = b2a_hex(chars[i].encode('utf-8')).decode('utf-8')
|
||||||
|
u = list(u)
|
||||||
|
for j in range(len(u)):
|
||||||
|
u[j] = u[j].upper()
|
||||||
|
if j % 2 != 0:
|
||||||
|
continue
|
||||||
|
u[j] = '%' + u[j]
|
||||||
|
chars[i] = "".join(u)
|
||||||
|
|
||||||
|
string = "".join(chars)
|
||||||
|
return string
|
90
CommandHandler.py
Normal file
90
CommandHandler.py
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
import os
|
||||||
|
from CommandExecutioner import (
|
||||||
|
CommandSort,
|
||||||
|
CommandPlaylist
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
argument_requirement = {
|
||||||
|
'sort':
|
||||||
|
['source', 'destination', 'attribute'],
|
||||||
|
'playlist':
|
||||||
|
['source', 'destination', 'attribute', 'playlist_name'],
|
||||||
|
}
|
||||||
|
|
||||||
|
attributes = ('title', 'album', 'artist')
|
||||||
|
|
||||||
|
|
||||||
|
class CommandHandler:
|
||||||
|
"""
|
||||||
|
Pre validation on arugments
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, kwargs):
|
||||||
|
# print(kwargs)
|
||||||
|
self.run(kwargs)
|
||||||
|
|
||||||
|
def run(self, kwargs):
|
||||||
|
kwargs['source'] = os.path.abspath(kwargs['source'])
|
||||||
|
kwargs['destination'] = os.path.abspath(kwargs['destination'])
|
||||||
|
try:
|
||||||
|
self.validate(kwargs)
|
||||||
|
except Exception:
|
||||||
|
exit(-1)
|
||||||
|
|
||||||
|
mode = kwargs['mode']
|
||||||
|
worker = None
|
||||||
|
if mode == 'sort':
|
||||||
|
worker = CommandSort(kwargs)
|
||||||
|
elif mode == 'playlist':
|
||||||
|
worker = CommandPlaylist(kwargs)
|
||||||
|
else:
|
||||||
|
# not likely
|
||||||
|
return
|
||||||
|
|
||||||
|
worker.work()
|
||||||
|
|
||||||
|
def validate(self, kwargs):
|
||||||
|
requirements = argument_requirement[kwargs['mode']]
|
||||||
|
for requirement in requirements:
|
||||||
|
if requirement == 'source':
|
||||||
|
if not os.path.isdir(kwargs['source']):
|
||||||
|
print(
|
||||||
|
'source:',
|
||||||
|
kwargs['source'],
|
||||||
|
'is not a valid directory')
|
||||||
|
print('directory not exist or is not a directory')
|
||||||
|
raise Exception
|
||||||
|
if requirement == 'destination':
|
||||||
|
if not os.path.isdir(kwargs['destination']):
|
||||||
|
print(
|
||||||
|
'destination:',
|
||||||
|
kwargs['destination'],
|
||||||
|
'is not a valid directory')
|
||||||
|
print('directory not exist or is not a directory')
|
||||||
|
raise Exception
|
||||||
|
elif requirement == 'attribute':
|
||||||
|
if kwargs['attribute'] not in attributes:
|
||||||
|
print(
|
||||||
|
'attribute:',
|
||||||
|
kwargs['attribute'],
|
||||||
|
'is not a valid attribute')
|
||||||
|
print('valid attributes are:', attributes)
|
||||||
|
raise Exception
|
||||||
|
elif requirement == 'playlist_name':
|
||||||
|
if kwargs['playlist'] is None:
|
||||||
|
print("Playlist name must be given")
|
||||||
|
raise Exception
|
||||||
|
pl_file = kwargs['destination'] + \
|
||||||
|
'/' + kwargs['playlist'] + '.m3u'
|
||||||
|
while True:
|
||||||
|
if not os.path.exists(pl_file):
|
||||||
|
break
|
||||||
|
overwrite = input(
|
||||||
|
'File path {} is existed, overwrite?(y/n) '
|
||||||
|
.format(pl_file))
|
||||||
|
if overwrite in ('y', 'Y'):
|
||||||
|
break
|
||||||
|
newname = input("New file name: ")
|
||||||
|
pl_file = kwargs['destination'] + '/' + newname + '.m3u'
|
||||||
|
kwargs['playlist'] = pl_file
|
46
run.py
Normal file
46
run.py
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
# from .commandparser import commandparser
|
||||||
|
import click
|
||||||
|
import os
|
||||||
|
from CommandHandler import CommandHandler
|
||||||
|
|
||||||
|
|
||||||
|
@click.group()
|
||||||
|
# @click.option('--verbose', '-v', is_flag=True, default=False)
|
||||||
|
def main():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@main.command()
|
||||||
|
@click.option('--source', '-src', default=os.getcwd())
|
||||||
|
@click.option('--destination', '-dst', default=os.getcwd() + '/dst/')
|
||||||
|
@click.option('--attribute', '-attr', default='album')
|
||||||
|
@click.option('--auto-overwrite', is_flag=True, default=False)
|
||||||
|
def sort(**kwargs):
|
||||||
|
commandparser('sort', **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
@main.command()
|
||||||
|
@click.option('--source', '-src', default=os.getcwd())
|
||||||
|
@click.option('--destination', '-dst', default=os.getcwd() + '/dst/')
|
||||||
|
@click.option('--attribute', '-attr', default='album')
|
||||||
|
@click.option('--auto-overwrite', is_flag=True, default=False)
|
||||||
|
@click.option('--playlist', '-name')
|
||||||
|
def playlist(**kwargs):
|
||||||
|
commandparser('playlist', **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
@main.command()
|
||||||
|
@click.option('--source', '-src', default=os.getcwd())
|
||||||
|
@click.option('--destination', '-dst', default=os.getcwd())
|
||||||
|
@click.option('--format', '-fmt')
|
||||||
|
def format(**kwargs):
|
||||||
|
commandparser('format', **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
def commandparser(mode, **kwargs):
|
||||||
|
kwargs['mode'] = mode
|
||||||
|
CommandHandler(kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in New Issue
Block a user