Dictionary filtering

utils.py

#!/bin/env python


# keys is a list of key
def filter_keys(keys, object):
  return dict((x,y) for (x, y) in object.iteritems() if x in keys)

# or with keys being one or more keys
def filter_keys2(object, *keys):
  return dict((x,y) for (x, y) in object.iteritems() if x in keys)

# or with keys being a dictionnary (from another object maybe)
def filter_keys3(object, keys):
  return dict((x,y) for (x, y) in object.iteritems() if x in keys.keys())



user_obj = dict( {'name':'matt lohier', 'first':'matt', 'last':'lohier', 'bsb':'12321', 'account':'4565'} )

print filter_keys(['first','last'], user_obj)

print filter_keys2(user_obj, 'first', 'last', 'more_attribute')

print filter_keys2(user_obj, *['first', 'last'])

some_other_dict = dict( {'first':'whocares', 'last':'whocares'} )
print filter_keys3(user_obj, some_other_dict)