itemProjectionRange.py

import re

import wx

import gui.fitCommands as cmd
import gui.mainFrame
from eos.saveddata.fighter import Fighter as EosFighter
from eos.saveddata.fit import Fit as EosFit
from eos.saveddata.module import Module as EosModule
from gui.contextMenu import ContextMenuCombined
from gui.fitCommands.helpers import getSimilarFighters, getSimilarModPositions
from gui.utils.themes import ThemedDialog, Themes
from service.fit import Fit

# noinspection PyPackageRequirements

_t = wx.GetTranslation


class ChangeItemProjectionRange(ContextMenuCombined):

    def __init__(self):
        self.mainFrame = gui.mainFrame.MainFrame.getInstance()

    def display(self, callingWindow, srcContext, mainItem, selection):
        if srcContext not in ('projectedFit', 'projectedModule', 'projectedDrone', 'projectedFighter'):
            return False
        if mainItem is None:
            return False
        if getattr(mainItem, 'isExclusiveSystemEffect', False):
            return False
        return True

    def getText(self, callingWindow, itmContext, mainItem, selection):
        return _t('Change {} Range').format(itmContext)

    def activate(self, callingWindow, fullContext, mainItem, selection, i):
        fitID = self.mainFrame.getActiveFit()
        if isinstance(mainItem, EosFit):
            try:
                value = mainItem.getProjectionInfo(fitID).projectionRange
            except AttributeError:
                return
        else:
            value = mainItem.projectionRange
        if value is not None:
            value /= 1000
        with RangeChanger(self.mainFrame, value) as dlg:
            if dlg.ShowModal() == wx.ID_OK:
                cleanInput = re.sub(r'[^0-9.]', '', dlg.input.GetLineText(0).strip())
                if cleanInput:
                    try:
                        cleanInputFloat = float(cleanInput)
                    except ValueError:
                        return
                    newRange = cleanInputFloat * 1000
                else:
                    newRange = None
                fitID = self.mainFrame.getActiveFit()
                items = selection
                if wx.GetMouseState().GetModifiers() in (wx.MOD_ALT, wx.MOD_CONTROL):
                    if isinstance(mainItem, EosModule):
                        fit = Fit.getInstance().getFit(fitID)
                        positions = getSimilarModPositions(fit.projectedModules, mainItem)
                        items = [fit.projectedModules[p] for p in positions]
                    elif isinstance(mainItem, EosFighter):
                        fit = Fit.getInstance().getFit(fitID)
                        items = getSimilarFighters(fit.projectedFighters, mainItem)
                self.mainFrame.command.Submit(cmd.GuiChangeProjectedItemsProjectionRangeCommand(
                        fitID=fitID, items=items, projectionRange=newRange))


ChangeItemProjectionRange.register()


class RangeChanger(ThemedDialog):

    def __init__(self, parent, value):
        super().__init__(parent, title='Change Projection Range', style=wx.DEFAULT_DIALOG_STYLE)
        self.SetMinSize((346, 156))

        bSizer1 = wx.BoxSizer(wx.VERTICAL)

        bSizer2 = wx.BoxSizer(wx.VERTICAL)
        text = wx.StaticText(self, wx.ID_ANY, 'New Range, km:')
        bSizer2.Add(text, 0)

        bSizer1.Add(bSizer2, 0, wx.ALL, 10)

        self.input = wx.TextCtrl(self, wx.ID_ANY, style=wx.TE_PROCESS_ENTER)
        Themes.styleInput(self.input)
        if value is None:
            value = ''
        else:
            if value == int(value):
                value = int(value)
            value = str(value)
        self.input.SetValue(value)

        bSizer1.Add(self.input, 0, wx.LEFT | wx.RIGHT | wx.EXPAND, 15)

        bSizer3 = wx.BoxSizer(wx.VERTICAL)
        bSizer3.Add(wx.StaticLine(self, wx.ID_ANY), 0, wx.BOTTOM | wx.EXPAND, 15)

        bSizer3.Add(self.CreateStdDialogButtonSizer(wx.OK | wx.CANCEL), 0, wx.EXPAND)
        bSizer1.Add(bSizer3, 0, wx.ALL | wx.EXPAND, 10)

        self.input.Bind(wx.EVT_CHAR, self.onChar)
        self.input.Bind(wx.EVT_TEXT_ENTER, self.processEnter)
        self.SetSizer(bSizer1)
        self.Fit()
        self.CenterOnParent()
        self.input.SetFocus()
        self.input.SelectAll()

    def processEnter(self, evt):
        self.EndModal(wx.ID_OK)

    # checks to make sure it's valid number
    @staticmethod
    def onChar(event):
        key = event.GetKeyCode()

        acceptable_characters = '1234567890.'
        acceptable_keycode = [3, 22, 13, 8, 127]  # modifiers like delete, copy, paste
        if key in acceptable_keycode or key >= 255 or (key < 255 and chr(key) in acceptable_characters):
            event.Skip()
            return
        else:
            return False