From 3b003c10fc6aa3a2a60eac27cbef94c3c8c284e7 Mon Sep 17 00:00:00 2001
From: binary <me@rgoncalves.se>
Date: Sun, 14 Feb 2021 12:14:01 +0100
Subject: Add groups assignement based on hostname regexp

---
 dynamic.py             | 32 +++++++++++++++++++++++++++++---
 group_vars/servers.yml |  0
 2 files changed, 29 insertions(+), 3 deletions(-)
 create mode 100644 group_vars/servers.yml

diff --git a/dynamic.py b/dynamic.py
index 908e4d3..5354474 100755
--- a/dynamic.py
+++ b/dynamic.py
@@ -12,16 +12,29 @@
 # host/group/node is individually track on version control.
 
 import os
-import yaml
 import logging
 import json
+import re
+import yaml
 
-from typing import Final
 from optparse import OptionParser
+from typing import Final
 
 
 HOST_DIR: Final = "./host_vars"
 GROUP_DIR: Final = "./group_vars"
+GROUP_PATTERNS: Final = {
+    "servers": [
+        "^dc[0-9]*",
+        "^rt[0-9]*",
+        "^st[0-9]*",
+        "^stack[0-9]*",
+        "^emb[0-9]*"
+    ],
+    "workstation": [
+        "^ws*"
+    ]
+}
 
 
 class AnsibleInput:
@@ -63,6 +76,16 @@ def create_nodes(directory):
     return nodes
 
 
+def group_regexp(hostname, hostgroups):
+    for groupname, group in GROUP_PATTERNS.items():
+        for pattern in group:
+            regexp = re.compile(pattern)
+            if regexp.search(hostname):
+                logging.info(f"hostname { hostname } matched with"
+                             f"{ pattern } for group { groupname }")
+                hostgroups.append(groupname)
+
+
 def get_hostvars(host):
     """
     Retrieve and build dynamic variables for a specific host.
@@ -79,7 +102,7 @@ def get_subgroups(groups, og):
     Get and return all children of a specific group, in a given subset.
     """
 
-    # specific return for group "all" 
+    # specific return for group "all"
     if og.name in "all":
         groups = list(groups.keys())
         groups.remove("all")
@@ -130,6 +153,9 @@ def get_list(hosts, groups):
         # retrieve _groups variables and force conversion to list
         cleangr = host.data["_groups"]
         cleangr = [cleangr] if isinstance(cleangr, str) else cleangr
+        # group assignement based on regexp
+        group_regexp(hostname, cleangr)
+        # final assignement with correct groupname
         for group in cleangr:
             if group not in groups:
                 logging.error(f"group { group } does not exist!")
diff --git a/group_vars/servers.yml b/group_vars/servers.yml
new file mode 100644
index 0000000..e69de29
-- 
cgit v1.2.3