def sentencia_episodio(id_episodio):
total_sentences = []
sent_episodios = list(
collection_sentencia.find({"Episode": f"{id_episodio}"}, {
'Name': 1,
"Episode": 1,
"Sentence": 1
}))
sent_episodios = json.loads(json_util.dumps(sent_episodios))
for item in sent_episodios:
episodio = json.loads(
json_util.dumps(
list(
collection_episodios.find(
{"_id": ObjectId(item['Episode'])}, {'Episode': 1}))))
name = json.loads(
json_util.dumps(
list(
collection_personajes.find({"_id": ObjectId(item['Name'])},
{'Name': 1}))))
sentence = {
"personaje_id": item['_id']['$oid'],
'Name': name[0]['Name'],
'Episode': episodio[0]['Episode'],
'Episode_id': item["Episode"],
'Sentence': item["Sentence"]
}
total_sentences.append(sentence)
return total_sentences
def analyzeResultPersonajeEpisodio(episodio_id, personaje_id):
all_sentences = list(collection_sentencia.find({"$and":[ {"Episode":{"$eq": episodio_id}}, {"Name":{"$eq" : personaje_id}}]}))
all_sentences = json.loads(json_util.dumps(all_sentences))
total_score = {'episodio_id': episodio_id, "personaje_id":personaje_id , 'neg': 0.0, 'neu': 0.0, 'pos': 0.0, 'compound': 0.0}
for i in all_sentences:
score = sia.polarity_scores(i['Sentence'])
total_score['neg'] += score['neg']
total_score['pos'] += score['pos']
total_score['neu'] += score['neu']
total_score['compound'] += score['compound']
return total_score
def analyzeResultSerie():
all_sentences = list(collection_sentencia.find({}))
all_sentences = json.loads(json_util.dumps(all_sentences))
total_score = {'neg': 0.0, 'neu': 0.0, 'pos': 0.0, 'compound': 0.0}
for i in all_sentences:
score = sia.polarity_scores(i['Sentence'])
total_score['neg'] += score['neg']
total_score['pos'] += score['pos']
total_score['neu'] += score['neu']
total_score['compound'] += score['compound']
return total_score
def analyzeResultPersonaje(personaje_id):
all_sentences = list(collection_sentencia.find({"Name":{"$eq" : personaje_id}}))
name = json.loads(json_util.dumps(list(collection_personajes.find({"_id": ObjectId(personaje_id)},{'Name':1}))))
all_sentences = json.loads(json_util.dumps(all_sentences))
total_score = {"personaje_id":personaje_id , 'Personaje': name[0]['Name'], 'neg': 0.0, 'neu': 0.0, 'pos': 0.0, 'compound': 0.0}
for i in all_sentences:
score = sia.polarity_scores(i['Sentence'])
total_score['neg'] += score['neg']
total_score['pos'] += score['pos']
total_score['neu'] += score['neu']
total_score['compound'] += score['compound']
return total_score
def analyzeResultallPersonajes():
all_personajes = list(collection_personajes.find({}))
all_personajes = json.loads(json_util.dumps(all_personajes))
total = []
for item in all_personajes:
all_sentences = list(collection_sentencia.find({"Name": item['_id']['$oid']}))
all_sentences = json.loads(json_util.dumps(all_sentences))
name = json.loads(json_util.dumps(list(collection_personajes.find({"_id": ObjectId(item['_id']['$oid'])},{'Name':1}))))
total_score = {"personaje_id": item['_id']['$oid'] ,'Name': name[0]['Name'], 'neg': 0.0, 'neu': 0.0, 'pos': 0.0, 'compound': 0.0}
for i in all_sentences:
score = sia.polarity_scores(i['Sentence'])
total_score['neg'] += score['neg']
total_score['pos'] += score['pos']
total_score['neu'] += score['neu']
total_score['compound'] += score['compound']
total.append(total_score)
return total
def mensajespersonaje(id_personaje):
query = {"Name": f"{id_personaje}"}
frases = list(collection_sentencia.find(query, {"_id": 0}))
return frases