Method reserve

hyperopt/mongoexp.py:483–522

(self, host_id, cond=None, exp_key=None)

        return self.delete_all(cond={"state": JOB_STATE_ERROR})

    def reserve(self, host_id, cond=None, exp_key=None):
        now = coarse_utcnow()
        if cond is None:
            cond = {}
        else:
            cond = copy.copy(
                cond
            )  # copy is important, will be modified, but only the top-level

        if exp_key is not None:
            cond["exp_key"] = exp_key

        # having an owner of None implies state==JOB_STATE_NEW, so this effectively
        # acts as a filter to make sure that only new jobs get reserved.
        if cond.get("owner") is not None:
            raise ValueError("refusing to reserve owned job")
        else:
            cond["owner"] = None
            cond[
                "state"
            ] = JOB_STATE_NEW  # theoretically this is redundant, theoretically

        try:
            rval = self.jobs.find_and_modify(
                cond,
                {
                    "$set": {
                        "owner": host_id,
                        "book_time": now,
                        "state": JOB_STATE_RUNNING,
                        "refresh_time": now,
                    }
                },
                new=True,
                upsert=False,
            )
        except pymongo.errors.OperationFailure as e:
            logger.error("Error during reserve_job: %s" % str(e))
            rval = None
        return rval

    def refresh(self, doc):
        self.update(doc, dict(refresh_time=coarse_utcnow()))