account type, table setup initialisation

Author: OJASisLive

accounttype.py

@@ -0,0 +1,32 @@
+def acctype():
+    while True:
+        print("--------------Account Selector Menu--------------")
+        print("1.Admin.")
+        print("2.Employee.")
+        print("3.Client.")
+        print("Enter 0 to end process.")
+        a=int(input("\nEnter your account type:"))
+        
+        if a==1:
+            b=input("\nEnter admin password:")
+            if b=="admin123":
+                return 1
+            else:
+                print("\nWrong password!\n") 
+            
+        elif a==2:
+            b=input("\nEnter employee password:")
+            if b=="emp123":
+                return 2
+            else:
+                print("\nWrong password!\n")
+        
+        elif a==3:
+            return 3
+        
+        elif a==0:
+            print("\nShutting down the program.")
+            break
+        
+        else:
+            print("\nWrong input!")

setup.py

@@ -1,4 +1,5 @@
 import check
+import sqltables
 
 import pickle
 import mysql.connector
@@ -42,6 +43,7 @@ def querycheck():
             print("Connection established successfully.")
             with open("firsttime.txt","w") as f:
                 f.write("False")
+                sqltables.tables()
                 ans=True
 
     if not ans:

sqltables.py

@@ -0,0 +1,36 @@
+import setup
+import mysql.connector
+from mysql.connector import errorcode
+conn=setup.setup()
+cursor=conn.cursor()
+
+DB_NAME = setup.sqldb()
+
+TABLES = {}
+TABLES['employees'] = (
+    "CREATE TABLE `employees` ("
+    "  `emp_no` int(5) NOT NULL AUTO_INCREMENT,"
+    "  `birth_date` date NOT NULL,"
+    "  `first_name` varchar(15) NOT NULL,"
+    "  `last_name` varchar(15) NOT NULL,"
+    "  `gender` enum('M','F') NOT NULL,"
+    "  `hire_date` date NOT NULL,"
+    "  `age` int(2) NOT NULL"
+    "  PRIMARY KEY (`emp_no`)"
+    ") ")
+
+TABLES['client'] = (
+    "CREATE TABLE `clients` ("
+    "  `acc_no` int(5) NOT NULL AUTO_INCREMENT,"
+    "  `acc_type` enum('S','C') NOT NULL"
+    "  `first_name` varchar(15) NOT NULL,"
+    "  `last_name` varchar(15) NOT NULL,"
+    "  `gender` enum('M','F') NOT NULL,"
+    "  `birth_date` date NOT NULL,"
+    "  `acc_creation_date` date NOT NULL,"
+    "  PRIMARY KEY (`acc_no`)"
+)
+
+#https://education.github.com/git-cheat-sheet-education.pdf
+#https://dev.mysql.com/doc/connector-python/en/connector-python-example-ddl.html
+