Merge pull request #9 from zweckj/0.0.17

Fix: catch missing field

Author: zweckj

onedrive_personal_sdk/clients/large_file_upload.py

@@ -2,18 +2,19 @@
 
 import asyncio
 import logging
+from collections.abc import Awaitable, Callable
 from datetime import datetime, timezone
-from collections.abc import Callable, Awaitable
 
 from aiohttp import ClientSession
+from mashumaro.exceptions import MissingField
 
 from onedrive_personal_sdk.clients.base import OneDriveBaseClient
-from onedrive_personal_sdk.const import GRAPH_BASE_URL, HttpMethod, ConflictBehavior
+from onedrive_personal_sdk.const import GRAPH_BASE_URL, ConflictBehavior, HttpMethod
 from onedrive_personal_sdk.exceptions import (
+    ExpectedRangeNotInBufferError,
     HashMismatchError,
     HttpRequestException,
     OneDriveException,
-    ExpectedRangeNotInBufferError,
     UploadSessionExpired,
 )
 from onedrive_personal_sdk.models.items import File
@@ -84,8 +85,8 @@ async def upload(
         while retries < self._max_retries:
             try:
                 return await self.start_upload(upload_session, validate_hash)
-            except UploadSessionExpired:
-                _LOGGER.debug("Session expired/not found, restarting")
+            except (UploadSessionExpired, MissingField):
+                _LOGGER.debug("Session expired/not found/broken, restarting")
                 self._buffer = UploadBuffer()
                 self._upload_result = LargeFileChunkUploadResult(
                     datetime.now(timezone.utc), ["0-"]
@@ -131,8 +132,9 @@ async def start_upload(
             if self._buffer.length >= self._upload_chunk_size:
                 uploaded_chunks = 0
                 while (
-                    self._buffer.length - uploaded_chunks * UPLOAD_CHUNK_SIZE
-                ) > self._upload_chunk_size:  # Loop in case the buffer is >= UPLOAD_CHUNK_SIZE * 2
+                    (self._buffer.length - uploaded_chunks * UPLOAD_CHUNK_SIZE)
+                    > self._upload_chunk_size
+                ):  # Loop in case the buffer is >= UPLOAD_CHUNK_SIZE * 2
                     slice_start = uploaded_chunks * self._upload_chunk_size
                     try:
                         chunk_result = await self._async_upload_chunk(

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "onedrive-personal-sdk"
-version = "0.0.16"
+version = "0.0.17"
 license = {text = "MIT"}
 description = "A package to interact with the Microsoft Graph API for personal OneDrives."
 readme = "README.md"