Как создать исключение в поиске?

Question

[Qvint01]

Приветствую, в коде иногда возникает ошибка, а если быть конкретнее то в парсере( парсит url случайного фото из гугла)
Нужно создать исключение, чтобы выдавало только url с окончанием на .jpg, .png и .jpeg

сам код:

import requests
from json import loads
from bs4 import BeautifulSoup
from random import choice

s = requests.session()
s.headers.update({'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:69.0) Gecko/20100101 Firefox/69.0'})
 
r = s.get('https://www.google.ru/search?q=мем&tbm=isch')
 
soup = BeautifulSoup(r.text, "html.parser")

elems = soup.findAll(attrs={'class': 'rg_meta notranslate'})
print(loads(choice(elems).text)['ou'])

Заранее спасибо.

Comments

[Qvint01]

like-a-boss, Не сердись, мне нужен был ответ ( просто забыл отметить ). Спасибо за уделенное время.

Answer 1

[Пашенька]

import re

def getImg( imgURL ):
    pattern = re.compile(r'^.*\.(jp(e)?g|png)$')
    
    if bool( re.match(pattern, imgURL) ):
        return imgURL
    else:
        return getImg( loads(choice(elems).text)['ou'] )

print( getImg( loads(choice(elems).text)['ou'] ) )