def increase_negative_resource(node_id, trust_mod): try: with db.transaction(): LocalRank.create(node_id=node_id, negative_resource=trust_mod) except IntegrityError: LocalRank.update(negative_resource=LocalRank.negative_resource + trust_mod, modified_date=str(datetime.datetime.now())) \ .where(LocalRank.node_id == node_id).execute()
def increase_positive_payment(node_id, trust_mod): try: with db.transaction(): LocalRank.create(node_id=node_id, positive_payment=trust_mod) except IntegrityError: LocalRank.update(positive_payment=LocalRank.positive_payment + trust_mod, modified_date=str(datetime.datetime.now())) \ .where(LocalRank.node_id == node_id).execute()
def increase_wrong_computed(node_id, trust_mod): try: with db.transaction(): LocalRank.create(node_id=node_id, wrong_computed=trust_mod) except IntegrityError: LocalRank.update(wrong_computed=LocalRank.wrong_computed + trust_mod, modified_date=str(datetime.datetime.now())) \ .where(LocalRank.node_id == node_id).execute()
def test_default_fields(self): r = LocalRank() self.assertGreaterEqual(datetime.now(), r.created_date) self.assertGreaterEqual(datetime.now(), r.modified_date) self.assertEqual(0, r.positive_computed) self.assertEqual(0, r.negative_computed) self.assertEqual(0, r.wrong_computed) self.assertEqual(0, r.positive_requested) self.assertEqual(0, r.negative_requested) self.assertEqual(0, r.positive_payment) self.assertEqual(0, r.negative_payment) self.assertEqual(0, r.positive_resource) self.assertEqual(0, r.negative_resource)
def get_local_rank_for_all(): return LocalRank.select()
def get_local_rank(node_id): return LocalRank.select().where(LocalRank.node_id == node_id).first()