def editServicePrompt(name=""):
'''
Edit a given service. Old data will be kept if it is not changed.
'''
name = name if name else getUserInput("Current service name to edit: ")
if not checkIfServiceExists(name):
print("Service does not exist.")
return False
# store the old stuff
oldName = getServiceData(name, 'service')
oldUserName = getServiceData(name, 'serviceUserName')
oldUrl = getServiceData(name, 'serviceUrl')
oldPassword = getServiceData(name, 'servicePassword')
# get the new stuff
newName = getEditData(oldName, "service name")
newUserName = getEditData(oldUserName, "service username")
newUrl = getEditData(oldUrl, "service url")
newPassword = getUserInput("Change password? (y/N)")
if newPassword == 'y':
newPassword = getUserInput("Password [leave blank to generate]: ",
True)
if newPassword == "":
newPassword = generatePasswordPrompt()
else:
newPassword = oldPassword
result = updateService(oldName, newName, newPassword, newUrl, newUserName)
return result
def getEditData(oldData, dataDescription):
'''
Used in the editServicePrompt. Shows old data and prompts for
confirmation to change it.
'''
print("Current {}: {}".format(dataDescription, oldData))
newData = getUserInput("Change {}? (y/N)".format(dataDescription))
if newData == 'y':
newData = getUserInput("Enter the new {}".format(dataDescription))
else:
newData = oldData # new is always better
return newData
def generatePasswordPrompt():
'''
Generates a random password to the users specifications
'''
lc = getUserInput("Include lowercase? (y/n/E)")
uc = getUserInput("Incldue Uppercase? (y/n/E)")
dig = getUserInput("Include numbers? (y/n/E)")
punc = getUserInput("Include special characters? (y/n/E)")
spc = getUserInput("Include spaces? (y/n/E)")
if lc == 'n' and uc == 'n' and dig == 'n' and punc == 'n' and spc == 'n':
# gotta have something...
print("No character set chosen")
return ""
siz = getUserInput("Password length (default: 30)")
while not siz.isdecimal() or siz == 0:
print("Not a number")
siz = getUserInput("Password length (default: 30)")
size = int(siz) if siz else 30
while size < 5:
print("Minimum length is 5")
siz = getUserInput("Password length (default: 30)")
# Any character combination that isn't 'y' or 'n' counts as E
charlist = string.ascii_lowercase if not lc == 'n' else ''
charlist += string.ascii_uppercase if not uc == 'n' else ''
charlist += string.digits if not dig == 'n' else ''
charlist += string.punctuation if not punc == 'n' else ''
charlist += ' ' if not spc == 'n' else ''
matched = False
while not matched:
# Find your next date on match.com
password = ''
for _ in range(size):
password += ''.join(random.SystemRandom().choice(charlist))
matched = True
if matched and lc == 'y' and not any(char.islower()
for char in password):
matched = False
if matched and (uc == 'y') and not any(char.isupper()
for char in password):
matched = False
if matched and (dig == 'y') and not any(char.isdigit()
for char in password):
matched = False
if matched and (punc == 'y') and not any(char in string.punctuation
for char in password):
matched = False
if matched and (spc == 'y') and not any(char == ' '
for char in password):
matched = False
return password
def handleOfflineMenu():
'''
Only read-related commands are available in offline mode
'''
prompt = "What do you want to do?\n\n" \
+ "(1) List all services\n" \
+ "(2) Get password for a service\n" \
+ "(3) Get username for a service\n" \
+ "(4) Get URL for a service\n\n"
option = getUserInput(prompt)
if option == "1":
listServicesOffline()
handleOfflineMenu()
elif option == "2":
getPasswordOffline()
handleOfflineMenu()
elif option == "3":
getUserNameOffline()
handleOfflineMenu()
elif option == "4":
getURLOffline()
handleOfflineMenu()
else:
print("Didn't get that...\n")
handleOfflineMenu()
def showMenu():
'''
Show the main menu for command-line usage
'''
prompt = "What do you want to do?\n\n" \
+ "(1) List all services\n" \
+ "(2) Add a new service\n" \
+ "(3) Remove an existing service\n" \
+ "(4) Edit an existing service\n" \
+ "(5) Get password for a service\n" \
+ "(6) Get username for a service\n" \
+ "(7) Get URL for a service\n" \
+ "(8) Change master password\n\n"
option = getUserInput(prompt)
if option == "1":
listServicesPrompt()
elif option == "2":
addServicePrompt()
elif option == "3":
removeServicePrompt()
elif option == "4":
editServicePrompt()
elif option == "5":
getPassPrompt()
elif option == "6":
getNamePrompt()
elif option == "7":
getUrlPrompt()
elif option == "8":
changeMasterPrompt()
else:
print("Didn't get that...\n")
showMenu()
def signUpUser():
'''
Handles sign up for new users
'''
if not checkConnection("test"):
print("Sorry - cannot create user without internet connection")
quit()
username = getUserInput("Please enter your username")
pw = getUserInput("Please enter your password", True)
if addUser(username, pw):
setDBUsername(pw, username)
pullDatabase()
return True
else:
print("Sorry, that username is already taken")
signUpUser()
def getOfflinePassword(data):
'''
Checks a password against that stored in the local database
'''
pw = getUserInput("Please enter your password", True)
key = hashlib.sha256(pw.encode()).digest()
hashedpw = hashlib.sha512(pw.encode('utf-8')).hexdigest()
inc = 0
while hashedpw != data['password'] and inc < 2:
print("Wrong password")
pw = getUserInput("Please enter your password", True)
key = hashlib.sha256(pw.encode()).digest()
hashedpw = hashlib.sha512(pw.encode('utf-8')).hexdigest()
inc += 1
if inc >= 2: # three strikes; you're out
quit()
return key
def getUrlPrompt(sname=""):
'''
Copy a service's URL to the clipboard
'''
if sname == "":
inc = 0
sname = getUserInput("Enter service name: ")
while (not checkIfServiceExists(sname)) and inc < 2:
print("Service not found.")
sname = getUserInput("Enter service name: ")
if inc >= 2: # three strikes; you're out
print("Returning to menu")
showMenu()
else:
if not checkIfServiceExists(sname):
print("Service not found.")
return False
clipboard(getServiceData(sname, 'serviceUrl'), True, True)
return True
def getOfflineUsername(username=""):
'''
Checks if a user has a local database saved. Reprompts for new
username if none is found.
'''
file_path = os.path.expanduser("~/.passman/{}.json".format(username))
while not os.path.isfile(file_path):
print("Sorry, that doesn't match our records")
username = getUserInput("Please enter your username")
file_path = os.path.expanduser("~/.passman/{}.json".format(username))
return username
def getUserNameOffline(sname=""):
'''
Similar to menu.py's get username function, but pulls from local
JSON file instead
'''
sname = sname if sname else getUserInput("Enter service name: ")
inc = 0
while (not getServiceDataOffline(sname)) and inc < 2:
print("Service not found.")
sname = getUserInput("Enter service name: ")
inc += 1
if inc >= 2: # three strikes; you're out
print("Returning to menu")
handleOfflineMenu()
username = getServiceDataOffline(sname)['serviceUserName']
clipboard(username, False, True)
return True
def addServicePrompt(name="", usname="", url=""):
'''
Add a service. Takes optional arguments to avoid prompts for
username, url, etc
'''
name = name if name else getUserInput("Service name: ")
if checkIfServiceExists(name):
print("Service already exists.")
# get out
return False
usname = usname if usname else getUserInput("Service username: "******"Password [leave blank to generate]: ", True)
while password == "":
password = generatePasswordPrompt()
url = url if url else getUserInput("Service URL: ")
result = addService(name, password, url, usname)
return result
def changeMasterPrompt():
'''
Menu for changing the master password to the passman application
'''
oldPass = getUserInput("Enter your old password", True)
isUser = checkUserCredentials(oldPass) # check the name/pw
inc = 0
while (not isUser) and inc < 2:
print("That didn't work")
oldPass = getUserInput("Enter your old password", True)
isUser = checkUserCredentials(oldPass)
inc += 1
if (inc >= 2): # three strikes; you're out
quit()
newPass = getUserInput("Enter your new password", True)
newPassCheck = getUserInput("Once more, with feeling", True)
while newPass != newPassCheck:
print("Sorry. Those didn't match.")
newPass = getUserInput("Enter your new password", True)
newPassCheck = getUserInput("Once more, with feeling", True)
changePassword(newPass)
return True
def removeServicePrompt(sname=""):
'''
Remove a service. Takes optional arguments from one-off command
line arguments to avoid prompts for service name
'''
sname = sname if sname else getUserInput("Enter service to be deleted: ")
if not checkIfServiceExists(sname):
print("Service not found.")
return False
print("Delete ", sname, ". This cannot be undone.", sep="")
confirm = getUserInput("Are you sure? (y/N)")
if confirm == "y" or confirm == "Y": # capitalize me, cap'n
success = removeService(sname)
if success:
print(sname, "Successfully deleted.")
else:
print("Remove failed, unknown error occured.") #spookyerrors
else:
print("Something terrible happened. Aborting")
success = False
return success
def loginUser(username=""):
'''
Handles login for online database
'''
isCommandLine = username
username = username if username else getUserInput("Please enter your username")
if not checkConnection("test"):
handleOfflineLogin(username)
pw = getUserInput("Please enter your password", True)
inc = 0
while not checkUserCredentials(pw, username) and inc < 2:
print("Sorry, that doesn't match our records")
pw = getUserInput("Please enter your password", True)
inc += 1
if inc >= 2: # three strikes; you're out
quit()
setDBUsername(pw, username)
pullDatabase()
return True
def handleLogin():
'''
Handles main menu login/signup functionality
'''
prompt = "Do you want to log in or start a new account?\n"\
+ "(Enter the number of your choice)\n\n" \
+ "(1) Log In\n"\
+ "(2) Start New Account\n"
option = getUserInput(prompt)
if option == "1":
loginUser()
elif option == "2":
signUpUser()
else:
print("Please enter a valid option\n")
handleLogin() # back to the beginning