# =============================================================================
# Copyright (C) 2010 Diego Duclos, Lucas Thode
#
# This file is part of pyfa.
#
# pyfa is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pyfa is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
# =============================================================================

# noinspection PyPackageRequirements
import wx

from eos.saveddata.cargo import Cargo
from eos.saveddata.drone import Drone
from service.price import Price as ServicePrice
from gui.viewColumn import ViewColumn
from gui.bitmapLoader import BitmapLoader
from gui.utils.numberFormatter import formatAmount


class Price(ViewColumn):
    name = "Price"

    def __init__(self, fittingView, params):
        ViewColumn.__init__(self, fittingView)
        self.mask = wx.LIST_MASK_IMAGE
        self.bitmap = BitmapLoader.getBitmap("totalPrice_small", "gui")
        self.imageId = fittingView.imageList.GetImageIndex("totalPrice_small", "gui")

    def getText(self, stuff):
        if stuff.item is None or stuff.item.group.name == "Ship Modifiers":
            return ""

        if hasattr(stuff, "isEmpty"):
            if stuff.isEmpty:
                return ""

        price = stuff.item.price.price

        if not price:
            return ""

        if isinstance(stuff, Drone) or isinstance(stuff, Cargo):
            price *= stuff.amount

        return formatAmount(price, 3, 3, 9, currency=True)

    def delayedText(self, mod, display, colItem):
        sPrice = ServicePrice.getInstance()

        def callback(item):
            price = item.item.price
            text = formatAmount(price.price, 3, 3, 9, currency=True) if price.price else ""
            if price.failed:
                text += " (!)"
            colItem.SetText(text)

            display.SetItem(colItem)

        sPrice.getPrices([mod.item], callback, True)

    def getImageId(self, mod):
        return -1

    def getToolTip(self, mod):
        return self.name


Price.register()