implement upload result

Author: shemogumbe

src/msgraph_core/models/__init__.py

@@ -1,4 +1,5 @@
 from .page_result import PageResult
 from .large_file_upload_session import LargeFileUploadSession
+from .upload_result import UploadResult, UploadSessionDataHolder
 
-__all__ = ['PageResult', 'LargeFileUploadSession']
+__all__ = ['PageResult', 'LargeFileUploadSession', 'UploadResult', 'UploadSessionDataHolder']

src/msgraph_core/models/upload_result.py

@@ -0,0 +1,61 @@
+from typing import Any, Callable, Dict, List, Optional, TypeVar
+
+from dataclasses import dataclass
+from datetime import datetime
+
+from kiota_abstractions.serialization import (
+    AdditionalDataHolder, Parsable, ParseNode, SerializationWriter
+)
+
+T = TypeVar('T')
+
+
+@dataclass
+class UploadSessionDataHolder(AdditionalDataHolder, Parsable):
+    expiration_datetime: Optional[datetime] = None
+    next_expected_ranges: Optional[List[str]] = None
+    upload_url: Optional[str] = None
+
+    def get_field_deserializers(self, ) -> Dict[str, Callable[[ParseNode], None]]:
+        """
+        The deserialization information for the current model
+        Returns: Dict[str, Callable[[ParseNode], None]]
+        """
+        fields: Dict[str, Callable[[Any], None]] = {
+            "expirationDateTime":
+            lambda n: setattr(self, 'expiration_date_time', n.get_datetime_value()),
+            "nextExpectedRanges":
+            lambda n:
+            setattr(self, 'next_expected_ranges', n.get_collection_of_primitive_values(str)),
+            "@odata.type":
+            lambda n: setattr(self, 'odata_type', n.get_str_value()),
+            "uploadUrl":
+            lambda n: setattr(self, 'upload_url', n.get_str_value()),
+        }
+        return fields
+
+    def serialize(self, writer: SerializationWriter) -> None:
+        """
+        Serializes information the current object
+        param writer: Serialization writer to use to serialize this model
+        Returns: None
+        """
+        if not writer:
+            raise TypeError("writer cannot be null.")
+        writer.write_datetime_value("expirationDateTime", self.expiration_date_time)
+        writer.write_collection_of_primitive_values("nextExpectedRanges", self.next_expected_ranges)
+        writer.write_str_value("@odata.type", self.odata_type)
+        writer.write_str_value("uploadUrl", self.upload_url)
+        writer.write_additional_data_value(self.additional_data)
+
+
+class UploadResult:
+
+    def __init__(self):
+        self.upload_session: Optional[UploadSessionDataHolder] = None
+        self.item_response: Optional[T] = None
+        self.location: Optional[str] = None
+
+    @property
+    def upload_succeeded(self) -> bool:
+        return self.item_response is not None or self.location is not None

src/msgraph_core/tasks/large_file_upload.py

@@ -13,7 +13,7 @@
 
 from kiota_abstractions.request_adapter import RequestAdapter
 
-from msgraph_core.models import LargeFileUploadSession  # check imports
+from msgraph_core.models import LargeFileUploadSession, UploadResult, UploadSessionDataHolder  # check imports
 
 
 class LargeFileUploadTask:
@@ -123,7 +123,15 @@ async def upload(self, after_chunk_upload: Optional[Callable] = None):
                 logging.error("Error uploading chunk  %s", error)
             finally:
                 self.chunks -= 1
-        return session
+        upload_result = UploadResult()
+        upload_result.upload_session = UploadSessionDataHolder(
+            expiration_datetime=self.upload_session.expiration_date_time,
+            next_expected_ranges=self.upload_session.next_expected_ranges,
+            upload_url=self.upload_session.upload_url
+        )
+        upload_result.item_response = session
+        upload_result.location = self.upload_session.upload_url
+        return upload_result
 
     @property
     def next_range(self):