Copy Enums to const.py

Switching to IntEnum where necessary.

eos/const.py

@@ -1,5 +1,5 @@
 from eos.enum import Enum
-
+from enum import IntEnum
 
 
 class Slot(Enum):
@@ -24,3 +24,56 @@ class Slot(Enum):
     FS_LIGHT = 13
     FS_SUPPORT = 14
     FS_HEAVY = 15
+
+
+class ImplantLocation(Enum):
+    FIT = 0
+    CHARACTER = 1
+
+
+class CalcType(Enum):
+    LOCAL = 0
+    PROJECTED = 1
+    COMMAND = 2
+
+class State(Enum):
+    OFFLINE = -1
+    ONLINE = 0
+    ACTIVE = 1
+    OVERHEATED = 2
+
+class Hardpoint(Enum):
+    NONE = 0
+    MISSILE = 1
+    TURRET = 2
+
+class LoginMethod(Enum):
+    SERVER = 0
+    MANUAL = 1
+
+class SsoMode(Enum):
+    AUTO = 0
+    CUSTOM = 1
+
+class ESIEndpoints(Enum):
+    CHAR = "/v4/characters/{character_id}/"
+    CHAR_SKILLS = "/v4/characters/{character_id}/skills/"
+    CHAR_FITTINGS = "/v1/characters/{character_id}/fittings/"
+    CHAR_DEL_FIT = "/v1/characters/{character_id}/fittings/{fitting_id}/"
+
+class MultiBuy_ItemType(IntEnum):
+    IMPLANTS = 1
+    CARGO = 2
+    LOADED_CHARGES = 3
+
+class Options(IntEnum):
+    IMPLANTS = 1
+    MUTATIONS = 2
+    LOADED_CHARGES = 3
+
+class RigSize(Enum):
+    # Matches to item attribute "rigSize" on ship and rig items
+    SMALL = 1
+    MEDIUM = 2
+    LARGE = 3
+    CAPITAL = 4

eos/saveddata/fit.py

@@ -29,6 +29,7 @@ import eos.db
 from eos import capSim
 from eos.effectHandlerHelpers import HandledModuleList, HandledDroneCargoList, HandledImplantBoosterList, HandledProjectedDroneList, HandledProjectedModList
 from eos.enum import Enum
+from eos.const import ImplantLocation, CalcType
 from eos.saveddata.ship import Ship
 from eos.saveddata.drone import Drone
 from eos.saveddata.character import Character
@@ -36,20 +37,10 @@ from eos.saveddata.citadel import Citadel
 from eos.saveddata.module import Module, State, Slot, Hardpoint
 from eos.utils.stats import DmgTypes
 from logbook import Logger
+
 pyfalog = Logger(__name__)
 
 
-class ImplantLocation(Enum):
-    FIT = 0
-    CHARACTER = 1
-
-
-class CalcType(Enum):
-    LOCAL = 0
-    PROJECTED = 1
-    COMMAND = 2
-
-
 class Fit(object):
     """Represents a fitting, with modules, ship, implants, etc."""
 

eos/saveddata/module.py

@@ -23,7 +23,7 @@ from logbook import Logger
 from sqlalchemy.orm import reconstructor, validates
 
 import eos.db
-from eos.const import Slot
+from eos.const import Slot, State, Hardpoint
 from eos.effectHandlerHelpers import HandledCharge, HandledItem
 from eos.enum import Enum
 from eos.modifiedAttributeDict import ChargeAttrShortcut, ItemAttrShortcut, ModifiedAttributeDict
@@ -35,14 +35,6 @@ from eos.utils.stats import DmgTypes
 
 pyfalog = Logger(__name__)
 
-
-class State(Enum):
-    OFFLINE = -1
-    ONLINE = 0
-    ACTIVE = 1
-    OVERHEATED = 2
-
-
 ProjectedMap = {
     State.OVERHEATED: State.ACTIVE,
     State.ACTIVE: State.OFFLINE,
@@ -67,12 +59,6 @@ ProjectedSystem = {
 }
 
 
-class Hardpoint(Enum):
-    NONE = 0
-    MISSILE = 1
-    TURRET = 2
-
-
 class Module(HandledItem, HandledCharge, ItemAttrShortcut, ChargeAttrShortcut):
     """An instance of this class represents a module together with its charge and modified attributes"""
     MINING_ATTRIBUTES = ("miningAmount",)