mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-12 11:02:21 -05:00
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
|
# engine/util.py
|
||
|
|
# Copyright (C) 2005-2014 the SQLAlchemy authors and contributors <see AUTHORS file>
|
||
|
|
#
|
||
|
|
# This module is part of SQLAlchemy and is released under
|
||
|
|
# the MIT License: http://www.opensource.org/licenses/mit-license.php
|
||
|
|
|
||
|
|
from .. import util
|
||
|
|
|
||
|
|
def connection_memoize(key):
|
||
|
|
"""Decorator, memoize a function in a connection.info stash.
|
||
|
|
|
||
|
|
Only applicable to functions which take no arguments other than a
|
||
|
|
connection. The memo will be stored in ``connection.info[key]``.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@util.decorator
|
||
|
|
def decorated(fn, self, connection):
|
||
|
|
connection = connection.connect()
|
||
|
|
try:
|
||
|
|
return connection.info[key]
|
||
|
|
except KeyError:
|
||
|
|
connection.info[key] = val = fn(self, connection)
|
||
|
|
return val
|
||
|
|
|
||
|
|
return decorated
|
||
|
|
|
||
|
|
|
||
|
|
def py_fallback():
|
||
|
|
def _distill_params(multiparams, params):
|
||
|
|
"""Given arguments from the calling form *multiparams, **params,
|
||
|
|
return a list of bind parameter structures, usually a list of
|
||
|
|
dictionaries.
|
||
|
|
|
||
|
|
In the case of 'raw' execution which accepts positional parameters,
|
||
|
|
it may be a list of tuples or lists.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
if not multiparams:
|
||
|
|
if params:
|
||
|
|
return [params]
|
||
|
|
else:
|
||
|
|
return []
|
||
|
|
elif len(multiparams) == 1:
|
||
|
|
zero = multiparams[0]
|
||
|
|
if isinstance(zero, (list, tuple)):
|
||
|
|
if not zero or hasattr(zero[0], '__iter__') and \
|
||
|
|
not hasattr(zero[0], 'strip'):
|
||
|
|
# execute(stmt, [{}, {}, {}, ...])
|
||
|
|
# execute(stmt, [(), (), (), ...])
|
||
|
|
return zero
|
||
|
|
else:
|
||
|
|
# execute(stmt, ("value", "value"))
|
||
|
|
return [zero]
|
||
|
|
elif hasattr(zero, 'keys'):
|
||
|
|
# execute(stmt, {"key":"value"})
|
||
|
|
return [zero]
|
||
|
|
else:
|
||
|
|
# execute(stmt, "value")
|
||
|
|
return [[zero]]
|
||
|
|
else:
|
||
|
|
if hasattr(multiparams[0], '__iter__') and \
|
||
|
|
not hasattr(multiparams[0], 'strip'):
|
||
|
|
return multiparams
|
||
|
|
else:
|
||
|
|
return [multiparams]
|
||
|
|
|
||
|
|
return locals()
|
||
|
|
try:
|
||
|
|
from sqlalchemy.cutils import _distill_params
|
||
|
|
except ImportError:
|
||
|
|
globals().update(py_fallback())
|