data-post-processing-v2/db_connection.py at main · open-data-rescue/data-post-processing-v2 · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# -*- coding: utf-8 -*-
import pymysql
import sqlalchemy
from sqlalchemy import URL
import os


def connect():

    ## ECCC server database
    # dbHost = os.getenv("ECCC_db_host")
    # dbPort = int(os.getenv("ECCC_db_port"))
    # dbUser = os.getenv("ECCC_db_user")
    # dbPasswd = os.getenv("ECCC_db_passwd")
    # dbDB = os.getenv("ECCC_db_database")

    # localhost
    # dbHost = os.getenv("localhost")
    # dbPort = 3306
    # dbUser = os.getenv("db_root")
    # dbPasswd = os.getenv("db_password_root")
    # dbDB = "Canada_wx"

    #dbHost = "54.39.21.6"
    #dbPort = 3306
    #dbUser = "mysql"
    #dbPasswd = "3589a8dea043af14"
    #dbDB = "eccc_db"

    dbHost =  os.getenv("localhost")
    dbPort = 3306
    dbUser =  os.getenv("db_root")
    dbPasswd = os.getenv("db_password_root")
    dbDB = "DRAW"

    mydb = pymysql.connect(
        host=dbHost,
        port=dbPort,
        user=dbUser,
        password=dbPasswd,
        database=dbDB
        )

    url = URL.create(
        "mysql+mysqlconnector",
        username=dbUser,
        host=dbHost,
        port=dbPort,
        password=dbPasswd,
        database=dbDB
        )

    engine = sqlalchemy.create_engine(url)

    return (mydb, mydb.cursor(), engine)