Merge pull request #766 from DataDog/0.17.1-dev

v0.17.1

Author: Kyle-Verhoog

ddtrace/__init__.py

@@ -4,7 +4,7 @@
 from .tracer import Tracer
 from .settings import config
 
-__version__ = '0.17.0'
+__version__ = '0.17.1'
 
 # a global tracer instance with integration settings
 tracer = Tracer()

ddtrace/api.py

@@ -18,14 +18,15 @@
                       'compatibility_mode': False,
                       'fallback': 'v0.3'},
              'v0.3': {'traces': '/v0.3/traces',
-                     'services': '/v0.3/services',
+                      'services': '/v0.3/services',
                       'compatibility_mode': False,
                       'fallback': 'v0.2'},
              'v0.2': {'traces': '/v0.2/traces',
-                     'services': '/v0.2/services',
+                      'services': '/v0.2/services',
                       'compatibility_mode': True,
                       'fallback': None}}
 
+
 def _parse_response_json(response):
     """
     Parse the content of a response object, and return the right type,
@@ -48,6 +49,7 @@ def _parse_response_json(response):
         except (ValueError, TypeError) as err:
             log.debug("unable to load JSON '%s': %s" % (body, err))
 
+
 class API(object):
     """
     Send data to the trace agent using the HTTP protocol and JSON format

ddtrace/commands/ddtrace_run.py

@@ -33,6 +33,7 @@
     DATADOG_PRIORITY_SAMPLING=true|false : (default: false): enables Priority Sampling.
 """ # noqa
 
+
 def _ddtrace_root():
     from ddtrace import __file__
     return os.path.dirname(__file__)
@@ -46,8 +47,7 @@ def _add_bootstrap_to_pythonpath(bootstrap_dir):
     python_path = os.environ.get('PYTHONPATH', '')
 
     if python_path:
-        new_path = "%s%s%s" % (bootstrap_dir, os.path.pathsep,
-                os.environ['PYTHONPATH'])
+        new_path = "%s%s%s" % (bootstrap_dir, os.path.pathsep, os.environ['PYTHONPATH'])
         os.environ['PYTHONPATH'] = new_path
     else:
         os.environ['PYTHONPATH'] = bootstrap_dir

ddtrace/contrib/bottle/patch.py

@@ -1,4 +1,3 @@
-
 import os
 
 from .trace import TracePlugin
@@ -7,6 +6,7 @@
 
 import wrapt
 
+
 def patch():
     """Patch the bottle.Bottle class
     """
@@ -16,6 +16,7 @@ def patch():
     setattr(bottle, '_datadog_patch', True)
     wrapt.wrap_function_wrapper('bottle', 'Bottle.__init__', traced_init)
 
+
 def traced_init(wrapped, instance, args, kwargs):
     wrapped(*args, **kwargs)
 

ddtrace/contrib/bottle/trace.py

@@ -10,6 +10,7 @@
 
 SPAN_TYPE = 'web'
 
+
 class TracePlugin(object):
     name = 'trace'
     api = 2

ddtrace/contrib/cassandra/session.py

@@ -25,22 +25,26 @@
 # Original connect connect function
 _connect = cassandra.cluster.Cluster.connect
 
+
 def patch():
     """ patch will add tracing to the cassandra library. """
     setattr(cassandra.cluster.Cluster, 'connect',
             wrapt.FunctionWrapper(_connect, traced_connect))
     Pin(service=SERVICE, app=SERVICE, app_type="db").onto(cassandra.cluster.Cluster)
 
+
 def unpatch():
     cassandra.cluster.Cluster.connect = _connect
 
+
 def traced_connect(func, instance, args, kwargs):
     session = func(*args, **kwargs)
     if not isinstance(session.execute, wrapt.FunctionWrapper):
         # FIXME[matt] this should probably be private.
         setattr(session, 'execute_async', wrapt.FunctionWrapper(session.execute_async, traced_execute_async))
     return session
 
+
 def _close_span_on_success(result, future):
     span = getattr(future, CURRENT_SPAN, None)
     if not span:
@@ -54,11 +58,13 @@ def _close_span_on_success(result, future):
         span.finish()
         delattr(future, CURRENT_SPAN)
 
+
 def traced_set_final_result(func, instance, args, kwargs):
     result = args[0]
     _close_span_on_success(result, instance)
     return func(*args, **kwargs)
 
+
 def _close_span_on_error(exc, future):
     span = getattr(future, CURRENT_SPAN, None)
     if not span:
@@ -76,11 +82,13 @@ def _close_span_on_error(exc, future):
         span.finish()
         delattr(future, CURRENT_SPAN)
 
+
 def traced_set_final_exception(func, instance, args, kwargs):
     exc = args[0]
     _close_span_on_error(exc, instance)
     return func(*args, **kwargs)
 
+
 def traced_start_fetching_next_page(func, instance, args, kwargs):
     has_more_pages = getattr(instance, 'has_more_pages', True)
     if not has_more_pages:
@@ -106,11 +114,12 @@ def traced_start_fetching_next_page(func, instance, args, kwargs):
     setattr(instance, CURRENT_SPAN, span)
     try:
         return func(*args, **kwargs)
-    except:
+    except Exception:
         with span:
             span.set_exc_info(*sys.exc_info())
         raise
 
+
 def traced_execute_async(func, instance, args, kwargs):
     cluster = getattr(instance, 'cluster', None)
     pin = Pin.get_from(cluster)
@@ -161,11 +170,12 @@ def traced_execute_async(func, instance, args, kwargs):
         )
         result.clear_callbacks()
         return result
-    except:
+    except Exception:
         with span:
             span.set_exc_info(*sys.exc_info())
         raise
 
+
 def _start_span_and_set_tags(pin, query, session, cluster):
     service = pin.service
     tracer = pin.tracer
@@ -175,6 +185,7 @@ def _start_span_and_set_tags(pin, query, session, cluster):
     span.set_tags(_extract_cluster_metas(cluster))
     return span
 
+
 def _extract_session_metas(session):
     metas = {}
 
@@ -185,6 +196,7 @@ def _extract_session_metas(session):
 
     return metas
 
+
 def _extract_cluster_metas(cluster):
     metas = {}
     if deep_getattr(cluster, "metadata.cluster_name"):
@@ -194,6 +206,7 @@ def _extract_cluster_metas(cluster):
 
     return metas
 
+
 def _extract_result_metas(result):
     metas = {}
     if result is None:
@@ -230,6 +243,7 @@ def _extract_result_metas(result):
 
     return metas
 
+
 def _sanitize_query(span, query):
     # TODO (aaditya): fix this hacky type check. we need it to avoid circular imports
     t = type(query).__name__
@@ -250,7 +264,7 @@ def _sanitize_query(span, query):
     elif t == 'str':
         resource = query
     else:
-        resource = 'unknown-query-type' # FIXME[matt] what else do to here?
+        resource = 'unknown-query-type'  # FIXME[matt] what else do to here?
 
     span.resource = stringify(resource)[:RESOURCE_MAX_LENGTH]
 

ddtrace/contrib/celery/signals.py

@@ -13,10 +13,10 @@
     retrieve_span,
 )
 
-
 log = logging.getLogger(__name__)
 SPAN_TYPE = 'worker'
 
+
 def trace_prerun(*args, **kwargs):
     # safe-guard to avoid crashes in case the signals API
     # changes in Celery

ddtrace/contrib/celery/task.py

@@ -18,6 +18,7 @@ def patch_task(task, pin=None):
     patch_app(task.app)
     return task
 
+
 def unpatch_task(task):
     """Deprecated API. The new API uses signals that can be deactivated
     via unpatch() API. This API is now a no-op implementation so it doesn't

