123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- import cmd
- import sys
- if sys.version_info.major > 2:
- raw_input = input
- try:
- import readline
- _readline_available = True
- except ImportError as e:
- import pyreadline as readline
- _readline_available = True
- from pwman.ui.baseui import BaseCommands
- from pwman import (get_conf_options, get_db_version, version, website,
- parser_options, has_cryptography, calculate_client_info,
- is_latest_version)
- from pwman.ui.tools import CLICallback
- from pwman.data import factory
- from pwman.exchange.importer import Importer
- from pwman.util.crypto_engine import CryptoEngine
- class PwmanCli(cmd.Cmd, BaseCommands):
- """
- Inherit from the BaseCommands and Aliases
- """
- undoc_header = "Aliases:"
- def __init__(self, db, hasxsel, callback, config_parser, **kwargs):
- """
- initialize CLI interface, set up the DB
- connecion, see if we have xsel ...
- """
- super(PwmanCli, self).__init__(**kwargs)
- self.intro = "%s %s (c) visit: %s" % ('pwman3', version, website)
- self._historyfile = config_parser.get_value("Readline", "history")
- self.hasxsel = hasxsel
- self.config = config_parser
- try:
- enc = CryptoEngine.get()
- enc.callback = callback()
- self._db = db
-
- self._db.open()
- except Exception as e:
- self.error(e)
- sys.exit(1)
- try:
- readline.read_history_file(self._historyfile)
- except IOError as e:
- pass
- self.prompt = "pwman> "
- def get_ui_platform(platform):
- if 'darwin' in platform:
- from pwman.ui.mac import PwmanCliMac as PwmanCli
- OSX = True
- elif 'win' in platform:
- from pwman.ui.win import PwmanCliWin as PwmanCli
- OSX = False
- else:
- from pwman.ui.cli import PwmanCli
- OSX = False
- return PwmanCli, OSX
- def main():
- args = parser_options().parse_args()
- PwmanCli, OSX = get_ui_platform(sys.platform)
- xselpath, dbtype, config = get_conf_options(args, OSX)
- dburi = config.get_value('Database', 'dburi')
- if config.get_value('Updater',
- 'supress_version_check').lower() != 'yes':
- client_info = config.get_value('Updater', 'client_info')
- if not client_info:
- client_info = calculate_client_info()
- config.set_value('Updater', 'client_info', client_info)
- _, latest = is_latest_version(version, client_info)
- if not latest:
- print("A newer version of Pwman3 was release, you should consider updating")
- if not has_cryptography:
- import colorama
- if config.get_value('Crypto', 'supress_warning').lower() != 'yes':
- print("{}WARNING: You are not using PyCrypto!!!\n"
- "WARNING: You should install PyCrypto for better security and "
- "perfomance\nWARNING: You can supress this warning by editing "
- "pwman config file.{}".format(colorama.Fore.RED,
- colorama.Style.RESET_ALL))
- print(dburi)
- dbver = get_db_version(config, args)
- CryptoEngine.get()
- db = factory.createdb(dburi, dbver)
- if args.file_delim:
- importer = Importer((args, config, db))
- importer.run()
- sys.exit(0)
- cli = PwmanCli(db, xselpath, CLICallback, config)
- try:
- cli.cmdloop()
- except KeyboardInterrupt as e:
- print(e)
- finally:
- config.save()
|