from decimal import Decimal

from masters_registry.constant import field_const


def ownership_info_conflicts_enabled():
    return [
        {
            field_const.ISRC: 'USA371690671',
            field_const.LOCKED_TERRITORIES: {},
            field_const.TERRITORIES: {
                'QA': [
                    {field_const.TUID: Decimal('321')},
                    {field_const.TUID: Decimal('789')}
                ],
                'CA': {'tuid': Decimal('12345')},
            },
            'timestamp': Decimal('12'),
        },
        {
            field_const.ISRC: 'USJTQ1109709',
            field_const.LOCKED_TERRITORIES: {},
            field_const.TERRITORIES: {
                'US': [{field_const.TUID: 789}, {field_const.TUID: 321}],
            },
        },
    ]


def audit_table_conflicts_enabled():
    return {
        'USA371690671': [
            {
                'opcode': 'ADD',
                'isrc': 'USA371690671',
                'territories': {'CA': Decimal('11'), 'US': Decimal('11')},
                'user': '1234',
                'correlation_id': '123',
            },
            {
                'opcode': 'CONFLICT_CREATED',
                'isrc': 'USA371690671',
                'source': 'BULK_UPDATE_UPC',
                'territories': {'US': Decimal('11'), 'CA': Decimal('11')},
                'conflict': {
                    'conflicting_tuid': [Decimal('12345'), Decimal('11')],
                    'status': 'CONFLICT_CREATED'
                },
                'correlation_id': '123',
                'user': '1234'}
        ],
        'USA371690672': [
            {
                'opcode': 'ADD',
                'isrc': 'USA371690672',
                'territories': {
                    'CA': Decimal('12345'),
                    'US': Decimal('12345')
                },
                'user': '1234',
                'correlation_id': '123',
            }
        ],
        'USJTQ1109709': [
            {
                'opcode': 'REMOVE',
                'isrc': 'USJTQ1109709',
                'territories': {'US': Decimal('789')},
                'user': '1234',
                'correlation_id': '123'
            },
            {
                'opcode': 'CONFLICT_RESOLVED',
                'isrc': 'USJTQ1109709',
                'territories': {'US': Decimal('789')},
                'source': 'BULK_UPDATE_UPC',
                'conflict': {
                    'resolved': Decimal('1'),
                    'status': 'CONFLICT_RESOLVED',
                    'conflicting_tuid': []
                },
                'correlation_id': '123',
                'user': '1234',
             }
        ],
        'USJTQ1109710': [
            {
                'opcode': 'REMOVE',
                'isrc': 'USJTQ1109710',
                'territories': ['CA'],
                'user': '1234',
                'correlation_id': '123'
            }
        ]
    }


def active_table_conflicts_enabled():
    return {
        'USA371690671': {
            'isrc': 'USA371690671',
            'locked_territories': {},
            'territories': {
                'CA': [{'tuid': Decimal('12345')}, {'tuid': Decimal('11')}],
                'QA': [{field_const.TUID: 321}, {field_const.TUID: 789}],
                'US': [{'tuid': Decimal('11')}],
            }
        },
        'USA371690672': {
            'isrc': 'USA371690672',
            'locked_territories': {},
            'territories': {
                'US': [{'tuid': Decimal('12345')}],
                'CA': [{'tuid': Decimal('12345')}]
            },
        },
        'USJTQ1109709': {
            'isrc': 'USJTQ1109709',
            'territories': {'US': [{'tuid': Decimal('321')}]},
            'locked_territories': {}
        },
        'USJTQ1109710': {},
    }


def task_result_conflicts_enabled():
    return [
        {
            'upc': '767767806329',
            'error_count': 0,
            'warning_count': 0,
            'status_report': [
                {
                    'success': True,
                    'warning': True,
                    'resolved_conflict': True,
                    'removed_territories': ['US'],
                    'isrc': 'USJTQ1109709',
                    'tuid': 789,
                    'vendor_id': 9,
                    'upc': '767767806329',
                    'error_message': ''
                },
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'removed_territories': [],
                    'isrc': 'USJTQ1109710',
                    'tuid': 7891,
                    'vendor_id': 9,
                    'upc': '767767806329',
                    'error_message': ''
                }
            ],
            'success_count': 2
        },
        {
            'upc': '886788771130',
            'error_count': 0,
            'warning_count': 1,
            'status_report': [
                {
                    'success': True,
                    'warning': True,
                    'resolved_conflict': False,
                    'isrc': 'USA371690671',
                    'tuid': 11,
                    'vendor_id': 9,
                    'upc': '886788771130',
                    'locked_territories': None,
                    'claimed_by_another_owner': [('CA', [Decimal('12345')])],
                    'claimed_by_the_same_label': [],
                    'substore_carveouts': [],
                    'error_message': ''
                },
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'isrc': 'USA371690672',
                    'tuid': 12345,
                    'vendor_id': 9,
                    'upc': '886788771130',
                    'locked_territories': None,
                    'claimed_by_another_owner': [],
                    'claimed_by_the_same_label': [],
                    'substore_carveouts': [('453', set())],
                    'error_message': ''}
            ],
            'success_count': 2
        }
    ]


def ownership_info_conflicts_disabled():
    return [
        {
            field_const.ISRC: 'USA371690671',
            field_const.LOCKED_TERRITORIES: {},
            field_const.TERRITORIES: {
                'QA': [{field_const.TUID: 321}],
            },
            'timestamp': Decimal('12'),
        },
        {
            field_const.ISRC: 'USJTQ1109709',
            field_const.LOCKED_TERRITORIES: {},
            field_const.TERRITORIES: {
                'US': {field_const.TUID: 789},
            },
        },
    ]


def active_table_conflicts_disabled():
    return {
        'USA371690671': {
            'isrc': 'USA371690671',
            'locked_territories': {},
            'territories': {
                'CA': {'tuid': Decimal('11')},
                'QA': [{'tuid': Decimal('321')}],
                'US': {'tuid': Decimal('11')}},
        },
        'USA371690672': {
            'isrc': 'USA371690672',
            'locked_territories': {},
            'territories': {
                'US': {'tuid': Decimal('12345')},
                'CA': {'tuid': Decimal('12345')}
            },
        },
        'USJTQ1109709': {
            'isrc': 'USJTQ1109709',
            'territories': {},
            'locked_territories': {}
        },
        'USJTQ1109710': {},
    }


def audit_table_conflicts_disabled():
    return {
        'USA371690671': [
            {
                'opcode': 'ADD',
                'isrc': 'USA371690671',
                'territories': {'CA': Decimal('11'), 'US': Decimal('11')},
                'user': '1234',
                'correlation_id': '123'
            }
        ],
        'USA371690672': [
            {
                'opcode': 'ADD',
                'isrc': 'USA371690672',
                'territories': {
                    'CA': Decimal('12345'),
                    'US': Decimal('12345')
                },
                'user': '1234',
                'correlation_id': '123'
            }
        ],
        'USJTQ1109709': [
            {
                'opcode': 'REMOVE',
                'isrc': 'USJTQ1109709',
                'territories': ['US'],
                'user': '1234',
                'correlation_id': '123'
            }
        ],
        'USJTQ1109710': [
            {
                'opcode': 'REMOVE',
                'isrc': 'USJTQ1109710',
                'territories': ['CA'],
                'user': '1234',
                'correlation_id': '123'
            }
        ]
    }


def task_result_conflicts_disabled():
    return [
        {
            'upc': '767767806329',
            'error_count': 0,
            'warning_count': 0,
            'status_report': [
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'removed_territories': ['US'],
                    'isrc': 'USJTQ1109709',
                    'tuid': 789,
                    'vendor_id': 9,
                    'upc': '767767806329',
                    'error_message': ''
                },
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'removed_territories': [],
                    'isrc': 'USJTQ1109710',
                    'tuid': 7891,
                    'vendor_id': 9,
                    'upc': '767767806329',
                    'error_message': ''
                }
            ],
            'success_count': 2
        },
        {
            'upc': '886788771130',
            'error_count': 0,
            'warning_count': 0,
            'status_report': [
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'isrc': 'USA371690671',
                    'tuid': 11,
                    'vendor_id': 9,
                    'upc': '886788771130',
                    'locked_territories': None,
                    'claimed_by_another_owner': [],
                    'claimed_by_the_same_label': [],
                    'substore_carveouts': [],
                    'error_message': ''
                },
                {
                    'success': True,
                    'warning': False,
                    'resolved_conflict': False,
                    'isrc': 'USA371690672',
                    'tuid': 12345,
                    'vendor_id': 9,
                    'upc': '886788771130',
                    'locked_territories': None,
                    'claimed_by_another_owner': [],
                    'claimed_by_the_same_label': [],
                    'substore_carveouts': [('453', set())],
                    'error_message': ''}
            ],
            'success_count': 2
        }
    ]