# -*- coding: utf-8 -*-

from typing import List, Dict, Union

from sqlalchemy import text

from website.db_mysql import get_session, to_json_list, to_json


class EnterpriseNodeAlertRepository(object):

    def get_all(self) -> List[Dict]:
        with get_session() as session:
            sql = text(
                """SELECT * FROM `enterprise_alert` WHERE `enterprise_node_id`=:enterprise_node_id ORDER BY `created_at` DESC;"""
            )
            cursor = session.execute(
                sql, {"enterprise_node_id": self.enterprise_node_id}
            )
            return to_json_list(cursor)

    def get_one(
            self, entity_suid: str, node_id: int
    ) -> Union[Dict, None]:
        with get_session() as session:
            sql = text(
                """SELECT * FROM `enterprise_alert` WHERE `entity_suid`=:entity_suid and `node_id`=:node_id;"""
            )
            cursor = session.execute(
                sql, {"entity_suid": entity_suid, "node_id": node_id}
            )
            return to_json(cursor)

    def update_or_inert(self, data: Dict) -> None:
        with get_session() as session:
            sql = text(
                """
                INSERT INTO `enterprise_alert` (`entity_suid`, `node_id`, `node_suid`, `is_sms`, `sms_to`, `is_email`, `email_to`, `freq`) 
                VALUES (:entity_suid, :node_id, :node_suid, :is_sms, :sms_to, :is_email, :email_to, :freq) 
                ON DUPLICATE KEY UPDATE 
                `node_suid`=:node_suid, `is_sms`=:is_sms, `sms_to`=:sms_to, `is_email`=:is_email, `email_to`=:email_to, `freq`=:freq 
                """
            )
            session.execute(sql, data)
            session.commit()