Uncompyle6 of tunable_handler.py

# uncompyle6 version 2.9.4
# Python bytecode 3.3 (3230)
# Decompiled from: Python 3.3.4 (v3.3.4:7ff62415e426, Feb 10 2014, 18:13:51) [MSC v.1600 64 bit (AMD64)]
# Embedded file name: .\WickedWhims 2.2.3.102.3\wickedwoohoo\exhibitionism\tunable_handler.py
# Compiled at: 2017-04-27 04:10:18
# Size of source mod 2**32: 9956 bytes
import services
from event_testing.test_variants import SimInfoTest, UserRunningInteractionTest
from event_testing.tests import CompoundTestList, TestList
from interactions import ParticipantType
from sims.phone_tuning import PhoneTuning
from sims.sim_info_types import Age
from sims4.collections import make_immutable_slots_class, ListSet
from sims4.localization import TunableLocalizedStringFactory
from sims4.resources import Types, get_resource_key
from sims4.tuning.instance_manager import InstanceManager
from wickedwoohoo.global_manager import register_game_load_listener
from wickedwoohoo.utils_enums import SimInteraction
from wickedwoohoo.utils_injector import injector
from wickedwoohoo.utils_tunings import get_sim_info_age_test, create_impossible_sim_info_age_test
NUDE_CLOTHING_BUFF = 128807
NUDE_TEENS_TEST_SETS = (129064)
NUDITY_AFFORDANCE_LIST = (120455, 120453, 128625, 129163, 117339, 26878)
ALL_SIMS_AFFORDANCE_LIST = (125127, 125129)
DISABLE_AFFORDANCE_LIST = (128922, 129168, 128960, 142210)

@register_game_load_listener()
def _add_test_to_phone_tuning(is_first_load):
    if is_first_load is False:
        return
    affordance_manager = services.get_instance_manager(Types.INTERACTION)
    affordances = {
     affordance_manager.get(get_resource_key(SimInteraction.WW_SEX_INTERACTION, Types.INTERACTION)), affordance_manager.get(get_resource_key(SimInteraction.WW_SETUP_SEX, Types.INTERACTION)),
     affordance_manager.get(get_resource_key(SimInteraction.WW_WAIT_FOR_SEX_PARTNER, Types.INTERACTION)), affordance_manager.get(get_resource_key(SimInteraction.WW_GO_TO_GROUP_SEX_DESTINATION, Types.INTERACTION)),
     affordance_manager.get(get_resource_key(SimInteraction.WW_GO_TO_JOIN_GROUP_SEX_DESTINATION, Types.INTERACTION)), affordance_manager.get(get_resource_key(SimInteraction.WW_GO_TO_SEX_DESTINATION, Types.INTERACTION))}
    test = UserRunningInteractionTest(participant=ParticipantType.Actor, affordances=affordances, affordance_lists=list(), test_for_not_running=False, all_participants_running=False)
    immutable_slots_class = make_immutable_slots_class([
     'test', 'tooltip'])
    immutable_slots = immutable_slots_class(dict(test=test, tooltip=TunableLocalizedStringFactory._Wrapper(1181912093)))
    items_list = list(PhoneTuning.DISABLE_PHONE_TESTS)
    items_list.append(immutable_slots)
    PhoneTuning.DISABLE_PHONE_TESTS = ListSet(items_list)


@injector(InstanceManager, 'load_data_into_class_instances')
def _wickedwoohoo_load_data_and_disable_interactions(original, self, *args, **kwargs):
    global ALL_SIMS_AFFORDANCE_LIST
    result = original(self, *args, **kwargs)
    if self.TYPE != Types.INTERACTION:
        return result
    else:
        for affordance_id in ALL_SIMS_AFFORDANCE_LIST:
            affordance_key = get_resource_key(affordance_id, Types.INTERACTION)
            affordance_instance = self._tuned_classes.get(affordance_key)
            if affordance_instance is None:
                continue
            if hasattr(affordance_instance, 'test_globals') and affordance_instance.test_globals:
                affordance_tests_list = []
                for test in affordance_instance.test_globals:
                    test = get_sim_info_age_test(test, add_age=(Age(2), Age(4),
                     Age(8), Age(16), Age(32), Age(64)))
                    affordance_tests_list.append(test)

                affordance_instance.test_globals = TestList(affordance_tests_list)
                continue

        return result


