Add backwards compatible dict.get() fallback value (singer-io#95)

judahrand · Oct 27, 2020 · d209713 · d209713
1 parent aea7b22
commit d209713
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/singer/bookmarks.py b/singer/bookmarks.py
@@ -22,8 +22,8 @@ def reset_stream(state, tap_stream_id):
     state['bookmarks'][tap_stream_id] = {}
     return state
 
-def get_bookmark(state, tap_stream_id, key):
-    return state.get('bookmarks', {}).get(tap_stream_id, {}).get(key)
+def get_bookmark(state, tap_stream_id, key, default=None):
+    return state.get('bookmarks', {}).get(tap_stream_id, {}).get(key, default)
 
 def set_offset(state, tap_stream_id, offset_key, offset_value):
     state = ensure_bookmark_path(state, ['bookmarks', tap_stream_id, "offset", offset_key])
@@ -35,12 +35,12 @@ def clear_offset(state, tap_stream_id):
     state['bookmarks'][tap_stream_id]["offset"] = {}
     return state
 
-def get_offset(state, tap_stream_id):
-    return state.get('bookmarks', {}).get(tap_stream_id, {}).get("offset")
+def get_offset(state, tap_stream_id, default=None):
+    return state.get('bookmarks', {}).get(tap_stream_id, {}).get("offset", default)
 
 def set_currently_syncing(state, tap_stream_id):
     state['currently_syncing'] = tap_stream_id
     return state
 
-def get_currently_syncing(state):
-    return state.get('currently_syncing')
+def get_currently_syncing(state, default=None):
+    return state.get('currently_syncing', default)