# 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 $
-#
from PLC.Parameter import Parameter
from PLC.Table import Row, Table
+from PLC.Filter import Filter
class Message(Row):
"""
- Representation of a row in the messages table.
+ Representation of a row in the messages table.
"""
-
+
table_name = 'messages'
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"),
}
-
+
class Messages(Table):
"""
- Representation of row(s) from the messages table in the database.
+ Representation of row(s) from the messages table in the database.
"""
- def __init__(self, api, message_ids, enabled = None):
- self.api = api
-
- sql = "SELECT %s from messages" % ", ".join(Message.fields)
+ def __init__(self, api, message_filter = None, columns = None, enabled = None):
+ Table.__init__(self, api, Message, columns)
+
+ 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, int)):
+ message_filter = Filter(Message.fields, {'message_id': message_filter})
+ sql += " AND (%s) %s" % message_filter.sql(api, "OR")
+ elif isinstance(message_filter, dict):
+ message_filter = Filter(Message.fields, message_filter)
+ sql += " AND (%s) %s" % message_filter.sql(api, "AND")
+ else:
+ raise PLCInvalidArgument("Wrong message filter %r"%message_filter)
- for row in rows:
- self[row['message_id']] = Message(api, row)
+ self.selectall(sql)