def blacklist()

in data/question-gen/engine.py [0:0]

• 24 lines of code
• 14 McCabe index (conditional complexity)

    def blacklist(self, *args):
        assert self.ent_queue != None

        ent = self.ent_queue

        if ent['type'] == 'objects':
            template = args[0]

            names = [x.name for x in ent['elements']]
            idx = [
                i for i, v in enumerate(names)
                if v in self.blacklist_objects[template]
            ]
            for i in idx[::-1]:
                del ent['elements'][i]
        elif ent['type'] == 'rooms':
            names = [x.name for x in ent['elements']]
            idx = [
                i for i, v in enumerate([
                    any([k for k in x if k in self.blacklist_rooms])
                    for x in names
                ]) if v == True
            ]
            for i in idx[::-1]:
                del ent['elements'][i]

        self.ent_queue = ent
        return self.ent_queue