even more logging

This commit is contained in:
Joshua Burman 2022-02-09 17:25:23 -05:00
parent 3f7b3cc0b6
commit 78331577dc

View File

@ -1,6 +1,8 @@
from pydantic import BaseModel from pydantic import BaseModel
from typing import List, Optional from typing import List, Optional
import logging
from models.item import Item from models.item import Item
from models.constraint import Constraint from models.constraint import Constraint
from models.irt_model import IRTModel from models.irt_model import IRTModel
@ -31,6 +33,7 @@ class SolverRun(BaseModel):
return True return True
def generate_bundles(self): def generate_bundles(self):
logging.info('Generating Bundles...')
bundle_constraints = (constraint.reference_attribute for constraint in self.constraints if constraint.reference_attribute.type == 'bundle') bundle_constraints = (constraint.reference_attribute for constraint in self.constraints if constraint.reference_attribute.type == 'bundle')
for bundle_constraint in bundle_constraints: for bundle_constraint in bundle_constraints:
@ -64,5 +67,7 @@ class SolverRun(BaseModel):
type=type_attribute type=type_attribute
)] )]
logging.info('Bundles Generated...')
def get_constraint(self, name): def get_constraint(self, name):
return next((constraint for constraint in self.constraints if constraint.reference_attribute.id == name), None) return next((constraint for constraint in self.constraints if constraint.reference_attribute.id == name), None)