wyj/starbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Fixed the error in generating word clouds when the text became empty after stop word filtering

Browse Source
This commit is contained in:
LWR
2023-06-03 01:08:02 +08:00
parent 7697951f6b
commit cf39287afb
1 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
starbot/painter/LiveReportGenerator.py
View File

@@ -368,29 +368,28 @@ class LiveReportGenerator:
if model.danmu_cloud:
all_danmu = param.get('all_danmu', [])
if all_danmu:
if config.get("DANMU_CLOUD_DICT"):
try:
jieba.load_userdict(config.get("DANMU_CLOUD_DICT"))
except Exception:
logger.error("载入弹幕词云自定义词典失败, 请检查配置的词典路径是否正确")
all_danmu_str = " ".join(all_danmu)
words = list(jieba.cut(all_danmu_str))
counts = dict(Counter(words))
if config.get("DANMU_CLOUD_STOP_WORDS"):
try:
with open(config.get("DANMU_CLOUD_STOP_WORDS"), "r", encoding="utf-8") as f:
stop_words = set(line.strip() for line in f)
for sw in stop_words:
counts.pop(sw, None)
except Exception:
logger.error("载入弹幕词云停用词失败, 请检查配置的停用词路径是否正确")
if len(counts) > 0:
pic.draw_section("弹幕词云")
if config.get("DANMU_CLOUD_DICT"):
try:
jieba.load_userdict(config.get("DANMU_CLOUD_DICT"))
except Exception:
logger.error("载入弹幕词云自定义词典失败, 请检查配置的词典路径是否正确")
all_danmu_str = " ".join(all_danmu)
words = list(jieba.cut(all_danmu_str))
counts = dict(Counter(words))
stop_words = {}
if config.get("DANMU_CLOUD_STOP_WORDS"):
try:
with open(config.get("DANMU_CLOUD_STOP_WORDS"), "r", encoding="utf-8") as f:
stop_words = set(line.strip() for line in f)
except Exception:
logger.error("载入弹幕词云停用词失败, 请检查配置的停用词路径是否正确")
for sw in stop_words:
counts.pop(sw, None)
font_base_path = os.path.dirname(os.path.dirname(__file__))
word_cloud = WordCloud(width=900,
height=450,