Allow passing a custom template registry

1 files changed, 14 insertions, 0 deletions

diff --git a/wikimini/templates/__init__.py b/wikimini/templates/__init__.py
index 360b3fa..ebe3bbc 100644
--- a/wikimini/templates/__init__.py
+++ b/wikimini/templates/__init__.py
@@ -6,6 +6,7 @@ A template is a function that takes the :class:`~wikimini.Wikimini` instance
 and the :class:`~mwparserfromhell.nodes.template.Template` node to convert, and
 returns a string with the template output (see :const:`Template`).
 """
+import copy
 from typing import Callable, Optional
 
 import mwparserfromhell as mwp
@@ -49,6 +50,19 @@ class Registry:
         """
         self.templates[name] = template
 
+    def copy(self) -> "Registry":
+        """Returns a copy of the registry.
+
+        This is useful if you want to use the global registry as a starting
+        point to add your own templates, without modifying the global instance.
+
+        Returns:
+            A copy of the registry.
+        """
+        copied = Registry()
+        copied.templates = copy.copy(self.templates)
+        return copied
+
 
 #: The global template registry.
 registry = Registry()