✨ split events into single files

Author: yanyongyu

codegen/__init__.py

@@ -33,6 +33,10 @@ class Data:
     payload_types: list[str]
 
 
+def module_name(event: str, action: Data) -> str:
+    return f"{event}_{action.action}" if action.action else event
+
+
 def pascal_case(*names: str) -> str:
     words = chain.from_iterable(name.split("_") for name in names)
     return "".join(word if word.isupper() else word.capitalize() for word in words)
@@ -82,6 +86,7 @@ def build_event():
     output_dir.mkdir(parents=True, exist_ok=True)
 
     env = Environment(loader=PackageLoader("codegen"))
+    env.globals["filename"] = module_name
 
     # generate base model
     template = env.get_template("_base.py.jinja")
@@ -96,13 +101,13 @@ def build_event():
     # generate event models
     template = env.get_template("event.py.jinja")
     for event, actions in event_types.items():
-        event_text = template.render(
-            event=event,
-            actions=actions,
-            message_events=MESSAGE_EVENTS,
-            has_message_events=HAS_MESSAGE_EVENTS,
-        )
-        (output_dir / f"{event}.py").write_text(event_text)
+        for action in actions:
+            event_text = template.render(
+                action=action,
+                message_events=MESSAGE_EVENTS,
+                has_message_events=HAS_MESSAGE_EVENTS,
+            )
+            (output_dir / f"{module_name(event, action)}.py").write_text(event_text)
 
     # generate init file
     template = env.get_template("__init__.py.jinja")

codegen/templates/__init__.py.jinja

@@ -5,18 +5,18 @@ if TYPE_CHECKING:
     from ._types import events as events
     {% for event, actions in types.items() %}
     {% for action in actions %}
-    from .{{ event }} import {{ action.class_name }} as {{ action.class_name }}
+    from .{{ filename(event, action) }} import {{ action.class_name }} as {{ action.class_name }}
     {% endfor %}
     {% endfor %}
 else:
     __lazy_vars__ = {
         {% for event, actions in types.items() %}
-        ".{{ event }}": (
-            {% for action in actions %}
+        {% for action in actions %}
+        ".{{ filename(event, action) }}": (
             "{{ action.class_name }}",
-            {% endfor %}
         ),
         {% endfor %}
+        {% endfor %}
         "._base": (
             "Event",
         ),

codegen/templates/event.py.jinja

@@ -2,18 +2,14 @@ from typing import Union
 from functools import cached_property
 from typing_extensions import override
 
-{% for action in actions %}
 {% for model in action.payload_types %}
 from githubkit.versions.latest.models import {{ model }}
 {% endfor %}
-{% endfor %}
 
 from ..message import Message
 
 from ._base import Event
 
-{% for action in actions %}
-
 class {{ action.class_name }}(Event):
     {% if action.payload_types | length > 1 %}
     payload: Union[{{ action.payload_types | join(", ") }}]
@@ -40,4 +36,3 @@ class {{ action.class_name }}(Event):
         keep_untouched = (cached_property,)
 
     {% endif %}
-{% endfor %}

nonebot/adapters/github/event/__init__.py

@@ -1,5 +1,4 @@
 from githubkit.versions.latest.models import (
-    WebhookBranchProtectionConfigurationEnabled,
     WebhookBranchProtectionConfigurationDisabled,
 )
 
@@ -9,8 +8,3 @@
 class BranchProtectionConfigurationDisabled(Event):
 
     payload: WebhookBranchProtectionConfigurationDisabled
-
-
-class BranchProtectionConfigurationEnabled(Event):
-
-    payload: WebhookBranchProtectionConfigurationEnabled

…event/branch_protection_configuration.py …nch_protection_configuration_disabled.pynonebot/adapters/github/event/branch_protection_configuration.py renamed to nonebot/adapters/github/event/branch_protection_configuration_disabled.py nonebot/adapters/github/event/branch_protection_configuration.py renamed to nonebot/adapters/github/event/branch_protection_configuration_disabled.py

@@ -0,0 +1,8 @@
+from githubkit.versions.latest.models import WebhookBranchProtectionConfigurationEnabled
+
+from ._base import Event
+
+
+class BranchProtectionConfigurationEnabled(Event):
+
+    payload: WebhookBranchProtectionConfigurationEnabled

nonebot/adapters/github/event/branch_protection_configuration_enabled.py

@@ -0,0 +1,8 @@
+from githubkit.versions.latest.models import WebhookBranchProtectionRuleCreated
+
+from ._base import Event
+
+
+class BranchProtectionRuleCreated(Event):
+
+    payload: WebhookBranchProtectionRuleCreated

nonebot/adapters/github/event/branch_protection_rule.py

@@ -0,0 +1,8 @@
+from githubkit.versions.latest.models import WebhookBranchProtectionRuleDeleted
+
+from ._base import Event
+
+
+class BranchProtectionRuleDeleted(Event):
+
+    payload: WebhookBranchProtectionRuleDeleted

nonebot/adapters/github/event/branch_protection_rule_created.py

@@ -0,0 +1,8 @@
+from githubkit.versions.latest.models import WebhookBranchProtectionRuleEdited
+
+from ._base import Event
+
+
+class BranchProtectionRuleEdited(Event):
+
+    payload: WebhookBranchProtectionRuleEdited