# Tony Mack <tmack@cs.princeton.edu>
# Copyright (C) 2006 The Trustees of Princeton University
#
-# $Id$
-# $URL$
-#
from PLC.Parameter import Parameter
from PLC.Table import Row, Table
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 = {
'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_filter = None, columns = None, enabled = None):
Table.__init__(self, api, Message, columns)
-
+
sql = "SELECT %s from messages WHERE True" % \
", ".join(self.columns)
sql += " AND enabled IS %s" % enabled
if message_filter is not None:
- if isinstance(message_filter, (list, tuple, set)):
+ 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)
self.selectall(sql)