lambda-tutorial
lambda-tutorial copied to clipboard
Modifications to initial db connect code
A few minor updates in the Setting Environment variables are needed to get this to run properly (mainly steps for less savvy learners).
- Add
import pymysqlto the import list. Add the correspondingpipcommands as well to install on the environment - It looks like the
dotenvpackage is actuallypython-dotenv - Most of the code to retrieve environment variables could be written as the following:
import os
import logging
import pymysql
from dotenv import load_dotenv
from flask import Flask, Response, json, request
app = Flask(__name__)
# update environment just in case
load_dotenv()
# set globals
RDS_HOST = os.getenv("DB_HOST")
RDS_PORT = int(os.getenv("DB_PORT"))
NAME = os.getenv("DB_USERNAME")
PASSWORD = os.getenv("DB_PASSWORD")
DB_NAME = os.getenv("DB_NAME")
# we need to instantiate the logger
logger = logging.getLogger()
logger.setLevel(logging.INFO)