Use full paths to access .config files for tko/db.py

Signed-off-by: Martin J. Bligh <mbligh@google.com>



git-svn-id: svn://test.kernel.org/autotest/trunk@742 592f7852-d20e-0410-864c-8624ca9c26a4

1 files changed, 8 insertions, 4 deletions

diff --git a/tko/db.py b/tko/db.py
index ed6350ed..4a35482b 100644
--- a/tko/db.py
+++ b/tko/db.py
@@ -3,9 +3,11 @@ import MySQLdb, re, os, sys
 class db:
 	def __init__(self, debug = False):
 		self.debug = debug
-			
+
+		path = os.path.dirname(os.path.abspath(sys.argv[0]))
 		try:
-			db_prefs = open('.database', 'r')
+			file = os.path.join(path, '.database')
+			db_prefs = open(path, 'r')
 			host = db_prefs.readline().rstrip()
 			database = db_prefs.readline().rstrip()
 		except:
@@ -13,12 +15,14 @@ class db:
 			database = 'tko'
 	
 		try:
-			login = open('.priv_login', 'r')
+			file = os.path.join(path, '.priv_login')
+			login = open(file, 'r')
 			user = login.readline().rstrip()
 			password = login.readline().rstrip()
 		except:	
 			try:
-				login = open('.unpriv_login', 'r')
+				file = os.path.join(path, '.unpriv_login')
+				login = open(path, 'r')
 				user = login.readline().rstrip()
 				password = login.readline().rstrip()
 			except: