archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Allow current_app.db('collections-name')

Browse Source 
This mimics the use in Flamenco (current_flamenco.db('collection_name')),
and makes calling code a bit nicer (db('coll') instead of db()['coll'])
This commit is contained in:
Sybren A. Stüvel
2017-05-10 12:08:11 +02:00
parent 319f815985
commit a6617cae68
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pillar/__init__.py
View File

@@ -551,12 +551,14 @@ class PillarServer(Eve):
pprint(links) pprint(links)
def db(self): def db(self, collection_name: str=None):
"""Returns the MongoDB database. """Returns the MongoDB database.
:rtype: flask_pymongo.PyMongo :rtype: flask_pymongo.PyMongo
""" """
if collection_name:
return self.data.driver.db[collection_name]
return self.data.driver.db return self.data.driver.db
def extension_sidebar_links(self, project): def extension_sidebar_links(self, project):