X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=PLC%2FMessages.py;h=898047eddd584110db9459ee2b0b193eb5c73257;hb=fce6d4f0d14af7bd41d534d942b6f6e293b89b24;hp=4c7299e187e19e0fb944ff43459cfd49c901ef25;hpb=f5cd72e35593c96b9996ab4e49674cccc7525c48;p=plcapi.git

diff --git a/PLC/Messages.py b/PLC/Messages.py
index 4c7299e..898047e 100644
--- a/PLC/Messages.py
+++ b/PLC/Messages.py
@@ -14,9 +14,9 @@ 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 = {
@@ -25,15 +25,15 @@ class Message(Row):
         '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)
 
@@ -41,11 +41,13 @@ class Messages(Table):
             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, long)):
                 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)