@injector(InstanceManager, 'load_data_into_class_instances')
def _wickedwoohoo_load_data_and_disable_interactions(original, self, *args, **kwargs):
    global DISABLE_AFFORDANCE_LIST
    result = original(self, *args, **kwargs)
    if self.TYPE != Types.INTERACTION:
        return result
    else:
        for affordance_id in DISABLE_AFFORDANCE_LIST:
            affordance_key = get_resource_key(affordance_id, Types.INTERACTION)
            affordance_instance = self._tuned_classes.get(affordance_key)
            if affordance_instance is None:
                continue
            if hasattr(affordance_instance, 'test_globals') and affordance_instance.test_globals:
                affordance_tests_list = []
                has_age_test = False
                for test in affordance_instance.test_globals:
                    if isinstance(test, SimInfoTest):
                        test = create_impossible_sim_info_age_test()
                        has_age_test = True
                    affordance_tests_list.append(test)

                if has_age_test is False:
                    affordance_tests_list.append(create_impossible_sim_info_age_test())
                affordance_instance.test_globals = TestList(affordance_tests_list)
                continue

        return result


@injector(InstanceManager, 'load_data_into_class_instances')
def _wickedwoohoo_load_data_into_nudity_buff(original, self, *args, **kwargs):
    global NUDE_CLOTHING_BUFF
    result = original(self, *args, **kwargs)
    if self.TYPE != Types.BUFF:
        return result
    else:
        key = get_resource_key(NUDE_CLOTHING_BUFF, Types.BUFF)
        buff_instance = self._tuned_classes.get(key)
        if buff_instance is None:
            return result
        buff_instance.broadcaster = None
        return result


@injector(InstanceManager, 'load_data_into_class_instances')
def _wickedwoohoo_load_data_into_nudity_testsets(original, self, *args, **kwargs):
    global NUDE_TEENS_TEST_SETS
    result = original(self, *args, **kwargs)
    if self.TYPE != Types.SNIPPET:
        return result
    else:
        for snippet_id in NUDE_TEENS_TEST_SETS:
            snippet_key = get_resource_key(snippet_id, Types.SNIPPET)
            snippet_instance = self._tuned_classes.get(snippet_key)
            if snippet_instance is None:
                continue
            testset_groups_list = []
            for test_group in snippet_instance.test:
                testset_tests_list = []
                for test in test_group:
                    test = get_sim_info_age_test(test, add_age=(Age(1), Age(2), Age(4), Age(8)))
                    testset_tests_list.append(test)

                testset_groups_list.append(tuple(testset_tests_list))

            snippet_instance.test = CompoundTestList(list(testset_groups_list))

        return result


@injector(InstanceManager, 'load_data_into_class_instances')
def _wickedwoohoo_load_data_into_nudity_interactions(original, self, *args, **kwargs):
    global NUDITY_AFFORDANCE_LIST
    result = original(self, *args, **kwargs)
    if self.TYPE != Types.INTERACTION:
        return result
    else:
        for affordance_id in NUDITY_AFFORDANCE_LIST:
            affordance_key = get_resource_key(affordance_id, Types.INTERACTION)
            affordance_instance = self._tuned_classes.get(affordance_key)
            if affordance_instance is None:
                continue
            if hasattr(affordance_instance, 'test_globals') and affordance_instance.test_globals:
                affordance_tests_list = []
                for test in affordance_instance.test_globals:
                    test = get_sim_info_age_test(test, add_age=(Age(8),))
                    affordance_tests_list.append(test)

                affordance_instance.test_globals = TestList(affordance_tests_list)
            if hasattr(affordance_instance, 'sim_tests') and affordance_instance.sim_tests:
                simtests_group_list = []
                for test_group in affordance_instance.sim_tests:
                    simtest_tests_list = []
                    for test in test_group:
                        test = get_sim_info_age_test(test, add_age=(Age(8),))
                        simtest_tests_list.append(test)

                    simtests_group_list.append(tuple(simtest_tests_list))

                affordance_instance.sim_tests = CompoundTestList(simtests_group_list)
            if hasattr(affordance_instance, 'tests') and affordance_instance.tests:
                tests_group_list = []
                for test_group in affordance_instance.tests:
                    test_tests_list = []
                    for test in test_group:
                        test = get_sim_info_age_test(test, add_age=(Age(8),))
                        test_tests_list.append(test)

                    tests_group_list.append(tuple(test_tests_list))

                affordance_instance.tests = CompoundTestList(tests_group_list)
                continue

        return result