debian-koji/koji/context.py
Pavol Babincak 9e89ebb9a4 Ran reindent.py from cpython tools on all python scripts
Citing from reindent docs:

    Change Python (.py) files to use 4-space indents and no hard tab
    characters. Also trim excess spaces and tabs from ends of lines, and
    remove empty lines at the end of files.  Also ensure the last line
    ends with a newline.

Citing from PEP 8:

    Use 4 spaces per indentation level.

    Python 2 code indented with a mixture of tabs and spaces should be
    converted to using spaces exclusively.

    Don't write string literals that rely on significant trailing
    whitespace. Such trailing whitespace is visually indistinguishable
    and some editors (or more recently, reindent.py) will trim them.

Also PyLint recommends not to have trailing whitespace on any line.
2015-05-13 16:39:53 -04:00

112 lines
3.1 KiB
Python
Executable file

#!/usr/bin/python
# Copyright (c) 2005-2014 Red Hat, Inc.
#
# Koji is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation;
# version 2.1 of the License.
#
# This software 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this software; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#
# Authors:
# Mike McLean <mikem@redhat.com>
# This modules provides a thread-safe way of passing
# request context around in a global way
# - db connections
# - request data
# - auth data
import thread
class _data(object):
pass
class ThreadLocal(object):
def __init__(self):
object.__setattr__(self, '_tdict', {})
# should probably be getattribute, but easier to debug this way
def __getattr__(self, key):
id = thread.get_ident()
tdict = object.__getattribute__(self, '_tdict')
if not tdict.has_key(id):
raise AttributeError(key)
data = tdict[id]
return object.__getattribute__(data, key)
def __setattr__(self, key, value):
id = thread.get_ident()
tdict = object.__getattribute__(self, '_tdict')
if not tdict.has_key(id):
tdict[id] = _data()
data = tdict[id]
return object.__setattr__(data,key,value)
def __delattr__(self, key):
id = thread.get_ident()
tdict = object.__getattribute__(self, '_tdict')
if not tdict.has_key(id):
raise AttributeError(key)
data = tdict[id]
ret = object.__delattr__(data, key)
if len(data.__dict__) == 0:
del tdict[id]
return ret
def __str__(self):
id = thread.get_ident()
tdict = object.__getattribute__(self, '_tdict')
return "(current thread: %s) {" % id + \
", ".join([ "%s : %s" %(k,v.__dict__) for (k,v) in tdict.iteritems() ]) + \
"}"
def _threadclear(self):
id = thread.get_ident()
tdict = object.__getattribute__(self, '_tdict')
if not tdict.has_key(id):
return
del tdict[id]
context = ThreadLocal()
if __name__ == '__main__':
#testing
#context.foo = 1
#context.bar = 2
print context
#del context.bar
print context
import random
import time
def test():
context.foo=random.random()
time.sleep(1.5+random.random())
context._threadclear()
print context
for x in xrange(1,10):
thread.start_new_thread(test,())
time.sleep(4)
print
print context
context.foo = 1
context.bar = 2
print context.foo,context.bar
print context
context._threadclear()
print context