from utils.db_connectors.base_connector import BaseConnector
from utils.db_objects.art_relations import EncodingOrderDetail, \
    EncodingOrderDetailDms, EncodingOrder


class EncodingOrderConnector(BaseConnector):

    def insert_encoding_order(self, encoding_order):
        session = self.Session()
        session.add(encoding_order)
        session.commit()
        session.refresh(encoding_order)
        eo_id = encoding_order.encoding_order_id
        eo_detail_id = self.insert_encoding_order_detail(eo_id)
        self.insert_encoding_order_detail_dms(eo_detail_id)
        session.close()
        return encoding_order

    def insert_encoding_order_detail(self, encoding_order_id):
        session = self.Session()
        eo_detail = EncodingOrderDetail(encoding_order_id)
        session.add(eo_detail)
        session.commit()
        session.refresh(eo_detail)
        encoding_order_detail_id = eo_detail.encoding_order_detail_id
        session.close()
        return encoding_order_detail_id

    def insert_encoding_order_detail_dms(self, eo_detail_id):
        session = self.Session()
        eo_detail_dms = EncodingOrderDetailDms(eo_detail_id)
        session.add(eo_detail_dms)
        session.commit()
        session.close()

    def update_encoding_order_id(self, eo_id):
        session = self.Session()
        order = session.query(EncodingOrder).filter_by(
            encoding_order_id=eo_id).first()
        if order:
            order.order_status = 'close'
            session.commit()
        session.close()