badwitch/badwitch.py

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# vim:fenc=utf-8:ft=python
#
# Bad Witch
# Copyright © 2020 Vintage Salt <rehashedsalt@cock.li>
#
# 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()