def get_state(self, payment_info): """ Return state of a payment for given task that should be / was made to given node :return str|None: return state of payment or none if such payment don't exist in database """ # FIXME: Remove this method try: return Payment.get( Payment.subtask == payment_info.subtask_id).status except Payment.DoesNotExist: logger.warning( "Payment for subtask {} to node {} does not exist".format( payment_info.subtask_id, payment_info.computer.key_id)) return None
def get_payment_for_subtask(subtask_id): try: return Payment.get(Payment.subtask == subtask_id).value except Payment.DoesNotExist: logger.debug("Can't get payment value - payment does not exist") return 0