def query(args):
path = os.path.realpath(os.path.expanduser(args.filepath))
if not os.path.exists(path):
args.parser.error("FILE {} DOES NOT EXIST".format(path))
password = util.get_master_password(args)
database = None
try:
database = passdb.PassDB.read_file(path, password)
filters = []
if hasattr(args, "hostname") and args.hostname:
filters.append(("hostname", args.hostname))
if hasattr(args, "username") and args.username:
filters.append(("username", args.username))
results = database.search(filters)
util.print_dictrows(
results,
(
''
'Username',
'Hostname',
'Date Modifed'
)
)
except ValueError:
args.parser.error("INCORRECT PASSWORD")
return
def write_entry(args, update_only:bool = True):
path = os.path.realpath(os.path.expanduser(args.filepath))
if not os.path.exists(path):
args.parser.error("FILE {} DOES NOT EXIST".format(path))
# print(args)
password = util.get_master_password(args)
database = None
try:
database = passdb.PassDB.read_file(path, password)
index, entry = database.get_entry(args.username, args.hostname)
if update_only:
if index is None:
args.parser.error("Cannot update: account does not exist")
else:
database.set_entry(args.username, args.hostname, util.get_user_password(args, account=args.username))
else:
if index is None:
database.set_entry(args.username, args.hostname, util.get_user_password(args, account=args.username))
else:
args.parser.error("Cannot define: account already exists")
database.save(password)
except ValueError:
args.parser.error("Incorrect Password")
finally:
print("Database saved at {}".format(database.path))
def save(args, database):
if args.filepath:
password = util.get_master_password(args, verify=True)
path = os.path.realpath(os.path.expanduser(args.filepath))
if (not os.path.exists(path)) or util.confirm_overwrite_file(path):
database.save_as(path, password)
elif database.path:
password = util.get_master_password(args)
if args.sess_salt and hashlib.sha256(
(password +
args.sess_salt).encode('utf-8')).digest() == args.pass_hash:
database.save(password)
print('Database saved at %s' % database.path)
else:
args.parser.error(
"\nPassword Mismatch: Please enter the same password used to open the database, or specify a file path."
)
else:
args.parser.error(
"\n Error: Database path unknown. Cannot save without file path. (-d/--dest)"
)
def init(args):
# check file path
path = os.path.realpath(os.path.expanduser(args.filepath))
if os.path.exists(path):
if args.force :
if not util.confirm_overwrite_file(path):
args.parser.error("ACTION ABORTED: INIT AT {}".format(path))
else:
args.parser.error("FILE {} ALREADY EXISTS".format(path))
# get user password
password = util.get_master_password(args, verify=True)
# create a new database and save it to the path, encrypting with password
passdb.PassDB().save_as(path, password)
print("Database initialized at %s" % path)
def close(args, database):
pyperclip.copy("")
if args.save:
password = util.get_master_password(args)
if args.sess_salt and hashlib.sha256(
(password +
args.sess_salt).encode('utf-8')).digest() == args.pass_hash:
database.save(password)
print('Database saved at %s' % database.path)
sys.exit()
else:
args.parser.error(
"\nPassword Mismatch: Please enter the same password used to open the database."
)
else:
sys.exit()
def read(args):
path = os.path.realpath(os.path.expanduser(args.filepath))
if not os.path.exists(path):
args.parser.error("Error: File {} does not exist.".format(path))
password = util.get_master_password(args)
database = None
try:
database = passdb.PassDB.read_file(path, password)
if args.index:
if args:
pass
index, entry = database.get_entry(args.username, args.hostname)
password = database.get_password((index, entry))
util.print_pass_entry(index, entry)
pyperclip.copy('')
except ValueError:
args.parser.error("Incorrect Password")
def session(args):
# print(args)
database = None
if args.filepath is not None:
path = os.path.realpath(os.path.expanduser(args.filepath))
if not os.path.exists(path):
args.parser.error("FILE {} DOES NOT EXIST".format(path))
password=util.get_master_password(args)
session_salt = base64.b64encode(
Random.new().read(64)
).decode("utf-8")
args.pass_hash = hashlib.sha256((password + session_salt).encode('utf-8')).digest()
args.sess_salt = session_salt
try:
database = passdb.PassDB.read_file(path, password)
except ValueError:
print("pypassman: error: INCORRECT PASSWORD")
exit(1)
else:
database = passdb.PassDB()
userSession.run(database, args)