Add github source link transform

AA-Turner · AA-Turner · commit daf9f6447d3c · 2020-08-15T07:21:52.000+01:00
diff --git a/pep_extensions/pep_processor/pep_target_notes.py b/pep_extensions/pep_processor/pep_target_notes.py
@@ -5,6 +5,7 @@
 from docutils.transforms import misc
 from docutils.transforms import references
 
+import pep_extensions.config as pep_config
 
 class PEPTargetNotes(transforms.Transform):
 
@@ -17,7 +18,8 @@ class PEPTargetNotes(transforms.Transform):
     default_priority = 520
 
     def apply(self):
-        if not Path(self.document["source"]).match("pep-*"):
+        pep_source_path = Path(self.document["source"])
+        if not pep_source_path.match("pep-*"):
             # not a PEP file
             return
 
@@ -49,6 +51,8 @@ def apply(self):
         reference_section.append(pending)
         self.document.note_pending(pending, 1)
 
+        self.add_source_link(pep_source_path)
+
     @staticmethod
     def cleanup_callback(pending):
         """
@@ -58,3 +62,9 @@ def cleanup_callback(pending):
         """
         if len(pending.parent) == 2:    # <title> and <pending>
             pending.parent.parent.remove(pending.parent)
+
+    def add_source_link(self, pep_source_path: Path) -> None:
+        source_link = pep_config.pep_vcs_url.format(pep_source_path.name)
+        link_node = nodes.reference("", source_link, refuri=source_link)
+        span_node = nodes.inline("", "Source: ", link_node)
+        self.document.append(span_node)
