From 97e4a4933b63e6e6947585b657a5ad9ecc089afe Mon Sep 17 00:00:00 2001 From: Holger Levsen Date: Tue, 22 Jan 2013 18:46:45 +0100 Subject: [PATCH] use cidb only when using master branch, else use cidb_dev --- cidb.py | 27 ++++++++++++++++++++++----- 1 file changed, 22 insertions(+), 5 deletions(-) diff --git a/cidb.py b/cidb.py index 2238713..108f1aa 100644 --- a/cidb.py +++ b/cidb.py @@ -1,17 +1,34 @@ #!/usr/bin/python +import subprocess import psycopg2 import psycopg2.extras +import logging -DB_USER = "cidb" -DB_PASSWD = "cidb" -DB_NAME = "cidb" -DB_HOST = "jenkins" -DB_PORT = 5432 +# get current git branch +p = subprocess.Popen(["git", "rev-parse", "--symbolic-full-name", + "--abbrev-ref", "HEAD"], stdout=subprocess.PIPE) +p.wait() +branch = p.stdout.read().rstrip() +if branch == "master": + DB_USER = "cidb" + DB_PASSWD = "cidb" + DB_NAME = "cidb" + DB_HOST = "jenkins" + DB_PORT = 5432 +else: + DB_USER = "cidb_dev" + DB_PASSWD = "cidb_dev" + DB_NAME = "cidb_dev" + DB_HOST = "jenkins" + DB_PORT = 5432 + +logger = logging.getLogger() def db_connect(): """ Connects to the db and returns the Connection object. """ + logger.info("Connecting to database %s" % DB_NAME) return psycopg2.connect(database=DB_NAME, user=DB_USER, password=DB_PASSWD, host=DB_HOST, port=DB_PORT) -- 2.39.5