Added creation_date,portal_url and email_id in savepass

paint-it · Jan 13, 2020 · f2c86a8 · f2c86a8
1 parent dba29cf
commit f2c86a8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/genpass/database.py b/genpass/database.py
@@ -23,7 +23,7 @@
 import sqlite3  # library for database
 
 
-class DatabaseConnection(object):
+class DatabaseConnection():
     """ Class of database entries for user's information."""
 
     def __init__(self):
@@ -44,10 +44,10 @@ def create_table(self):
     def insert_data(self, portal_name, password, creation_date, email, portal_url):
         """Adding values into database"""
         try:
-            self.portal_name = portal_name
             self.password = password
             self.creation_date = creation_date
             self.email = email
+            self.portal_name = portal_name
             self.portal_url = portal_url
             self.cursor_obj.execute(
                 """INSERT INTO passwords

diff --git a/genpass/password.py b/genpass/password.py
@@ -52,8 +52,11 @@ def savepass():
     """Used to take portal name and password from user"""
     portal_name = click.prompt("Enter portal name", default="None")
     pwd = click.prompt("Enter your password", default="None", hide_input=True)
+    creation_date = date.today()
+    email = click.prompt("Enter email id", default="None")
+    portal_url = click.prompt("Enter portal url", default="None")
     db_obj.create_table()
-    db_obj.insert_data(portal_name=portal_name, password=pwd)
+    db_obj.insert_data(portal_name=portal_name, password=pwd, creation_date=creation_date, email=email, portal_url=portal_url)
 
 
 @click.command(help="Create new password")