Browse Source
Multiple changes:
Multiple changes:
1) Makes mixin for API views. 2) Makes device monitoring notification. And little bit fixesdevel
bashmak
8 years ago
16 changed files with 332 additions and 60 deletions
-
68agent/monitoring_agent.py
-
7chatbot/telebot.py
-
2devapp/forms.py
-
44devapp/locale/ru/LC_MESSAGES/django.po
-
34devapp/migrations/0006_auto_20180129_1625.py
-
17devapp/models.py
-
6devapp/templates/devapp/add_dev.html
-
2devapp/templates/devapp/dev.html
-
20devapp/templates/devapp/devices.html
-
6devapp/urls.py
-
112devapp/views.py
-
57djing/global_base_views.py
-
6djing/settings.py
-
4djing/views.py
-
1requirements.txt
-
6taskapp/handle.py
@ -0,0 +1,68 @@ |
|||
#!/usr/bin/env python3 |
|||
import sys |
|||
import re |
|||
from hashlib import sha256 |
|||
import requests |
|||
|
|||
API_AUTH_SECRET = 'asihdfaoisydoiayosidyaoisydoiasydaisydasd' |
|||
|
|||
SERVER_DOMAIN = 'http://localhost:8000' |
|||
|
|||
|
|||
IP_REGEXP = r'^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.' \ |
|||
r'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.' \ |
|||
r'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.' \ |
|||
r'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$' |
|||
|
|||
|
|||
def calc_hash(data): |
|||
if type(data) is str: |
|||
result_data = data.encode('utf-8') |
|||
else: |
|||
result_data = bytes(data) |
|||
return sha256(result_data).hexdigest() |
|||
|
|||
|
|||
def check_sign(get_list, sign): |
|||
hashed = '_'.join(get_list) |
|||
my_sign = calc_hash(hashed) |
|||
return sign == my_sign |
|||
|
|||
|
|||
def validate(regexp, string): |
|||
if not bool(re.match(regexp, string)): |
|||
raise ValueError |
|||
return string |
|||
|
|||
|
|||
def validate_status(text): |
|||
if not text in ('UP', 'DOWN', 'UNREACHABLE'): |
|||
raise ValueError |
|||
return text |
|||
|
|||
|
|||
def send_request(ip, status, sign): |
|||
r = requests.get( |
|||
"%(domain)s/dev/on_device_down/" % {'domain': SERVER_DOMAIN}, |
|||
params={ |
|||
'ip': ip, |
|||
'status': status, |
|||
'sign': sign |
|||
}) |
|||
if r.status_code == 200: |
|||
print(r.json()) |
|||
else: |
|||
print('Status:', r.status_code, r.text) |
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
if len(sys.argv) < 3: |
|||
print('You forget parameters, example of usage:\n' |
|||
'$ python3 ./monitoring_agent.py 192.168.0.100 DOWN|UP|UNREACHABLE') |
|||
exit(0) |
|||
dev_ip = validate(IP_REGEXP, sys.argv[1]) |
|||
status = validate_status(sys.argv[2]) |
|||
|
|||
sign = calc_hash('_'.join((dev_ip, status, API_AUTH_SECRET))) |
|||
|
|||
send_request(dev_ip, status, sign) |
|||
@ -0,0 +1,34 @@ |
|||
# -*- coding: utf-8 -*- |
|||
# Generated by Django 1.11 on 2018-01-29 16:25 |
|||
from __future__ import unicode_literals |
|||
|
|||
from django.db import migrations, models |
|||
|
|||
|
|||
class Migration(migrations.Migration): |
|||
|
|||
dependencies = [ |
|||
('devapp', '0005_device_snmp_item_num'), |
|||
] |
|||
|
|||
operations = [ |
|||
migrations.AlterModelOptions( |
|||
name='device', |
|||
options={'ordering': ['comment'], 'permissions': (('can_view_device', 'Can view device'),), 'verbose_name': 'Device', 'verbose_name_plural': 'Devices'}, |
|||
), |
|||
migrations.AddField( |
|||
model_name='device', |
|||
name='is_noticeable', |
|||
field=models.BooleanField(default=False, verbose_name='Send notify when monitoring state changed'), |
|||
), |
|||
migrations.AddField( |
|||
model_name='device', |
|||
name='status', |
|||
field=models.CharField(choices=[('und', 'Undefined'), ('up', 'Up'), ('unr', 'Unreachable'), ('dwn', 'Down')], default='und', max_length=3, verbose_name='Status'), |
|||
), |
|||
migrations.AlterField( |
|||
model_name='device', |
|||
name='snmp_item_num', |
|||
field=models.PositiveSmallIntegerField(blank=True, default=0, verbose_name='SNMP Number'), |
|||
), |
|||
] |
|||
@ -0,0 +1,57 @@ |
|||
from hashlib import sha256 |
|||
from django.views.generic.base import View |
|||
from django.http.response import HttpResponseForbidden |
|||
from django.conf import settings |
|||
from netaddr import IPNetwork, IPAddress |
|||
|
|||
|
|||
API_AUTH_SECRET = getattr(settings, 'API_AUTH_SECRET') |
|||
API_AUTH_SUBNET = getattr(settings, 'API_AUTH_SUBNET') |
|||
|
|||
|
|||
class HashAuthView(View): |
|||
|
|||
@staticmethod |
|||
def calc_hash(data): |
|||
if type(data) is str: |
|||
result_data = data.encode('utf-8') |
|||
else: |
|||
result_data = bytes(data) |
|||
return sha256(result_data).hexdigest() |
|||
|
|||
@staticmethod |
|||
def check_sign(get_list, sign): |
|||
hashed = '_'.join(get_list) |
|||
my_sign = HashAuthView.calc_hash(hashed) |
|||
return sign == my_sign |
|||
|
|||
def __init__(self, *args, **kwargs): |
|||
if API_AUTH_SECRET is None: |
|||
raise ImportError('You must specified API_AUTH_SECRET is settings') |
|||
else: |
|||
super(HashAuthView, self).__init__(*args, **kwargs) |
|||
|
|||
def dispatch(self, request, *args, **kwargs): |
|||
sign = request.GET.get('sign') |
|||
|
|||
# Transmittent get list without sign |
|||
get_values = request.GET.copy() |
|||
del get_values['sign'] |
|||
if HashAuthView.check_sign(list(get_values.values()) + [API_AUTH_SECRET], sign): |
|||
return super(HashAuthView, self).dispatch(request, *args, **kwargs) |
|||
else: |
|||
return HttpResponseForbidden('Access Denied') |
|||
|
|||
|
|||
class AllowedSubnetMixin(object): |
|||
|
|||
def dispatch(self, request, *args, **kwargs): |
|||
""" |
|||
Check if user ip in allowed subnet. |
|||
Return 403 denied otherwise. |
|||
""" |
|||
ip = IPAddress(request.META.get('REMOTE_ADDR')) |
|||
if ip in IPNetwork(API_AUTH_SUBNET): |
|||
return super(AllowedSubnetMixin, self).dispatch(request, *args, **kwargs) |
|||
else: |
|||
return HttpResponseForbidden('Access Denied') |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue