|
@@ -170,14 +170,11 @@ class db():
|
|
|
# FIXME, categories
|
|
# FIXME, categories
|
|
|
return ret
|
|
return ret
|
|
|
|
|
|
|
|
- def add_keyword(self, kw):
|
|
|
|
|
- if 'keyword' not in kw or 'weight' not in kw:
|
|
|
|
|
- return False
|
|
|
|
|
-
|
|
|
|
|
- kw["pattern"] = self.to_regex(kw["keyword"].split("|"))
|
|
|
|
|
|
|
+ def add_keyword(self, kw, weight):
|
|
|
|
|
+ kw["pattern"] = self.to_regex(kw.split("|"))
|
|
|
kwdb = self.client[self.DB]["keywords"]
|
|
kwdb = self.client[self.DB]["keywords"]
|
|
|
try:
|
|
try:
|
|
|
- kwdb.insert_one(kw)
|
|
|
|
|
|
|
+ kwdb.insert_one({'keyword': kw, 'weight': weight})
|
|
|
return True
|
|
return True
|
|
|
except errors.DuplicateKeyError:
|
|
except errors.DuplicateKeyError:
|
|
|
print("Dup KW")
|
|
print("Dup KW")
|