ddtrace/contrib/dbapi/__init__.py

@@ -153,6 +153,7 @@ def rollback(self, *args, **kwargs):
         span_name = '{}.{}'.format(self._self_datadog_name, 'rollback')
         return self._trace_method(self.__wrapped__.rollback, span_name, {}, *args, **kwargs)
 
+
 def _get_vendor(conn):
     """ Return the vendor (e.g postgres, mysql) of the given
         database.

ddtrace/contrib/django/cache.py

@@ -54,8 +54,7 @@ def _trace_operation(fn, method_name):
         def wrapped(self, *args, **kwargs):
             # get the original function method
             method = getattr(self, DATADOG_NAMESPACE.format(method=method_name))
-            with tracer.trace('django.cache',
-                    span_type=TYPE, service=cache_service_name) as span:
+            with tracer.trace('django.cache', span_type=TYPE, service=cache_service_name) as span:
                 # update the resource name and tag the cache backend
                 span.resource = _resource_from_cache_prefix(method_name, self)
                 cache_backend = '{}.{}'.format(self.__module__, self.__class__.__name__)
@@ -93,6 +92,7 @@ def _wrap_method(cls, method_name):
         for method in TRACED_METHODS:
             _wrap_method(cache, method)
 
+
 def unpatch_method(cls, method_name):
     method = getattr(cls, DATADOG_NAMESPACE.format(method=method_name), None)
     if method is None:
@@ -101,6 +101,7 @@ def unpatch_method(cls, method_name):
     setattr(cls, method_name, method)
     delattr(cls, DATADOG_NAMESPACE.format(method=method_name))
 
+
 def unpatch_cache():
     cache_backends = set([cache['BACKEND'] for cache in django_settings.CACHES.values()])
     for cache_module in cache_backends: