ciocore 9.1.0b1__py2.py3-none-any.whl → 9.1.0b2__py2.py3-none-any.whl

ciocore/VERSION

@@ -1 +1 @@
-9.1.0-beta.1
+9.1.0-beta.2

ciocore/uploader/_uploader.py

@@ -26,13 +26,12 @@ from ciocore import (
     exceptions,
 )
 
+from . import thread_queue_job
+
 from .upload_stats import UploadStats
 
 logger = logging.getLogger("{}.uploader".format(loggeria.CONDUCTOR_LOGGER_NAME))
 
-SINGLEPART = "singlepart"
-MULTIPART = "multipart"
-
 
 class MD5Worker(worker.ThreadWorker):
     """
@@ -330,28 +329,61 @@ class FileStatWorker(worker.ThreadWorker):
         """
 
         if job:
+
+            kms_key_name = job.get('kmsKeyName')
+
             # iterate through singlepart urls
             for singlepart_upload in job.get("singlePartURLs", []):
                 path = singlepart_upload["filePath"]
                 file_size = singlepart_upload["fileSize"]
                 upload_url = singlepart_upload["preSignedURL"]
+                md5 = self.metric_store.get_dict("file_md5s", path)
 
                 self.metric_store.increment("bytes_to_upload", file_size, path)
                 self.metric_store.increment("num_files_to_upload")
                 logger.debug("Singlepart, adding task %s", path)
 
-                self.put_job((path, file_size, upload_url, SINGLEPART))
+                upload_tq_job = thread_queue_job.UploadThreadQueueJob(path,
+                                                                      file_size,
+                                                                      presigned_url=upload_url,
+                                                                      file_md5=md5,
+                                                                      upload_id=None,
+                                                                      part_size=file_size,
+                                                                      part_index=1,
+                                                                      kms_key_name=kms_key_name)
+
+                self.put_job(upload_tq_job)
 
             # iterate through multipart
             for multipart_upload in job.get("multiPartURLs", []):
                 path = multipart_upload["filePath"]
                 file_size = multipart_upload["fileSize"]
 
+                part = multipart_upload
+                total_parts = len(multipart_upload['parts'])
+                md5 = self.metric_store.get_dict("file_md5s", path)
+
+                for chunk in multipart_upload['parts']:
+                    logger.debug("Multipart, adding task %s (part %s)", path, chunk['partNumber'])
+
+                    upload_tq_job = thread_queue_job.UploadThreadQueueJob(
+                                                        path=path,
+                                                        file_size=file_size,
+                                                        presigned_url=chunk['url'],
+                                                        file_md5=md5,
+                                                        upload_id=multipart_upload['uploadID'],
+                                                        part_size=multipart_upload['partSize'],
+                                                        total_parts=total_parts,
+                                                        part_index=chunk['partNumber'],
+                                                        kms_key_name=kms_key_name)
+
+
+                    part['parts'] = chunk
+                    self.put_job(upload_tq_job)
+
                 self.metric_store.increment("bytes_to_upload", file_size, path)
                 self.metric_store.increment("num_files_to_upload")
-                logger.debug("Multipart, adding task %s", path)
-                self.put_job((path, file_size, multipart_upload, MULTIPART))
-
+ 
         # make sure we return None, so no message is automatically added to the out_queue
         return None
 
@@ -383,45 +415,30 @@ class UploadWorker(worker.ThreadWorker):
                 self.metric_store.increment("bytes_uploaded", len(data), filename)
 
     def do_work(self, job, thread_int):
+        
         if job:
-            kms_key_name = None
+        
+            md5 = self.metric_store.get_dict("file_md5s", job.path)
 
             try:
-                filename = job[0]
-                file_size = job[1]
-                upload = job[2]
-                upload_type = job[3]
-
-            except Exception:
-                logger.error("Issue with job (%s): %s", len(job), job)
-                raise
-
-            if len(job) > 4:
-                kms_key_name = job[4]
-
-            md5 = self.metric_store.get_dict("file_md5s", filename)
-
-            try:
-                if upload_type == SINGLEPART:
-                    return self.do_singlepart_upload(
-                        upload, filename, file_size, md5, kms_key_name
-                    )
-                elif upload_type == MULTIPART:
-                    return self.do_multipart_upload(upload, filename, md5)
-
-                raise Exception(
-                    "upload_type is '%s' expected %s or %s"
-                    % (upload_type, SINGLEPART, MULTIPART)
-                )
+                if job.is_multipart():
+                    return self.do_multipart_upload(job)
+                    
+                else:
+                    return self.do_singlepart_upload(job)
 
             except Exception as err_msg:
