Esempio n. 1
0
def get_clusters(userId):
    db = MongoInit().initialize()
    messageService = MessageService(db)
    messages = messageService.getMessagesByUser(userId)

    nlpService = NLPService()
    clusters = nlpService.processMessages(messages)

    result = []

    for key, values in clusters.iteritems():
        clusterId = int(key)
        clusteredMessages = [{'subject': v.subject, 'sender': v.sender, 'content': v.originalContent, 'date': v.date} for v in values[:10]]
        cluster = {'clusterId': clusterId, 'messages': clusteredMessages}
        result.append(cluster)
    
    return make_response(jsonify({'clusters':result}), 200)