From cea539b5d5b24d43e8e0b18f5d3c88bb24fc5e02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ma=C3=ABl=20Kervella?= <kervella.mael@gmail.com>
Date: Sat, 26 May 2018 21:04:14 +0000
Subject: [PATCH] Check and update need_regen info

---
 main.py | 73 ++++++++++++++++++++++++++++++---------------------------
 re2oapi |  2 +-
 2 files changed, 40 insertions(+), 35 deletions(-)

diff --git a/main.py b/main.py
index f211888..7bc7e55 100644
--- a/main.py
+++ b/main.py
@@ -1,43 +1,48 @@
-import time
-t1 = time.time()
-
 from configparser import ConfigParser
+import socket
 
 from re2oapi import Re2oAPIClient
 
 config = ConfigParser()
 config.read('config.ini')
 
-hostname = config.get('Re2o', 'hostname')
-password = config.get('Re2o', 'password')
-username = config.get('Re2o', 'username')
-
+api_hostname = config.get('Re2o', 'hostname')
+api_password = config.get('Re2o', 'password')
+api_username = config.get('Re2o', 'username')
+
+def regen_dhcp(api_client):
+    for hmi in api_client.list_hostmacip():
+        if hmi['extension'] not in host_mac_ip.keys():
+            host_mac_ip[hmi['extension']] = []
+        host_mac_ip[hmi['extension']].append((hmi['hostname'],
+                                              hmi['mac_address'],
+                                              hmi['ipv4']))
+
+    template = ("host {hostname}{extension} {{\n"
+                "    hardware ethernet {mac_address};\n"
+                "    fixed-address {ipv4};\n"
+                "}}")
+
+    for extension, hmi_list in host_mac_ip.items():
+        dhcp_leases_content = '\n\n'.join(template.format(
+                hostname=hostname,
+                extension=extension,
+                mac_address=mac_address,
+                ipv4=ipv4
+            ) for hostname, mac_address, ipv4 in hmi_list)
+
+        filename = 'dhcp{extension}.list'.format(extension=extension)
+        with open(filename, 'w+') as f:
+            f.write(dhcp_leases_content)
 
 host_mac_ip = {}
-api_client = Re2oAPIClient(hostname, username, password)
-
-for hmi in api_client.list_hostmacip(params={'page_size': 'all'}):
-    if hmi['extension'] not in host_mac_ip.keys():
-        host_mac_ip[hmi['extension']] = []
-    host_mac_ip[hmi['extension']].append((hmi['hostname'],
-                                          hmi['mac_address'],
-                                          hmi['ipv4']))
-    
-template = ("host {hostname}{extension} {{\n"
-            "    hardware ethernet {mac_address};\n"
-            "    fixed-address {ipv4};\n"
-            "}}")
-
-for extension, hmi_list in host_mac_ip.items():
-    dhcp_leases_content = '\n\n'.join(template.format(
-            hostname=hostname,
-            extension=extension,
-            mac_address=mac_address,
-            ipv4=ipv4
-        ) for hostname, mac_address, ipv4 in hmi_list)
-    
-    filename = 'dhcp{extension}.list'.format(extension=extension)
-    with open(filename, 'w+') as f:
-        f.write(dhcp_leases_content)
-
-print(time.time() - t1)
+api_client = Re2oAPIClient(api_hostname, api_username, api_password)
+
+client_hostname = socket.gethostname().split('.', 1)[0]
+
+for service in api_client.list_servicesregen():
+    if service['hostname'] == client_hostname and \
+            service['service_name'] == 'dhcp' and \
+            service['need_regen']:
+        regen_dhcp(api_client)
+        api_client.patch(service['api_url'], data={'need_regen': False})
diff --git a/re2oapi b/re2oapi
index 74fa513..7bcd092 160000
--- a/re2oapi
+++ b/re2oapi
@@ -1 +1 @@
-Subproject commit 74fa513e9a0348e6d4ce030c2364a40268c8eef0
+Subproject commit 7bcd0924b823cc11a5714b96cc96fe26638ac9c7