| author | Sandro Knauß <knauss@netzguerilla.net> |
| Mon, 13 Feb 2012 19:50:01 +0100 | |
| branch | devel |
| changeset 151 | ff8448fb07e7 |
| parent 123 | 2d685c3c10b2 |
| child 155 | ff1edf7c1329 |
| permissions | -rw-r--r-- |
| 117 | 1 |
from decorator import FunctionMaker |
|
151
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
2 |
import sqlalchemy |
| 117 | 3 |
|
4 |
from .pool import runInDBPool |
|
5 |
from .utils import WithSession |
|
6 |
||
7 |
import inspect |
|
8 |
||
9 |
class DBDefer(object): |
|
10 |
'''a twisted sqlalchemy connector this Decorator adds a session parameter, with a valid session connection''' |
|
11 |
def __init__(self, engine, autocommit=False): |
|
12 |
self.autocommit=autocommit |
|
13 |
self.engine = engine |
|
14 |
||
15 |
def __call__(self, func): |
|
16 |
@runInDBPool |
|
|
123
2d685c3c10b2
dbdefer now can handle session argument on different positions
Sandro Knauß <knauss@netzguerilla.net>
parents:
117
diff
changeset
|
17 |
def wrapper(func,*a, **kw): |
|
151
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
18 |
i = argspec.args.index("session") |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
19 |
ab = a[:i] |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
20 |
ae = a[i:-1] |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
21 |
if isinstance(a[-1],sqlalchemy.orm.session.Session): |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
22 |
al = ab + (a[-1],) + ae |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
23 |
ret = func(*al, **kw) |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
24 |
if self.autocommit: |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
25 |
a[-1].commit() |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
26 |
return ret |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
27 |
else: |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
28 |
with WithSession(self.engine, self.autocommit) as session: |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
29 |
al = ab + (session,) + ae |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
30 |
return func(*al, **kw) |
| 117 | 31 |
|
32 |
caller=func |
|
|
123
2d685c3c10b2
dbdefer now can handle session argument on different positions
Sandro Knauß <knauss@netzguerilla.net>
parents:
117
diff
changeset
|
33 |
argspec = inspect.getargspec(caller) |
|
2d685c3c10b2
dbdefer now can handle session argument on different positions
Sandro Knauß <knauss@netzguerilla.net>
parents:
117
diff
changeset
|
34 |
args =[i for i in argspec.args if i != "session" ] |
| 117 | 35 |
sargs=", ".join(args) |
|
151
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
36 |
sargs+=", session" |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
37 |
defaults = argspec.defaults |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
38 |
if not defaults: |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
39 |
defaults = (None,) |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
40 |
else: |
|
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
41 |
defaults += (None,) |
| 117 | 42 |
evaldict = caller.func_globals.copy() |
43 |
evaldict['_call_'] = func |
|
44 |
evaldict['decorator'] = wrapper |
|
45 |
wrap = FunctionMaker.create( |
|
46 |
'%s(%s)' % (caller.__name__, sargs), |
|
47 |
'return decorator(_call_, %s)' % sargs, |
|
|
151
ff8448fb07e7
runInDBThread and dbdefer can now handle if a valid session is an arg
Sandro Knauß <knauss@netzguerilla.net>
parents:
123
diff
changeset
|
48 |
evaldict, defaults=defaults, undecorated=caller, __wrapped__=caller, |
| 117 | 49 |
doc=caller.__doc__, module=caller.__module__, addsource=True) |
50 |
return wrap |
|
51 |
||
52 |
dbdefer=DBDefer(None) |
|
53 |
||
54 |
def setEngine(engine,autocommit=False): |
|
55 |
dbdefer.engine = engine |
|
56 |
dbdefer.autocommit = autocommit |
|
57 |
||
58 |