-                real_md5 = common.get_base64_md5(filename)
+                real_md5 = common.get_base64_md5(job.path)
+
+                exc_tb = sys.exc_info()[2]
+                exception_line_num = exc_tb.tb_lineno
+                exception_file = pathlib.Path(exc_tb.tb_frame.f_code.co_filename).name
 
                 if isinstance(err_msg, requests.exceptions.HTTPError):
-                    error_message = f"Upload of {filename} failed with a response code {err_msg.response.status_code} ({err_msg.response.reason}) (expected '{md5}', got '{real_md5}')"
+                    error_message = f"Upload of {job.path} failed with a response code {err_msg.response.status_code} ({err_msg.response.reason}) (expected '{job.md5}', got '{real_md5}')"
                 else:
                     error_message = (
-                        f"Upload of {filename} failed. (expected '{md5}', got '{real_md5}') {str(err_msg)}"
+                        f"Upload of {job.path} failed. (expected '{job.file_md5}', got '{real_md5}') {str(err_msg)} [{exception_file}-{exception_line_num}]"
                     )
 
                 logger.error(error_message)
@@ -430,9 +447,7 @@ class UploadWorker(worker.ThreadWorker):
         return worker.EMPTY_JOB
 
     @common.DecRetry(retry_exceptions=api_client.CONNECTION_EXCEPTIONS, tries=5)
-    def do_singlepart_upload(
-        self, upload_url, filename, file_size, md5, kms_key_name=None
-    ):
+    def do_singlepart_upload(self, job):
         """
         Note that for GCS we don't rely on the make_request's own retry mechanism because we need to
         recreate the chunked_reader generator before retrying the request. Instead, we wrap this
@@ -442,19 +457,23 @@ class UploadWorker(worker.ThreadWorker):
         headers that S3 does not accept.
         """
 
-        if ("amazonaws" in upload_url) or ("coreweave" in upload_url):
+        tq_job = thread_queue_job.MultiPartThreadQueueJob( md5=job.file_md5,
+                                                           path=job.path,
+                                                           total_parts=job.total_parts)
+
+        if job.is_vendor_aws() or job.is_vendor_cw():
             # must declare content-length ourselves due to zero byte bug in requests library.
             # api_client.make_prepared_request docstring.
             headers = {
                 "Content-Type": "application/octet-stream",
-                "Content-Length": str(file_size),
+                "Content-Length": str(job.file_size),
             }
 
