sharedkernel 2.2.6__py3-none-any.whl → 2.3.0__py3-none-any.whl

sharedkernel/multipart_upload.py

@@ -0,0 +1,69 @@
+class MultipartUploadSession:
+    MIN_PART_SIZE = 5 * 1024 * 1024  # 5MB
+
+    def __init__(self, s3_client, bucket, object_key, acl="private"):
+        self.s3 = s3_client
+        self.bucket = bucket
+        self.key = object_key
+        self.acl = acl
+
+        self.upload_id = None
+        self.parts = []
+        self.part_number = 1
+        self.buffer = bytearray()
+
+        self._start()
+
+    def _start(self):
+        resp = self.s3.create_multipart_upload(
+            Bucket=self.bucket,
+            Key=self.key,
+            ACL=self.acl,
+        )
+        self.upload_id = resp["UploadId"]
+
+    def upload_chunk(self, chunk: bytes):
+        """
+        Receives small chunks (KB) and buffers them
+        """
+        self.buffer.extend(chunk)
+
+        if len(self.buffer) >= self.MIN_PART_SIZE:
+            self._upload_part(bytes(self.buffer))
+            self.buffer.clear()
+
+    def _upload_part(self, data: bytes):
+        resp = self.s3.upload_part(
+            Bucket=self.bucket,
+            Key=self.key,
+            UploadId=self.upload_id,
+            PartNumber=self.part_number,
+            Body=data,
+        )
+
+        self.parts.append({
+            "ETag": resp["ETag"],
+            "PartNumber": self.part_number
+        })
+        self.part_number += 1
+
+    def complete(self):
+        # upload remaining buffer (can be < 5MB)
+        if self.buffer:
+            self._upload_part(bytes(self.buffer))
+            self.buffer.clear()
+
+        self.s3.complete_multipart_upload(
+            Bucket=self.bucket,
+            Key=self.key,
+            UploadId=self.upload_id,
+            MultipartUpload={"Parts": self.parts},
+        )
+
+    def abort(self):
+        if self.upload_id:
+            self.s3.abort_multipart_upload(
+                Bucket=self.bucket,
+                Key=self.key,
+                UploadId=self.upload_id,
+            )

sharedkernel/s3_uploader.py

@@ -1,62 +1,64 @@
+
 import requests
 import boto3
 from io import BytesIO
 import uuid
 import os
+from sharedkernel.multipart_upload import MultipartUploadSession
 
 class S3Uploader:
     def __init__(self, endpoint_url, bucket, access_key, secret_key):
-        """
-        Initializes the S3Uploader with credentials and optional endpoint/region.
-
-        :param endpoint_url: Custom endpoint URL (for S3-compatible services like MinIO)
-        :param bucket: Name of S3 bucket
-        :param access_key: Access key
-        :param secret_key: Secret key
-        """
         self.endpoint_url = endpoint_url
         self.bucket = bucket
         self.access_key = access_key
         self.secret_key = secret_key
 
-        # Initialize the S3 client
         self.s3 = boto3.client(
             's3',
             endpoint_url=self.endpoint_url,
             aws_access_key_id=self.access_key,
             aws_secret_access_key=self.secret_key,
         )
-
-    def upload_file_object(self, file_obj, object_name=None, file_extension=None, folder_name=None):
-        """
-        Uploads a file object to an S3 bucket and returns its URL.
-
-        :param file_obj: File object to upload
-        :param object_name: S3 file name to save as (optional)
-        :param file_extension: File extension to append to object name (optional)
-        :param folder_name: Optional folder name to save the file in
-        :return: URL of the uploaded file if successful, else False
-        """
-        # Use uuid4 to generate a unique object_name if not provided
+    
+    @staticmethod
+    def __generate_object_name(object_name=None, file_extension=None, folder_name=None):
         if object_name is None:
             object_name = str(uuid.uuid4())
 
-        # Append the file extension to the object_name
         if file_extension:
             object_name += file_extension
 
-        # Prefix the object name with the folder if provided
         if folder_name:
             object_name = f"{folder_name}/{object_name}"
+        
+        return object_name
+
 
-        # Upload the file object
-        self.s3.upload_fileobj(file_obj, self.bucket, object_name, ExtraArgs={'ACL':'public-read'})
+    def upload_file_object(self, file_obj, object_name=None, file_extension=None, folder_name=None, public_read=True):
+        object_name = self.__generate_object_name(object_name, file_extension, folder_name)
+
+        acl = "public-read" if public_read else "private"
+
+        self.s3.upload_fileobj(file_obj, self.bucket, object_name, ExtraArgs={'ACL': acl})
 
-        # Construct the URL of the uploaded file
         file_url = f"{self.endpoint_url}/{self.bucket}/{object_name}"
 
         return file_url
 
