#!/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