PyTibia/src/gameplay/resolvers.py at e9447f4aae1404bd561d65bcfb0fae9233703f98 · lucasmonstrox/PyTibia · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from typing import Union
from src.shared.typings import Waypoint
from .core.tasks.common.base import BaseTask
from .core.tasks.common.vector import VectorTask
from .core.tasks.depositGold import DepositGoldTask
from .core.tasks.depositItems import DepositItemsTask
from .core.tasks.dropFlasks import DropFlasksTask
from .core.tasks.logout import LogoutTask
from .core.tasks.lure import LureWaypointTask
from .core.tasks.refill import RefillTask
from .core.tasks.refillChecker import RefillCheckerTask
from .core.tasks.singleWalk import SingleWalkTask
from .core.tasks.useRopeWaypoint import UseRopeWaypointTask
from .core.tasks.useShovelWaypoint import UseShovelWaypointTask
from .core.tasks.walkToWaypoint import WalkToWaypointTask


# TODO: add unit tests
def resolveTasksByWaypoint(waypoint: Waypoint) -> Union[BaseTask, VectorTask]:
    if waypoint['type'] == 'depositGold':
        return DepositGoldTask()
    elif waypoint['type'] == 'depositItems':
        return DepositItemsTask(waypoint)
    elif waypoint['type'] == 'dropFlasks':
        return DropFlasksTask()
    elif waypoint['type'] == 'logout':
        return LogoutTask()
    elif waypoint['type'] == 'lure':
        return LureWaypointTask(waypoint)
    elif waypoint['type'] == 'moveDown':
        return SingleWalkTask(waypoint['type'], waypoint['options']['direction'])
    elif waypoint['type'] == 'moveUp':
        return SingleWalkTask(waypoint['type'], waypoint['options']['direction'])
    elif waypoint['type'] == 'refill':
        return RefillTask(waypoint)
    elif waypoint['type'] == 'refillChecker':
        return RefillCheckerTask(waypoint)
    elif waypoint['type'] == 'useRope':
        return UseRopeWaypointTask(waypoint)
    elif waypoint['type'] == 'useShovel':
        return UseShovelWaypointTask(waypoint)
    elif waypoint['type'] == 'walk':
        return WalkToWaypointTask(waypoint['coordinate'])