Add ALLOWED_URL_SCHEMES constant (#55)

messense · Sep 22, 2024 · 98ca727 · 98ca727
1 parent 2f7f481
commit 98ca727
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/nh3.pyi b/nh3.pyi
@@ -2,6 +2,7 @@ from typing import Callable, Dict, Optional, Set
 
 ALLOWED_TAGS: Set[str]
 ALLOWED_ATTRIBUTES: Dict[str, Set[str]]
+ALLOWED_URL_SCHEMES: Set[str]
 
 def clean(
     html: str,

diff --git a/src/lib.rs b/src/lib.rs
@@ -240,5 +240,6 @@ fn nh3(_py: Python, m: &Bound<PyModule>) -> PyResult<()> {
     let a = ammonia::Builder::default();
     m.add("ALLOWED_TAGS", a.clone_tags())?;
     m.add("ALLOWED_ATTRIBUTES", a.clone_tag_attributes())?;
+    m.add("ALLOWED_URL_SCHEMES", a.clone_url_schemes())?;
     Ok(())
 }