[ccloud] adds tftpmap for pxe

sapcc · May 4, 2022 · bee6355 · bee6355
1 parent 662791d
commit bee6355
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/ironic/pxe/__init__.py b/ironic/pxe/__init__.py
diff --git a/ironic/pxe/tftpmap.py b/ironic/pxe/tftpmap.py
@@ -0,0 +1,15 @@
+import re, os, stat
+
+patterns = [ (re.compile(a), b) for a, b in [
+    (r'^(/tftpboot/)', r'/tftpboot/'),
+    (r'^/tftpboot/', r'/tftpboot/'),
+    (r'^(^/)', r'/tftpboot/\1'),
+    (r'^([^/])', r'/tftpboot/\1')
+]]
+
+# filename -> (stream, size)
+def handle(self, filename):
+    for pattern, replacement in patterns:
+        if pattern.search(filename):
+            new_filename = pattern.sub(replacement, filename)
+            return open(new_filename, 'rb'), os.stat(new_filename)[stat.ST_SIZE]