Initial commit.

Author: ranrib

Diff for: .gitignore

@@ -0,0 +1,10 @@
+.idea
+*.pyc
+.pypirc
+.coverage
+.cache
+dist
+.eggs
+.pytest_cache
+build
+epsagon.egg-info

Diff for: .pylintrc

@@ -0,0 +1,19 @@
+# Based on:
+# https://github.com/kpreid/shinysdr/blob/master/pylintrc
+
+[MASTER]
+# XXX should be changed to your project folder
+# see http://stackoverflow.com/a/37238692/3828891 for explanation
+init-hook='base_dir="epsagon"; import sys,os,re; _re=re.search(r".+\/" + base_dir, os.getcwd()); project_dir = _re.group() if _re else os.path.join(os.getcwd(), base_dir); sys.path.append(project_dir)'
+
+[MESSAGES CONTROL]
+# Find available symbolic names in:
+# https://docs.pylint.org/features.html
+disable=duplicate-code,too-few-public-methods,too-many-arguments,fixme,too-many-instance-attributes,bad-continuation,useless-object-inheritance
+
+[FORMAT]
+# Maximum number of characters on a single line.
+max-line-length=80
+
+# Maximum number of lines in a module
+max-module-lines=500

Diff for: .travis.yml

@@ -0,0 +1,13 @@
+language: python
+notifications:
+    slack: epsagon:Q0XcrHSJRI5YkYftfUzl83Lx
+python:
+  - "2.7"
+  - "3.6"
+install:
+  - pip install -r requirements-dev.txt
+  - pip install pylint
+before_script:
+  - "pylint --msg-template='{path}:{line}: [{msg_id}({symbol}) {obj}] {msg}' epsagon/"
+script:
+  - pytest

Diff for: LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2018 Epsagon
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Diff for: MANIFEST.in

@@ -0,0 +1,2 @@
+include *.py *.md *.txt LICENSE
+recursive-include epsagon

Diff for: README.md

