final

import requests
import re
import sys
from bs4 import BeautifulSoup as bs

# Setup
url = requests.get("https://www.itnwwe.com/mma/ufc-rankings/")
soup = bs(url.text, 'html.parser')


def rankings(element):
    # CSS Selector As Element
    elem = soup.select(element)
    content = elem[0].text
    content = content.replace("(", " ")
    content = content.replace(")", " ")
    # Find Pattern
    pattern = re.compile(r"(?:Champion:|\d+\.)\s+\w+\s\w+\s+\d+-\d+-\d+")
    matches = pattern.findall(content)
    fighter_ranks = '\n'.join(matches)

    return fighter_ranks


# Assign Weight Classes, (m)en & (w)omen
weight_class_rankings = {
    "m_p4p": rankings(".kadence-column_b8bbf4-03 > div:nth-child(1)"),
    "m_heavyweight": rankings(".kt-svg-icon-list-items_68bb27-ed > ul:nth-child(1)"),
    "m_light_heavyweight": rankings(".kt-svg-icon-list-items_659b4f-00 > ul:nth-child(1)"),
    "m_middleweight": rankings(".kt-svg-icon-list-items_2743c7-33 > ul:nth-child(1)"),
    "m_welterweight": rankings(".kt-svg-icon-list-items_82cd81-fc > ul:nth-child(1)"),
    "m_lightweight": rankings(".kt-svg-icon-list-items_00ea45-f0 > ul:nth-child(1)"),
    "m_featherweight": rankings(".kt-svg-icon-list-items_91412b-2d > ul:nth-child(1)"),
    "m_bantamweight": rankings(".kt-svg-icon-list-items_195a09-5b > ul:nth-child(1)"),
    "m_flyweight": rankings(".kt-svg-icon-list-items_62f802-5b > ul:nth-child(1)"),
    "w_p4p": rankings(".kadence-column_82ab3b-c7 > div:nth-child(1) > div:nth-child(3)"),
    "w_bantamweight": rankings(".kt-svg-icon-list-items_1c7b33-e3 > ul:nth-child(1)"),
    "w_flyweight": rankings(".kt-svg-icon-list-items_1400c1-04 > ul:nth-child(1)"),
    "w_strawweight": rankings(".kt-svg-icon-list-items_acf1a0-3a > ul:nth-child(1)")
    }

try:
    print(weight_class_rankings[sys.argv[1]])
except KeyError:
    print("Weight Class Does Not Exist! Please Try Again.")
except IndexError:
    v = input("Please Enter A Weight Class: ")
    try:
        print(weight_class_rankings[v])
    except KeyError:
        print("Weight Class Does Not Exist! Please Try Again.")