fix: wrong mask

Author: SamuNatsu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nonebot-plugin-deer-pipe"
-version = "0.2.0"
+version = "0.2.1"
 description = "A deer-pipe attendance nonebot2 plugin"
 authors = [
     {name = "SNRainiar", email = "[email protected]"},

src/nonebot_plugin_deer_pipe/__init__.py

@@ -37,6 +37,7 @@ async def handle(user_info: UserInfo = EventUserInfo()) -> None:
 
   now: datetime = datetime.now()
   ok, deer = await attend(now, user_info.user_id)
+  print(deer)
   img: bytes = generate_image(now, name, deer)
 
   await UniMessage.text(

src/nonebot_plugin_deer_pipe/database.py

@@ -19,13 +19,13 @@ class User(SQLModel, table=True):
 initialized: bool = False
 
 # Attendance
-def get_seq(mask: str) -> list[int]:
+def get_seq(mask: int) -> list[int]:
   return list(
     map(
       lambda x: x[0] + 1,
       filter(
         lambda x: x[1] == '1',
-        enumerate(mask)
+        enumerate(bin(mask)[2:][::-1])
       )
     )
   )
@@ -45,12 +45,12 @@ async def attend(now: datetime, user_id: str) -> tuple[bool, Sequence[int]]:
 
     mask: int = int(user.mask)
     if (mask >> (now.day - 1)) & 1 == 1:
-      return (False, get_seq(user.mask))
+      return (False, get_seq(mask))
     else:
       mask |= (1 << (now.day - 1))
       user.mask = str(mask)
 
       session.add(user)
       await session.commit()
 
-      return (True, get_seq(str(mask)))
+      return (True, get_seq(mask))