Be consistent in the API.

[plcapi.git] / PLC / Leases.py

diff --git a/PLC/Leases.py b/PLC/Leases.py
index 4b967c6..7211516 100644 (file)
--- a/PLC/Leases.py
+++ b/PLC/Leases.py
@@ -80,10 +80,7 @@ class Leases(Table):
 
 
         if lease_filter is not None:
-
-            if isinstance (lease_filter, int):
-                lease_filter = Filter (Lease.fields, {'lease_id': lease_filter})
-            elif isinstance(lease_filter, (list, tuple, set)):
+            if isinstance(lease_filter, (list, tuple, set, int, long)):
                 lease_filter = Filter(Lease.fields, {'lease_id': lease_filter})
             elif isinstance(lease_filter, dict):
                 lease_filter = LeaseFilter(Lease.fields, lease_filter)