@@ -0,0 +1,43 @@
+# Epsagon Instrumentation for Python
+[![Build Status](https://travis-ci.com/epsagon/epsagon-python.svg?token=wsveVqcNtBtmq6jpZfSf&branch=master)](https://travis-ci.com/epsagon/epsagon-python)
+[![Pyversions](https://img.shields.io/pypi/pyversions/epsagon.svg?style=flat)](https://pypi.org/project/epsagon/)
+[![PypiVersions](https://img.shields.io/pypi/v/epsagon.svg)](https://pypi.org/project/epsagon/)
+
+This package provides an instrumentation to Python code running on functions for collection of distributed tracing and performance monitoring.
+
+
+## Installation
+
+From your project directory:
+
+```
+$ pip install epsagon -t .
+
+# If running in virtualenv:
+$ pip install epsagon
+```
+
+More details about lambda deployments are available in the [AWS documentation](https://docs.aws.amazon.com/lambda/latest/dg/lambda-python-how-to-create-deployment-package.html).
+
+## Basic Usage
+
+Simply use our decorator to report metrics:
+
+```python
+import epsagon
+epsagon.init(
+    token='my-secret-token',
+    app_name='my-app-name',  # Optional, used in the dashboard
+    metadata_only=False,  # Optional, send more trace data
+)
+
+@epsagon.lambda_wrapper
+def handler(event, context):
+  pass
+```
+
+## Copyright
+
+Provided under the MIT license. See LICENSE for details.
+
+Copyright 2018, Epsagon.

Diff for: epsagon/__init__.py

@@ -0,0 +1,56 @@
+"""
+Epsagon's init.
+"""
+
+from __future__ import absolute_import
+import os
+from .utils import init
+from .patcher import patch_all
+from .constants import __version__
+from .trace import tracer
+
+
+def dummy_wrapper(func):
+    """
+    A dummy wrapper for when Epsagon is disabled
+    :param func: The function to wrap
+    :return: The same function, unchanged
+    """
+    return func
+
+
+if os.environ.get('DISABLE_EPSAGON') == 'TRUE':
+    os.environ['DISABLE_EPSAGON_PATCH'] = 'TRUE'
+    lambda_wrapper = dummy_wrapper  # pylint: disable=C0103
+    step_lambda_wrapper = dummy_wrapper  # pylint: disable=C0103
+    azure_wrapper = dummy_wrapper  # pylint: disable=C0103
+    python_wrapper = dummy_wrapper  # pylint: disable=C0103
+    EpsagonFlask = dummy_wrapper  # pylint: disable=C0103
+else:
+    # Environments.
+    from .wrappers import (
+        lambda_wrapper,
+        step_lambda_wrapper,
+        azure_wrapper,
+        python_wrapper
+    )
+
+    # Frameworks.
+    try:
+        from .wrappers.flask import FlaskWrapper as flask_wrapper
+    except ImportError:
+        flask_wrapper = dummy_wrapper
+
+
+# pylint: disable=C0103
+log = tracer.add_log
+# pylint: disable=C0103
+error = tracer.add_error
+
+__all__ = ['lambda_wrapper', 'azure_wrapper', 'python_wrapper', 'init',
+           'step_lambda_wrapper', 'flask_wrapper', 'log', 'error']
+
+
+# The modules are patched only if DISABLE_EPSAGON_PATCH variable is NOT 'TRUE'
+if os.environ.get('DISABLE_EPSAGON_PATCH') != 'TRUE':
+    patch_all()

Diff for: epsagon/common.py

@@ -0,0 +1,17 @@
+"""Common objects"""
+
+
+class ErrorCode(object):
+    """
+    Error codes enum
+    """
+    OK = 0
+    ERROR = 1
+    EXCEPTION = 2
+
+
+class EpsagonWarning(Warning):
+    """
+    An Epsagon warning.
+    """
+    pass

Diff for: epsagon/constants.py

@@ -0,0 +1,16 @@
+"""General constants"""
+
+import os
+
+__version__ = '1.0.0'
+
+DEFAULT_REGION = 'us-east-1'
+REGION = os.environ.get('AWS_REGION', DEFAULT_REGION)
+
+TRACE_COLLECTOR_URL = "{protocol}{region}.tc.epsagon.com"
+COLD_START = True
+
+# How long we try to send traces.
+SEND_TIMEOUT = 1
+
+MAX_MESSAGE_SIZE = 512

Diff for: epsagon/event.py

@@ -0,0 +1,104 @@
+"""
+Base Event class
+"""
+
+from __future__ import absolute_import
+import time
+from .common import ErrorCode
+
+
+class BaseEvent(object):
+    """
+    Represents base trace's event
+    """
+
+    ORIGIN = 'base'
+    RESOURCE_TYPE = 'base'
+
+    def __init__(self, start_time):
+        """
+        Initialize.
+        :param start_time: event's start time (epoch)
+        """
+
+        self.start_time = start_time
+        self.event_id = ''
+        self.origin = self.ORIGIN
+        self.duration = 0.0
+        self.error_code = ErrorCode.OK
+        self.exception = {}
+
+        self.resource = {
+            'type': self.RESOURCE_TYPE,
+            'name': '',
+            'operation': '',
+            'metadata': {},
+        }
+
+    @staticmethod
+    def load_from_dict(event_data):
+        """
+        Load Event object from dict.
+        :param event_data: dict
+        :return: Event
+        """
+
+        event = BaseEvent(event_data['start_time'])
+        event.event_id = event_data['id']
+        event.origin = event_data['origin']
+        event.duration = event_data['duration']
+        event.error_code = event_data['error_code']
+        event.resource = event_data['resource']
+        if event.error_code == ErrorCode.EXCEPTION:
+            event.exception = event_data['exception']
+
+        return event
+
+    def to_dict(self):
+        """
+        Converts Event to dict.
+        :return: dict
+        """
+
+        self_as_dict = {
+            'id': self.event_id,
+            'start_time': self.start_time,
+            'duration': self.duration,
+            'origin': self.origin,
+            'error_code': self.error_code,
+            'resource': self.resource,
+        }
+
+        if self.error_code == ErrorCode.EXCEPTION:
+            self_as_dict['exception'] = self.exception
+
+        return self_as_dict
+
+    def terminate(self):
+        """
+        Sets duration time.
+        :return: None
+        """
+        self.duration = time.time() - self.start_time
+
+    def set_error(self):
+        """
+        Sets general error.
+        :return: None
+        """
+
+        self.error_code = ErrorCode.ERROR
+
+    def set_exception(self, exception, traceback_data):
+        """
+        Sets exception data on event.
+        :param exception: Exception object
+        :param traceback_data: traceback string
+        :return: None
+        """
+
+        self.error_code = ErrorCode.EXCEPTION
+        self.exception['type'] = type(exception).__name__
+        self.exception['message'] = str(exception)
+        self.exception['traceback'] = traceback_data
+        self.exception['time'] = time.time()