from oscar.templatetags.category_tags import (
    get_annotated_list,
)
from oscar.templatetags.category_tags import (
    register as category_tags_library,
)
from oscar.templatetags.purchase_info_tags import (
    purchase_info_for_line,
    purchase_info_for_product,
)
from oscar.templatetags.purchase_info_tags import (
    register as purchase_info_tags_library,
)

from ocyan.core.utils import RequestLibrary

category_overide = RequestLibrary(category_tags_library)
category_overide.simple_tag(get_annotated_list, name="category_tree")

purchase_info_override = RequestLibrary(purchase_info_tags_library)
purchase_info_override.simple_tag(purchase_info_for_product, takes_request=True)
purchase_info_override.simple_tag(purchase_info_for_line, takes_request=True)
