Only convert db port if it's there and use asint to do it (better errors if failing)

This commit is contained in:
Christopher Allan Webber 2011-05-08 07:16:50 -05:00
parent 0ce26c370c
commit a1eb1f6051
2 changed files with 9 additions and 3 deletions

View File

@ -18,7 +18,7 @@ import urllib
import routes
import mongokit
from paste.deploy.converters import asbool
from paste.deploy.converters import asint
from webob import Request, exc
from mediagoblin import routing, util, models, storage, staticdirect
@ -114,7 +114,9 @@ class MediaGoblinApp(object):
def paste_app_factory(global_config, **app_config):
# Get the database connection
port = int(app_config.get('db_port'))
port = app_config.get('db_port')
if port:
port = asint(port)
connection = mongokit.Connection(
app_config.get('db_host'), port)

View File

@ -18,6 +18,7 @@ import os
import mongokit
from paste.deploy.loadwsgi import NicerConfigParser
from paste.deploy.converters import asint
from mediagoblin import storage, models
from mediagoblin.celery_setup import setup_celery_from_config
@ -67,8 +68,11 @@ def setup_self(setup_globals_func=setup_globals):
settings_module=OUR_MODULENAME,
set_environ=False)
port = mgoblin_section.get('db_port')
if port:
port = asint(port)
connection = mongokit.Connection(
mgoblin_section.get('db_host'), mgoblin_section.get('db_port'))
mgoblin_section.get('db_host'), port)
db = connection[mgoblin_section.get('db_name', 'mediagoblin')]
models.register_models(connection)