todo.py

from os import environ
from functools import partial

from pymlconf import DeferredRoot
from easycli import Root, Argument, SubCommand


__version__ = '0.1.0'
HOME = environ['HOME']
BUILTINSETTINGS = f'''
database:
  filename: {HOME}/.cache/todo.csv
'''
settings = DeferredRoot()


ItemArgument = partial(
    Argument,
    'item',
    help='item name'
)


ListArgument = partial(
    Argument,
    'description',
    nargs='?',
    default='',
    help='description name'
)


class DB:
    def __init__(self):
        with open(self.filename) as dbfile:
            self.items = [line.strip() for line in dbfile]

    @property
    def filename(self):
        return settings.database.filename

    def append(self, item):
        self.items.append(item)
        with open(self.filename, 'w') as f:
            for i in self.items:
                f.write(i)
                f.write('\n')

    def delete(self, item):
        data = []
        for line in self.items:
            if item != line.strip().split(':', 1)[0]:
                data.append(line)

        with open(self.filename, 'w') as f:
            for i in data:
                f.write(i)
                f.write('\n')


def initialize_settings(configfile=None):
    settings.initialize(BUILTINSETTINGS)
    if configfile is not None:
        settings.loadfile(configfile)


class Search(SubCommand):
    __command__ = 'search'
    __aliases__ = ['s']
    __help__ = 'To print searched items(no argument print the whole list).'
    __arguments__ = [
        ItemArgument(
            nargs='?',
            default=''
        )
    ]

    def __call__(self, args):
        db = DB()
        if args.item == '':
            for i in db.items:
                print(i)

        else:
            for i in db.items:
                if i.strip().split(':', 1)[0] == args.item:
                    print(i)


class Delete(SubCommand):
    __command__ = 'delete'
    __aliases__ = ['d', 'remove', 'r']
    __help__ = 'To delete an item and its desctiption.'
    __arguments__ = [ItemArgument()]

    def __call__(self, args):
        db = DB()
        deleteitem = args.item.split(':', 1)[0]
        db.delete(deleteitem)


class Append(SubCommand):
    __command__ = 'append'
    __aliases__ = ['add', 'a']
    __help__ = 'Append item and description to the todo list.'
    __arguments__ = [
        ItemArgument(),
        ListArgument()
    ]

    def __call__(self, args):
        db = DB()
        appendline = f'{args.item.strip()}:\t{args.description.strip()}'
        db.append(appendline)


class List(SubCommand):
    __command__ = 'list'
    __aliases__ = ['l']
    __help__ = 'Print the todo list.'

    def __call__(self, args):
        db = DB()
        for i in db.items:
            print(i)


class Todo(Root):
    __help__ = 'Simple todo list'
    __completion__ = True
    __arguments__ = [
        Argument(
            '-v', '--version',
            action='store_true',
            help='Show version'
        ),
        Argument(
            '-c', '--configfile',
            metavar='FILENAME',
            help='Configuration file to load.'
        ),
        List,
        Append,
        Delete,
        Search
    ]

    def _execute_subcommand(self, args):
        initialize_settings(args.configfile)
        return super()._execute_subcommand(args)

    def __call__(self, args):
        if args.version:
            print(__version__)
            return

        return super().__call__(args)