# -*- coding: utf_8 -*-
# 表名: businesses
# 作者: 陈磊
# 时间: 2019-11-25


from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, VARCHAR, SMALLINT, DATETIME, DECIMAL, BigInteger, FLOAT, NUMERIC, TEXT
from sqlalchemy.orm.exc import MultipleResultsFound, NoResultFound
from sqlalchemy import or_

from WorkUtils.UtilsLog import UtilsLog
from WorkUtils.UtilsDataBase import UtilsDataBase

import time


Base = declarative_base()


class HostTypes(Base):
    """
    设备类型
    """
    __tablename__ = "host_types"

    id = Column(BigInteger, primary_key=True)
    title = Column(VARCHAR(255), comment="标题")
    parent_id = Column(BigInteger, comment="父级ID")
    template_ids = Column(VARCHAR(255), comment="模板ID集合")
    interfaces = Column(VARCHAR(255), comment="取数方式集合")
    description = Column(VARCHAR(255), comment="描述")
    sortid = Column(Integer, comment="排序")
    create_time = Column(BigInteger, comment="创建时间")


class DataHostTypes(object):
    def __init__(self):
        self.log = UtilsLog()
        self.log.info(self.__class__)

        self.table = HostTypes

    def select_all_from_allKeys(self, session, _id=None, title=None, parent_id=None):
        """
        :param session: 指针
        :param _id:
        :param title:
        :param parent_id:
        :return: 查询结果
        """
        self.log.debug("查询数据库:")
        try:
            base = UtilsDataBase()
            base.add_param(_key="id", value=_id)
            base.add_param(_key="title", value=title)
            base.add_param(_key="parent_id", value=parent_id)
            sql_rep = session.query(self.table).filter_by(**base.param).all()
            self.log.debug(sql_rep)
            session.close()
            return sql_rep
        except UtilsDataBase().errors as error:
            self.log.error("异常:")
            self.log.error(error)
            session.close()
            return error

    def delete_like_title(self, session, title):
        """
        :param session: 指针
        :param title:
        :return:
        """
        self.log.debug("查询数据库:")
        try:
            self.log.debug("删除相关测试数据")
            session.query(self.table).filter(self.table.title.like(str(title) + "%")).delete(synchronize_session=False)
            # session.delete(sql)
            session.commit()
            session.close()
            self.log.debug("删除成功")
            return True
        except UtilsDataBase().errors as error:
            self.log.error("异常:")
            self.log.error(error)
            session.close()
            return error
