pyHeatpump/pyheatpump/db.py

#!/usr/bin/env python3
import sqlite3
from subprocess import Popen
from .config import config
import sys
from pprint import pprint

from pyheatpump.logger import logger_init
logger = logger_init()

conn = None

class DB():
    Conn = None

    @staticmethod
    def connect():
        if DB.Conn:
            try:
                DB.Conn.cursor()
                return True
            except sqlite3.ProgrammingError:
                pass
            except Exception as e:
                print(e)
                sys.exit(1)

        logger.info('Will connect to database {}'.format(
            config['heatpump']['database']))
        DB.Conn = sqlite3.connect(
            config['heatpump']['database'],
            isolation_level=None)
        DB.Conn.row_factory = sqlite3.Row
        return True


    def commit():
        """
            Default mode of connecting is autocommit
        """
        pass


    @staticmethod
    def initialize(filename):
        if DB.Conn:
            DB.Conn.close()

        p = Popen(
            '/usr/bin/env sqlite3 -init {} {}'.format(filename, config['heatpump']['database']),
            shell=True)

        return True if p.wait() == 0 else False


    @staticmethod
    def sql(query, params={}):
        logger.debug(query, params)

        if not DB.connect():
            raise Exception('Can\'t connect to DB')

        return DB.Conn.execute(query, params or {})

    @staticmethod
    def sqlf(filename):
        with open(filename) as f:
            if not DB.connect():
                raise Exception('Can\'t connect to DB')

            try:
                return DB.Conn.execute((' '.join(f.readlines())).replace('\n', ''))
            except StopIteration:
                print('failed')
                return []
            except sqlite3.IntegrityError as e:
                print(e)
                return []
            except sqlite3.OperationalError as e:
                raise Exception('Can\'t find DB file %s'.format(filename)) from e



class RowClass(object):
    def __init__(self, **kwargs):
        for key in kwargs.keys():
            if hasattr(self, key):
                setattr(self, key, kwargs[key])

    def select(self, key, tablename):
        attr = getattr(self, key)
        if type(attr) == str:
            q = ("SELECT * FROM {tablename} WHERE ? LIKE '?'", (key, attr))
        elif type(attr) == int:
            q = ("SELECT * FROM {tablename} WHERE ? LIKE ?", (key, attr))

        res = []
        for row in DB.sql(*q):
            res.append(res)

        return res