-
Notifications
You must be signed in to change notification settings - Fork 0
/
mysql_connector_env.py
38 lines (29 loc) · 1.24 KB
/
mysql_connector_env.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import mysql.connector
# DB_HOST is where the params for pricing are held, DB_HOST_CALIBRATE is where the total db is held
from ao.ao_codes import DB_HOST, DATABASE, DB_USER
class MysqlConnectorEnv:
def __init__( self
, host : str = DB_HOST
, database : str = DATABASE
, user : str = DB_USER
, password : str = None ):
""" Initializes the MySQL connector environment.
:param host: host mysql connection
:param database: database where to connect
:param user: user used for connection
:param password: potential password used.
"""
self.host = host
self.database = database
self.user = user
self.password = password
# cached value
self._connection = None
def __enter__(self):
self._connection = mysql.connector.connect( host = self.host
, database = self.database
, user = self.user
, password = self.password )
return self._connection
def __exit__(self, *args):
self._connection.close()