From b579f4065ad71970e6ab127933897b2b7b68ad2b Mon Sep 17 00:00:00 2001
From: bashmak <nerosketch@gmail.com>
Date: Thu, 30 Mar 2017 16:31:56 +0300
Subject: [PATCH] =?UTF-8?q?=D0=AD=D1=82=D0=BE=D1=82=20=D1=81=D0=BA=D1=80?=
 =?UTF-8?q?=D0=B8=D0=BF=D1=82=20=D1=81=D0=BE=D0=B5=D0=B4=D0=B8=D0=BD=D0=B8?=
 =?UTF-8?q?=D0=B5=D1=82=20dhcp=20=D1=81=D0=B5=D1=80=D0=B2=D0=B5=D1=80=20?=
 =?UTF-8?q?=D1=81=20=D0=B1=D0=B8=D0=BB=D0=BB=D0=B8=D0=BD=D0=B3=D0=BE=D0=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dhcp_lever.py | 92 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100755 dhcp_lever.py

diff --git a/dhcp_lever.py b/dhcp_lever.py
new file mode 100755
index 0000000..f168f39
--- /dev/null
+++ b/dhcp_lever.py
@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+import os
+import sys
+import django
+from django.core.exceptions import MultipleObjectsReturned, ValidationError
+from django.utils.translation import ugettext as _
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "djing.settings")
+django.setup()
+from agent import NasFailedResult, NasNetworkError
+from abonapp.models import Abon, Opt82
+from ip_pool.models import IpPoolItem
+
+
+def die(text):
+    print(text)
+    exit(1)
+
+
+def get_82_opts(switch_mac, switch_port):
+    try:
+        opt82 = Opt82.objects.get(mac=switch_mac, port=switch_port)
+    except MultipleObjectsReturned:
+        Opt82.objects.filter(mac=switch_mac, port=switch_port)[1:].delete()
+        return get_82_opts(switch_mac, switch_port)
+    except Opt82.DoesNotExist:
+        opt82 = Opt82.objects.create(mac=switch_mac, port=switch_port)
+    return opt82
+
+
+def get_or_create_pool_item(ip):
+    try:
+        ip_item = IpPoolItem.objects.get(ip=ip)
+    except IpPoolItem.DoesNotExist:
+        ip_item = IpPoolItem.objects.create(ip=ip)
+    except MultipleObjectsReturned:
+        IpPoolItem.objects.filter(ip=ip)[1:].delete()
+        return get_or_create_pool_item(ip)
+    return ip_item
+
+
+def dhcp_commit(client_ip, client_mac, switch_mac, switch_port):
+    opt82 = get_82_opts(switch_mac, switch_port)
+    if opt82 is None:
+        print(_("ERROR: opt82 with mac:%s and port:%d does not exist in db") % (switch_mac, switch_port))
+        return
+    try:
+        abon = Abon.objects.get(opt82=opt82)
+        abon.ip_address = get_or_create_pool_item(client_ip)
+        Abon.is_dhcp = True
+        abon.save(update_fields=['ip_address'])
+        print(_('Ip address update for %s successfull') % abon.get_short_name())
+    except Abon.DoesNotExist:
+        print('ERROR: abon with option82(%s-%d) does not exist' % (opt82.mac, opt82.port))
+
+
+def dhcp_expiry(client_ip):
+    try:
+        ip_item = IpPoolItem.objects.get(ip=client_ip)
+        abon = Abon.objects.get(ip_address=ip_item)
+        abon.ip_address = None
+        abon.save(update_fields=['ip_address'])
+    except IpPoolItem.DoesNotExist:
+        pass
+    except Abon.DoesNotExist:
+        pass
+
+
+def dhcp_release(client_ip):
+    dhcp_expiry(client_ip)
+
+
+def main(argv):
+    if len(argv) < 3:
+        die(_('Too few arguments, exiting...'))
+    action = argv[1]
+    if action == 'commit':
+        if len(argv) < 6:
+            die(_('Too few arguments, exiting...'))
+        dhcp_commit(argv[2], argv[3], argv[4], int(argv[5]))
+    elif action == 'expiry':
+        dhcp_expiry(argv[2])
+    elif action == 'release':
+        dhcp_release(argv[2])
+
+
+if __name__ == "__main__":
+    try:
+        main(sys.argv)
+    except (NasNetworkError, NasFailedResult) as e:
+        print('NAS:', e)
+    except (ValidationError, ValueError) as e:
+        print('ERROR:', e)