Merge ~gavin.lin/cc-lab-manager:restart_dhcp into cc-lab-manager:master

[Gavin Lin]

Apply DHCP config automatically.

[Kevin Yeh (kevinyeh)]

lgtm +1

review: Approve

diff --git a/cc_lab_manager/commands/cc_lab_manager.py b/cc_lab_manager/commands/cc_lab_manager.py
index a8324d2..3b78d8e 100644
--- a/cc_lab_manager/commands/cc_lab_manager.py
+++ b/cc_lab_manager/commands/cc_lab_manager.py
@@ -6,9 +6,11 @@ from cc_lab_manager.gsheet_db import gsheet_db
 from cc_lab_manager.gsheet_db import db_gsheet
 from cc_lab_manager.c3_db import c3_db
 from cc_lab_manager.dhcp import dhcp_config_generator
+from cc_lab_manager.dhcp import restart_dhcp_servers
 from cc_lab_manager.maas import create_maas_node
 from cc_lab_manager.gen_config import gen_agent_tf_config
 
+
 logger = logging.getLogger('cc-lab-manager')
 
 def main():
@@ -18,10 +20,9 @@ def main():
     # Get needed information from c3
     c3_db.main()
 
-    # Generate DHCP config for each subnets
-    # TODO: Currently these config files need to be copied
-    # manually, then MAAS will apply it automatically
+    # Generate and apply DHCP config for each subnet
     dhcp_config_generator.main()
+    restart_dhcp_servers.main()
 
     # Create MAAS nodes for DUTs
     create_maas_node.main()
diff --git a/cc_lab_manager/dhcp/restart_dhcp_servers.py b/cc_lab_manager/dhcp/restart_dhcp_servers.py
new file mode 100644
index 0000000..22986f7
--- /dev/null
+++ b/cc_lab_manager/dhcp/restart_dhcp_servers.py
@@ -0,0 +1,42 @@
+import logging
+import argparse
+import os
+import re
+import time
+import subprocess
+
+from cc_lab_manager.cc_lab_manager_maptable import lab_dhcp_servers
+
+logger = logging.getLogger('cc-lab-manager')
+
+def apply_dhcp_conf(cc_tool_box):
+    dhcp_conf_location = os.path.join(cc_tool_box, 'config', 'dhcp')
+    r = re.compile('cert-.*l-l.*-dhcpd-pool.conf')
+    for dirpath, _, files in os.walk(dhcp_conf_location):
+        for file in files:
+            if r.match(file) is not None:
+                lab_id = file.split('-')[1] + '-' + file.split('-')[2]
+                if lab_id in lab_dhcp_servers:
+                    subprocess.run(['scp', os.path.join(dirpath, file), lab_dhcp_servers[lab_id] + ':~'])
+                    time.sleep(1)
+                    subprocess.run(['ssh', lab_dhcp_servers[lab_id], 'sudo systemctl restart maas-dhcpd.service'])
+                    time.sleep(5)
+                    subprocess.run(['ssh', lab_dhcp_servers[lab_id], 'sudo systemctl status maas-dhcpd.service'])
+
+
+def environ_or_required(key):
+    if os.environ.get(key):
+        return {'default': os.environ.get(key)}
+    else:
+        return {'required': True}
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--cctoolbox', help="Path to cc-tool-box for storing dhcp config", **environ_or_required('CCLM_CC_TOOL_BOX'))
+    args = parser.parse_args()
+
+    apply_dhcp_conf(args.cctoolbox)
+
+if __name__ == '__main__':
+    main()