11import StealthIM
22from StealthIM import User
33from StealthIM .apis .group import GroupInfoResult , GroupPublicInfoResult , InviteGroupResult , \
4- GroupMemberType , SetMemberRoleResult , KickMemberResult , ChangeGroupNameResult , ChangeGroupPasswordResult
4+ GroupMemberType , SetMemberRoleResult , KickMemberResult , ChangeGroupNameResult , ChangeGroupPasswordResult , \
5+ JoinGroupResult , CreateGroupResult
56from StealthIM .apis .message import SendMessageResult
67
78
@@ -11,7 +12,7 @@ def __init__(self, user: User, group_id: int):
1112 self .group_id = group_id
1213
1314 @classmethod
14- async def create (cls , user : User , group_name : str ) -> "Group" :
15+ async def create (cls , user : User , group_name : str ) -> CreateGroupResult :
1516 """
1617 Create a new Group.
1718
@@ -25,19 +26,14 @@ async def create(cls, user: User, group_name: str) -> "Group":
2526 Raises:
2627 RuntimeError: If the request failed.
2728 """
28- res = await StealthIM .apis .group .create_group (user .server .url , user .session , group_name )
29- if res .result .code != 800 :
30- raise RuntimeError (res .result .msg )
31- return cls (user , res .groupid )
29+ return await StealthIM .apis .group .create_group (user .server .url , user .session , group_name )
3230
33- @classmethod
34- async def join (cls , user : User , group_id : int , password : str ) -> "Group" :
31+ async def join (self , password : str ) -> JoinGroupResult :
3532 """
3633 Join a Group.
3734
3835 Args:
3936 user (User): The user to join.
40- group_id (int): The ID of the group.
4137 password (str): The password.
4238
4339 Returns:
@@ -46,10 +42,8 @@ async def join(cls, user: User, group_id: int, password: str) -> "Group":
4642 Raises:
4743 RuntimeError: If the request failed.
4844 """
49- res = await StealthIM .apis .group .join_group (user .server .url , user .session , group_id , password )
50- if res .result .code != 800 :
51- raise RuntimeError (res .result .msg )
52- return cls (user , group_id )
45+ return await StealthIM .apis .group .join_group (self .user .server .url , self .user .session , self .group_id , password )
46+
5347
5448 async def get_members (self ) -> GroupInfoResult :
5549 """
0 commit comments