From a6617cae68b2669eac0cde38a434e2826886b807 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Wed, 10 May 2017 12:08:11 +0200 Subject: [PATCH] Allow current_app.db('collections-name') This mimics the use in Flamenco (current_flamenco.db('collection_name')), and makes calling code a bit nicer (db('coll') instead of db()['coll']) --- pillar/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pillar/__init__.py b/pillar/__init__.py index f97e4d4e..83d265e2 100644 --- a/pillar/__init__.py +++ b/pillar/__init__.py @@ -551,12 +551,14 @@ class PillarServer(Eve): pprint(links) - def db(self): + def db(self, collection_name: str=None): """Returns the MongoDB database. :rtype: flask_pymongo.PyMongo """ + if collection_name: + return self.data.driver.db[collection_name] return self.data.driver.db def extension_sidebar_links(self, project):