pwman3/scripts/pwman3

#!/usr/bin/env python
#============================================================================
# This file is part of Pwman3.
#
# Pwman3 is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License, version 2
# as published by the Free Software Foundation;
#
# Pwman3 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Pwman3; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#============================================================================
# Copyright (C) 2012 Oz Nahum <nahumoz@gmail.com>
#============================================================================
# Copyright (C) 2006 Ivan Kelly <ivan@ivankelly.net>
#============================================================================
from __future__ import print_function
import os
import argparse
import sys
import re


_saveconfig = True

_db_warn = ("\n*** WARNNING: You are using the old database format"
            " which is insecure."
            " Please upgrade to the new database "
            " format. Do note: support for this DB format will be dropped in"
            " v0.5. This  database format is on hold. No bugs are fixead"
            " Check the help (pwman3 -h) or look at the manpage which"
            " explains how to proceed. ***")

def parser_options():
    parser = argparse.ArgumentParser(description=('pwman3 - a command line '
                                                  'password manager.'))
    parser.add_argument('-c', '--config', dest='cfile',
                        default=os.path.expanduser("~/.pwman/config"),
                        help='cofiguration file to read')
    parser.add_argument('-d', '--database', dest='dbase')
    parser.add_argument('-e', '--encryption', dest="algo",
                        help=("Possible options are: AES(default), ARC2, ARC4,"
                              " Blowfish, CAST, DES, DES3, IDEA, RC5"))
    parser.add_argument('-k', '--convert', dest='dbconvert',
                        action='store_true', default=False,
                        # os.path.expanduser('~/.pwman/pwman.db'),
                        help=("Convert old DB format to version >= 0.4."
                              " The database that will be converted is the"
                              " one found in the config file, or the one given"
                              " as command line argument."))
    parser.add_argument('-O', '--output', dest='output',
                        default=os.path.expanduser('~/.pwman/pwman-newdb.db'),
                        help=("The name of the newly created database after "
                              "converting."))
    parser.add_argument('-t', '--test', help=("Run pwman from current"
                                              " directory"
                                              " without installation"),
                        action="store_true")
    return parser


def get_conf():
    config_dir = os.path.expanduser("~/.pwman")

    if not os.path.isdir(config_dir):
        os.mkdir(config_dir)

    if not os.path.exists(args.cfile):
        config.set_defaults(default_config)
    else:
        config.load(args.cfile)

    return config


def set_xsel(config, OSX):
    if not OSX:
        xselpath = which("xsel")
        config.set_value("Global", "xsel", xselpath)
    elif OSX:
        pbcopypath = which("pbcopy")
        config.set_value("Global", "xsel", pbcopypath)


def set_win_colors(config):
    if 'win' in sys.platform:
        try:
            import colorama
            colorama.init()
        except ImportError:
            config.set_value("Global", "colors", 'no')


def set_umask(config):
    # set umask before creating/opening any files
    try:
        umask = config.get_value("Global", "umask")
        if re.search(r'^\d{4}$', umask):
            os.umask(int(umask))
        else:
            raise ValueError
    except ValueError:
        print("Could not determine umask from config!")
        sys.exit(2)


def set_db(args):
    global _saveconfig
    if args.dbase:
        config.set_value("Database", "filename", args.dbase)
        _saveconfig = False


def set_algorithm(args):
    global _saveconfig
    if args.algo:
        config.set_value("Encryption", "algorithm", args.algo)
        _saveconfig = False


def get_conf_options(args, OSX):

    config = get_conf()
    xselpath = config.get_value("Global", "xselpath")
    if not xselpath:
        set_xsel(config, OSX)

    set_win_colors(config)
    set_db(args)
    set_umask(config)
    dbtype = config.get_value("Database", "type")
    if not dbtype:
        print("Could not read the Database type from the config!")
        sys.exit(1)

    return xselpath, dbtype


def get_db_version(config, dbtype):
    if os.path.exists(config.get_value("Database", "filename")):
        dbver = pwman.data.factory.check_db_version(dbtype)
        if dbver < 0.4:
            print(_db_warn)
    else:
        dbver = 0.4
    return dbver


def main(args):
    PwmanCliNew, OSX = get_ui_platform(sys.platform)
    xselpath, dbtype = get_conf_options(args, OSX)
    enc = CryptoEngine.get()
    dbver = get_db_version(config, dbtype)

    if args.dbconvert:
        dbconvertor = PwmanConvertDB(args, config)
        dbconvertor.run()
        sys.exit(0)

    if dbver < 0.4:
        sys.exit(1)

    db = pwman.data.factory.create(dbtype, dbver)
    cli = PwmanCliNew(db, xselpath, CLICallback)

    try:
        cli.cmdloop()
    except KeyboardInterrupt, e:
        print(e)

    if _saveconfig:
        try:
            config.save(args.cfile)
        except Exception, e:
            print ("Error: %s" % e)
            sys.exit(-1)

if __name__ == '__main__':
    args = parser_options().parse_args()
    if args.test:
        sys.path.insert(0, os.getcwd())

    from pwman import default_config, which
    from pwman.ui import get_ui_platform
    from pwman.ui.tools import CLICallback
    import pwman.util.config as config
    import pwman.data.factory
    from pwman.data.convertdb import PwmanConvertDB
    from pwman.util.crypto import CryptoEngine
    main(args)