Initial code drop

koji/context.py

@@ -0,0 +1,109 @@
+#!/usr/bin/python
+# Copyright (c) 2005-2007 Red Hat
+#
+#    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
+
+# 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