mypy

crud.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import List
 
 from lnbits.db import Database
 from lnbits.helpers import urlsafe_short_hash
@@ -8,7 +8,7 @@
 db = Database("ext_satspot")
 
 
-async def create_satspot(data: CreateSatspot, wallet, user) -> Optional[Satspot]:
+async def create_satspot(data: CreateSatspot, wallet, user) -> List[Satspot]:
     satspot = Satspot(**data.dict(), id=urlsafe_short_hash(), wallet=wallet, user=user)
     await db.insert("satspot.satspot", satspot)
     return await get_satspots(user)
@@ -19,15 +19,15 @@ async def update_satspot(satspot: Satspot) -> Satspot:
     return satspot
 
 
-async def get_satspot(satspot_id: str) -> Optional[Satspot]:
+async def get_satspot(satspot_id: str) -> Satspot:
     return await db.fetchone(
         "SELECT * FROM satspot.satspot WHERE id = :id",
         {"id": satspot_id},
         Satspot,
     )
 
 
-async def get_satspots(user: str) -> Optional[Satspot]:
+async def get_satspots(user: str) -> List[Satspot]:
     return await db.fetchall(
         "SELECT * FROM satspot.satspot WHERE user = :user",
         {"user": user},

models.py

@@ -14,7 +14,7 @@ class CreateSatspot(BaseModel):
 
 class Satspot(BaseModel):
     id: Optional[str] = None
-    wallet: Optional[str] = None
+    wallet: str
     user: Optional[str] = None
     name: str
     closing_date: datetime

pyproject.toml

@@ -7,6 +7,7 @@ authors = ["Alan Bits <[email protected]>"]
 [tool.poetry.dependencies]
 python = "^3.10 | ^3.9"
 lnbits = {version = "*", allow-prereleases = true}
+mypy = "^1.13.0"
 
 [tool.poetry.group.dev.dependencies]
 black = "^24.3.0"