23 lines
771 B
Python
23 lines
771 B
Python
from typing import Tuple
|
|
|
|
from models.item import Item
|
|
|
|
def sanctify(solved_items: [Item]) -> Tuple[list]:
|
|
sacred_ids = []
|
|
enemy_ids = []
|
|
|
|
# get all enemies
|
|
for item in solved_items:
|
|
# if it has enemies, check if it exists as part of the solved items
|
|
for enemy_id in item.enemies:
|
|
# if it does, it's a true enemy
|
|
if enemy_id in (item.id for item in solved_items):
|
|
enemy_ids.append(enemy_id)
|
|
# remove enemy from solved items,
|
|
# lest it has this sacred item added to enemies
|
|
solved_items = [i for i in solved_items if i.id != enemy_id]
|
|
|
|
# the item is cleansed, now it's sacred
|
|
sacred_ids.append(item.id)
|
|
|
|
return sacred_ids, enemy_ids |