MrTuxedo/telegram-moderator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add basic sentiment analysis

Browse Source
This commit is contained in:
Josh Fraser 2020-01-27 20:26:26 -08:00
parent 79ec4c1abf
commit d6bdb9655a
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bot.py
View File

@ -19,6 +19,7 @@ import re
import unidecode import unidecode
from mwt import MWT from mwt import MWT
from googletrans import Translator from googletrans import Translator
from textblob import TextBlob
class TelegramMonitorBot: class TelegramMonitorBot:
@ -303,20 +304,26 @@ class TelegramMonitorBot:
return bool_set return bool_set
def log_message(self, user_id, user_message, chat_id): def log_message(self, user_id, user_message, chat_id):
try: try:
s = session() s = session()
language_code = english_message = "" language_code = english_message = ""
polarity = subjectivity = 0.0
try: try:
# translate to English & log the original language
translator = Translator() translator = Translator()
translated = translator.translate(user_message) translated = translator.translate(user_message)
language_code = translated.src language_code = translated.src
english_message = translated.text english_message = translated.text
# run basic sentiment analysis on the translated English string
analysis = TextBlob(english_message)
polarity = analysis.sentiment.polarity
subjectivity = analysis.sentiment.subjectivity
except Exception as e: except Exception as e:
print(e.message) print(e.message)
msg1 = Message(user_id=user_id, message=user_message, msg1 = Message(user_id=user_id, message=user_message, chat_id=chat_id,
chat_id=chat_id, language_code=language_code, english_message=english_message) language_code=language_code, english_message=english_message, polarity=polarity,
subjectivity=subjectivity)
s.add(msg1) s.add(msg1)
s.commit() s.commit()
s.close() s.close()