Mitja HORVAT
44a0466553
Change the usage header from _Passgeny - Password Generator_ to _Passgeny - Not a Passowrd Manager_
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import argparse
|
|
import getpass
|
|
import os
|
|
import sys
|
|
|
|
from passgeny import passgeny
|
|
|
|
pargs = argparse.ArgumentParser("Passgeny - Not a Passowrd Manager")
|
|
pargs.add_argument("domain", help="Domain or unique site identifier")
|
|
pargs.add_argument("user", help="Username or unique user identifier")
|
|
pargs.add_argument("tokens", nargs='*', help="Additional tokens")
|
|
pargs.add_argument("--verbose", "-v", action='store_true', help="Verbose")
|
|
pargs.add_argument("--pattern", "-p", help="Set pattern")
|
|
|
|
pargs_pass = pargs.add_mutually_exclusive_group()
|
|
pargs_pass.add_argument("--stdin", "-s", action='store_true', help="Read password from stdin")
|
|
pargs_pass.add_argument("--env", "-e", help="Read password from environment")
|
|
|
|
popt = pargs.parse_args()
|
|
|
|
# Read master password
|
|
if popt.env:
|
|
mpw = os.getenv(popt.env)
|
|
if mpw is None:
|
|
raise Exception("Environment {} not defined.".format(popt.env))
|
|
elif popt.stdin:
|
|
mpw = sys.stdin.readline().rstrip('\n\r')
|
|
else:
|
|
mpw = getpass.getpass("Master password: ")
|
|
|
|
pg = passgeny.Passgeny(mpw)
|
|
del mpw
|
|
|
|
if popt.pattern:
|
|
pg.set_pattern(popt.pattern)
|
|
|
|
print(pg.generate(popt.domain, popt.user, *popt.tokens))
|