crate.py 432 B

123456789101112131415
  1. from dataclasses import dataclass, field
  2. import itertools
  3. from data_models.workpiece import Workpiece
  4. @dataclass
  5. class Crate:
  6. id: int = field(default_factory=itertools.count().__next__, init=False) # autogen ids on creation
  7. workpiece: Workpiece | None = None
  8. def to_dict(self) -> dict:
  9. return {
  10. "id": self.id,
  11. "workpiece": self.workpiece.to_dict() if self.workpiece else None
  12. }