fixed major bugs, added password for employees

Author: OJASisLive

check.py

@@ -1,6 +1,11 @@
 def check():
-    with open("firsttime.txt","r") as a:
-        if a.read().strip()=="True":
-            return True
-        else:
-            return False
+    try:
+        with open("firsttime.txt","r") as a:
+            if a.read().strip()=="True":
+                return True
+            else:
+                return False
+    except FileNotFoundError:
+        with open("firsttime.txt","w") as a:
+            a.write("True")
+            return True

editemployee.py

@@ -52,6 +52,7 @@ def ap3():
         print("4.last-name:",results1[3])
         print("5.gender:",results1[4])
         print("6.hire_date:",results1[5])
+        print("7.password")
         birth_date=results1[1]
         hire_date=results1[5]
         f2()
@@ -268,9 +269,43 @@ def f2():
                 traceback.print_exc()
             else:
                 if age(birth_date)-age(hire_date)>=20:
-                    break
+                    try:
+                        cur.execute("update employees set hire_date='{}' where emp_no={}".format(hire_date,emp_no))
+                        conn.commit()
+                    except mysql.connector.Error as err:
+                        print(err.msg)
+                        print("-----------Value addition was unsuccessful!!!!-------------")
+                        break
+                    else:
+                        print("Updated hire date...")
+                        break
                 else:
                     print("Employee must atleast be 20 years of age when hired!!")
-
+    if a=='7':
+        print("1.Show the password")
+        print("2.Change the password")
+        ans=input("Enter your choice (1,2):")
+        if ans=='1':
+            cur.execute("SELECT pass from empass where emp_no={}".format(emp_no))
+            result=cur.fetchall()
+            print(result[0][0], "is the password.")
+        elif ans=='2':
+            while True:
+                password=input("Enter employee login password(max 8 characters, min 4): ")
+                lp=len(password)
+                if lp>8:
+                    print("Max 8 characters only.")
+                elif lp<4:
+                    print("Minimum 4 characters to be entered.")
+                else:
+                    try:
+                        cur.execute("UPDATE empass set pass=LPAD({},{},'0') where emp_no={}".format(password,lp,emp_no))
+                        conn.commit()
+                    except mysql.connector.Error as err:
+                        print(err.msg)
+                        print("-----------Password change was unsuccessful!!!!-------------")
+                    else:
+                        print("Password changed successfully!!!")
+                        break
     cur.close()
     conn.close()

fireemployee.py

@@ -22,7 +22,7 @@ def ap2():
         else:
             print("Maximum length is 5!")
 
-    query="delete from employees where emp_no = {}".format(emp_no)
+    query="delete from employees,empass where emp_no = {}".format(emp_no)
     cur.execute("select emp_no from employees")
     record=cur.fetchall()
     changed=False
@@ -34,7 +34,7 @@ def ap2():
                 changed=True
             except mysql.connector.Error as err:
                 print(err.msg)
-                print("-----------Value addition was unsuccessful!!!!-------------\n")
+                print("-----------Value deletion was unsuccessful!!!!-------------\n")
             else:
                 print("Employee fired successfully...\n")
     if not changed:

hireemployee.py

@@ -181,5 +181,21 @@ def ap1():
         print("-----------Value addition was unsuccessful!!!!-------------")
     else:
         print("Values added successfully!!")
+        while True:
+            password=input("Enter employee login password(max 8 characters, min 4): ")
+            if len(password)>8:
+                print("Max 8 characters only.")
+            elif len(password)<4:
+                print("Minimum 4 characters to be entered.")
+            else:
+                try:
+                    cur.execute("INSERT INTO empass values({},LPAD({},{},'0')".format(emp_no,password,len(password)))
+                    query.commit()
+                except mysql.connector.Error as err:
+                    print(err.msg)
+                    print("-----------Password addition was unsuccessful!!!!-------------")
+                else:
+                    print("Password added successfully!!!")
+                    break
     cur.close()
     query.close()

main.py

@@ -1,6 +1,9 @@
 import setup
 import check
 import accounttype
-conn=setup.setup()
-if not check.check():
-    accounttype.acctype()
+while True:
+    if not check.check():
+        accounttype.acctype()
+        break
+    else:
+        setup.setup()

setup.py

@@ -5,6 +5,10 @@
 
 from mysql.connector import errorcode
 existing=0
+
+conn=None
+cursor=None
+
 TABLES = {}
 TABLES['employees'] = (
     "CREATE TABLE `employees` ("
@@ -27,10 +31,21 @@
     "  `birth_date` date NOT NULL,"
     "  `acc_creation_date` date NOT NULL,"
     "  `mobile_no` int(10) NOT NULL,"
-    "  `email_id` varchar(25) NOT NULL"
+    "  `email_id` varchar(25) NOT NULL,"
+    "  `pass` varchar(8) NOT NULL"
+    ") "
+)
+
+TABLES['empass'] = (
+    "CREATE TABLE `empass` ("
+    "  `emp_no` int(5) NOT NULL,"
+    "  `pass` varchar(8) NOT NULL,"
+    "  PRIMARY KEY (`emp_no`),"
+    "  FOREIGN KEY(`emp_no`) REFERENCES employees(emp_no)"
     ") "
 )
 
+
 TABLES['savings'] = (
     "CREATE TABLE `savings` ("
     "  `acc_no` int(5) NOT NULL,"
@@ -108,6 +123,8 @@ def connectionquery():
     return query
 
 def querycheck():
+    global conn
+    global cursor
     global existing
     conn=connectionquery()
     ans=False
@@ -130,8 +147,6 @@ def querycheck():
                             print(err.msg())
                     else:
                         print("OK")
-                        with open("firsttime.txt","w") as f:
-                            f.write("False")
             ans=True
 
     if not ans:
@@ -152,6 +167,8 @@ def mysqlsetup():
     querycheck()
 
 def setup():
+    global cursor
+    global conn
     global existing
     while check.check():
         print("\n\n-----------------Welcome to the Project!!!-------------------")
@@ -165,7 +182,7 @@ def setup():
             break
         if ans2=="1":
             mysqlsetup()
-            if existing==6:
+            if existing==7:
                 with open("firsttime.txt","w") as f:
                     f.write("False")
                 continue
@@ -175,4 +192,4 @@ def setup():
             print("\nWrong input, (1/2).........")
     else: 
         if querycheck():
-            connectionquery()
+            return True