quickfix

crud.py

@@ -3,16 +3,17 @@
 from typing import List, Optional, Union
 
 from lnbits.db import Database
-from loguru import logger
+from lnbits.helpers import urlsafe_short_hash
 
-from .models import MyExtension
+from .models import CreateMyExtensionData, MyExtension
 
 db = Database("ext_myextension")
 
 
-async def create_myextension(data: MyExtension) -> MyExtension:
+async def create_myextension(data: CreateMyExtensionData) -> MyExtension:
+    data.id = urlsafe_short_hash()
     await db.insert("myextension.maintable", data)
-    return data
+    return MyExtension(**data.dict())
 
 
 async def get_myextension(myextension_id: str) -> Optional[MyExtension]:
@@ -27,7 +28,6 @@ async def get_myextensions(wallet_ids: Union[str, List[str]]) -> List[MyExtensio
     if isinstance(wallet_ids, str):
         wallet_ids = [wallet_ids]
     q = ",".join([f"'{w}'" for w in wallet_ids])
-    logger.debug(q)
     return await db.fetchall(
         f"SELECT * FROM myextension.maintable WHERE wallet IN ({q}) ORDER BY id",
         model=MyExtension,