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")
