• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-scarface-master
  • start.command
  • requirements-test.txt
  • mkdocs.yml
  • CHANGELOG.MD
  • scarface
    • apps.py
    • exceptions.py
    • models.py
    • views.py
    • forms.py
    • tests.py
    • __init__.py
    • utils.py
    • migrations
      • 0002_auto_20151217_1017.py
      • 0001_initial.py
      • 0003_auto_20151217_1101.py
      • __init__.py
      • 0004_auto_20151217_1131.py
    • settings.py
    • admin.py
    • platform_strategy.py
    • signals.py
    • management
      • commands
        • extract_keys.py
        • __init__.py
      • __init__.py
  • LICENSE
  • manage.py
  • setup.py
  • __init__.py
  • .travis.yml
  • README.md
  • pytest.ini
  • requirements.txt
  • .gitignore
  • docs
    • index.md
  • MANIFEST.in
  • django_scarface
    • wsgi.py
    • pep8.rc
    • pylint.rc
    • urls.py
    • __init__.py
    • settings
      • unit_tests.py
      • __init__.py
      • test.py
      • base.py
  • tox.ini

# -*- coding: utf-8 -*-
import inspect
from functools import partial

from boto import sns
from django.conf import settings

__author__ = 'dreipol GmbH'


class Decorator(object):
    def __init__(self, function, *args, **kwargs):
        self.function = function

    @property
    def original_function(self):
        function = self.function
        while isinstance(function, Decorator):
            function = function.function
        return function


class DefaultConnection(Decorator):
    def __get__(self, obj, objtype=None):
        if obj is None:
            return self.function
        return partial(self, obj)

    def __call__(self, *args, **kwargs):
        connection_keyword = 'connection'
        if len(args) + len(kwargs) == 0:
            call_kwargs = dict()
        else:
            call_kwargs = inspect.getcallargs(
                self.original_function,
                *args,
                **kwargs
            )
        if not call_kwargs.get(connection_keyword, None):
            call_kwargs[connection_keyword] = get_sns_connection()
        return self.function(**call_kwargs)


class PushLogger(Decorator):
    def __get__(self, obj, objtype=None):
        if obj is None:
            return self.function
        self.obj = obj
        return partial(self, obj)

    def __call__(self, *args, **kwargs):
        call_kwargs = inspect.getcallargs(self.original_function, *args,
            **kwargs)
        push_message = call_kwargs.get('push_message')
        self.obj.sign(push_message)
        if logging_enabled():
            push_message.save()
        return self.function(*args, **kwargs)


def get_sns_connection():
    """
    Creates a new AWS connection based upon the credentials defined in the django configuration
    :param region: the region of the DynamoDB, defaults to Ireland
    :return: a new dynamodb2 connection
    """

    region = settings.SCARFACE_REGION_NAME if hasattr(settings, "SCARFACE_REGION_NAME") else 'eu-west-1'

    return sns.connect_to_region(
        region, aws_access_key_id=settings.AWS_ACCESS_KEY,
        aws_secret_access_key=settings.AWS_SECRET_ACCESS_KEY
    )


def logging_enabled():
    return settings.SCARFACE_LOGGING_ENABLED if hasattr(
        settings,
        'SCARFACE_LOGGING_ENABLED'
    ) else True