fabaccess/fab_access/sql_handler.py

import psycopg2
from psycopg2 import sql, extensions
import json

from config import Config

class SQLHandler:
    @staticmethod
    def setup():
        SQLHandler.cursor = None
        SQLHandler.conn = None
        SQLHandler.conn = psycopg2.connect(host=Config.db_host_name, user=Config.db_user_name, port=Config.db_port, password=Config.db_password, dbname=Config.db_database)
        # get the isolation leve for autocommit
        autocommit = extensions.ISOLATION_LEVEL_AUTOCOMMIT
        print ("ISOLATION_LEVEL_AUTOCOMMIT:", extensions.ISOLATION_LEVEL_AUTOCOMMIT)

        # set the isolation level for the connection's cursors
        # will raise ActiveSqlTransaction exception otherwise
        SQLHandler.conn.set_isolation_level( autocommit )
        SQLHandler.cursor = SQLHandler.conn.cursor()

    @staticmethod
    def get_machine_data(reader_id):
            # TODO Database
    # - get machine_id from database
    # - get last_user from database
    # - get machine_status from database, is 0 for off, 1 for on
    # - get plug_id from database
        SQLHandler.cursor.execute("SELECT * FROM readerplug WHERE readerplug.reader_id = %s;", (reader_id,))
        data = [row for row in SQLHandler.cursor.fetchall()]
        if(len(data) > 0):
            return {
                'machine_id': data[0][2],
                'last_user': data[0][4],
                'machine_status': data[0][3],
                'plug_id': data[0][0]
            }
        else:
            print("No maching Card Reader found in db")
            return "Error"

    @staticmethod
    def update_machine(reader_id,last_user,machine_status):
        SQLHandler.cursor.execute("UPDATE readerplug SET machine_status = %s WHERE readerplug.reader_id = %s", (False if machine_status else True,reader_id))
        SQLHandler.cursor.execute("UPDATE readerplug SET last_user      = %s WHERE readerplug.reader_id = %s", (last_user,reader_id))
        SQLHandler.conn.commit()

    @staticmethod
    def init_db():
        SQLHandler.cursor.execute("SELECT datname FROM pg_database;")
        dbs = [row[0] for row in SQLHandler.cursor.fetchall()]
        if not (Config.db_database in dbs):
            print(f"Missing database ({Config.db_database}) -> creating new db")
            SQLHandler.cursor.execute(sql.SQL("CREATE DATABASE {};").format(sql.Identifier( Config.db_database )))
        else:
            print(f"Found DB {Config.db_database} -> Using existing one")

        SQLHandler.cursor.execute("SELECT * FROM pg_catalog.pg_tables\
            WHERE schemaname != 'pg_catalog' AND \
            schemaname != 'information_schema';")
        tables = [row[1] for row in SQLHandler.cursor.fetchall()]
        if not ("readerplug" in tables):
            print("Missing table -> creating new table in db")
            SQLHandler.cursor.execute("\
                CREATE TABLE readerplug (\
                    reader_id        int             NOT NULL, \
                    plug_id          varchar(255)    NOT NULL, \
                    machine_id       varchar(255)    NOT NULL, \
                    machine_status   boolean         NOT NULL, \
                    last_user        varchar(255)    NOT NULL  \
                )")
        else:
            print("Found Table -> Using existing one")
        SQLHandler.conn.commit()

        SQLHandler.cursor.execute("SELECT * FROM readerplug;")
        if(len(SQLHandler.cursor.fetchall()) < 1):
            print("Found no machines in table, adding machines from config")
            machines = json.loads(Config.machines)
            for machine in machines:
                SQLHandler.cursor.execute("INSERT INTO readerplug (reader_id, plug_id, machine_id, machine_status, last_user) VALUES (%s, %s, %s, False, 'no_user');", (machine[0],machine[1],machine[2]))
        SQLHandler.conn.commit()