+    def generate_presigned_url(self, object_name=None, file_extension=None, folder_name=None, expire_time: int = 3600):
+        object_name = self.__generate_object_name(object_name, file_extension, folder_name)
+
+        temp_url = self.s3.generate_presigned_url(
+           'get_object',
+           Params={
+               'Bucket': self.bucket,
+               'Key': object_name
+           },
+           ExpiresIn=expire_time
+       )
+        
+        return temp_url
+        
     def upload_file_from_url(self, file_url, object_name=None, folder_name=None):
         """
         Downloads a file from a URL and uploads it to an S3 bucket, returning its URL.
@@ -80,3 +82,25 @@ class S3Uploader:
         # Step 2: Upload the file to S3
         file_obj = BytesIO(response.content)
         return self.upload_file_object(file_obj=file_obj, object_name=object_name, file_extension=file_extension, folder_name=folder_name)
+
+    def create_multipart_session(
+        self,
+        object_name=None,
+        file_extension=None,
+        folder_name=None,
+        public_read=True,
+    ):
+        object_key = self.__generate_object_name(
+            object_name, file_extension, folder_name
+        )
+
+        acl = "public-read" if public_read else "private"
+
+        return MultipartUploadSession(
+            s3_client=self.s3,
+            bucket=self.bucket,
+            object_key=object_key,
+            acl=acl,
+        )
+
+

{sharedkernel-2.2.6.dist-info → sharedkernel-2.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sharedkernel
-Version: 2.2.6
+Version: 2.3.0
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -29,6 +29,10 @@ Dynamic: summary
 this is a shared kernel package
 
 # Change Log
+### Version 2.3.0
+- Implement multipart uploader for S3
+### Version 2.2.7
+- add presigned url in s3 uploader
 ### Version 2.2.6
 - implement api key
 ### Version 2.2.5

{sharedkernel-2.2.6.dist-info → sharedkernel-2.3.0.dist-info}/RECORD

@@ -3,8 +3,9 @@ sharedkernel/data_format_converter.py,sha256=GWGbfhKJBifkz-cfnqKAFjJM43WC0qdq9KS
 sharedkernel/date_converter.py,sha256=Cjd4ewm0pIfQzv7nlgAAB_EYrr-VvXxQGehJCNphgXc,4491
 sharedkernel/diff_utils.py,sha256=mtwJmc05GAXUOB0ZLtqAhfBT1kGoSQ7qmP5N44P73ho,2564
 sharedkernel/jwt_service.py,sha256=KSkrpXVqmKMGdaoDg0DqhOfzR9CIGVTg7HfOlAaz1Zo,1611
+sharedkernel/multipart_upload.py,sha256=JVlCBlznB9dWh2_spjAqzLOqQT1CHUTvrR4m7ug8qaM,1877
 sharedkernel/regex_masking.py,sha256=zQrgteP8Cuq1EC9B7QUJqAXUxK9ISD9kWMYK2AbRfw0,3288
-sharedkernel/s3_uploader.py,sha256=9Tzi5siceHPKab244tOAmnY_hNOeTn9yvsQevNdz7Gk,3313
+sharedkernel/s3_uploader.py,sha256=VWgN-RVHmLXMDuxgZux3M-iFwWk5zhRnUIECKS0auW4,3637
 sharedkernel/string_extentions.py,sha256=ld02W06gd0Ql80GQU6nlqPAeUSfOe2Yr8cCzf3lJgQY,98
 sharedkernel/database/__init__.py,sha256=AtIbU7pDKwY6YCg_J8PX62WcYld1lAvBcVFW62MdiFg,241
 sharedkernel/database/audit_model.py,sha256=SMAYrvMb7XvPi4076TDgkOLaxh_3Jg_tfE12qKOy8RA,364
@@ -27,7 +28,7 @@ sharedkernel/objects/json_string_model.py,sha256=j63tnoqiok0EmBP6T-ChYuQYKPw7mLq
 sharedkernel/objects/jwt_model.py,sha256=XQHQhTbg7PT8XiUh5fd9MwRH4ldPsesI_hfbjaSqdKg,134
 sharedkernel/objects/result.py,sha256=I_9hX5TPEO1oStzuFLjFh1rtimXorz7ml-OaW_2BMvc,680
 sharedkernel/objects/user_info.py,sha256=51WyspRxlIWzK7Lfxgqg4D6mylXeHe9ZSenf-RhYTdA,286
-sharedkernel-2.2.6.dist-info/METADATA,sha256=PnmhL6-MWa8qXSRjiaMlkF-uI6sXyUpXtj3Kxz1nc1w,3230
-sharedkernel-2.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sharedkernel-2.2.6.dist-info/top_level.txt,sha256=TVTOnV1MItSSlpSjqkiijuHkoVsGHS4CArpsM-lylkE,13
-sharedkernel-2.2.6.dist-info/RECORD,,
+sharedkernel-2.3.0.dist-info/METADATA,sha256=sJZef9JTB6R7oQ5kGWKCQPuV6d7vIztvimh2VWN6Uh4,3339
+sharedkernel-2.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sharedkernel-2.3.0.dist-info/top_level.txt,sha256=TVTOnV1MItSSlpSjqkiijuHkoVsGHS4CArpsM-lylkE,13
+sharedkernel-2.3.0.dist-info/RECORD,,