From: Holger Levsen <holger@layer-acht.org>
Date: Tue, 22 Jan 2013 17:46:45 +0000 (+0100)
Subject: use cidb only when using master branch, else use cidb_dev
X-Git-Url: https://git.uhu-banane.org/?a=commitdiff_plain;h=97e4a4933b63e6e6947585b657a5ad9ecc089afe;p=profitbricks%2Fjenkins-build-scripts.git

use cidb only when using master branch, else use cidb_dev
---

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)