Merge pull request #55 from tom-pollak/dev/default-get-fix

fixes #52

Author: jph00

fastlite/kw.py

@@ -56,7 +56,7 @@ def ids_and_rows_where(
         yield row.pop('__rid'), row
 
 @patch
-def get(self:Table, pk_values: list|tuple|str|int, as_cls:bool=True, xtra:dict|None=None)->Any:
+def get(self:Table, pk_values: list|tuple|str|int, as_cls:bool=True, xtra:dict|None=None, default:Any=UNSET)->Any:
     if not isinstance(pk_values, (list, tuple)): pk_values = [pk_values]
     last_pk = pk_values[0] if len(self.pks) == 1 else pk_values
     if not xtra: xtra = getattr(self, 'xtra_id', {})
@@ -65,7 +65,9 @@ def get(self:Table, pk_values: list|tuple|str|int, as_cls:bool=True, xtra:dict|N
     if len(pks)!=len(vals): raise NotFoundError(f"Need {len(pks)} pk")
     wheres = ["[{}] = ?".format(pk_name) for pk_name in pks]
     item = first(self.ids_and_rows_where(" and ".join(wheres), vals))
-    if not item: raise NotFoundError
+    if not item:
+        if default is UNSET: raise NotFoundError()
+        return default
     rid,row = item
     self.last_pk,self.last_rowid = last_pk,rid
     if as_cls and hasattr(self,'cls'): row = self.cls(**row)