Move remaining enumerations to different files
This makes it possible to split the central script up further. Importing enum is no longer necessary for the central script, so it has been removed.
This commit is contained in:
parent
a17135d85d
commit
b0075ab519
|
@ -5,7 +5,6 @@ __author__ = "Jeremy Saklad"
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import curses
|
import curses
|
||||||
from enum import Enum
|
|
||||||
from functools import reduce
|
from functools import reduce
|
||||||
from os import cpu_count
|
from os import cpu_count
|
||||||
|
|
||||||
|
@ -16,7 +15,10 @@ from data.appendages import Appendage
|
||||||
from data.buyers import Buyer
|
from data.buyers import Buyer
|
||||||
from data.costs import Cost
|
from data.costs import Cost
|
||||||
from data.declarations import Declaration
|
from data.declarations import Declaration
|
||||||
|
from data.diplomat_fascinations import DiplomatFascination
|
||||||
from data.embellishments import Embellishment
|
from data.embellishments import Embellishment
|
||||||
|
from data.fluctuations import Fluctuation
|
||||||
|
from data.occasional_buyers import OccasionalBuyer
|
||||||
from data.skulls import Skull
|
from data.skulls import Skull
|
||||||
from data.torsos import Torso
|
from data.torsos import Torso
|
||||||
from objects.enumaction import EnumAction
|
from objects.enumaction import EnumAction
|
||||||
|
@ -74,40 +76,6 @@ cp_model.CpModel.AddGeneralMultiplicationEquality = AddGeneralMultiplicationEqua
|
||||||
del AddGeneralMultiplicationEquality
|
del AddGeneralMultiplicationEquality
|
||||||
|
|
||||||
|
|
||||||
class Fluctuation(Enum):
|
|
||||||
"""Which skeleton attribute is currently boosted."""
|
|
||||||
|
|
||||||
ANTIQUITY = 1
|
|
||||||
AMALGAMY = 2
|
|
||||||
MENACE = 3
|
|
||||||
|
|
||||||
class OccasionalBuyer(Enum):
|
|
||||||
"""Which of several unusual buyers are available."""
|
|
||||||
|
|
||||||
AN_ENTHUSIAST_IN_SKULLS = [Buyer.AN_ENTHUSIAST_IN_SKULLS]
|
|
||||||
|
|
||||||
A_DREARY_MIDNIGHTER = [Buyer.A_DREARY_MIDNIGHTER]
|
|
||||||
|
|
||||||
A_COLOURFUL_PHANTASIST = [
|
|
||||||
Buyer.A_COLOURFUL_PHANTASIST_BAZAARINE,
|
|
||||||
Buyer.A_COLOURFUL_PHANTASIST_NOCTURNAL,
|
|
||||||
Buyer.A_COLOURFUL_PHANTASIST_CELESTIAL,
|
|
||||||
]
|
|
||||||
|
|
||||||
AN_INGENUOUS_MALACOLOGIST = [Buyer.AN_INGENUOUS_MALACOLOGIST]
|
|
||||||
|
|
||||||
AN_ENTERPRISING_BOOT_SALESMAN = [Buyer.AN_ENTERPRISING_BOOT_SALESMAN]
|
|
||||||
|
|
||||||
|
|
||||||
DiplomatFascination = Enum(
|
|
||||||
'DiplomatFascination',
|
|
||||||
( (diplomat.name[22:], diplomat) for diplomat in Buyer
|
|
||||||
if diplomat.name.startswith('THE_TRIFLING_DIPLOMAT_') ),
|
|
||||||
module = __name__
|
|
||||||
)
|
|
||||||
DiplomatFascination.__doc__ = "The current fascination of the Trifling Diplomat."
|
|
||||||
|
|
||||||
|
|
||||||
def Solve(shadowy_level, bone_market_fluctuations = None, zoological_mania = None, occasional_buyer = None, diplomat_fascination = None, desired_buyers = [], maximum_cost = cp_model.INT32_MAX, maximum_exhaustion = cp_model.INT32_MAX, time_limit = float('inf'), workers = cpu_count(), stdscr = None):
|
def Solve(shadowy_level, bone_market_fluctuations = None, zoological_mania = None, occasional_buyer = None, diplomat_fascination = None, desired_buyers = [], maximum_cost = cp_model.INT32_MAX, maximum_exhaustion = cp_model.INT32_MAX, time_limit = float('inf'), workers = cpu_count(), stdscr = None):
|
||||||
model = cp_model.CpModel()
|
model = cp_model.CpModel()
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,14 @@
|
||||||
|
__all__ = ['DiplomatFascination']
|
||||||
|
__author__ = "Jeremy Saklad"
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
from data.buyers import Buyer
|
||||||
|
|
||||||
|
DiplomatFascination = Enum(
|
||||||
|
'DiplomatFascination',
|
||||||
|
( (diplomat.name[22:], diplomat) for diplomat in Buyer
|
||||||
|
if diplomat.name.startswith('THE_TRIFLING_DIPLOMAT_') ),
|
||||||
|
module = __name__
|
||||||
|
)
|
||||||
|
DiplomatFascination.__doc__ = "The current fascination of the Trifling Diplomat."
|
|
@ -0,0 +1,11 @@
|
||||||
|
__all__ = ['Fluctuation']
|
||||||
|
__author__ = "Jeremy Saklad"
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
class Fluctuation(Enum):
|
||||||
|
"""Which skeleton attribute is currently boosted."""
|
||||||
|
|
||||||
|
ANTIQUITY = 1
|
||||||
|
AMALGAMY = 2
|
||||||
|
MENACE = 3
|
|
@ -0,0 +1,23 @@
|
||||||
|
__all__ = ['OccasionalBuyer']
|
||||||
|
__author__ = "Jeremy Saklad"
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
from data.buyers import Buyer
|
||||||
|
|
||||||
|
class OccasionalBuyer(Enum):
|
||||||
|
"""Which of several unusual buyers are available."""
|
||||||
|
|
||||||
|
AN_ENTHUSIAST_IN_SKULLS = [Buyer.AN_ENTHUSIAST_IN_SKULLS]
|
||||||
|
|
||||||
|
A_DREARY_MIDNIGHTER = [Buyer.A_DREARY_MIDNIGHTER]
|
||||||
|
|
||||||
|
A_COLOURFUL_PHANTASIST = [
|
||||||
|
Buyer.A_COLOURFUL_PHANTASIST_BAZAARINE,
|
||||||
|
Buyer.A_COLOURFUL_PHANTASIST_NOCTURNAL,
|
||||||
|
Buyer.A_COLOURFUL_PHANTASIST_CELESTIAL,
|
||||||
|
]
|
||||||
|
|
||||||
|
AN_INGENUOUS_MALACOLOGIST = [Buyer.AN_INGENUOUS_MALACOLOGIST]
|
||||||
|
|
||||||
|
AN_ENTERPRISING_BOOT_SALESMAN = [Buyer.AN_ENTERPRISING_BOOT_SALESMAN]
|
Loading…
Reference in New Issue