forked from hkc/mastoposter
Added mention filter
This commit is contained in:
parent
4d7a9be45f
commit
d19a3d2005
|
@ -1,2 +1,4 @@
|
||||||
from .base import BaseFilter # NOQA
|
from .base import BaseFilter # NOQA
|
||||||
from mastoposter.filters.boost_filter import BoostFilter # NOQA
|
from mastoposter.filters.boost import BoostFilter # NOQA
|
||||||
|
from mastoposter.filters.combined import CombinedFilter # NOQA
|
||||||
|
from mastoposter.filters.mention import MentionFilter # NOQA
|
||||||
|
|
|
@ -0,0 +1,29 @@
|
||||||
|
from configparser import SectionProxy
|
||||||
|
from re import Pattern, compile as regexp
|
||||||
|
from typing import ClassVar
|
||||||
|
from fnmatch import fnmatch
|
||||||
|
from mastoposter.filters.base import BaseFilter
|
||||||
|
from mastoposter.types import Status
|
||||||
|
|
||||||
|
|
||||||
|
class MentionFilter(BaseFilter, filter_name="mention"):
|
||||||
|
MENTION_REGEX: ClassVar[Pattern] = regexp(r"@([^@]+)(@([^@]+))?")
|
||||||
|
|
||||||
|
def __init__(self, section: SectionProxy):
|
||||||
|
super().__init__(section)
|
||||||
|
self.list = section.get("list", "").split()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def check_account(cls, acct: str, mask: str):
|
||||||
|
return fnmatch(acct, mask)
|
||||||
|
|
||||||
|
def __call__(self, status: Status) -> bool:
|
||||||
|
return any(
|
||||||
|
(
|
||||||
|
any(
|
||||||
|
self.check_account(mention.acct, mask)
|
||||||
|
for mask in self.list
|
||||||
|
)
|
||||||
|
for mention in status.mentions
|
||||||
|
)
|
||||||
|
)
|
|
@ -262,6 +262,7 @@ class Status:
|
||||||
reblogs_count: int
|
reblogs_count: int
|
||||||
favourites_count: int
|
favourites_count: int
|
||||||
replies_count: int
|
replies_count: int
|
||||||
|
mentions: List[Mention]
|
||||||
application: Optional[Application] = None
|
application: Optional[Application] = None
|
||||||
url: Optional[str] = None
|
url: Optional[str] = None
|
||||||
in_reply_to_id: Optional[str] = None
|
in_reply_to_id: Optional[str] = None
|
||||||
|
@ -298,6 +299,7 @@ class Status:
|
||||||
card=data.get("card"),
|
card=data.get("card"),
|
||||||
language=data.get("language"),
|
language=data.get("language"),
|
||||||
text=data.get("text"),
|
text=data.get("text"),
|
||||||
|
mentions=[Mention.from_dict(m) for m in data.get("mentions", [])],
|
||||||
)
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|
Loading…
Reference in New Issue