Update main.py

main.py

@@ -599,8 +599,6 @@ def mindmapDemo():
     print(data)
     return get_initial_data(), 200
 
-
-
 @app.route('/roadmap', methods=['POST'])
 def roadmap():
     if request.method == 'POST':
@@ -725,7 +723,6 @@ Generate topics for all 7 days in a week and reference links only links(no heade
             # return jsonify({'success': True, 'roadmapData': roads_data_demo, 'activeDays': active_days})
             
     
-
 @app.route('/roadmapmodder', methods=['POST'])
 def roadmapmodder():
     if request.method == 'POST':
@@ -786,7 +783,11 @@ def roadmapmodder():
             )
             
             if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['year'] != day_data['year']:
-                updateActiveStreak(obj_id, day_data)
+                temp_data['activeDays'].append(day_data)
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "activeDays": temp_data['activeDays'] }}
+                )
             
             return jsonify({'success': True, 'youtubeUpdate': temp})
             
@@ -801,8 +802,12 @@ def roadmapmodder():
                 {"$set": { "data": temp_data['data'] }}
             )
             
-            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
-                updateActiveStreak(obj_id, day_data)
+            if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
+                temp_data['activeDays'].append(day_data)
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "activeDays": temp_data['activeDays'] }}
+                )
 
             return jsonify({'success': True})
 
@@ -817,8 +822,12 @@ def roadmapmodder():
                 {"$set": { "data": temp_data['data'] }}
             )
             
-            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
-                updateActiveStreak(obj_id, day_data)
+            if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['year'] != day_data['year']:
+                temp_data['activeDays'].append(day_data)
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "activeDays": temp_data['activeDays'] }}
+                )
 
         elif data.get('type') == 22:                # Update Reference link Visit
             ref_link_index = data.get('refLinkIndex')
@@ -831,8 +840,12 @@ def roadmapmodder():
                 {"$set": { "data": temp_data['data'] }}
             )
 
-            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
-                updateActiveStreak(obj_id, day_data)
+            if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['year'] != day_data['year']:
+                temp_data['activeDays'].append(day_data)
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "activeDays": temp_data['activeDays'] }}
+                )
 
         # Add Mindmaps change in mindap generation fucntion such that if roadmapId, WeekNum, DayNum is sent while generation of mindmap then associate it directly with the roadmap.
 
@@ -845,21 +858,14 @@ def roadmapmodder():
                 {"$set": { "data": temp_data['data'] }}
             )
             
-            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
-                updateActiveStreak(obj_id, day_data)
+            if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['year'] != day_data['year']:
+                temp_data['activeDays'].append(day_data)
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "activeDays": temp_data['activeDays'] }}
+                )
 
         return jsonify({'success': True})
-    
-
-def updateActiveStreak(object_id, day_data):       # Update Active Streak
-    temp_data = roadmap_collection.find_one({"_id": ObjectId(object_id)})
-    temp_data['activeDays'].append(day_data)
-
-    roadmap_collection.update_one(
-        {"_id": ObjectId(object_id)},
-        {"$set": { "activeDays": temp_data['activeDays'] }}
-    )
-
 
 
 if __name__ == '__main__':