#! /usr/bin/env python3 # -*- coding: utf-8 -*- # vim:fenc=utf-8:ft=python # # Bad Witch # Copyright © 2020 Vintage Salt # # Distributed under terms of the MIT license. # from appdirs import AppDirs from pathlib import Path import argparse import json import sys class Library: # A thing full of albums def __init__(self, file, albums={}): self.albums = albums self.file = file # Load from file def load(self, file): with open(self.file, 'r+') as libfd: libfd.seek(0) return # Save to file def save(self): with open(self.file, 'w+') as libfd: libfd.seek(0) json.dump(self.albums, libfd, indent='\t') return def download(self): for album in self.albums: return class BadWitch: # Our program def __init__(self): # Set up arguments self.argparser = argparse.ArgumentParser(description='Manage a declarative music library through YouTube scraping') self.argparser.add_argument('action', metavar='action', nargs=1, choices=['download', 'add', 'remove', 'list', 'test'], 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') self.argparser.add_argument('-v', '--verbose', action='store_true', help='Show more status messages') # Set up appdirs self.dirs = AppDirs('badwitch', 'rehashedsalt') Path(self.dirs.user_data_dir).mkdir(parents=True, exist_ok=True) def execute(self): self.args = self.argparser.parse_args() # Initialize library libfile = self.args.library[0] or self.dirs.user_data_dir + '/lib.json' lib = Library(file=libfile) # Perform action for action in self.args.action: if action == 'download': return elif action == 'add': return elif action == 'remove': return elif action == 'list': for album in lib.albums: for song in lib.songs: print(song.title + ' by ' + song.artist + ', released ' + song.releasedate) return elif action == 'test': # Set up a test album lib.albums['Bad Witch'] = { 'Shit Mirror': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=yeqjz5mXrLQ' }, 'Ahead of Ourselves': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=4Ab1O-i4ep4' }, 'Play the Goddamned Part': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=85UgvBkMfr8' }, 'God Break Down the Door': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=eeJ_DzRJUI4' }, 'I\'m Not From This World': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=9fjbcSUSt9w' }, 'Over and Out': { 'artist': 'Nine Inch Nails', 'source': 'https://www.youtube.com/watch?v=h-XlN3N2fis' } } lib.save() return badwitch = BadWitch() badwitch.execute()