Feature/find_or

src/masoniteorm/exceptions.py

@@ -32,3 +32,7 @@ class InvalidUrlConfiguration(Exception):
 
 class MultipleRecordsFound(Exception):
     pass
+
+
+class InvalidArgument(Exception):
+    pass

src/masoniteorm/models/Model.py

@@ -183,6 +183,7 @@ class Model(TimeStampsMixin, ObservesEvents, metaclass=ModelMeta):
             "doesnt_exist",
             "doesnt_have",
             "exists",
+            "find_or",
             "find_or_404",
             "find_or_fail",
             "first_or_fail",

src/masoniteorm/query/QueryBuilder.py

@@ -1,15 +1,15 @@
 import inspect
 from copy import deepcopy
 from datetime import datetime
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Callable
 
 from ..collection.Collection import Collection
 from ..config import load_config
 from ..exceptions import (
     HTTP404,
     ConnectionNotRegistered,
     ModelNotFound,
-    MultipleRecordsFound,
+    MultipleRecordsFound, InvalidArgument,
 )
 from ..expressions.expressions import (
     AggregateExpression,
@@ -1789,6 +1789,30 @@ def find(self, record_id):
 
         return self.where(self._model.get_primary_key(), record_id).first()
 
+    def find_or(self, record_id: int, callback: Callable, args=None):
+        """Finds a row by the primary key ID (Requires a model) or raise a ModelNotFound exception.
+
+        Arguments:
+            record_id {int} -- The ID of the primary key to fetch.
+            callback {Callable} -- The function to call if no record is found.
+
+        Returns:
+            Model|Callable
+        """
+
+        if not callable(callback):
+            raise InvalidArgument("A callback must be callable.")
+
+        result = self.find(record_id=record_id)
+
+        if not result:
+            if not args:
+                return callback()
+            else:
+                return callback(*args)
+
+        return result
+
     def find_or_fail(self, record_id):
         """Finds a row by the primary key ID (Requires a model) or raise a ModelNotFound exception.