Example #1
0
def addMessage(chat_id):
    new_idMessage = max(coll.distinct("idMessage")) + 1
    idUser = int(request.forms.get("idUser"))
    message = str(request.forms.get("text"))
    fields = list(
        coll.find({"idUser": idUser}, {
            "userName": 1,
            "idUser": 1,
            "_id": 0,
            "idChat": 1
        }))
    print(fields)
    name = fields[0]["userName"]
    for f in fields:
        if f["userName"] == name:
            new_idUser = f["idUser"]
        else:
            name = name
    new_message = {
        "idUser": idUser,
        "userName": name,
        "idChat": int(chat_id),
        "idMessage": new_idMessage,
        "text": message
    }
    print(new_message)
    new_user = {"idUser": new_idUser, "userName": name}
    print(new_user)
    coll.insert_one(new_message)
Example #2
0
def getSentiment(chat_id):
    query = list(
        coll.find({'idChat': int(chat_id)}, {
            "userName": 1,
            "idUser": 1,
            "text": 1,
            "_id": 0
        }))
    sid = SentimentIntensityAnalyzer()
    total_info = []
    for text in query:
        info = {}
        info["userName"] = text["userName"]
        info["idChat"] = int(chat_id)
        info["text"] = text["text"]
        info["sentiment"] = sid.polarity_scores(text["text"])
        total_info.append(info)
    print(total_info)
    return dumps(total_info)
    """comp_sent = [s['sentiment']['compound'] for s in total_info]
Example #3
0
def conver(tipo):
    return dumps(coll.find({'userName': tipo}))
Example #4
0
def index():
    return dumps(coll.find())
Example #5
0
def get_user(user):
    return dumps(coll.find({"userName": str(user)}, {"text": 0}))
Example #6
0
def get_chat(chat_id):
    return dumps(coll.find({"idChat": int(chat_id)}, {"_id": 0}))