Exemplo n.º 1
0
    def remove(self, item):
        item = map_item(item)
        bots = mongo.db.bots

        class_id = item.get('class_id')
        instance_id = item.get('instance_id')
        amount = int(item['amount'])
        if amount <= 0:
            return

        query = {
            'class_id': class_id,
            'instance_id': instance_id,
        }
        bots.update(
            {
                '_id': self.bot['_id'],
                'inventory': {'$elemMatch': query},
            },
            {'$inc': {'inventory.$.amount': -amount}},
        )

        query['amount'] = {'$lte': 0}
        bots.update(
            {'_id': self.bot['_id']},
            {'$pull': {'inventory': query}},
        )
Exemplo n.º 2
0
def user_inventory(_id):
    if request.method == 'GET':
        inventory = (mongo.db.users.find_one(
            {'_id': _id}, {'_id': 0, 'inventory': 1}
        ) or {}).get('inventory', {})
        return fill_items(inventory)
    elif request.json:
        inventory = request.json.get('inventory', {})
        if inventory:
            items = []
            for app, ctx, item, desc in flatten_inventory(inventory):
                item.update({
                    'app_id': app,
                    'context_id': ctx,
                })
                items.append(map_item(item))

            mongo.db.users.update(
                {'_id': _id},
                {'$set': {
                    'inventory': items,
                    'times.inventory': time.time(),
                }},
            )
            insert_items(inventory, foreign=True)

        return {}
Exemplo n.º 3
0
    def add(self, item, sync=False):
        item = map_item(item)
        bots = mongo.db.bots

        app_id = item.get('app_id')
        context_id = item.get('context_id')
        class_id = item.get('class_id')
        instance_id = item.get('instance_id')
        amount = int(item['amount'])

        new = {
            'app_id': app_id,
            'context_id': context_id,
            'class_id': class_id,
            'instance_id': instance_id,
            'amount': 0,
        }

        query = {
            'class_id': class_id,
            'instance_id': instance_id,
        }

        bots.update(
            {
                '_id': self.bot['_id'],
                'inventory': {'$not': {'$elemMatch': query}},
            },
            {'$push': {'inventory': new}},
        )

        if sync:
            mod = '$set'
        else:
            mod = '$inc'

        old = bots.find_and_modify(
            {
                '_id': self.bot['_id'],
                'inventory': {'$elemMatch': query},
            },
            {mod: {'inventory.$.amount': amount}},
            fields={'inventory': 1},
        )

        if sync:
            # boo, can't properly filter the results of the find_and_modify
            # so we'll do it here
            if old:
                old_amount = 0
                key = (class_id, instance_id)
                for i in old['inventory']:
                     if (i['class_id'], i['instance_id']) == key:
                        old_amount = int(i['amount'])
                        break

                item['amount'] -= old_amount

        return item