A5T5SQLite.py

import sqlite3, time

''' 
    A5T5SQLite.py
    Created by Brad Conklin, Mashiad Hasan and Klyde Pausang
    For CMPUT 291 - Assignment 5
    University of Alberta
    Winter 2021
'''

conn = None
c = None

def connect(path):
    '''
    Connect to the sqlite3 database located at path
    '''
    global conn, c
    
    conn = sqlite3.connect(path)
    conn.row_factory = sqlite3.Row
    c = conn.cursor()
    c.execute(' PRAGMA foreign_keys=ON; ')
    conn.commit()
    return

def handle_query(n):
    '''
    run the query 20 times and 
    find the average running time in ms
    '''
    
    start_time = time.time()
    
    for i in range (0,20):
        c.execute('''
        SELECT AVG(price) as avg_price
        FROM Listings
        WHERE neighbourhood = ?
        ''',(n,))

    q = ((time.time() - start_time)*1000)/20
    q = "{:.3f}".format(q)

    print('*'*70)

    '''
    output the task result
    '''
    rows = c.fetchone()
    for i in rows:
        print('Average price for the area is: ', round(rows[0]))

    
    '''
    display the running time
    '''
    print('Average running time for the SQLite Query:',q,'ms')
    
    return

def main():
    
    global conn, c
    n = input("Enter a neighbourhood: ")

    
    path = "./A5.db"
    connect(path)
    
    handle_query(n)
    
    conn.commit()
    conn.close()
    return

if __name__ == "__main__":
    main()