xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-27 07:54:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

style: introduce ruff as linter and formatter (#1356)

Browse Source 
* style: remove necessary imports

* style: fix F841

* style: fix F401

* style: fix F811

* style: fix E402

* style: fix E721

* style: fix E722

* style: fix E722

* style: fix F541

* style: ruff format

* style: all passed

* style: add ruff in deps

* style: more ignores in ruff.toml

* style: add pre-commit
This commit is contained in:
Junyan Qin (Chin)
2025-04-29 17:24:07 +08:00
committed by GitHub
parent 09e70d70e9
commit 209f16af76
240 changed files with 5307 additions and 4689 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/pipeline/resprule/entities.py
-1
View File
@@ -4,7 +4,6 @@ from ...platform.types import message as platform_message
class RuleJudgeResult(pydantic.BaseModel):
matching: bool = False
replacement: platform_message.MessageChain = None
pkg/pipeline/resprule/resprule.py
+17 -15
View File
@@ -1,16 +1,18 @@
from __future__ import annotations
from ...core import app
from . import entities as rule_entities, rule
from .rules import atbot, prefix, regexp, random
from . import rule
from .. import stage, entities
from ...core import entities as core_entities
from ...config import manager as cfg_mgr
from ...utils import importutil
from . import rules
importutil.import_modules_in_pkg(rules)
@stage.stage_class("GroupRespondRuleCheckStage")
@stage.stage_class('GroupRespondRuleCheckStage')
class GroupRespondRuleCheckStage(stage.PipelineStage):
"""群组响应规则检查器
@@ -21,8 +23,7 @@ class GroupRespondRuleCheckStage(stage.PipelineStage):
"""检查器实例"""
async def initialize(self, pipeline_config: dict):
"""初始化检查器
"""
"""初始化检查器"""
self.rule_matchers = []
@@ -31,12 +32,12 @@ class GroupRespondRuleCheckStage(stage.PipelineStage):
await rule_inst.initialize()
self.rule_matchers.append(rule_inst)
async def process(self, query: core_entities.Query, stage_inst_name: str) -> entities.StageProcessResult:
async def process(
self, query: core_entities.Query, stage_inst_name: str
) -> entities.StageProcessResult:
if query.launcher_type.value != 'group': # 只处理群消息
return entities.StageProcessResult(
result_type=entities.ResultType.CONTINUE,
new_query=query
result_type=entities.ResultType.CONTINUE, new_query=query
)
rules = query.pipeline_config['trigger']['group-respond-rules']
@@ -48,7 +49,9 @@ class GroupRespondRuleCheckStage(stage.PipelineStage):
# use_rule = rules[str(query.launcher_id)]
for rule_matcher in self.rule_matchers: # 任意一个匹配就放行
res = await rule_matcher.match(str(query.message_chain), query.message_chain, use_rule, query)
res = await rule_matcher.match(
str(query.message_chain), query.message_chain, use_rule, query
)
if res.matching:
query.message_chain = res.replacement
@@ -56,8 +59,7 @@ class GroupRespondRuleCheckStage(stage.PipelineStage):
result_type=entities.ResultType.CONTINUE,
new_query=query,
)
return entities.StageProcessResult(
result_type=entities.ResultType.INTERRUPT,
new_query=query
result_type=entities.ResultType.INTERRUPT, new_query=query
)
pkg/pipeline/resprule/rule.py
+6 -5
View File
@@ -10,17 +10,19 @@ from ...platform.types import message as platform_message
preregisetered_rules: list[typing.Type[GroupRespondRule]] = []
def rule_class(name: str):
def decorator(cls: typing.Type[GroupRespondRule]) -> typing.Type[GroupRespondRule]:
cls.name = name
preregisetered_rules.append(cls)
return cls
return decorator
class GroupRespondRule(metaclass=abc.ABCMeta):
"""群组响应规则的抽象类
"""
"""群组响应规则的抽象类"""
name: str
ap: app.Application
@@ -37,8 +39,7 @@ class GroupRespondRule(metaclass=abc.ABCMeta):
message_text: str,
message_chain: platform_message.MessageChain,
rule_dict: dict,
query: core_entities.Query
query: core_entities.Query,
) -> entities.RuleJudgeResult:
"""判断消息是否匹配规则
"""
"""判断消息是否匹配规则"""
raise NotImplementedError
pkg/pipeline/resprule/rules/atbot.py
+10 -10
View File
@@ -7,21 +7,24 @@ from ....core import entities as core_entities
from ....platform.types import message as platform_message
@rule_model.rule_class("at-bot")
@rule_model.rule_class('at-bot')
class AtBotRule(rule_model.GroupRespondRule):
async def match(
self,
message_text: str,
message_chain: platform_message.MessageChain,
rule_dict: dict,
query: core_entities.Query
query: core_entities.Query,
) -> entities.RuleJudgeResult:
if message_chain.has(platform_message.At(query.adapter.bot_account_id)) and rule_dict['at']:
if (
message_chain.has(platform_message.At(query.adapter.bot_account_id))
and rule_dict['at']
):
message_chain.remove(platform_message.At(query.adapter.bot_account_id))
if message_chain.has(platform_message.At(query.adapter.bot_account_id)): # 回复消息时会at两次，检查并删除重复的
if message_chain.has(
platform_message.At(query.adapter.bot_account_id)
): # 回复消息时会at两次，检查并删除重复的
message_chain.remove(platform_message.At(query.adapter.bot_account_id))
return entities.RuleJudgeResult(
@@ -29,7 +32,4 @@ class AtBotRule(rule_model.GroupRespondRule):
replacement=message_chain,
)
return entities.RuleJudgeResult(
matching=False,
replacement = message_chain
)
return entities.RuleJudgeResult(matching=False, replacement=message_chain)
pkg/pipeline/resprule/rules/prefix.py
+4 -10
View File
@@ -1,36 +1,30 @@
from .. import rule as rule_model
from .. import entities
from ....core import entities as core_entities
from ....platform.types import message as platform_message
@rule_model.rule_class("prefix")
@rule_model.rule_class('prefix')
class PrefixRule(rule_model.GroupRespondRule):
async def match(
self,
message_text: str,
message_chain: platform_message.MessageChain,
rule_dict: dict,
query: core_entities.Query
query: core_entities.Query,
) -> entities.RuleJudgeResult:
prefixes = rule_dict['prefix']
for prefix in prefixes:
if message_text.startswith(prefix):
# 查找第一个plain元素
for me in message_chain:
if isinstance(me, platform_message.Plain):
me.text = me.text[len(prefix):]
me.text = me.text[len(prefix) :]
return entities.RuleJudgeResult(
matching=True,
replacement=message_chain,
)
return entities.RuleJudgeResult(
matching=False,
replacement=message_chain
)
return entities.RuleJudgeResult(matching=False, replacement=message_chain)
pkg/pipeline/resprule/rules/random.py
+5 -7
View File
@@ -7,19 +7,17 @@ from ....core import entities as core_entities
from ....platform.types import message as platform_message
@rule_model.rule_class("random")
@rule_model.rule_class('random')
class RandomRespRule(rule_model.GroupRespondRule):
async def match(
self,
message_text: str,
message_chain: platform_message.MessageChain,
rule_dict: dict,
query: core_entities.Query
query: core_entities.Query,
) -> entities.RuleJudgeResult:
random_rate = rule_dict['random']
return entities.RuleJudgeResult(
matching=random.random() < random_rate,
replacement=message_chain
)
matching=random.random() < random_rate, replacement=message_chain
)
pkg/pipeline/resprule/rules/regexp.py
+4 -8
View File
@@ -7,15 +7,14 @@ from ....core import entities as core_entities
from ....platform.types import message as platform_message
@rule_model.rule_class("regexp")
@rule_model.rule_class('regexp')
class RegExpRule(rule_model.GroupRespondRule):
async def match(
self,
message_text: str,
message_chain: platform_message.MessageChain,
rule_dict: dict,
query: core_entities.Query
query: core_entities.Query,
) -> entities.RuleJudgeResult:
regexps = rule_dict['regexp']
@@ -27,8 +26,5 @@ class RegExpRule(rule_model.GroupRespondRule):
matching=True,
replacement=message_chain,
)
return entities.RuleJudgeResult(
matching=False,
replacement=message_chain
)
return entities.RuleJudgeResult(matching=False, replacement=message_chain)