X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=PLC%2FMessages.py;h=250ff12fe1ebec8de761b25cb960f23cadd72c52;hb=e43eb135483c45435c9d4a50d6dff744fb7aef85;hp=11b7433f9c6fa266fba1ddb11af83c5f0afcc293;hpb=e6f1cfc3e60c97f756858da04067e060d5d491d1;p=plcapi.git

diff --git a/PLC/Messages.py b/PLC/Messages.py
index 11b7433..250ff12 100644
--- a/PLC/Messages.py
+++ b/PLC/Messages.py
@@ -4,11 +4,12 @@
 # Tony Mack <tmack@cs.princeton.edu>
 # Copyright (C) 2006 The Trustees of Princeton University
 #
-# $Id: Messages.py,v 1.4 2006/10/31 21:46:14 mlhuang Exp $
+# $Id: Messages.py,v 1.4 2006/11/16 17:03:36 mlhuang Exp $
 #
 
 from PLC.Parameter import Parameter
 from PLC.Table import Row, Table
+from PLC.Filter import Filter
 
 class Message(Row):
     """
@@ -19,6 +20,7 @@ class Message(Row):
     primary_key = 'message_id'
     fields = {
         'message_id': Parameter(str, "Message identifier"),
+        'subject': Parameter(str, "Message summary", nullok = True),
         'template': Parameter(str, "Message template", nullok = True),
         'enabled': Parameter(bool, "Message is enabled"),
         }
@@ -28,15 +30,21 @@ class Messages(Table):
     Representation of row(s) from the messages table in the database. 
     """
 
-    def __init__(self, api, message_ids, enabled = None):
-        self.api = api
+    def __init__(self, api, message_filter = None, columns = None, enabled = None):
+        Table.__init__(self, api, Message, columns)
     
-        sql = "SELECT %s from messages" % ", ".join(Message.fields)
+        sql = "SELECT %s from messages WHERE True" % \
+              ", ".join(self.columns)
 
         if enabled is not None:
-            sql += " WHERE enabled IS %(enabled)s"
+            sql += " AND enabled IS %s" % enabled
 
-        rows = self.api.db.selectall(sql, locals())
+        if message_filter is not None:
+            if isinstance(message_filter, (list, tuple, set)):
+                message_filter = Filter(Message.fields, {'message_id': message_filter})
+                sql += " AND (%s)" % message_filter.sql(api, "OR")
+            elif isinstance(message_filter, dict):
+                message_filter = Filter(Message.fields, message_filter)
+                sql += " AND (%s)" % message_filter.sql(api, "AND")
 
-        for row in rows:
-            self[row['message_id']] = Message(api, row)
+        self.selectall(sql)