def connect():
""" Connect to MySQL database """
try:
conn = mysql.connector.connect(host='localhost',
database='mydb',
user='root',
password='')
if conn.is_connected():
print('connection established.')
else:
print('connection failed.')
return conn
except Error as e:
print(e)
finally:
conn.close()