typos, factoring out functionality
This commit is contained in:
23
app/helpers/problem_helper.py
Normal file
23
app/helpers/problem_helper.py
Normal file
@ -0,0 +1,23 @@
|
||||
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
|
Reference in New Issue
Block a user