salt/badwitch
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Cleaner args

Browse Source
This commit is contained in:
Salt 2020-03-03 19:25:40 -06:00
parent b3bd9c616b
commit d59cd47f36
1 changed files with 50 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
badwitch.py
View File

@ -41,15 +41,17 @@ class Library:
class BadWitch: class BadWitch:
# Our program # Our program
def __init__(self): def __init__(self):
# Set up arguments # Flags and arguments
self.argparser = argparse.ArgumentParser(description='Manage a declarative music library through YouTube scraping') self.argparser = argparse.ArgumentParser(
self.argparser.add_argument('action', metavar='action', nargs=1, description='Manage a declarative music library through YouTube scraping')
choices=['download', 'add', 'remove', 'list', 'test'], self.argparser.add_argument('-l', '--library', metavar='f', nargs='?',
help='Action to perform on the library')
self.argparser.add_argument('-l', '--library', metavar='f', nargs=1,
help='Override default library file with this one') help='Override default library file with this one')
self.argparser.add_argument('-v', '--verbose', action='store_true', self.argparser.add_argument('-v', '--verbose', action='store_true',
help='Show more status messages') help='Show more status messages')
# Subparsers
self.argparser.add_argument('action', metavar='action', nargs='?',
choices=['download', 'add', 'remove', 'list', 'test'],
help='Action to perform on the library')
# Set up appdirs # Set up appdirs
self.dirs = AppDirs('badwitch', 'rehashedsalt') self.dirs = AppDirs('badwitch', 'rehashedsalt')
Path(self.dirs.user_data_dir).mkdir(parents=True, exist_ok=True) Path(self.dirs.user_data_dir).mkdir(parents=True, exist_ok=True)
@ -57,25 +59,22 @@ class BadWitch:
def execute(self): def execute(self):
self.args = self.argparser.parse_args() self.args = self.argparser.parse_args()
# Initialize library # Initialize library
libfile = self.dirs.user_data_dir + '/lib.json' libfile = self.args.library or self.dirs.user_data_dir + '/lib.json'
if self.args.library is not None:
libfile = self.args.library[0]
lib = Library(file=libfile) lib = Library(file=libfile)
# Perform action # Perform action
for action in self.args.action: if self.args.action == 'download':
if action == 'download':
return return
elif action == 'add': elif self.args.action == 'add':
return return
elif action == 'remove': elif self.args.action == 'remove':
return return
elif action == 'list': elif self.args.action == 'list':
lib.load() lib.load()
for album, albumcontent in lib.albums.items(): for album, albumcontent in lib.albums.items():
for song, songcontent in albumcontent.items(): for song, songcontent in albumcontent.items():
print(album + ' - ' + song + ' by ' + songcontent['artist']) print(album + ' - ' + song + ' by ' + songcontent['artist'])
return return
elif action == 'test': elif self.args.action == 'test':
# Set up a test album # Set up a test album
lib.albums['Bad Witch'] = { lib.albums['Bad Witch'] = {
'Shit Mirror': { 'Shit Mirror': {