34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from __future__ import annotations
|
|
from typing import TYPE_CHECKING
|
|
|
|
from models.constraints.generic_constraint import *
|
|
|
|
if TYPE_CHECKING:
|
|
from models.solver_run import SolverRun
|
|
|
|
class TotalFormItemsConstraint(GenericConstraint):
|
|
@classmethod
|
|
def create(cls: Type[_T], total_items: int) -> _T:
|
|
return cls(
|
|
minimum=0,
|
|
maximum=0,
|
|
reference_attribute=Attribute(
|
|
value=total_items,
|
|
type='form_uniqueness',
|
|
id='form_uniqueness'
|
|
)
|
|
)
|
|
|
|
def build(self, problem_handler: Problem, **kwargs) -> None:
|
|
logging.info('Total Form Items Constraint Generating...')
|
|
|
|
problem_handler.problem += lpSum(
|
|
[
|
|
bundle.count * problem_handler.solver_bundles_var[bundle.id]
|
|
for bundle in problem_handler.bundles
|
|
] + [
|
|
1 * problem_handler.solver_items_var[item.id]
|
|
for item in problem_handler.items
|
|
]
|
|
) == self.reference_attribute.value, f'Total bundle form items for form'
|