@@ -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