merge-requests/30/head
Bob Mottram 2020-12-05 11:58:03 +00:00
parent 67e06f65c9
commit 4d46817350
1 changed files with 3 additions and 6 deletions

View File

@ -75,20 +75,17 @@ def guessHashtagCategory(tagName: str, hashtagCategories: {}) -> str:
This works by trying to find the longest similar hashtag This works by trying to find the longest similar hashtag
""" """
categoryMatched = '' categoryMatched = ''
tagMatched = ''
tagMatchedLen = 0 tagMatchedLen = 0
for categoryStr, hashtagList in hashtagCategories.items(): for categoryStr, hashtagList in hashtagCategories.items():
for hashtag in hashtagList: for hashtag in hashtagList:
if hashtag in tagName: if hashtag in tagName or tagName in hashtag:
if not tagMatched: if not categoryMatched:
tagMatched = hashtag tagMatchedLen = len(hashtag)
tagMatchedLen = len(tagMatched)
categoryMatched = categoryStr categoryMatched = categoryStr
else: else:
# match the longest tag # match the longest tag
if len(hashtag) > tagMatchedLen: if len(hashtag) > tagMatchedLen:
tagMatched = hashtag
categoryMatched = categoryStr categoryMatched = categoryStr
if not categoryMatched: if not categoryMatched:
return return