From 1185f7b8b51da1e217d32b9018185ea5b7e3eab1 Mon Sep 17 00:00:00 2001
From: "Sunil Sharma (OpenERP)" <sunilsharma.sharma07@gmail.com>
Date: Fri, 21 Mar 2014 11:16:42 +0530
Subject: [PATCH] [imp]:improve gamification challenge line category forum and
 add clss gamification challenge assign reward user

bzr revid: sunilsharma.sharma07@gmail.com-20140321054642-bjbgm4e92tur9f83
---
 .../website_forum/data/forum_badges_data.xml  | 68 +++++++++----------
 addons/website_forum/models/website_forum.py  | 33 +++++++++
 2 files changed, 67 insertions(+), 34 deletions(-)

diff --git a/addons/website_forum/data/forum_badges_data.xml b/addons/website_forum/data/forum_badges_data.xml
index a51dbd267ea..0bd852eac87 100644
--- a/addons/website_forum/data/forum_badges_data.xml
+++ b/addons/website_forum/data/forum_badges_data.xml
@@ -32,7 +32,7 @@
             <field name="reward_id" eval="ref('badge_1')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_configure_profile">
@@ -65,7 +65,7 @@
             <field name="reward_id" eval="ref('badge_3')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_cleanup">
@@ -98,7 +98,7 @@
             <field name="reward_id" eval="ref('badge_4')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_commentator">
@@ -131,7 +131,7 @@
             <field name="reward_id" eval="ref('badge_5')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_critic">
@@ -167,7 +167,7 @@
             <field name="reward_id" eval="ref('badge_6')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_disciplined">
@@ -200,7 +200,7 @@
             <field name="reward_id" eval="ref('badge_7')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_editor">
@@ -233,7 +233,7 @@
             <field name="reward_id" eval="ref('badge_8')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_enlightened">
@@ -275,7 +275,7 @@ result = get_counter(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_9')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_enthusiast">
@@ -317,7 +317,7 @@ result = get_counter(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_10')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_expert">
@@ -350,7 +350,7 @@ result = get_counter(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_11')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_famous_question">
@@ -365,7 +365,7 @@ result = get_counter(cr, uid, context=context)
             <field name="level">silver</field>
         </record>
 
-        <!--NOTE: passing default value 1 for res otherwise it gives error in gamification, will improve in gamification -->
+        <!--NOTE: passing default value 1 for res forumwise it gives error in gamification, will improve in gamification -->
         <record model="gamification.goal.definition" id="definition_favorite_question">
             <field name="name">Favorite Question</field>
             <field name="description">Ask a question which can be favorited by minimum 5 users</field>
@@ -392,7 +392,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_12')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_favorite_question">
@@ -425,7 +425,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_13')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_good_answer">
@@ -458,7 +458,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_14')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_good_question">
@@ -491,7 +491,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_15')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_great_answer">
@@ -524,7 +524,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_16')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_great_question">
@@ -557,7 +557,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_17')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_guru">
@@ -601,7 +601,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_18')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_necromancer">
@@ -634,7 +634,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_19')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_nice_answer">
@@ -667,7 +667,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_20')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_nice_question">
@@ -700,7 +700,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_21')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_notable_question">
@@ -733,7 +733,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_22')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_notable_question">
@@ -769,7 +769,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_23')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_peer_pressure">
@@ -802,7 +802,7 @@ result = get_posts(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_24')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_popular_question">
@@ -847,7 +847,7 @@ result = get_count(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_25')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_pundit">
@@ -881,7 +881,7 @@ result = get_count(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_26')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_scholar">
@@ -918,7 +918,7 @@ result = get_count(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_27')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_self_learner">
@@ -933,7 +933,7 @@ result = get_count(cr, uid, context=context)
             <field name="level">bronze</field>
         </record>
 
-        <!--NOTE: passing default value 1 for res otherwise it gives error in gamification, will improve in gamification -->
+        <!--NOTE: passing default value 1 for res forumwise it gives error in gamification, will improve in gamification -->
         <record model="gamification.goal.definition" id="definition_stellar_question">
             <field name="name">Stellar Question</field>
             <field name="description">Ask a question which can be favorited by minimum 25 users</field>
@@ -960,7 +960,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_28')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_stellar_question">
@@ -993,7 +993,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_29')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_associate_editor">
@@ -1026,7 +1026,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_30')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_student">
@@ -1059,7 +1059,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_31')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_supporter">
@@ -1092,7 +1092,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_32')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_taxonomist">
@@ -1125,7 +1125,7 @@ result = count_favorites(cr, uid, context=context)
             <field name="reward_id" eval="ref('badge_33')" />
             <field name="autojoin_group_id" eval="ref('base.group_user')" />
             <field name="state">inprogress</field>
-            <field name="category">other</field>
+            <field name="category">forum</field>
         </record>
 
         <record model="gamification.challenge.line" id="line_teacher">
diff --git a/addons/website_forum/models/website_forum.py b/addons/website_forum/models/website_forum.py
index 683eb0902d7..db14cd32a7e 100644
--- a/addons/website_forum/models/website_forum.py
+++ b/addons/website_forum/models/website_forum.py
@@ -353,3 +353,36 @@ class Tags(osv.Model):
             }
         ),
    }
+
+class gamification_challenge(osv.Model):
+    _inherit = 'gamification.challenge'
+
+    def _get_categories(self, cr, uid, context=None):
+        res = super(gamification_challenge, self)._get_categories(cr, uid,
+                context=context)
+        res.append(('forum', 'Website / Q.A'))
+        return res
+
+    _columns = {
+        'category': fields.selection(lambda s, *a, **k: s._get_categories(*a, **k),
+            string="Appears in", help="Define the visibility of the challenge through menus", required=True),
+    }
+    
+    def check_challenge_reward(self, cr, uid, ids, force=False, context=None):
+        if isinstance(ids, (int,long)):
+            ids = [ids]
+        context = context or {}
+        badge_user_obj = self.pool.get('gamification.badge.user');
+        for challenge in self.browse(cr, uid, ids, context=context):
+            if challenge.category == 'forum':
+                for user in challenge.user_ids:
+                    bages_user_ids = badge_user_obj.search(cr, uid, [('user_id', '=', user.id),('badge_id', '=', challenge.reward_id.id)],context=None)
+                    if not bages_user_ids:
+                        reached_goal_ids = self.pool.get('gamification.goal').search(cr, uid, [
+                            ('challenge_id', '=', challenge.id),
+                            ('user_id', '=', user.id),
+                            ('state', '=', 'reached'),
+                        ], context=context)
+                        if len(reached_goal_ids) == len(challenge.line_ids):
+                            super(gamification_challenge, self).check_challenge_reward(cr, uid, ids, force=False, context=context)
+                            self.reward_user(cr, uid, user.id, challenge.reward_id.id, context)