move qms specific parts to qms module

header.py

 import os
-from generator import format
+from generator import format, qms
 
 def remove_header(localroot, root, filename):
     nfirstlines = []
@@ -68,10 +68,8 @@ def generate_header(folder, permalink, shortcut, order, legacy_from, title, desc
     # generate specific redirects
     header = element_header(folder, "  - ", shortcut, header)
 
-    if "qms" in permalink or "policies" in permalink or "sops" in permalink:
-        qmsDoc = True
-    else:
-        qmsDoc = False
+    qmsDoc = qms.determine_qms(permalink)
+
 
     # include the legacy section in the redirect_from section
     if len(legacy_from) > 0:
@@ -94,17 +92,7 @@ def generate_header(folder, permalink, shortcut, order, legacy_from, title, desc
 
     # add title for QMS documents
     if qmsDoc:
-
-        if os.getenv('VERSION') is None:
-            ver = "latest"
-        else:
-            ver = os.getenv('VERSION')
-
-        header += "\n# " + title + "\n"
-        header += "<center>"
-        header += "<a href='https://qms.lcsb.uni.lu/" + str(ver) + "/" + str(name) + "_v" + nb.replace(".", "") +".pdf'>Download the signed version (Version v" + nb + ") here</a>"
-        header += "</center>"
-
+        header += qms.add_link(title, name, nb)
 
     print(title)
 

qms.py

@@ -27,4 +27,25 @@ def prepare_qms(localroot, root, filename):
         file.write(filedata)
 
     # change back to the local root
-    os.chdir(localroot)
\ No newline at end of file
+    os.chdir(localroot)
+
+def determine_qms(permalink):
+
+    if "qms" in permalink or "policies" in permalink or "sops" in permalink:
+        return True
+    else:
+        return False
+
+def add_link(title, name, nb):
+
+    if os.getenv('VERSION') is None:
+        ver = "latest"
+    else:
+        ver = os.getenv('VERSION')
+
+    header = "\n# " + title + "\n"
+    header += "<center>"
+    header += "<a href='https://qms.lcsb.uni.lu/" + str(ver) + "/" + str(name) + "_v" + nb.replace(".", "") +".pdf'>Download the signed document (version v" + nb + ")</a>"
+    header += "</center>"
+
+    return header
\ No newline at end of file