ZODBConnector -> _ZODBState and private

docs/index.rst

@@ -181,9 +181,6 @@ API Reference
 .. autoclass:: ZODB
   :members:
 
-.. autoclass:: ZODBConnector
-  :members:
-
 
 Persistent Objects
 """"""""""""""""""

flaskext/zodb.py

@@ -5,15 +5,15 @@
 from UserDict import IterableUserDict
 from ZODB.DB import DB
 from contextlib import contextmanager, closing
-from werkzeug import cached_property
+from werkzeug.utils import cached_property
 
 from BTrees.OOBTree import OOBTree as BTree
 from persistent import Persistent as Object
 from persistent.list import PersistentList as List
 from persistent.mapping import PersistentMapping as Dict
 
 
-__all__ = ['ZODB', 'ZODBConnector', 'Object', 'List', 'Dict', 'BTree']
+__all__ = ['ZODB', 'Object', 'List', 'Dict', 'BTree']
 
 
 class ZODB(IterableUserDict):
@@ -42,7 +42,7 @@ def init_app(self, app):
         """Configure a Flask application to use this ZODB extension."""
         assert 'zodb' not in app.extensions, \
                'app already initiated for zodb'
-        app.extensions['zodb'] = ZODBConnector(self, app)
+        app.extensions['zodb'] = _ZODBState(self, app)
         app.teardown_request(self.close_db)
 
     def close_db(self, exception):
@@ -91,14 +91,14 @@ def data(self):
         return self.connection.root()
 
 
-class ZODBConnector(object):
+class _ZODBState(object):
     """Adds a ZODB connection pool to a Flask application."""
 
-    def __init__(self, ext, app):
-        self.ext = ext
+    def __init__(self, zodb, app):
+        self.zodb = zodb
         self.app = app
 
     @cached_property
     def db(self):
         """Connection pool."""
-        return self.ext.create_db(self.app)
+        return self.zodb.create_db(self.app)

tests.py

@@ -29,8 +29,8 @@ def pytest_generate_tests(metafunc):
 
 def test_single_app_shortcut():
     app = Flask(__name__)
-    ext = ZODB(app)
-    assert app.extensions['zodb'].ext is ext
+    zodb = ZODB(app)
+    assert app.extensions['zodb'].zodb is zodb
 
 
 def test_connection(app):