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()