blob: d18346ebbdd9c0b3eba2a13a19fd8f741eea235d (
plain)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
import sys
import psycopg2
def setDatabase(dbhost_, dbport_, db_):
global dbhost, dbport, db
dbhost = dbhost_
dbport = dbport_
db = db_
assert "dbhost" in globals()
assert "dbport" in globals()
assert "db" in globals()
def databaseHost():
global dbhost
assert "dbhost" in globals()
return dbhost
def databasePort():
global dbport
assert "dbport" in globals()
return dbport
def database():
global db
assert "db" in globals()
return db
# Connects to the database and creates a global cursor:
def connectDatabase():
assert "dbhost" in globals()
assert "dbport" in globals()
assert "db" in globals()
default_dbhost = "bmc.test.qt.nokia.com" # Default unless overridden
default_dbport = 5432 # Default unless overridden
assert db != None # Needs to be explicitly specified for now
try:
conn = psycopg2.connect(
host=(dbhost if dbhost else default_dbhost),
port=(dbport if dbport else default_dbport),
database=db,
user="bmuser") # Hardcoded for now
except:
print "failed to connect to the database:", sys.exc_info()
sys.exit(1)
global cursor
cursor = conn.cursor()
# Executes a query against the database. Returns any result set iff
# fetch_results is true.
def execQuery(query, fetch_results = True):
if not "cursor" in globals():
connectDatabase()
assert "cursor" in globals()
try:
cursor.execute(query)
if fetch_results:
return cursor.fetchall()
except psycopg2.Error:
print "query failed: >" + query + "<"
print "reason:", str(sys.exc_info())
sys.exit(1)
# Commits everything that has been written to the database.
def commit():
if not "cursor" in globals():
return
cursor.connection.commit()
|