mirror
/
FairEmail
mirror of https://github.com/M66B/FairEmail.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Clear classifier data when folder deleted

This commit is contained in:
M66B 2021-09-16 15:26:53 +02:00
parent 027a81a772
commit ae0d96fbd8
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/src/main/java/eu/faircode/email/MessageClassifier.java
View File

@ -177,19 +177,20 @@ public class MessageClassifier {
private static String classify(EntityMessage message, @NonNull String currentClass, @NonNull List<String> texts, boolean added, @NonNull Context context) { private static String classify(EntityMessage message, @NonNull String currentClass, @NonNull List<String> texts, boolean added, @NonNull Context context) {
State state = new State(); State state = new State();
DB db = DB.getInstance(context);
// Check classes // Check classes
DB db = DB.getInstance(context); boolean clear = false;
for (String clazz : new ArrayList<>(classMessages.get(message.account).keySet())) { for (String clazz : new ArrayList<>(classMessages.get(message.account).keySet())) {
EntityFolder folder = db.folder().getFolderByName(message.account, clazz); EntityFolder folder = db.folder().getFolderByName(message.account, clazz);
if (folder == null) { if (folder == null) {
EntityLog.log(context, EntityLog.Type.Classification, message, EntityLog.log(context, EntityLog.Type.Classification, message,
"Classifier deleting folder class=" + message.account + ":" + clazz); "Classifier deleted folder=" + message.account + ":" + clazz);
classMessages.get(message.account).remove(clazz); clear = true;
for (String word : wordClassFrequency.get(message.account).keySet())
wordClassFrequency.get(message.account).get(word).remove(clazz);
} }
} }
if (clear)
clear(context);
Log.i("Classifier texts=" + texts.size()); Log.i("Classifier texts=" + texts.size());
for (String text : texts) { for (String text : texts) {