9th day of python challenges 139-142

challenges139-145/challenge-140.py

@@ -0,0 +1,84 @@
+import sqlite3
+from sqlite3 import Error
+
+
+def create_connection(db):
+    try:
+        conn = sqlite3.connect(db)
+        return conn
+    except Error as e:
+        print(e)
+    return None
+
+
+def show_all(conn):
+    cur = conn.cursor()
+    cur.execute("SELECT * FROM PhoneBook")
+    rows = cur.fetchall()
+    for row in rows:
+        print(row)
+
+
+def view_phone_book(conn, phone_id):
+    cur = conn.cursor()
+    cur.execute(str("SELECT * FROM PhoneBook WHERE id = " + str(phone_id)))
+    print(cur.fetchone())
+    cur.close()
+
+
+def add_phone_book(conn, phone_book):
+    sql = ''' INSERT INTO PhoneBook(first_name,surname,phone)
+              VALUES(?,?,?) '''
+    cur = conn.cursor()
+    cur.execute(sql, phone_book)
+    print(cur.fetchone())
+    cur.close()
+
+
+def surname_in_phone_book(conn, surname):
+    sql = "SELECT * FROM  PhoneBook WHERE surname like '%" + surname + "%'"
+    cur = conn.cursor()
+    cur.execute(sql)
+    rows = cur.fetchall()
+    for row in rows:
+        print(row)
+
+
+def delete_phone_book(conn, phone_id):
+    cur = conn.cursor()
+    cur.execute(str("DELETE FROM tasks WHERE id = " + str(phone_id)))
+    cur.close()
+
+
+def main():
+
+    conn = create_connection('python-by-example.db')
+
+    with conn:
+        print('\n\nMain Menu\n\n\n')
+        print('1) View phone book ')
+        print('2) Add to phone book')
+        print('3) Search for surname')
+        print('4) Delete person from phone book')
+        print('5) Quit')
+        operation_ask = int(input('Enter your selection : '))
+        if operation_ask == 1:
+            ask_id = int(input('Enter id to show : '))
+            view_phone_book(conn, ask_id)
+        elif operation_ask == 2:
+            ask_name = str(input('Enter name : '))
+            ask_surname = str(input('Enter surname : '))
+            ask_phone = str(input('Enter phone : '))
+            phone_book = (ask_name, ask_surname, ask_phone)
+            add_phone_book(conn, phone_book)
+        elif operation_ask == 3:
+            ask_surname = str(input('Enter surname to show : '))
+            surname_in_phone_book(conn, ask_surname)
+        elif operation_ask == 4:
+            ask_id = int(input('Enter id to delete : '))
+            delete_phone_book(conn, ask_id)
+        else:
+            print('Invalid input, please try again.')
+
+
+main()