-
Notifications
You must be signed in to change notification settings - Fork 0
/
DB.py
52 lines (42 loc) · 1.18 KB
/
DB.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
39
40
41
42
43
44
45
46
47
48
49
50
51
#!/usr/bin/env python
import os
import sys
import psycopg2
# tables for current season
starts = 'starts2024'
bat = 'bat2024'
pit = 'pit2024'
teams = 'teams2024'
sched = 'sched2024'
extra = 'extra2024'
inj = 'inj2024'
usage = 'usage2024'
def bin_dir():
return os.path.dirname( os.path.realpath(__file__) )
def connect():
name='ibl_stats'
user='ibl'
try:
import DBpasswd
host = 'host=' + DBpasswd.host
pwd = 'password=' + DBpasswd.pwd
except ImportError:
host=''
pwd=''
if 'IBL_DB' in list(os.environ.keys()):
name = os.environ.get('IBL_DB')
if 'IBL_DB_USER' in list(os.environ.keys()):
user = os.environ.get('IBL_DB_USER')
if 'IBL_DB_HOST' in list(os.environ.keys()):
host = 'host=' + os.environ.get('IBL_DB_HOST')
if 'IBL_DB_PWD' in list(os.environ.keys()):
pwd = 'password=' + os.environ.get('IBL_DB_PWD')
connstr = "dbname=%s user=%s %s %s" %\
( name, user, host, pwd )
try:
db = psycopg2.connect(connstr)
db.set_client_encoding("utf-8")
return db
except psycopg2.DatabaseError as err:
print(str(err))
sys.exit(1)