2008-12-18 15:01:59 -06:00
|
|
|
# Authors:
|
|
|
|
# Rob Crittenden <rcritten@redhat.com>
|
|
|
|
# Jason Gerard DeRose <jderose@redhat.com>
|
|
|
|
#
|
|
|
|
# Copyright (C) 2008 Red Hat
|
|
|
|
# see file 'COPYING' for use and warranty contextrmation
|
|
|
|
#
|
2010-12-09 06:59:11 -06:00
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
|
# (at your option) any later version.
|
2008-12-18 15:01:59 -06:00
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
2010-12-09 06:59:11 -06:00
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
2008-12-18 15:01:59 -06:00
|
|
|
|
|
|
|
"""
|
|
|
|
Per-request thread-local data.
|
|
|
|
"""
|
|
|
|
|
|
|
|
import threading
|
2015-07-31 03:15:01 -05:00
|
|
|
|
|
|
|
from ipalib.base import ReadOnly, lock
|
|
|
|
from ipalib.constants import OVERRIDE_ERROR, CALLABLE_ERROR
|
2008-12-18 15:01:59 -06:00
|
|
|
|
|
|
|
|
2008-12-18 17:58:48 -06:00
|
|
|
# Thread-local storage of most per-request information
|
2008-12-18 15:01:59 -06:00
|
|
|
context = threading.local()
|
|
|
|
|
|
|
|
|
2009-01-23 16:49:16 -06:00
|
|
|
class Connection(ReadOnly):
|
|
|
|
"""
|
|
|
|
Base class for connection objects stored on `request.context`.
|
|
|
|
"""
|
|
|
|
|
2009-01-23 19:02:32 -06:00
|
|
|
def __init__(self, conn, disconnect):
|
|
|
|
self.conn = conn
|
|
|
|
if not callable(disconnect):
|
|
|
|
raise TypeError(
|
|
|
|
CALLABLE_ERROR % ('disconnect', disconnect, type(disconnect))
|
|
|
|
)
|
|
|
|
self.disconnect = disconnect
|
2009-01-23 16:49:16 -06:00
|
|
|
lock(self)
|
|
|
|
|
|
|
|
|
|
|
|
def destroy_context():
|
|
|
|
"""
|
|
|
|
Delete all attributes on thread-local `request.context`.
|
|
|
|
"""
|
Use Python3-compatible dict method names
Python 2 has keys()/values()/items(), which return lists,
iterkeys()/itervalues()/iteritems(), which return iterators,
and viewkeys()/viewvalues()/viewitems() which return views.
Python 3 has only keys()/values()/items(), which return views.
To get iterators, one can use iter() or a for loop/comprehension;
for lists there's the list() constructor.
When iterating through the entire dict, without modifying the dict,
the difference between Python 2's items() and iteritems() is
negligible, especially on small dicts (the main overhead is
extra memory, not CPU time). In the interest of simpler code,
this patch changes many instances of iteritems() to items(),
iterkeys() to keys() etc.
In other cases, helpers like six.itervalues are used.
Reviewed-By: Christian Heimes <cheimes@redhat.com>
Reviewed-By: Jan Cholasta <jcholast@redhat.com>
2015-08-11 06:51:14 -05:00
|
|
|
# need to use a list of values, 'cos value.disconnect modifies the dict
|
|
|
|
for value in list(context.__dict__.values()):
|
2009-01-23 16:49:16 -06:00
|
|
|
if isinstance(value, Connection):
|
2009-01-23 19:02:32 -06:00
|
|
|
value.disconnect()
|
2010-03-08 21:42:26 -06:00
|
|
|
context.__dict__.clear()
|
2009-01-23 16:49:16 -06:00
|
|
|
|