ayaka.driver.msg_type

  1from .onebot import MessageSegment
  2
  3
  4class TypedMessageSegment(MessageSegment):
  5    __type__ = ""
  6
  7    @classmethod
  8    def check_type(cls, v: MessageSegment):
  9        return v.type == cls.__type__
 10
 11    @classmethod
 12    def __get_validators__(cls):
 13        yield cls.validate
 14
 15    @classmethod
 16    def validate(cls, v):
 17        if not isinstance(v, MessageSegment):
 18            raise TypeError('MessageSegment required')
 19        if not cls.check_type(v):
 20            raise ValueError('invalid MessageSegment format')
 21        return v
 22
 23    @classmethod
 24    def __modify_schema__(cls, field_schema):
 25        field_schema["msg_type"] = cls.__type__
 26
 27
 28class T_Anonymous(TypedMessageSegment):
 29    __type__ = "anonymous"
 30
 31
 32class T_At(TypedMessageSegment):
 33    __type__ = "at"
 34
 35
 36class T_Contact(TypedMessageSegment):
 37    __type__ = "contact"
 38
 39
 40class T_Dice(TypedMessageSegment):
 41    __type__ = "dice"
 42
 43
 44class T_Face(TypedMessageSegment):
 45    __type__ = "face"
 46
 47
 48class T_Forward(TypedMessageSegment):
 49    __type__ = "forward"
 50
 51
 52class T_Image(TypedMessageSegment):
 53    __type__ = "image"
 54
 55
 56class T_Json(TypedMessageSegment):
 57    __type__ = "json"
 58
 59
 60class T_Location(TypedMessageSegment):
 61    __type__ = "location"
 62
 63
 64class T_Music(TypedMessageSegment):
 65    __type__ = "music"
 66
 67
 68class T_Node(TypedMessageSegment):
 69    __type__ = "node"
 70
 71
 72class T_Poke(TypedMessageSegment):
 73    __type__ = "poke"
 74
 75
 76class T_Record(TypedMessageSegment):
 77    __type__ = "record"
 78
 79
 80class T_Reply(TypedMessageSegment):
 81    __type__ = "reply"
 82
 83
 84class T_Video(TypedMessageSegment):
 85    __type__ = "video"
 86
 87
 88class T_Rps(TypedMessageSegment):
 89    __type__ = "rps"
 90
 91
 92class T_Shake(TypedMessageSegment):
 93    __type__ = "shake"
 94
 95
 96class T_Share(TypedMessageSegment):
 97    __type__ = "share"
 98
 99
100class T_Text(TypedMessageSegment):
101    __type__ = "text"
102
103
104class T_Xml(TypedMessageSegment):
105    __type__ = "xml"