Script to update the values inside the database

sql/update_sg.sql

@@ -0,0 +1,3 @@
+UPDATE customers
+SET CITY = 'St. Gallen'
+WHERE CITY LIKE "St.%gallen";

start.py

@@ -23,6 +23,11 @@ def parse_args():
             default='table',
             help='prints all the cells from the table',
             )
+    parser.add_argument(
+            '-u', '--update',
+            action='store_true',
+            help='Updates the city names'
+            )
 
     return parser.parse_args()
 
@@ -39,6 +44,11 @@ def execute_sql(args):
         print("insert")
         with open ('sql/insert_data.sql', 'r') as sql_file:
                 cur.executescript(sql_file.read())
+    elif args.update:
+        print("Updating database")
+        with open ('sql/update_sg.sql', 'r') as sql_file:
+                cur.executescript(sql_file.read())
+
     elif args.print:
         cur.execute("SELECT * from customers")
         if args.print == 'table':