use our usual ndb context kwargs in arroba DatabaseStorage

for #1149
snarfed · Dec 18, 2024 · 36cbbdd · 36cbbdd
1 parent 91b268f
commit 36cbbdd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/atproto.py b/atproto.py
@@ -58,7 +58,8 @@
 
 logger = logging.getLogger(__name__)
 
-arroba.server.storage = DatastoreStorage(ndb_client=ndb_client)
+arroba.server.storage = DatastoreStorage(ndb_client=ndb_client,
+                                         ndb_context_kwargs=common.NDB_CONTEXT_KWARGS)
 
 appview = Client(f'https://{os.environ["APPVIEW_HOST"]}',
                  headers={'User-Agent': USER_AGENT})