Skip to content

Commit 0d4aafc

Browse files
committed
test file upload
1 parent 0f98833 commit 0d4aafc

File tree

4 files changed

+11
-11
lines changed

4 files changed

+11
-11
lines changed

src/msgraph_core/models/large_file_upload_create_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from kiota_abstractions.store.backing_store_factory_singleton import BackingStoreFactorySingleton
77

88

9-
class LargeFileUploadCreateSessionBody(Parsable, AdditionalDataHolder, BackedModel):
9+
class LargeFileUploadCreateSession(Parsable, AdditionalDataHolder, BackedModel):
1010

1111
def __init__(self):
1212
self.backing_store = BackingStoreFactorySingleton.get_instance().create_backing_store()

src/msgraph_core/models/page_result.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
Classes:
1010
PageResult: Represents a page of items in a paged response.
1111
"""
12+
from __future__ import annotations
1213
from typing import Any, List, Optional
1314
from dataclasses import dataclass
14-
from __future__ import annotations
1515

1616
from kiota_abstractions.serialization.parsable import Parsable # type: ignore
1717
from kiota_abstractions.serialization.serialization_writer \

src/msgraph_core/tasks/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
from .page_iterator import PageIterator
2+
from .large_file_upload import LargeFileUploadTask

src/msgraph_core/tasks/large_file_upload.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def __init__(
1818
self,
1919
upload_session: Parsable,
2020
request_adapter: RequestAdapter,
21-
stream: BytesIO,
21+
stream: BytesIO, # counter check this
2222
max_chunk_size: int = 4 * 1024 * 1024
2323
):
2424
self.upload_session = upload_session
@@ -36,16 +36,13 @@ def __init__(
3636
def get_upload_session(self) -> Parsable:
3737
return self.upload_session
3838

39-
def get_adapter(self) -> RequestAdapter:
40-
return self.request_adapter
41-
4239
def create_upload_session(
4340
self,
4441
request_body: LargeFileUploadSession,
4542
model: LargeFileUploadCreateSession,
4643
) -> Future:
4744
request_info = RequestInformation()
48-
request_info.url = ""
45+
request_info.url = request_info.url # check the URL
4946
request_info.http_method = Method.POST
5047
request_info.set_content_from_parsable(
5148
self.request_adapter, 'application/json', request_body
@@ -54,6 +51,9 @@ def create_upload_session(
5451

5552
return self.request_adapter.send_async(request_info, LargeFileUploadSession, {})
5653

54+
def get_adapter(self) -> RequestAdapter:
55+
return self.request_adapter
56+
5757
def get_chunks(self) -> int:
5858
return self.chunks
5959

@@ -80,7 +80,7 @@ def upload_session_expired(self, upload_session: Optional[Parsable] = None) -> b
8080
return False
8181

8282
async def upload(self, after_chunk_upload: Optional[Callable] = None) -> Future:
83-
# Rewinds to take care of failures.
83+
# Rewind to take care of failures.
8484
self.stream.seek(0)
8585
if self.upload_session_expired(self.upload_session):
8686
raise RuntimeError('The upload session is expired.')
@@ -90,7 +90,6 @@ async def upload(self, after_chunk_upload: Optional[Callable] = None) -> Future:
9090
self.stream, 0, max(0, min(self.max_chunk_size - 1, self.file_size - 1))
9191
)
9292
process_next = session
93-
# determine the range uploaded
9493
# includes when resuming existing upload sessions.
9594
range_parts = self.next_range[0].split("-") if self.next_range else ['0', '0']
9695
end = min(int(range_parts[0]) + self.max_chunk_size - 1, self.file_size)
@@ -250,5 +249,5 @@ def get_validated_upload_url(self, upload_session: Parsable) -> str:
250249
def get_next_range(self) -> Optional[str]:
251250
return self.next_range
252251

253-
def get_next_range(self) -> Optional[str]:
254-
return self.next_range
252+
def get_file_size(self) -> int:
253+
return self.file_size

0 commit comments

Comments
 (0)