from invoke import task

from .db import Data

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

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

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


@task
def list(c, store_id):
    items = Data().get_shopping_list(store_id)
    for item in items:
        print(item)

    
@task
def list_unsorted(c, store_id):
    items = Data().list_unsorted()
    for item in items:
        print(item)

@task
def assign(c, store_id, item_id):
    result = Data().assign_store(store_id, item_id)
    print(result)