# coding=utf-8
from api import app
from api.models.locale import Locale
from esales.models import Category, CategoryTree
from esales.utils import ImportFile


def update_designer(designer):
    category = Category(
        "root/designers/{}".format(designer.path),
        display_name=designer.name,
        visible_search=designer.visible_search,
        visible_list=designer.visible_list,
        visible_productpage=designer.visible_productpage,
        visible_designerpage=designer.visible_designerpage,
        send_to_bottom=designer.send_to_bottom,
        no_follow=designer.no_follow,
        hidden=1,
        path="/designers/{}".format(designer.path)
    )
    importfile = ImportFile()
    for locale in Locale.query.all():
        tree = CategoryTree("categories_{}".format(locale.value))
        tree.add_category(category)
        importfile.add_operation("update", tree)

    directory = app.config.get("ESALES_IMPORT_DIR")
    importfile.write(directory)


def update_category(category):
    importfile = ImportFile()
    for text in category.texts:
        locale = text.locale
        tree = CategoryTree("categories_{}".format(locale.value))
        tree.add_category(Category(
            key='root/{}'.format(category.path(locale.value)),
            display_name=category.name,
            wallpaper_description=text.wallpaper_description,
            wallpaper_h1=text.wallpaper_h1,
            canvas_description=text.canvas_description,
            canvas_h1=text.canvas_h1,
            design_wallpaper_description=text.design_wallpaper_description,
            design_wallpaper_h1=text.design_wallpaper_h1
        ))
        importfile.add_operation("update", tree)

    directory = app.config.get("ESALES_IMPORT_DIR")
    importfile.write(directory)