This commit is contained in:
Benoit Masson 2020-10-03 17:28:15 +02:00 committed by GitHub
commit c18cdb74ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 11 additions and 4 deletions

View File

@ -88,9 +88,11 @@ class SQLitePlugin(object):
if bottle.__version__.startswith('0.9'):
config = route['config']
_callback = route['callback']
argspec = inspect.getargspec(_callback).args
else:
config = route.config
_callback = route.callback
argspec = route.get_callback_args()
# Override global configuration with route-specific values.
if "sqlite" in config:
@ -109,10 +111,7 @@ class SQLitePlugin(object):
collations = g('collations', self.collations)
extensions = g('extensions', self.extensions)
# Test if the original callback accepts a 'db' keyword.
# Ignore it if it does not need a database handle.
argspec = inspect.getargspec(_callback)
if keyword not in argspec.args:
if keyword not in argspec:
return callback
def wrapper(*args, **kwargs):

View File

@ -28,6 +28,13 @@ class SQLiteTest(unittest.TestCase):
def tearDown(self):
os.unlink(self.plugin.dbfile)
def test_with_view(self):
@self.app.get('/')
@bottle.view('test_view')
def test(db):
self.assertEqual(type(db), type(sqlite3.connect(':memory:')))
self._request('/')
def test_with_keyword(self):
@self.app.get('/')
def test(db):

1
views/test_view.tpl Normal file
View File

@ -0,0 +1 @@
test_view