earthson / afewords_base Goto Github PK

def user_to_dict(user):
    '''
    user info to dict:
        {
            '_id': 'a241212dsf', # unicode, user id
            'name': 'afewords', # unicode, user name
            'draft': '', # int, user draft's num
            'notice': '',  # int, user notification's num
            'thumb': '', # unicode, user thumb

        }
    '''
    if user is None:
        return None
    ret = dict()
    ret['_id'] = unicode(user._id)
    ret['name'] = user.name
    ret['thumb'] = user.avatar.tbumb_name
    ret['draft'] = len(user.draft_lib.load_all())
    ret['notice'] = len([i for i in user.notification_list.load_all() if i is not None and i[1] == False])
    return ret

to_load = ['id', 'name', 'draft_count', 'notice_count', 'thumb_name']
return dict(zip(to_load, usr.get_propertys(*to_load)))

def attr_gen(obj, *keys):
    return dict(zip(keys, obj.get_propertys(*keys)))