Source code for pyrogram.raw.functions.aicompose.update_tone

#  Pyrogram - Telegram MTProto API Client Library for Python
#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
#
#  This file is part of Pyrogram.
#
#  Pyrogram is free software: you can redistribute it and/or modify
#  it under the terms of the GNU Lesser General Public License as published
#  by the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  Pyrogram is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU Lesser General Public License for more details.
#
#  You should have received a copy of the GNU Lesser General Public License
#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.

from io import BytesIO

from pyrogram.raw.core.primitives import Int, Long, Int128, Int256, Bool, Bytes, String, Double, Vector
from pyrogram.raw.core import TLObject
from pyrogram import raw
from typing import List, Optional, Any

# # # # # # # # # # # # # # # # # # # # # # # #
#               !!! WARNING !!!               #
#          This is a generated file!          #
# All changes made in this file will be lost! #
# # # # # # # # # # # # # # # # # # # # # # # #




class UpdateTone(TLObject):  # type: ignore
    """Edit a custom `AI composer tone » <https://core.telegram.org/api/ai#ai-compose-tones>`_ previously created by the current user. Only the fields whose flag is set will be modified.

    .. include:: /_includes/usable-by/users.rst

    Details:
        - Layer: ``229``
        - ID: ``903BCF59``

    Parameters:
        tone (:obj:`InputAiComposeTone <pyrogram.raw.base.InputAiComposeTone>`):
            The tone to edit

        display_author (``bool``, *optional*):
            If set, changes whether the current user is publicly credited as the author of the tone (ternary value, can be not set, set and true, set and false).

        emoji_id (``int`` ``64-bit``, *optional*):
            If set, the new `custom emoji ID <https://core.telegram.org/api/custom-emoji>`_ of the tone's icon

        title (``str``, *optional*):
            If set, the new human-readable tone name, up to `aicompose_tone_title_length_max » <https://core.telegram.org/api/config#aicompose-tone-title-length-max>`_ UTF-8 characters long

        prompt (``str``, *optional*):
            If set, the new prompt that describes how the AI should rephrase messages using this tone, up to `aicompose_tone_prompt_length_max » <https://core.telegram.org/api/config#aicompose-tone-prompt-length-max>`_ UTF-8 characters long

    Returns:
        :obj:`AiComposeTone <pyrogram.raw.base.AiComposeTone>`
    """

    __slots__: List[str] = ["tone", "display_author", "emoji_id", "title", "prompt"]

    ID = 0x903bcf59
    QUALNAME = "functions.aicompose.UpdateTone"

    def __init__(self, *, tone: "raw.base.InputAiComposeTone", display_author: Optional[bool] = None, emoji_id: Optional[int] = None, title: Optional[str] = None, prompt: Optional[str] = None) -> None:
        self.tone = tone  # InputAiComposeTone
        self.display_author = display_author  # flags.0?Bool
        self.emoji_id = emoji_id  # flags.1?long
        self.title = title  # flags.2?string
        self.prompt = prompt  # flags.3?string

    @staticmethod
    def read(b: BytesIO, *args: Any) -> "UpdateTone":
        
        flags = Int.read(b)
        
        tone = TLObject.read(b)
        
        display_author = Bool.read(b) if flags & (1 << 0) else None
        emoji_id = Long.read(b) if flags & (1 << 1) else None
        title = String.read(b) if flags & (1 << 2) else None
        prompt = String.read(b) if flags & (1 << 3) else None
        return UpdateTone(tone=tone, display_author=display_author, emoji_id=emoji_id, title=title, prompt=prompt)

    def write(self, *args) -> bytes:
        b = BytesIO()
        b.write(Int(self.ID, False))

        flags = 0
        flags |= (1 << 0) if self.display_author is not None else 0
        flags |= (1 << 1) if self.emoji_id is not None else 0
        flags |= (1 << 2) if self.title is not None else 0
        flags |= (1 << 3) if self.prompt is not None else 0
        b.write(Int(flags))
        
        b.write(self.tone.write())
        
        if self.display_author is not None:
            b.write(Bool(self.display_author))
        
        if self.emoji_id is not None:
            b.write(Long(self.emoji_id))
        
        if self.title is not None:
            b.write(String(self.title))
        
        if self.prompt is not None:
            b.write(String(self.prompt))
        
        return b.getvalue()