#!/usr/bin/env python3

import requests
import re
from bs4 import BeautifulSoup


class Models:
    def __init__(self):
        url_girls = "https://www.prosieben.de/tv/germanys-next-topmodel/models"

        html_girls = requests.get(url_girls)
        soup_girls = BeautifulSoup(html_girls.text, "html.parser")

        girls_in = soup_girls.findAll("a", class_="candidate-in")
        girls_out = soup_girls.findAll("a", class_="candidate-out")

        self.girls_in = {girl.get("title").lower(): girl for girl in girls_in}
        self.girls_out = {girl.get("title").lower(): girl for girl in girls_out}

        self.girls = {**self.girls_in, **self.girls_out}

    def get_in_names(self):
        return self.girls_in.keys()

    def get_out_names(self):
        return self.girls_out.keys()

    def get_image(self, name):
        style = self.girls[name.lower()].find("figure", class_="teaser-img").get("style")
        url = re.search(r"url\(.*\);", style).group()

        return url[4:-9] + "562x996"  # increase resolution