iro/database.py
changeset 56 3718abbfffd6
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/iro/database.py	Wed Nov 03 01:19:26 2010 +0100
@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import MySQLdb
+import sqlite3
+class Database(object):
+    def __init__(self,connection):
+        self.conn=None
+        self.cursor=None
+        self.connection=connection
+        self.testConnection()
+
+    def testConnection(self):
+        self.connect()
+        self.disconnect()
+
+    def getConn(self):
+        return self.conn
+
+    def getCursor(self):
+        return self.cursor
+
+    def connect(self):
+        if not self.getConn():
+            if (self.connection['type']=='mysql'):
+                self.conn=MySQLdb.connect(
+                    host   = self.connection["host"],
+                    db     = self.connection["db"],
+                    user   = self.connection["user"],
+                    passwd = self.connection["passwd"],
+                    )
+            if (self.connection['type']=='sqlite'):
+                self.conn=sqlite3.connect(self.connection['path'])
+
+        if not self.getCursor():
+            self.cursor = self.conn.cursor()
+
+    def disconnect(self):
+        if self.getCursor():
+            self.cursor.close()
+        self.cursor=None
+        if self.getConn():
+            self.conn.close()
+        self.conn=None
+
+