Added archived field to spools

Resolves #13

spoolman/api/v1/models.py

@@ -150,6 +150,7 @@ class Spool(BaseModel):
         description="Free text comment about this specific spool.",
         example="",
     )
+    archived: bool = Field(description="Whether this spool is archived and should not be used anymore.")
 
     @staticmethod
     def from_db(item: models.Spool) -> "Spool":
@@ -185,6 +186,7 @@ class Spool(BaseModel):
             location=item.location,
             lot_nr=item.lot_nr,
             comment=item.comment,
+            archived=item.archived if item.archived is not None else False,
         )
 
 

spoolman/api/v1/spool.py

@@ -49,6 +49,7 @@ class SpoolParameters(BaseModel):
         description="Free text comment about this specific spool.",
         example="",
     )
+    archived: bool = Field(default=False, description="Whether this spool is archived and should not be used anymore.")
 
 
 class SpoolUpdateParameters(SpoolParameters):
@@ -104,6 +105,11 @@ async def find(
         title="Lot/Batch Number",
         description="Partial case-insensitive search term for the spool lot number.",
     ),
+    allow_archived: bool = Query(
+        default=False,
+        title="Allow Archived",
+        description="Whether to include archived spools in the search results.",
+    ),
 ) -> list[Spool]:
     db_items = await spool.find(
         db=db,
@@ -114,6 +120,7 @@ async def find(
         vendor_id=vendor_id,
         location=location,
         lot_nr=lot_nr,
+        allow_archived=allow_archived,
     )
     return [Spool.from_db(db_item) for db_item in db_items]
 
@@ -168,6 +175,7 @@ async def create(  # noqa: ANN201
             location=body.location,
             lot_nr=body.lot_nr,
             comment=body.comment,
+            archived=body.archived,
         )
         return Spool.from_db(db_item)
     except ItemCreateError:

spoolman/database/models.py

@@ -57,3 +57,4 @@ class Spool(Base):
     location: Mapped[Optional[str]] = mapped_column(String(64))
     lot_nr: Mapped[Optional[str]] = mapped_column(String(64))
     comment: Mapped[Optional[str]] = mapped_column(String(1024))
+    archived: Mapped[Optional[bool]] = mapped_column()

spoolman/database/spool.py

@@ -30,6 +30,7 @@ async def create(
     location: Optional[str] = None,
     lot_nr: Optional[str] = None,
     comment: Optional[str] = None,
+    archived: bool = False,
 ) -> models.Spool:
     """Add a new spool to the database. Leave weight empty to assume full spool."""
     filament_item = await filament.get_by_id(db, filament_id)
@@ -55,6 +56,7 @@ async def create(
         location=location,
         lot_nr=lot_nr,
         comment=comment,
+        archived=archived,
     )
     db.add(db_item)
     await db.flush()
@@ -83,6 +85,7 @@ async def find(
     vendor_id: Optional[int] = None,
     location: Optional[str] = None,
     lot_nr: Optional[str] = None,
+    allow_archived: bool = False,
 ) -> list[models.Spool]:
     """Find a list of spool objects by search criteria."""
     stmt = (
@@ -105,6 +108,8 @@ async def find(
         stmt = stmt.where(models.Spool.location.ilike(f"%{location}%"))
     if lot_nr is not None:
         stmt = stmt.where(models.Spool.lot_nr.ilike(f"%{lot_nr}%"))
+    if not allow_archived:
+        stmt = stmt.where(models.Spool.archived != True)  # noqa: E712
 
     rows = await db.execute(stmt)
     return list(rows.scalars().all())