-            with open(filename, "rb") as fh:
+            with open(job.path, "rb") as fh:
                 # TODO: support chunked
                 response = self.api_client.make_prepared_request(
                     verb="PUT",
-                    url=upload_url,
+                    url=job.presigned_url,
                     headers=headers,
                     params=None,
                     data=fh,
@@ -468,25 +487,26 @@ class UploadWorker(worker.ThreadWorker):
                 response.close()
 
                 # report upload progress
-                self.metric_store.increment("bytes_uploaded", file_size, filename)
-
-                return response
+                self.metric_store.increment("bytes_uploaded", job.file_size, job.path)
+                
         else:
             headers = {"Content-MD5": md5, "Content-Type": "application/octet-stream"}
 
-            if kms_key_name:
-                headers["x-goog-encryption-kms-key-name"] = kms_key_name
+            if job.kms_key_name is not None:
+                headers["x-goog-encryption-kms-key-name"] = job.kms_key_name
 
-            return self.api_client.make_request(
-                conductor_url=upload_url,
+            self.api_client.make_request(
+                conductor_url=job.presigned_url,
                 headers=headers,
-                data=self.chunked_reader(filename),
+                data=self.chunked_reader(job.path),
                 verb="PUT",
                 tries=1,
                 use_api_key=True,
             )
+        
+        return tq_job
 
-    def do_multipart_upload(self, upload, filename, md5):
+    def do_multipart_upload(self, job):
         """
         Files will be split into partSize returned by the FileAPI and hydrated once all parts are
         uploaded. On successful part upload, response headers will contain an ETag. This value must
@@ -494,42 +514,32 @@ class UploadWorker(worker.ThreadWorker):
         """
         uploads = []
         complete_payload = {
-            "uploadID": upload["uploadID"],
-            "hash": md5,
+            "uploadID": job.upload_id,
+            "hash": job.file_md5,
             "completedParts": [],
             "project": self.project,
         }
 
-        # iterate over parts and upload
-        for part in upload["parts"]:
-            resp_headers = self._do_multipart_upload(
-                upload_url=part["url"],
-                filename=filename,
-                part_number=part["partNumber"],
-                part_size=upload["partSize"],
-            )
+        tq_job = thread_queue_job.MultiPartThreadQueueJob(path=job.path,
+                                                          md5=job.file_md5,
+                                                          total_parts=job.total_parts,
+                                                          part_index=job.part_index)
+        tq_job.upload_id = job.upload_id
+        tq_job.project = self.project
 
-            if resp_headers:
-                uploads.append(upload["uploadID"])
-                completed_part = {
-                    "partNumber": part["partNumber"],
-                    "etag": resp_headers["ETag"].strip('"'),
-                }
-                complete_payload["completedParts"].append(completed_part)
 
-        # Complete multipart upload in order to hydrate file for availability
-        uri_path = "/api/v2/files/multipart/complete"
-        headers = {"Content-Type": "application/json"}
-        self.api_client.make_request(
-            uri_path=uri_path,
-            verb="POST",
-            headers=headers,
-            data=json.dumps(complete_payload),
-            raise_on_error=True,
-            use_api_key=True,
+        resp_headers = self._do_multipart_upload(
+            upload_url=job.presigned_url,
+            filename=job.path,
+            part_number=job.part_index,
+            part_size=job.part_size,
         )
 
-        return uploads
+        if resp_headers:
+            tq_job.part = job.part_index
+            tq_job.etag = resp_headers["ETag"].strip('"')
+
+        return tq_job
 
     @common.DecRetry(retry_exceptions=api_client.CONNECTION_EXCEPTIONS, tries=5)
     def _do_multipart_upload(self, upload_url, filename, part_number, part_size):
@@ -564,6 +574,73 @@ class UploadWorker(worker.ThreadWorker):
 
             return response.headers
 
+
+class MultiPartSiphonWorker(worker.ThreadWorker):
+    def __init__(self, *args, **kwargs):
+        super(MultiPartSiphonWorker, self).__init__(*args, **kwargs)
+        
+        self.api_client = api_client.ApiClient()
+        self.multipart_siphon = {}
+
+    def do_work(self, job, thread_int):
+        """
+        Process files that have already been uploaded.
+
+        If it's a single-part file, add the job to the out queue, so that it can
+        be used to determine if the Upload entity is complete.
+
+        If it's a multi-part upload, collect all the parts together. Once all the
+        parts have been accumulated, mark it as complete and add the file to the
+        out queue.
+        """
+
+        if job:
+
+            if not job.is_multipart():
+                logger.debug("Job is not multipart (%s, %s)", job.total_parts, job.part_index)
+
+            else:
+
+                if job.md5 not in self.multipart_siphon:
+                    self.multipart_siphon[job.md5] = []
+                    
+                    # Add to the task count for this worker.
+                    # -1 because a task has already been added for a single file
+                    # but not all its parts.
+                    old_task_count = self.task_count
+                    self.task_count += job.total_parts - 1
+                    logger.debug("Incrementing task count to %s from %s", self.task_count, old_task_count)
+                
+                self.multipart_siphon[job.md5].append(job)
+
+                if len(self.multipart_siphon[job.md5]) == job.total_parts:
+
+                    complete_payload = {
+                        "uploadID": job.upload_id,
+                        "hash": job.md5,
+                        "completedParts": thread_queue_job.MultiPartThreadQueueJob.aggregate_parts(self.multipart_siphon[job.md5]),
+                        "project": job.project,
+                    }
+
+                    # Complete multipart upload in order to hydrate file for availability
+                    uri_path = "/api/v2/files/multipart/complete"
+                    headers = {"Content-Type": "application/json"}
+                    self.api_client.make_request(
+                        uri_path=uri_path,
+                        verb="POST",
+                        headers=headers,
+                        data=json.dumps(complete_payload),
+                        raise_on_error=True,
+                        use_api_key=True,
+                    )
+
+                    logger.debug("JSON payload: '%s'", json.dumps(complete_payload))
+
+            return job
+ 
+        # make sure we return None, so no message is automatically added to the out_queue
+        return None
+
     def is_complete(self):
         # Get the number of files already uploaded as they are not passed to the Upload
         # worker
@@ -581,11 +658,11 @@ class UploadWorker(worker.ThreadWorker):
                 self.task_count,
             )
 
-            return (queue_size + already_completed_uploads) >= self.task_count
+            return (queue_size) >= self.task_count
 
         else:
             logger.debug("Is complete?: files not initialized yet")
-            return False
+            return False         
 
 
 class Uploader(object):
@@ -639,6 +716,7 @@ class Uploader(object):
             ),
             (FileStatWorker, [], {"thread_count": 1}),
             (UploadWorker, [], {"thread_count": self.args["thread_count"]}),
+            (MultiPartSiphonWorker, [], {"thread_count": 1})
         ]
 
         manager = worker.JobManager(job_description)        

{ciocore-9.1.0b1.dist-info → ciocore-9.1.0b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ciocore
-Version: 9.1.0b1
+Version: 9.1.0b2
 Summary: Core functionality for Conductor's client tools
 Home-page: https://github.com/ConductorTechnologies/ciocore
 Author: conductor

{ciocore-9.1.0b1.dist-info → ciocore-9.1.0b2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ciocore/VERSION,sha256=TCUlj36m9-VV6atKK8EETx-2I8BQY_6CSKTVDGTHeqo,12
+ciocore/VERSION,sha256=V-Kb1TSOiqU51Ac0b71EzlP2Ib7nl3rdmaEL7bhM4iY,12
 ciocore/__init__.py,sha256=aTP7LeeosQA8BZE67gDV4jgfTK5zxmwZRjiTRu_ZWj0,646
 ciocore/api_client.py,sha256=SBxEwAiwn2XtH7T_ipefUbWhczXjoNdNbQBur1RV-Bw,32810
 ciocore/cli.py,sha256=jZ1lOKQiUcrMhsVmD9SVmPMFwHtgDF4SaoAf2-PBS54,15449
@@ -95,7 +95,7 @@ ciocore/downloader/perpetual_downloader.py,sha256=cD7lnBH75-c-ZVVPHZc1vSnDhgJOnG
 ciocore/downloader/registry.py,sha256=_JIOuqpWkJkgJGN33nt-DCvqN9Gw3xeFhzPq4RUxIoE,2903
 ciocore/downloader/reporter.py,sha256=p1NK9k6iQ-jt7lRvZR0xFz0cGb2yo8tQcjlvYKR9SWM,4501
 ciocore/uploader/__init__.py,sha256=hxRFJf5Lo86rtRObFXSjjot8nybQd-SebSfYCbgZwow,24
-ciocore/uploader/_uploader.py,sha256=Xz1sos76FK-BKsgPuErlKQ8HsCAtBKp3pLbCtRVrr9E,38250
+ciocore/uploader/_uploader.py,sha256=Kt4toITJHZDMjRLqRyw_lwe_HOoWz2AigMp2k5heHBI,42291
 ciocore/uploader/thread_queue_job.py,sha256=MzOcetttfWtDfwy-M0_ARwUf8_OjaGjyy-dA_WgNTPE,3416
 ciocore/uploader/upload_stats/__init__.py,sha256=Lg1y4zq1i0cwc6Hh2K1TAQDYymLff49W-uIo1xjcvdI,5309
 ciocore/uploader/upload_stats/stats_formats.py,sha256=giNirtObU66VALWghPFSRhg3q_vw5MvESsnXhb_I3y8,2402
@@ -122,8 +122,8 @@ tests/test_uploader.py,sha256=B1llTJt_fqR6e_V_Jxfw9z73QgkFlEPU87xLYGzt-TQ,2914
 tests/test_validator.py,sha256=2fY66ayNc08PGyj2vTI-V_1yeCWJDngkj2zkUM5TTCI,1526
 tests/mocks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/mocks/glob.py,sha256=J2MH7nqi6NJOHuGdVWxhfeBd700_Ckj6cLh_8jSNkfg,215
-ciocore-9.1.0b1.dist-info/METADATA,sha256=jd_61cbX9eiHy2SgX19hXCdf-gqvcvaOJP_gOwZVG-M,19085
-ciocore-9.1.0b1.dist-info/WHEEL,sha256=qUzzGenXXuJTzyjFah76kDVqDvnk-YDzY00svnrl84w,109
-ciocore-9.1.0b1.dist-info/entry_points.txt,sha256=cCqcALMYbC4d8545V9w0Zysfg9MVuKWhzDQ2er4UfGE,47
-ciocore-9.1.0b1.dist-info/top_level.txt,sha256=SvlM5JlqULzAz00JZWfiUhfjhqDzYzSWssA87zdJl0o,14
-ciocore-9.1.0b1.dist-info/RECORD,,
+ciocore-9.1.0b2.dist-info/METADATA,sha256=g67y_5StIUtJ3HXVWUsECxrGN5X6LrT8NU96amgI_fg,19085
+ciocore-9.1.0b2.dist-info/WHEEL,sha256=qUzzGenXXuJTzyjFah76kDVqDvnk-YDzY00svnrl84w,109
+ciocore-9.1.0b2.dist-info/entry_points.txt,sha256=cCqcALMYbC4d8545V9w0Zysfg9MVuKWhzDQ2er4UfGE,47
+ciocore-9.1.0b2.dist-info/top_level.txt,sha256=SvlM5JlqULzAz00JZWfiUhfjhqDzYzSWssA87zdJl0o,14
+ciocore-9.1.0b2.dist-info/RECORD,,