From c2bea64ca69cf964486e3bc22ee60cfd9c5f5bfa Mon Sep 17 00:00:00 2001
From: Jerry <isjerryxiao@outlook.com>
Date: Tue, 15 Jan 2019 15:25:55 +0800
Subject: [PATCH] change reward mechanism

---
 tgmsbot.py | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

diff --git a/tgmsbot.py b/tgmsbot.py
index e30fee8..a9ff044 100644
--- a/tgmsbot.py
+++ b/tgmsbot.py
@@ -199,25 +199,37 @@ def gen_reward(user, negative=True):
     # Negative rewards
     def restrict_mining(player):
         if player.immunity_cards >= 1:
-            player.immunity_cards -= 1
-            ret = "用去一张免疫卡，还剩{}张".format(player.immunity_cards)
+            if player.immunity_cards >= 10:
+                lost_cards = randint(2,4)
+            elif player.immunity_cards >= 5:
+                lost_cards = randint(1,3)
+            else:
+                lost_cards = 1
+            player.immunity_cards -= lost_cards
+            ret = "用去{}张免疫卡，还剩{}张".format(lost_cards, player.immunity_cards)
         else:
             now = int(time.time())
             seconds = randint(30, 120)
             player.restricted_until = now + seconds
-            ret = "被限制扫雷{}秒".format(seconds)
+            ret = "没有免疫卡了，被限制扫雷{}秒".format(seconds)
         player.save()
         return ret
     # Positive rewards
     def give_immunity_cards(player):
-        if player.immunity_cards >= 3 and choice((True, False)):
-            action = "没收"
-            player.immunity_cards -= 1
-        else:
-            action = "奖励"
-            player.immunity_cards += 1
+        rewarded_cards = 0
+        if player.immunity_cards <= 3:
+            rewarded_cards = randint(1, 2)
+        elif player.immunity_cards <= 10:
+            if randint(1, 5) == 5:
+                rewarded_cards = 1
+        elif randint(1, 10) == 10:
+            rewarded_cards = 1
+        player.immunity_cards += rewarded_cards
         player.save()
-        return "被{}了一张免疫卡，共有{}张".format(action, player.immunity_cards)
+        if rewarded_cards == 0:
+            return "共有{}张免疫卡".format(player.immunity_cards)
+        else:
+            return "被奖励了{}张免疫卡，共有{}张".format(rewarded_cards, player.immunity_cards)
 
     player = get_player(user.id)
     if negative: