nowpass.py

#!/usr/bin/env python3

import os
import sys
import logging
import nowpass.parser
import nowpass.element
import nowpass.engine
import nowpass.main_config
import errno

try:
 sys.setdefaultencoding('UTF8')
except Exception:
 pass

STORAGE_FOLDER = os.path.expanduser('~/.nowpass')
MAIN_CONFIG = STORAGE_FOLDER + '/main.conf'

# Testing work around for shortcuts
COMMANDS = {
 'del': 'delete', 'delete': 'delete',
 'a': 'add', 'add': 'add',
 'l': 'list', 'list': 'list',
 's': 'search', 'search': 'search',
 'e': 'edit', 'edit': 'edit'
}


def main():
 # Storage folder
 create_storage_folder(STORAGE_FOLDER)

 # Start args parser (needed for logging file)
 # Wrapper Class
 parser_class = nowpass.parser.Parser()
 parser = parser_class.get()

 args = parser.parse_args()

 # Logging
 logger = logging.getLogger('nowpass')
 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

 # create console handler with a higher log level
 ch = logging.StreamHandler()
 ch.setLevel(logging.INFO)
 ch.setFormatter(formatter)

 # File logging with debug
 fh = logging.FileHandler(os.path.expanduser(args.logfile))
 fh.setLevel(logging.DEBUG)
 fh.setFormatter(formatter)

 logger.addHandler(fh)
 logger.addHandler(ch)

 logger.debug('Starting nowpass CLI client')

 # Config
 main_config = nowpass.main_config.MainConfig(logger, MAIN_CONFIG)
 config = main_config.get_config()

 # Passphrase (stored or via input)
 if config['Encryption']['passphrase'] == '':
 config['Encryption']['passphrase'] = input('Enter your pass phrase: ')

 # Starting here we need a command
 if args.command is None:
 print('Please supply a command, check --help for more details.')
 exit(1)

 command = args.command

 logger.debug('Command: ' + command)

 # Get the Engine for processing tasks
 engine = nowpass.engine.Engine(logger, main_config)

 task_to_call = getattr(engine, COMMANDS[command])
 task_to_call(args)

 exit(0)


def create_storage_folder(path):
 try:
 os.makedirs(path)
 except OSError as exception:
 if exception.errno != errno.EEXIST:
 raise


if __name__ == "__main__":
 main()


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4