|
@@ -0,0 +1,131 @@
|
|
|
|
+#!/usr/bin/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) 2006 Ivan Kelly <ivan@ivankelly.net>
|
|
|
|
+#============================================================================
|
|
|
|
+from pwman.util.crypto import CryptoEngine
|
|
|
|
+import getopt
|
|
|
|
+import sys
|
|
|
|
+if 'darwin' in sys.platform:
|
|
|
|
+ from pwman.ui.cli import PwmanCliMac
|
|
|
|
+else:
|
|
|
|
+ from pwman.ui.cli import PwmanCli
|
|
|
|
+
|
|
|
|
+import pwman.util.config as config
|
|
|
|
+import pwman.data.factory
|
|
|
|
+import os
|
|
|
|
+import os.path
|
|
|
|
+import subprocess as sp
|
|
|
|
+_saveconfig = True
|
|
|
|
+
|
|
|
|
+
|
|
|
|
+def print_help():
|
|
|
|
+ print """Syntax: %s [options]
|
|
|
|
+
|
|
|
|
+ -c, --config FILE Read configuration from FILE
|
|
|
|
+ -d, --database FILE Use FILE as database
|
|
|
|
+ -e, --encryption ALGO Use ALGO to encrypt data
|
|
|
|
+ Possible options are:
|
|
|
|
+ AES, ARC2, ARC4, Blowfish(default),
|
|
|
|
+ CAST, DES, DES3, IDEA, RC5
|
|
|
|
+ -h, --help Display this help and exit
|
|
|
|
+
|
|
|
|
+Please report bugs at https://github.com/pwman3/pwman3
|
|
|
|
+""" % (sys.argv[0])
|
|
|
|
+
|
|
|
|
+
|
|
|
|
+def which(cmd):
|
|
|
|
+ _, cmdname = os.path.split(cmd)
|
|
|
|
+
|
|
|
|
+ for path in os.environ["PATH"].split(os.pathsep):
|
|
|
|
+ cmd = os.path.join(path, cmdname)
|
|
|
|
+ if os.path.isfile(cmd) and os.access(cmd, os.X_OK):
|
|
|
|
+ return cmd
|
|
|
|
+
|
|
|
|
+ return None
|
|
|
|
+
|
|
|
|
+try:
|
|
|
|
+ config_dir = os.path.expanduser("~/.pwman")
|
|
|
|
+ if not os.path.isdir(config_dir):
|
|
|
|
+ os.mkdir(config_dir)
|
|
|
|
+
|
|
|
|
+ config_file = os.path.join(config_dir, "config")
|
|
|
|
+
|
|
|
|
+ default_config = {'Global': {'umask': '0100', 'colors': 'yes'},
|
|
|
|
+ 'Database': {'type': 'SQLite',
|
|
|
|
+ 'filename': os.path.join(config_dir,
|
|
|
|
+ "pwman.db")},
|
|
|
|
+ 'Encryption': {'algorithm': 'Blowfish'},
|
|
|
|
+ 'Readline': {'history': os.path.join(config_dir,
|
|
|
|
+ "history")}
|
|
|
|
+ }
|
|
|
|
+ config.set_defaults(default_config)
|
|
|
|
+
|
|
|
|
+ opts, args = getopt.getopt(sys.argv[1:], 'c:d:e:h',
|
|
|
|
+ ["config=", "database=",
|
|
|
|
+ "encryption=", "help"])
|
|
|
|
+
|
|
|
|
+ for o in opts:
|
|
|
|
+ if o[0] == '-c' or o[0] == "--config":
|
|
|
|
+ config_file = os.path.expanduser(o[1])
|
|
|
|
+
|
|
|
|
+ if o[0] == '-h' or o[0] == "--help":
|
|
|
|
+ print_help()
|
|
|
|
+ sys.exit(0)
|
|
|
|
+ # if no config exists yet, look for xsel
|
|
|
|
+ if os.path.exists(config_file):
|
|
|
|
+ config.load(config_file)
|
|
|
|
+ xselpath = config.get_value("Global", "xselpath")
|
|
|
|
+ else:
|
|
|
|
+ xselpath = which("xsel")
|
|
|
|
+ config.set_value("Global", "xsel", xselpath)
|
|
|
|
+
|
|
|
|
+ for o in opts:
|
|
|
|
+ if o[0] == '-d' or o[0] == "--database":
|
|
|
|
+ config.set_value("Database", "filename",
|
|
|
|
+ os.path.expanduser(o[1]))
|
|
|
|
+ _saveconfig = False
|
|
|
|
+ if o[0] == '-e' or o[0] == "--encryption":
|
|
|
|
+ config.set_value("Encryption", "algorithm", o[1])
|
|
|
|
+ _saveconfig = False
|
|
|
|
+ # set umask before creating/opening any files
|
|
|
|
+ umask = int(config.get_value("Global", "umask"))
|
|
|
|
+ os.umask(umask)
|
|
|
|
+
|
|
|
|
+ enc = CryptoEngine.get()
|
|
|
|
+
|
|
|
|
+ dbtype = config.get_value("Database", "type")
|
|
|
|
+ db = pwman.data.factory.create(dbtype)
|
|
|
|
+ cli = PwmanCli(db, xselpath)
|
|
|
|
+except SystemExit, e:
|
|
|
|
+ sys.exit(e)
|
|
|
|
+except Exception, e:
|
|
|
|
+ print "Error: %s" % (e)
|
|
|
|
+ sys.exit(-1)
|
|
|
|
+
|
|
|
|
+try:
|
|
|
|
+ try:
|
|
|
|
+ cli.cmdloop()
|
|
|
|
+ except KeyboardInterrupt, e:
|
|
|
|
+ print
|
|
|
|
+finally:
|
|
|
|
+ try:
|
|
|
|
+ if _saveconfig:
|
|
|
|
+ config.save(config_file)
|
|
|
|
+ except Exception, e:
|
|
|
|
+ print "Error: %s" % (e)
|
|
|
|
+ sys.exit(-1)
|