parameterize Source class used in activitystreams.Handler

snarfed · Jul 22, 2012 · d19eb62 · d19eb62
1 parent e4ea68b
commit d19eb62
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/activitystreams.py b/activitystreams.py
@@ -70,7 +70,11 @@ class Handler(webapp2.RequestHandler):
 
   def __init__(self, *args, **kwargs):
     super(Handler, self).__init__(*args, **kwargs)
-    self.source = SOURCE(self)
+    self.source = self.source_class()(self)
+
+  def source_class(self):
+    """Return the Source subclass to use. May be overridden by subclasses."""
+    return SOURCE
 
   def get(self):
     """Handles an API GET.