# HG changeset patch
# User Todd Larsen <tlarsen@google.com>
# Date 1232487450 0
# Node ID 05e376c84e083ace352ee33004198c01ea0962fa
# Parent  c193ac0ef59337b1442a033120170522a7aafaa2
Return the ToS Document of the supplied Presence entity, if there is one.

Patch by: Todd Larsen
Review by: to-be-reviewed

diff -r c193ac0ef593 -r 05e376c84e08 app/soc/logic/models/presence.py
--- a/app/soc/logic/models/presence.py	Tue Jan 20 21:26:34 2009 +0000
+++ b/app/soc/logic/models/presence.py	Tue Jan 20 21:37:30 2009 +0000
@@ -41,5 +41,22 @@
     super(Logic, self).__init__(model, base_model=base_model,
                                 scope_logic=scope_logic)
 
+  def getToS(self, entity):
+    """Returns the ToS Document of the Presence entity, or None if no ToS.
+
+    Args:
+      entity:  Presence (or one of its sub-classes) entity that may or may
+        not have a ToS Document attached
+    """
+    if not entity:
+      return None
+
+    try:
+      tos_doc = entity.tos
+    except db.Error:
+      return None
+
+    return tos_doc
+
 
 logic = Logic()