import mysql.connector as mysql
from base import Base

class Mysql(Base):
    def __init__(self, database, user, password, host, port):
        if port is None:
            port = 3306
        try:
            self.conn = mysql.connect(database=database, user=user, password=password, host=host, port=port)
        except mysql.Error as e:
            raise Exception(e)