Refs mdowds#72, adding support for transactions argument

agherasim · Jul 9, 2023 · 6554df5 · 6554df5
1 parent 0de34b1
commit 6554df5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mockfirestore/document.py b/mockfirestore/document.py
@@ -1,7 +1,7 @@
 from copy import deepcopy
 from functools import reduce
 import operator
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 from mockfirestore import NotFound
 from mockfirestore._helpers import (
     Timestamp, Document, Store, get_by_path, set_by_path, delete_by_path
@@ -39,7 +39,7 @@ def read_time(self) -> Timestamp:
         timestamp = Timestamp.from_now()
         return timestamp
 
-    def get(self, field_path: str) -> Any:
+    def get(self, field_path: str, transaction=None) -> Any:
         if not self.exists:
             return None
         else: