Support boto3 S3 transfer manager about aws-xray-sdk-python HOT 14 OPEN

cannot find the current segment/subsegment, please make sure you have a segment open
---------------------------------------------------------------------------
SegmentNotFoundException                  Traceback (most recent call last)
<ipython-input-7-677aa688eccd> in <module>
      1 import tempfile
      2 with tempfile.NamedTemporaryFile(mode="r") as f:
----> 3     obj.download_file(f.name)
      4     assert "body" == open(f.name).read()

~/.virtualenvs/project/lib/python3.7/site-packages/boto3/s3/inject.py in object_download_file(self, Filename, ExtraArgs, Callback, Config)
    312     return self.meta.client.download_file(
    313         Bucket=self.bucket_name, Key=self.key, Filename=Filename,
--> 314         ExtraArgs=ExtraArgs, Callback=Callback, Config=Config)
    315 
    316 

~/.virtualenvs/project/lib/python3.7/site-packages/boto3/s3/inject.py in download_file(self, Bucket, Key, Filename, ExtraArgs, Callback, Config)
    170         return transfer.download_file(
    171             bucket=Bucket, key=Key, filename=Filename,
--> 172             extra_args=ExtraArgs, callback=Callback)
    173 
    174 

~/.virtualenvs/project/lib/python3.7/site-packages/boto3/s3/transfer.py in download_file(self, bucket, key, filename, extra_args, callback)
    305             bucket, key, filename, extra_args, subscribers)
    306         try:
--> 307             future.result()
    308         # This is for backwards compatibility where when retries are
    309         # exceeded we need to throw the same error from boto3 instead of

~/.virtualenvs/project/lib/python3.7/site-packages/s3transfer/futures.py in result(self)
    104             # however if a KeyboardInterrupt is raised we want want to exit
    105             # out of this and propogate the exception.
--> 106             return self._coordinator.result()
    107         except KeyboardInterrupt as e:
    108             self.cancel()

~/.virtualenvs/project/lib/python3.7/site-packages/s3transfer/futures.py in result(self)
    263         # final result.
    264         if self._exception:
--> 265             raise self._exception
    266         return self._result
    267 

~/.virtualenvs/project/lib/python3.7/site-packages/s3transfer/tasks.py in _main(self, transfer_future, **kwargs)
    253             # Call the submit method to start submitting tasks to execute the
    254             # transfer.
--> 255             self._submit(transfer_future=transfer_future, **kwargs)
    256         except BaseException as e:
    257             # If there was an exception raised during the submission of task

~/.virtualenvs/project/lib/python3.7/site-packages/s3transfer/download.py in _submit(self, client, config, osutil, request_executor, io_executor, transfer_future, bandwidth_limiter)
    343                 Bucket=transfer_future.meta.call_args.bucket,
    344                 Key=transfer_future.meta.call_args.key,
--> 345                 **transfer_future.meta.call_args.extra_args
    346             )
    347             transfer_future.meta.provide_transfer_size(

~/.virtualenvs/project/lib/python3.7/site-packages/botocore/client.py in _api_call(self, *args, **kwargs)
    355                     "%s() only accepts keyword arguments." % py_operation_name)
    356             # The "self" in this scope is referring to the BaseClient.
--> 357             return self._make_api_call(operation_name, kwargs)
    358 
    359         _api_call.__name__ = str(py_operation_name)

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/ext/botocore/patch.py in _xray_traced_botocore(wrapped, instance, args, kwargs)
     38         name=service,
     39         namespace='aws',
---> 40         meta_processor=aws_meta_processor,
     41     )

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/recorder.py in record_subsegment(self, wrapped, instance, args, kwargs, name, namespace, meta_processor)
    415                 self.context.put_segment(segment)
    416 
--> 417         subsegment = self.begin_subsegment(name, namespace)
    418 
    419         exception = None

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/recorder.py in begin_subsegment(self, name, namespace)
    282         """
    283 
--> 284         segment = self.current_segment()
    285         if not segment:
    286             log.warning("No segment found, cannot begin subsegment %s." % name)

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/recorder.py in current_segment(self)
    265         same thread.
    266         """
--> 267         entity = self.get_trace_entity()
    268         if self._is_subsegment(entity):
    269             return entity.parent_segment

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/recorder.py in get_trace_entity(self)
    367         A pass through method to ``context.get_trace_entity()``.
    368         """
--> 369         return self.context.get_trace_entity()
    370 
    371     def set_trace_entity(self, trace_entity):

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/context.py in get_trace_entity(self)
     91         """
     92         if not getattr(self._local, 'entities', None):
---> 93             return self.handle_context_missing()
     94 
     95         return self._local.entities[-1]

~/.virtualenvs/project/lib/python3.7/site-packages/aws_xray_sdk/core/context.py in handle_context_missing(self)
    116         if self.context_missing == 'RUNTIME_ERROR':
    117             log.error(MISSING_SEGMENT_MSG)
--> 118             raise SegmentNotFoundException(MISSING_SEGMENT_MSG)
    119         else:
    120             log.error(MISSING_SEGMENT_MSG)

SegmentNotFoundException: cannot find the current segment/subsegment, please make sure you have a segment open

from aws_xray_sdk import global_sdk_config

def disable_xray():
    os.environ['AWS_XRAY_SDK_ENABLED'] = "0"
    global_sdk_config.set_sdk_enabled(False)


def enable_xray():
    os.environ['AWS_XRAY_SDK_ENABLED'] = "1"
    global_sdk_config.set_sdk_enabled(True)

disable_xray()
do_s3_stuff()
enable_xray()