FilesExpand file tree

 main

/

db.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

29 lines (22 loc) · 801 Bytes

 main

/

db.py

Top

File metadata and controls

• Code
• Blame

29 lines (22 loc) · 801 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import logging

import pymysql

import os

import sys

# rds settings

user_name = os.environ['USER_NAME']

password = os.environ['PASSWORD']

rds_proxy_host = os.environ['RDS_PROXY_HOST']

db_name = os.environ['DB_NAME']

logger = logging.getLogger()

logger.setLevel(logging.INFO)

# create the database connection outside of the handler to allow connections to be

# re-used by subsequent function invocations.

def connect():

try:

return pymysql.connect(host=rds_proxy_host, user=user_name, passwd=password, db=db_name, connect_timeout=5)

except pymysql.MySQLError as e:

logger.error("ERROR: Unexpected error: Could not connect to MySQL instance.")

logger.error(e)

sys.exit(1)

conn = connect()

conn.autocommit(True)