Password Manager V.1 - Printable Version

Code:
import os, platform, base64, sqlite3, random, getpass
from simplepam import authenticate

class passmanager:

	def auth(username, password): #Unix Authentication using PAM

		if authenticate(username, password) == True:
			return 'pass'
		else:
			return 'fail'

	def encode(text): #Function to encrypt credentials details

		try:
			encoded = base64.b64encode(text.encode('utf8'))
			return encoded

		except TypeError:
			return "Wrong User Input"

	def decode(text): #Decodes the encrypted text after fetching credentials from database

		try:
			decoded = base64.b64decode(text)
			return decoded

		except TypeError:
			passmanager.clear()
			return "Wrong User Input"

	def checkdatabase(): #Function to  check whether the database exists or not

		try:
			with open('db.sql'): pass

		except IOError:
			passmanager.createdatabase()

	def createdatabase(): #Create the Database

		con = sqlite3.connect("db.sql")
		cur = con.cursor()
		cur.execute(""" CREATE TABLE passdb (username, password, description, dork BOOL) """)
		cur.close()
		cur.close()

	def insert(username, password, description): #Insert Credentials into the database

		try:
			rand_id = random.randint(0, 1000)
			if username == '': username = 'Unknown-%d' % rand_id
			if password == '': password = 'Unknown-%d' % rand_id
			if description == '': description = 'Unknown-%d' % rand_id
			con = sqlite3.connect('db.sql')
			cur = con.cursor()

			cur.execute(""" INSERT INTO passdb (username, password, description) VALUES (?,?,?) """, (passmanager.encode(username), passmanager.encode(password), passmanager.encode(description)))

			con.commit()
			cur.close()
			cur.close()

		except sqlite3.OperationalError:

			print('Database error, Make sure your database exists!')
			input('Press any key to Continue...')

		passmanager.clear()

	def viewdb(): #View the whole credential Database

		try:
			string = ['Username:', 'Password:', 'Description:']
			con = sqlite3.connect('db.sql')
			cur = con.cursor()

			cur.execute(""" SELECT * FROM passdb """)
			getdb = cur.fetchall()

			if len(getdb) == 0:

				print(' ')
				print('No Record Found!')
				print(' ')

			for items in getdb:

				print(' ')
				print("####################################")
				for i in range(len(string)):

					dec = passmanager.decode(items[i]).decode('utf8')
					print(string[i], dec)

				print("####################################")

			input('Press any key to continue')

		except sqlite3.OperationalError:

			print('Database error, Make sure your database exists!')
			input('Press any key to Continue...')



	def search(user): #Function to search for the credentials using the username

		try:

			string = ['Username:', 'Password:', 'Description:']
			con = sqlite3.connect('db.sql')
			cur = con.cursor()
			da = passmanager.encode(user)

			cur.execute("""SELECT * FROM passdb WHERE username LIKE ?""", (da,))

			find = cur.fetchall()

			if len(find) == 0:
				print(' ')
				print('No data Found')
				print(' ')

			con.commit()

			cur.close()

			cur.close()

			for terms in find:

				print(' ')
				print("####################################")
				for i in range(len(string)):

					dec = passmanager.decode(terms[i]).decode('utf8')
					print(string[i], dec)

				print("####################################")

			input('Press any key to continue...')

		except sqlite3.OperationalError:

			print('Database error, Make sure your database exists!')
			input('Press any key to Continue...')

			passmanager.clear()


	def delete(user): #Function to delete user from database

		da = passmanager.encode(user)
		con = sqlite3.connect('db.sql')
		cur = con.cursor()

		cur.execute("""DELETE FROM passdb WHERE username LIKE ?""", (da,))

		con.commit()
		cur.close()
		cur.close()

		input('Credential deleted!, Press any key to continue!')


	def clear(): #Clear Screen

		os.system('clear')

	def menu(): #The main menu

		passmanager.checkdatabase()

		username = input('Enter your UNIX Username: ')
		password = getpass.getpass()

		if passmanager.auth(username, password) == 'pass':

			passmanager.clear()

			while True:

				print("""
         #############################################################
         #                                                           #
         #                   Password Manager V.1                    #
         #     A small python database program to organize your      #
         #                       passwords                           #
         #                                                           #
         ############################################################# 
         # By Ex094 Of Hackcommunity #
         # http://ex094.blogspot.com # Type Help to view commands
         #############################""")

				cmd = input('>>> ')

				if cmd == 'insert':

					passmanager.clear()

					username = input('Your username: ').strip()
					password = input('Your password: ').strip()
					description = input('Credential Description: ').strip()

					passmanager.insert(username, password, description)

				if cmd == 'search':

					passmanager.clear()

					user = input('Type the username to search in the database:')

					passmanager.search(user)

				if cmd == 'delete':

					passmanager.clear()

					user = input('Input your Credential Username: ').strip()

					passmanager.delete(user)

				if cmd == 'view':

					passmanager.clear()

					passmanager.viewdb()

				if cmd == 'help':

					passmanager.clear()

					print("""
Password Manager
Version 1

Available Commands:
-------------------

Command    -     Function

insert          Insert credentials into the database

view            View the credential database

delete          Delete a credential based on username

search          Search for a credential detail using Username

exit            Quit program

Coded By Ex094
http://ex094.bogspot.com
http://www.hackcommunity.com
""")
					input('Press any key to continue...')


				if cmd == 'exit':

					break

				else:

					passmanager.clear()
		else:

			print('Sorry, Wrong Credentials')

			os.system('exit')

passmanager.menu()