def get_labels(file, format='dict'): if format=='dict': labels = dict() with open(file, 'r') as f: reader = csv.reader(f, dialect='excel') for row in reader: labels[row[0]] = row[1] elif format == 'array': labels = list() with open(file, 'r') as f: reader = csv.reader(f, dialect='excel') for row in reader: labels.append(row) else: print('Wrong Format', 'Choose between: array and dict') labels = None return labels
def bars_connection(): db = pymysql.connect( 'jake-thomas-database.cb00wwcerjsx.us-east-2.rds.amazonaws.com', 'StinkyboyNtoad', 'pleaseTakeMySocks', 'CS336Project', 3306) cursor = db.cursor() cursor.execute("DROP TABLE IF EXISTS bars") sql = "CREATE TABLE bars(name VARCHAR(100), location VARCHAR(50), opening_hours VARCHAR(10), closing_hours VARCHAR(10))" cursor.execute(sql) add_new_row = "INSERT INTO bills (ID, total_price, tip) VALUES (%s, %s, %s, %s)" with open('Bars.csv', 'r') as csvFile: csv_data = csv.reader(csvFile, delimiter=",") firstLine = True for row in csv_data: if firstLine: firstLine = False continue val = (row[0], row[1], row[2], row[3]) cursor.execute(add_new_row, val) db.commit() cursor.close() print("Load bars Complete")
def billed_to_connection(): db = pymysql.connect( 'jake-thomas-database.cb00wwcerjsx.us-east-2.rds.amazonaws.com', 'StinkyboyNtoad', 'pleaseTakeMySocks', 'CS336Project', 3306) cursor = db.cursor() cursor.execute("DROP TABLE IF EXISTS billed_to") sql = "CREATE TABLE billed_to(bill_ID int, drinker VARCHAR(50))" cursor.execute(sql) add_new_row = "INSERT INTO billed_to (bill_ID, drinker) VALUES (%s, %s)" with open('Billed_to.csv', 'r') as csvFile: csv_data = csv.reader(csvFile, delimiter=",") firstLine = True for row in csv_data: if firstLine: firstLine = False continue val = (row[0], row[1]) cursor.execute(add_new_row, val) db.commit() cursor.close() print("Load billed_to Complete")
def sells_connection(): db = pymysql.connect( 'jake-thomas-database.cb00wwcerjsx.us-east-2.rds.amazonaws.com', 'StinkyboyNtoad', 'pleaseTakeMySocks', 'CS336Project', 3306) cursor = db.cursor() cursor.execute("DROP TABLE IF EXISTS sells") sql = "CREATE TABLE sells(bar VARCHAR(50), item VARCHAR(50), price int)" cursor.execute(sql) add_new_row = "INSERT INTO sells (bar, item, price) VALUES (%s, %s, %s)" with open('Sells.csv', 'r') as csvFile: csv_data = csv.reader(csvFile, delimiter=",") firstLine = True for row in csv_data: if firstLine: firstLine = False continue val = (row[0], row[1], row[2]) cursor.execute(add_new_row, val) db.commit() cursor.close() print("Load sells Complete")