The code below works as is$.$ Your job is to expand the code and add some new features. You will also expand the given test code. There are five specific tasks for you to complete. Each task is shown in a comment in the code. After you have completed all the tasks, submit the finished, improved code as a $.$py file to this Canvas Dropbox.

import sqlite$3$

import csv

conn $=$ sqlite$3.$connect$("$test$.$db$")$

print$("$

Database connected."$)$

##TASK $1$: Add one more column to the table. Be sure to adjust all queries to reflect this extra column!

#CREATE A TABLE

conn $=$ sqlite$3.$connect$("$test$.$db$")$

conn.execute$(\text{'}\text{'}\text{'}$DROP TABLE IF EXISTS CONTACTS'$\text{'}\text{'})$

conn.execute$(\text{'}\text{'}\text{'}$CREATE TABLE CONTACTS

$($ID INT PRIMARY KEY NOT NULL,

NAME TEXT NOT NULL,

AGE INT NOT NULL,

ADDRESS CHAR$(50))$;$\text{'}\text{'}\text{'})$

print$("$Table created."$)$

conn.close$()$

##TASK $2$: Create an additional test record.

# ADD RECORDS

conn $=$ sqlite$3.$connect$("$test$.$db$")$

conn.execute$("$INSERT OR IGNORE INTO CONTACTS $($ID$,$NAME, AGE,ADDRESS$)\backslash$

VALUES $(10,$ 'Polly', $32,$ 'Seattle'$)")$;

conn.execute$("$INSERT or IGNORE INTO CONTACTS $($ID$,$NAME, AGE,ADDRESS$)\backslash$

VALUES $(11,$ 'Roger', $28,$ 'Bainbridge'$)")$;

conn.commit$()$

print$("$Records created.

$")$

conn.close$()$

#RETRIEVE RECORDS

conn $=$ sqlite$3.$connect$("$test$.$db$")$

cursor $=$ conn.execute$("$SELECT $*$ from CONTACTS"$)$

print$("$RECORDS:$")$

for row in cursor:

print $("$ID $="+$ str$($row$[0]))$

print $("$NAME $="+$ row$[1])$

print$("$AGE $="+$ str$($row$[2]))$

print $("$ADDRESS $="+$ str$($row$[3])+"$

$")$

conn.close$()$

##TASK $3$: Generalize the query so the user can input a column name.

#USER QUERY FROM CLI

query $=$ input$("$Enter a query term. Must be the ID$.")$

#column $=$ input $($TODO$)$

conn $=$ sqlite$3.$connect$("$test$.$db$")$

#Task $4-$ Place the query below into a "try: except:" block. The exception should say 'Query not valid'.

cursor $=$ conn.execute$("$SELECT $*$ from CONTACTS where ID $="+"\text{'}"+$ query $+"\text{'}")$

print$("$

SELECTED RECORD:"$)$

for row in cursor:

print $("$ID $="+$ str$($row$[0]))$

print $("$NAME $="+$ row$[1])$

print$("$AGE $="+$ str$($row$[2]))$

print $("$ADDRESS $="+$ str$($row$[3])+"$

$")$

conn.close$()$

#SAVE RECORDS TO A LIST

conn $=$ sqlite$3.$connect$(\text{'}$test$.$db$\text{'})$

cursor $=$ conn.execute$("$SELECT $*$ from CONTACTS"$)$

data$\_$list $=[]$

for row in cursor:

data$\_$list.append$($row$)$

conn.close$()$

print$("$RECORDS LIST:"$)$

print$($data$\_$list$)$

#Task $5-$ Let the user enter the name of the CSV file for data export.

#Include the selected file name in the final print statement.

#SAVE RECORDS TO A CSV FILE

with open$(\text{'}$contact$\_$file.csv$\text{'},$ mode$=\text{'}$w$\text{'})$ as contact$\_$file:

contact$\_$writer $=$ csv$.$writer$($contact$\_$file, delimiter$=\text{'},\text{'},$ quotechar$=\text{'}"\text{'},$ quoting$=$csv$.$QUOTE$\_$MINIMAL$)$

for row in data$\_$list:

contact$\_$writer.writerow$($row$)$

#Print statement should reflect the user$-$selected file name.

print$("$

CSV EXPORT COMPLETE"$)$