Follow PEP8

Author: Keniis0712

StealthIM/apis/__init__.py

@@ -1,4 +1,5 @@
 from . import common
 from . import user
 from . import util
-from . import group
+from . import group
+from . import message

StealthIM/apis/common.py

@@ -11,6 +11,7 @@ class Result:
     code: int
     msg: str
 
+
 @dataclasses.dataclass
 class NoValResult:
     result: Result
@@ -58,4 +59,4 @@ async def request(
 
                 return response_data
     logger.error(f"Request to {url} failed after {retry} retries.")
-    raise RuntimeError(f"Request failed with code: {response_data['result']['code']}")
+    raise RuntimeError(f"Request failed with code: {response_data['result']['code']}")

StealthIM/apis/file.py

@@ -1,41 +1,47 @@
 import dataclasses
 import enum
 
-from .common import request, Result, NoValResult
 from StealthIM import logger
+from .common import request, Result, NoValResult
 
 
 @dataclasses.dataclass
 class GetGroupsResult:
     groups: list[int]
     result: Result
 
+
 class GroupMemberType(enum.Enum):
     Member = 0
     Manager = 1
     Owner = 2
 
+
 @dataclasses.dataclass
 class GroupMember:
     name: str
     type: GroupMemberType
 
+
 @dataclasses.dataclass
 class GroupInfoResult:
     members: list[GroupMember]
     result: Result
 
+
 @dataclasses.dataclass
 class GroupPublicInfoResult:
     create_at: str
     name: str
     result: Result
 
+
 @dataclasses.dataclass
 class CreateGroupResult:
     groupid: int
     result: Result
 
+
 JoinGroupResult = NoValResult
 InviteGroupResult = NoValResult
 SetMemberRoleResult = NoValResult
@@ -65,6 +71,7 @@ async def get_groups(
         )
     )
 
+
 async def get_group_info(
         url: str,
         session: str,
@@ -94,6 +101,7 @@ async def get_group_info(
         )
     )
 
+
 async def get_group_public_info(
         url: str,
         session: str,
@@ -117,6 +125,7 @@ async def get_group_public_info(
         )
     )
 
+
 async def create_group(
         url: str,
         session: str,
@@ -142,6 +151,7 @@ async def create_group(
         )
     )
 
+
 async def join_group(
         url: str,
         session: str,
@@ -167,6 +177,7 @@ async def join_group(
         )
     )
 
+
 async def invite_group(
         url: str,
         session: str,
@@ -192,6 +203,7 @@ async def invite_group(
         )
     )
 
+
 async def set_user_role(
         url: str,
         session: str,
@@ -218,6 +230,7 @@ async def set_user_role(
         )
     )
 
+
 async def kick_user(
         url: str,
         session: str,
@@ -240,6 +253,7 @@ async def kick_user(
         )
     )
 
+
 async def change_group_name(
         url: str,
         session: str,
@@ -265,6 +279,7 @@ async def change_group_name(
         )
     )
 
+
 async def change_group_password(
         url: str,
         session: str,

StealthIM/apis/group.py

@@ -15,10 +15,15 @@ class MessageType(enum.Enum):
     LargeEmoji = 2
     Emoji = 3
     File = 4
+    Card = 5
+    InnerLink = 6
+    Recall = 16
+
 
 SendMessageResult = NoValResult
 RecallMessageResult = NoValResult
 
+
 @dataclasses.dataclass
 class Message:
     groupid: int
@@ -57,6 +62,7 @@ async def send_message(
         )
     )
 
+
 async def recall_message(
         url: str,
         session: str,
@@ -91,7 +97,7 @@ async def get_message(
         direction: int
 ):
     # noinspection PyStatementEffect
-    direction # currently unused
+    direction  # currently unused
 
     api_address = f'{url}/api/v1/message/{groupid}?msgid={from_id}'
     logger.debug(f"Called API get_message with url {api_address}")
@@ -125,4 +131,3 @@ async def get_message(
                             username=msg["username"],
                             hash=msg.get("hash", None),
                         )
-

StealthIM/apis/message.py

@@ -1,12 +1,12 @@
 import dataclasses
 from typing import Optional
 
-from .common import Result, request, NoValResult
 from StealthIM import logger
-
+from .common import Result, request, NoValResult
 
 RegisterResult = NoValResult
 
+
 @dataclasses.dataclass
 class UserInfo:
     result: Result
@@ -16,17 +16,20 @@ class UserInfo:
     phone_number: str
     username: str
 
+
 @dataclasses.dataclass
 class LoginResult:
     result: Result
     session: Optional[str] = None
     user_info: Optional[UserInfo] = None
 
+
 @dataclasses.dataclass
 class UserPublicInfo:
     result: Result
     nickname: str
 
+
 ChangePasswordResult = NoValResult
 ChangeEmailResult = NoValResult
 ChangeNicknameResult = NoValResult
@@ -78,6 +81,7 @@ async def register(
         )
     )
 
+
 async def login(
         url: str,
         username: str,
@@ -122,7 +126,8 @@ async def login(
         ),
         session=response_data.get("session"),
         user_info=info
-)
+    )
+
 
 async def get_self_info(
         url: str,
@@ -158,6 +163,7 @@ async def get_self_info(
         username=response_data["user_info"]["username"]
     )
 
+
 async def get_user_info(
         url: str,
         session: str,
@@ -194,6 +200,7 @@ async def get_user_info(
         nickname=nickname
     )
 
+
 async def change_password(
         url: str,
         session: str,
@@ -228,6 +235,7 @@ async def change_password(
         )
     )
 
+
 async def change_email(
         url: str,
         session: str,
@@ -262,6 +270,7 @@ async def change_email(
         )
     )
 
+
 async def change_nickname(
         url: str,
         session: str,
@@ -296,6 +305,7 @@ async def change_nickname(
         )
     )
 
+
 async def change_phone_number(
         url: str,
         session: str,
@@ -330,6 +340,7 @@ async def change_phone_number(
         )
     )
 
+
 async def delete(
         url: str,
         session: str

StealthIM/apis/user.py

@@ -2,6 +2,7 @@
 
 from StealthIM import logger
 
+
 async def ping(url: str) -> bool:
     """
     Ping the server to check if it is reachable.
@@ -23,10 +24,9 @@ async def ping(url: str) -> bool:
                     return False
                 response_data = await response.json()
         logger.debug(f"Response data: {response_data}")
-        if response_data != {"message":"Hello, StealthIM!"}:
+        if response_data != {"message": "Hello, StealthIM!"}:
             return False
     except aiohttp.ClientError as e:
         logger.error(f"Ping failed with error: {e}")
         return False
     return True
-