pubs/papers/commands/init_cmd.py

# init command

import os

from .. import databroker
from ..configs import config
from ..uis import get_ui
from .. import color


def parser(subparsers):
    parser = subparsers.add_parser('init',
                                   help="initialize the papers directory")
    parser.add_argument('-p', '--pubsdir', default=None,
                        help='path to papers directory (if none, ~/.papers is used)')
    parser.add_argument('-d', '--docsdir', default='docsdir://',
                        help=('path to document directory (if not specified, documents will'
                              'be stored in /path/to/pubsdir/doc/)'))
    return parser


def command(args):
    """Create a .papers directory"""

    ui = get_ui()
    pubsdir = args.pubsdir
    docsdir = args.docsdir

    if pubsdir is None:
        pubsdir = '~/.papers'
    
    pubsdir = os.path.normpath(os.path.abspath(os.path.expanduser(pubsdir)))

    if os.path.exists(pubsdir) and len(os.listdir(pubsdir)) > 0:
            ui.error('directory {} is not empty.'.format(
                                 color.dye(pubsdir, color.filepath)))
            ui.exit()

    ui.print_('Initializing papers in {}.'.format(
              color.dye(pubsdir, color.filepath)))

    config().pubsdir = pubsdir
    config().docsdir = docsdir
    config().save()

    databroker.DataBroker(pubsdir, create=True)