Hexor/desubot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/desubot.git synced 2025-07-08 13:04:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Mystem wrapper reworked.

Browse Source
This commit is contained in:
AB
2020-12-10 14:46:19 +03:00
parent 375015efff
commit cfe510029e
5 changed files with 101 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/utils.rs
View File

@ -8,8 +8,6 @@ use uuid::Uuid;
use crate::db;
use crate::errors;
extern crate reqwest;
use serde_json::Value;
use subprocess::Exec;
pub(crate) fn get_title(message: &Message) -> String {
match &message.chat {
@ -98,31 +96,3 @@ pub(crate) async fn get_files(
};
Ok(file_count)
}
pub(crate) async fn stemming(message: &Message) -> Result<Vec<String>, errors::Error> {
let mut words: Vec<String> = vec![];
let proc = Exec::shell("mystem -d --format json -l");
match proc
.stdin(message.text().unwrap().as_str())
.communicate()
.unwrap()
.read_string()
.unwrap()
.0
{
Some(line) => {
let v: Vec<Value> = match serde_json::from_str(line.as_str()) {
Ok(val) => val,
Err(_) => return Ok(vec![]),
};
for i in v {
words.push(i["analysis"][0]["lex"].to_string().replace("\"", ""));
}
words.retain(|x| x != "null");
//debug!("Parsed words: {}.", words.join(", "));
}
None => return Ok(vec![]),
};
Ok(words)
}