SoloLearn-Code-Challenges/M_youtube-link-finder.py at main · f4T1H21/SoloLearn-Code-Challenges · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
"""
YouTube Link Finder

You and your friends like to share YouTube links all throughout the day. You want to keep track of all the videos you watch in your own personal notepad, but you find that keeping the entire link is unnecessary.
Keep the video ID (the combination of letters and numbers at the end of the link) in your notepad to slim down the URL.

Task:
Create a program that parses through a link, extracts and outputs the YouTube video ID.

Input Format:
A string containing the URL to a YouTube video. The format of the string can be in "https://www.youtube.com/watch?v=kbxkq_w51PM" or the shortened "https://youtu.be/KMBBjzp5hdc" format.

Output Format:
A string containing the extracted YouTube video id.

Sample Input:
https://www.youtube.com/watch?v=RRW2aUSw5vU

Sample Output:
RRW2aUSw5vU
"""

import re
from urllib import parse

url = input()
pattern1 = r"^v=(.{11})"
pattern2 = r"^/(.{11})"

def ytparse(link):
    parsed = parse.urlsplit(url)
    if match := re.match(pattern1, parsed.query):
        return match.group(1)
    elif match := re.match(pattern2, parsed.path):
        return match.group(1)
    else:
        return None

print(ytparse(url))