From 736221ff203c389311b750b407ee74b704ae4052 Mon Sep 17 00:00:00 2001
From: Kelly Christensen <kelly.christensen@sciencespo.fr>
Date: Thu, 29 Feb 2024 11:08:20 +0100
Subject: [PATCH] add type to pyi and to normalize url function

---
 ural/youtube.py  | 4 ++++
 ural/youtube.pyi | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/ural/youtube.py b/ural/youtube.py
index 281b433..bd4a8b0 100644
--- a/ural/youtube.py
+++ b/ural/youtube.py
@@ -429,4 +429,8 @@ def normalize_youtube_url(url):
 
         return YOUTUBE_CHANNEL_NAME_URL_TEMPLATE % parsed.name
 
+    if isinstance(parsed, YoutubeShort):
+        if parsed.id is not None:
+            return YOUTUBE_SHORT_URL_TEMPLATE % parsed.id
+
     raise TypeError("normalize_youtube_url: impossible path reached")
diff --git a/ural/youtube.pyi b/ural/youtube.pyi
index 6fac1ce..f305c02 100644
--- a/ural/youtube.pyi
+++ b/ural/youtube.pyi
@@ -12,6 +12,9 @@ class YoutubeChannel(NamedTuple):
     id: str
     name: str
 
+class YoutubeShort(NamedTuple):
+    id: str
+
 def is_youtube_url(url: AnyUrlTarget) -> bool: ...
 def is_youtube_video_id(value: str) -> bool: ...
 def is_youtube_channel_id(value: str) -> bool: ...