import json
from ..param_types import *
from ..helper_functions.helpers_email import *
from ..constants import MESSAGES_URL
from ..microsoft_base_request import MicrosoftBaseRequest
class MicrosoftFlagRequests(MicrosoftBaseRequest):
"""
Handles flag management for Microsoft Outlook emails via Microsoft Graph API.
This class provides methods to set or update the flag status of an email message using the Microsoft Graph API.
Inherits from MicrosoftBaseRequest to manage authentication and token retrieval.
"""
@MicrosoftBaseRequest.handle_microsoft_errors
def manage_flags_microsoft_api(self, email_id: str, flag: str):
"""
Sets or updates the flag status of an email message.
Args:
email_id (str): The unique identifier of the email message to flag.
flag (str): The flag status to set. Must be one of 'complete', 'notFlagged', or 'flagged'.
Returns:
str: A JSON-formatted string containing the API response or an error message if the flag is invalid.
"""
url = MESSAGES_URL + f"/{email_id}"
if flag not in ["complete", "notFlagged", "flagged"]:
return json.dumps({"error": "Not valid flag submited"}, indent=2)
data = {"flag": {"flagStatus": f"{flag}"}}
status_code, response = self.microsoft_patch(
url, self.token_manager.get_token(), data=data
)
response = microsoft_simplify_message(response)
return json.dumps(response, indent=2)
