"""Functions for working with worlds.""" from __future__ import annotations from random import Random from tcod.ecs import Registry from game.components import Gold, Graphic, Position from game.tags import IsActor, IsItem, IsPlayer def new_world() -> Registry: """Return a freshly generated world.""" world = Registry() rng = world[None].components[Random] = Random() player = world[object()] player.components[Position] = Position(5, 5) player.components[Graphic] = Graphic(ord("@")) player.components[Gold] = 0 player.tags |= {IsPlayer, IsActor} for _ in range(10): gold = world[object()] gold.components[Position] = Position(rng.randint(0, 20), rng.randint(0, 20)) gold.components[Graphic] = Graphic(ord("$"), fg=(255, 255, 0)) gold.components[Gold] = rng.randint(1, 10) gold.tags |= {IsItem} return world