capitalize check also for mkdocs.yml

scripts/titlemd_test.py

@@ -29,13 +29,15 @@ def main(location):
       prev_line = lines.pop(0)
       disabled = 0
       echo_filename = False
+      if location.find("mkdocs.yml") != -1:
+          disabled = 1
       for line in lines:
-          if line.startswith("``") and disabled == 0:
+          if (line.startswith("``") or line.startswith("extra:")) and disabled == 0:
               disabled = 1
           else:
-              if line.startswith("``") and disabled == 1:
+              if (line.startswith("``") or prev_line.startswith("pages:")) and disabled == 1:
                   disabled = 0
-          if line.startswith('#') and disabled == 0:
+          if line.startswith('#') and disabled == 0 and (location.find("mkdocs.yml") == -1):
             if line != titlecase(line[:(line.find("]"))], callback=abbreviations)+line[(line.find("]")):]:
               if return_value == 0 and echo_filename == False:
                 print("%s" % location)
@@ -53,6 +55,15 @@ def main(location):
               print("+"+titlecase(prev_line[:(prev_line.find("]"))], callback=abbreviations)+prev_line[(prev_line.find("]")):],end="")
               print()
               return_value = 1
+          if ((location.find("mkdocs.yml") != -1) and not line.startswith('#') and disabled == 0):
+            if line != titlecase(line[:(line.find(":"))], callback=abbreviations)+line[(line.find(":")):]:
+              if return_value == 0 and echo_filename == False:
+                print("%s" % location)
+                echo_filename = True
+              print("-"+line,end="")
+              print("+"+titlecase(line[:(line.find(":"))], callback=abbreviations)+line[(line.find(":")):],end="")
+              print()
+              return_value = 1
           prev_line = line
       exit(return_value)
 if __name__ == "__main__":