1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011 MediaGoblin contributors. See AUTHORS.
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
19 from paste
.deploy
.converters
import asint
20 from mediagoblin
.db
import models
23 def connect_database_from_config(app_config
, use_pymongo
=False):
25 Connect to the main database, take config from app_config
27 Optionally use pymongo instead of mongokit for the connection.
29 port
= app_config
.get('db_port')
34 connection
= pymongo
.Connection(
35 app_config
.get('db_host'), port
)
37 connection
= mongokit
.Connection(
38 app_config
.get('db_host'), port
)
42 def setup_connection_and_db_from_config(app_config
, use_pymongo
=False):
44 Setup connection and database from config.
46 Optionally use pymongo instead of mongokit.
48 connection
= connect_database_from_config(app_config
, use_pymongo
)
49 database_path
= app_config
['db_name']
50 db
= connection
[database_path
]
53 models
.register_models(connection
)
55 return (connection
, db
)