Add album_ids to Story

Author: SpEcHiDe

pyrogram/methods/stories/post_story.py

@@ -38,6 +38,7 @@ async def post_story(
         protect_content: bool = None,
         business_connection_id: str = None,
         privacy_settings: "types.StoryPrivacySettings" = None,
+        album_ids: list[int] = None,
         from_story_chat_id: Union[int, str] = None,
         from_story_id: int = None,
         progress: Callable = None,
@@ -90,6 +91,9 @@ async def post_story(
                 The privacy settings for the story; ignored for stories sent to supergroup and channel chats.
                 Defaults to :obj:`~pyrogram.types.StoryPrivacySettingsEveryone`.
 
+            album_ids (List of ``int``, *optional*):
+                Identifiers of story albums to which the story will be added upon posting. An album can have up to ``stories_album_stories_limit``.
+
             from_story_chat_id (``int`` | ``str``, *optional*):
                 Full identifier of the original story, which content was used to create the story; pass None if the story isn't repost of another story.
                 Identifier of the chat that posted the story.
@@ -230,6 +234,7 @@ async def post_story(
                             caption=message,
                             entities=entities,
                             period=active_period,
+                            albums=album_ids,
                             # fwd_modified=True if from_story_id else None,
                             fwd_from_id=await self.resolve_peer(from_story_chat_id) if from_story_chat_id else None,
                             fwd_from_story=from_story_id,

pyrogram/types/stories/story.py

@@ -102,6 +102,9 @@ class Story(Object, Update):
             The story is deleted.
             A story can be deleted in case it was deleted or you tried to retrieve a story that doesn't exist yet.
         
+        album_ids (List of ``int``):
+            Identifiers of story albums to which the story is added; only for manageable stories.
+
         link (``str``, *property*):
             Generate a link to this story, only for Telegram Premium chats having usernames. Can be None if the story cannot have a link.
 
@@ -132,6 +135,7 @@ def __init__(
         forwards: int = None,
         skipped: bool = None,
         deleted: bool = None,
+        album_ids: list[int] = None,
         _raw = None
     ):
         super().__init__(client)
@@ -157,6 +161,7 @@ def __init__(
         self.forwards = forwards
         self.skipped = skipped
         self.deleted = deleted
+        self.album_ids = album_ids
         self._raw = _raw
 
     @staticmethod
@@ -182,6 +187,7 @@ def _parse_story_item(
         is_visible_only_for_self = None
         areas = None
         privacy_settings = None
+        album_ids = None
 
         if isinstance(story_item, raw.types.StoryItemDeleted):
             deleted = True
@@ -235,6 +241,8 @@ def _parse_story_item(
                         area,
                     ) for area in story_item.media_areas
                 ]
+            
+            album_ids = story_item.albums
 
         return (
             date,
@@ -255,6 +263,7 @@ def _parse_story_item(
             is_visible_only_for_self,
             areas,
             privacy_settings,
+            album_ids,
         )
 
     @staticmethod
@@ -292,6 +301,7 @@ async def _parse(
         areas = None
         privacy_settings = None
         repost_info = None
+        album_ids = None
 
         if story_media:
             rawupdate = story_media
@@ -375,6 +385,7 @@ async def _parse(
                 is_visible_only_for_self,
                 areas,
                 privacy_settings,
+                album_ids,
             ) = Story._parse_story_item(client, story_item)
 
             if not chat and story_item.from_id:
@@ -416,6 +427,7 @@ async def _parse(
             areas=areas,
             privacy_settings=privacy_settings,
             repost_info=repost_info,
+            album_ids=album_ids,
         )
 
     async def react(