from invoke import task

from .db import Data

@task
def setup(c):
    result = Data().setup()
    print(result)

@task
def add_goal(c, name):
    result = Data().add_goal(name)
    print(result)

@task
def add_task(c, goal_id, name):
    result = Data().add_task(goal_id=goal_id, name=name)
    print(result)


@task
def list(c):
    goals = Data().list_goals()
    for goal in goals:
        print(goal)

@task
def print_goal(c, id):
    result = Data().print_goal(id)
    print(result)