HYTOS

# coding: utf-8

""" This is AppDatabase module """

import sqlite3

import pymssql

class AppDatabase:

    """ This is AppDatabase class """

    def __init__(self, type, host, user, password, db_path):

        self._DBType = type

        self._host = host

        self._user = user

        self._password = password

        self._db_path = db_path 

    @property

    def db_type(self):

        """ return database type """

        return self._DBType

    @db_type.setter

    def db_type(self, value):

        self._DBType = value

    @property

    def host(self):

        """ return host for mssql """

        return self._host

    @host.setter

    def host(self, value):

        self._host = value

    @property

    def user(self):

        """ return user for mssql """

        return self._user

    @user.setter

    def user(self, value):

        self._user = value

    @property

    def password(self):

        """ return password for mssql """

        return self._password

    @password.setter

    def password(self, value):

        self._password = value

    @property

    def db_name(self):

        """ return database name """

        return self._db_path

    @property

    def file_path(self):

        """ return sqlite database file path """

        return self._db_path

    @file_path.setter

    def file_path(self, value):

        self._db_path = value 

    @property

    def place_holder(self):

        """ return database placeholder """

        return '?' if self.db_type == 'SQLite' else '%s' if self.db_type == 'MSSQL' else None

    def connect(self):

        """ return database connection depends on database type """

        conn = None

        if self._DBType == 'SQLite':

            conn = sqlite3.connect(self.file_path, isolation_level=None)

            conn.row_factory = sqlite3.Row

            #cursor = conn.cursor()

        elif self._DBType == 'MSSQL':

            conn = pymssql.connect(host=self._host, user=self._user, password=self._password, database=self.db_name, charset='utf8', autocommit=False)

            #cursor = conn.cursor(as_dic=True)

        return conn

    def to_sql(self, sql):

        """ convert given sql string for database """

        return sql.replace('?', '%s') if self.db_type == "MSSQL" else sql