MediaMath/t1-python

# -*- coding: utf-8 -*-
"""Provides strategy targeting segment object."""

from __future__ import absolute_import
from .. import t1types
from ..entity import Entity
from ..errors import ClientError


class StrategyTargetingSegment(Entity):
    """Object for strategy contextual targeting provided by third-party vendors."""
    collection = 'strategy_targeting_segments'
    resource = 'strategy_targeting_segment'
    _relations = {
        'targeting_segment',
        'strategy',
    }
    _pull = {
        'created_on': t1types.strpt,
        'group_identifier': None,
        'id': int,
        'operator': None,
        'restriction': None,
        'strategy_id': int,
        'targeting_segment_id': int,
        'type': None,
        'updated_on': t1types.strpt,
        'user_cpm': float,
        'version': int,
    }

    def __init__(self, session, properties=None, **kwargs):
        super(StrategyTargetingSegment, self).__init__(session, properties, **kwargs)

    def save(self, *args, **kwargs):
        raise ClientError('This object is not editable.')

    def remove(self):
        """Unassign the strategy targeting segment from the strategy."""
        url = '/'.join(['strategies',
                        str(self.parent_id),
                        'targeting_segments'])
        data = {
            'segments.1.id': str(-1 * self.targeting_segment_id),
            'segments.1.restriction': self.restriction,
            'segments.1.group_identifier': self.group_identifier,
            'exclude_op': 'OR',
            'include_op': 'OR',
        }
        self._post(self._get_service_path(), rest=url, data=data)
        for item in list(self._properties.keys()):
            del self._properties[item]