user_server/graph_controller.py at master · Ashprakash/user_server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
from flask import Flask, request
from flask import Response
from flask import jsonify
from flask_pymongo import PyMongo
from bson.json_util import dumps
from bson import json_util, ObjectId
import json
import datetime
import bcrypt
from bson.objectid import ObjectId
import study_global

app = Flask(__name__)
app.config["MONGO_URI"] = study_global.URI
mongo = PyMongo(app)

def get_tagcloud():
    dict_tag_count = {}

    results = mongo.db.Post.find()

    for each_result in results:
        list_tag_name = each_result['tags']
        for tag_name in list_tag_name:
            if tag_name not in dict_tag_count:
                dict_tag_count[tag_name] = 0
            dict_tag_count[tag_name] += 1

    return Response(dumps(dict_tag_count), status=200)


def get_circlegraph():
    dict_user_course = {}

    results = mongo.db.Post.find()

    for each_result in results:
    #print(each_result['created_user']['user_name'])
        user_name = each_result['created_user']['user_name']
        if user_name not in dict_user_course:
            dict_user_course[user_name] = {}
        course_code = each_result['code']
        if course_code not in dict_user_course[user_name]:
            dict_user_course[user_name][course_code] = 0
        dict_user_course[user_name][course_code] += 1

        list_strings = []
        for each_user in dict_user_course:
            for each_course in dict_user_course[each_user]:
                temp_dict = {}
                temp_dict['from'] = each_user
                temp_dict['to'] = each_course
                temp_dict['value'] = dict_user_course[each_user][each_course]
                #temp_string = "{\"from\":\"" + each_user + "\",\"to\":\"" + each_course + "\",\"value\":" + str(dict_user_course[each_user][each_course]) + "}"
                list_strings.append(temp_dict)

    return Response(dumps(list_strings), status=200)