9th day of python challenges 139-142

challenges139-145/challenge-139.py

@@ -1,12 +1,54 @@
 import sqlite3
+from sqlite3 import Error
 
-with sqlite3.connect('python-by-example') as db:
-    cursor = db.cursor()
-    cursor.execute("""CREATE TABLE IF NOT EXISTS employees(
-                                                    id integer PRIMARY KEY,
-                                                    name text NOT NULL,
-                                                    dept text NOT NULL,
-                                                    salary integer);""")
-    cursor.execute("""INSERT INTO employees(id,name,dept,salary)
-                        VALUES("1","Bob","Sales","25000")""")
-    db.commit()
+
+def create_connection(db):
+    try:
+        conn = sqlite3.connect(db)
+        return conn
+    except Error as e:
+        print(e)
+    return None
+
+
+def create_phone_book_table(conn):
+    sql_create_projects_table = """ CREATE TABLE IF NOT EXISTS PhoneBook (
+                                        id integer PRIMARY KEY,
+                                        first_name text NOT NULL,
+                                        surname text,
+                                        phone integer
+                                    ); """
+    cur = conn.cursor()
+    cur.execute(sql_create_projects_table)
+    cur.close()
+
+
+def create_phone_book_record(conn, phone_book):
+    sql = ''' INSERT INTO PhoneBook(first_name,surname,phone)
+              VALUES(?,?,?) '''
+    cur = conn.cursor()
+    cur.execute(sql, phone_book)
+    cur.close()
+    return cur.lastrowid
+
+
+def main():
+    conn = create_connection('python-by-example.db')
+
+    with conn:
+        create_phone_book_table(conn)
+
+        phone_book_record_1 = ('Simon',  'Howels',  '01223349752')
+        phone_book_record_2 = ('Karen',  'Philips', '01954295773')
+        phone_book_record_3 = ('Darren', 'Smith',   '01583749012')
+        phone_book_record_4 = ('Anne',   'Jones',   '01323567322')
+        phone_book_record_5 = ('Mark',   'Smith',   '01223855534')
+
+        create_phone_book_record(conn, phone_book_record_1)
+        create_phone_book_record(conn, phone_book_record_2)
+        create_phone_book_record(conn, phone_book_record_3)
+        create_phone_book_record(conn, phone_book_record_4)
+        create_phone_book_record(conn, phone_book_record_5)
+
+
+main()