From 4ed45e5af57007ca77e1e6922c618f1681ea32e1 Mon Sep 17 00:00:00 2001
From: Ryan Barrett <git@ryanb.org>
Date: Sun, 31 Mar 2024 18:44:38 -0700
Subject: [PATCH] atproto: pass new ndb_client kwarg to DatastoreStorage
 constructor

for snarfed/lexrpc#8
---
 atproto.py       | 3 ++-
 requirements.txt | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/atproto.py b/atproto.py
index f6c1c13..eaa046c 100644
--- a/atproto.py
+++ b/atproto.py
@@ -21,6 +21,7 @@ from granary import as1, bluesky
 from lexrpc import Client
 import requests
 from requests import RequestException
+from oauth_dropins.webutil.appengine_config import ndb_client
 from oauth_dropins.webutil.appengine_info import DEBUG
 from oauth_dropins.webutil import util
 from oauth_dropins.webutil.util import json_dumps, json_loads
@@ -40,7 +41,7 @@ from protocol import Protocol
 
 logger = logging.getLogger(__name__)
 
-arroba.server.storage = DatastoreStorage()
+arroba.server.storage = DatastoreStorage(ndb_client=ndb_client)
 
 LEXICONS = Client('https://unused').defs
 
diff --git a/requirements.txt b/requirements.txt
index ac99975..a706582 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -57,6 +57,7 @@ html5lib==1.1
 humanfriendly==10.0
 humanize==4.9.0
 idna==3.6
+iterators==0.2.0
 itsdangerous==2.1.2
 Jinja2==3.1.3
 jsonschema==4.21.1