<commit_before>import requests
from bs4 import BeautifulSoup


def dados_concurso(bsObj):
    div = bsObj.find('div', {'class': 'title-bar clearfix'})
    concurso = str(div.h2).split()
    c_num = concurso[4]
    c_data = str(concurso[5])[1:11]
    return [c_num, c_data]


def parse_concurso(concurso):

    url = 'http://loterias.caixa.gov.br/wps/portal/loterias/landing/' + concurso
    html = requests.request('GET', url)
    bsObj = BeautifulSoup(html.text, 'html.parser')

    return bsObj


def result_loterias(concurso, bsObj):

    nums_sorteados = []
    cells = []

    if concurso == 'lotomania':

        table = bsObj.table

        for row in table.findAll('tr'):
            cells.append(row.findAll('td'))

        for cell in cells:
            for i in range(len(cell)):
                nums_sorteados.append(str(cell[i])[4:6])

    elif concurso == 'megasena':

        ul = bsObj.find('ul', {'class': 'numbers mega-sena'})
        for i, li in enumerate(ul.findAll('li')):
            cells.append(li)
            nums_sorteados.append(str(cells[i])[4:6])
        '''
        for i, cell in enumerate(cells):
            nums_sorteados.append(str(cell[i])[4:6])
        '''
    return [concurso, nums_sorteados]


if __name__ == '__main__':

    print('Resultado de Loteria')
    ent_concurso = int(input('Você quer saber o resultado de qual loteria (1) Mega-Sena (2) Lotomania: '))
    if ent_concurso == 1:
        concurso = 'megasena'
    else:
        concurso = 'lotomania'
    bsObj = parse_concurso(concurso)
    concurso_num, concurso_data = dados_concurso(bsObj)

    concurso, nums_sorteados = result_loterias(concurso, bsObj)
    print('=' * 30)
    print(concurso.upper())
    print(f'Concurso: {concurso_num} de {concurso_data}')
    print(f'Os números sorteados foram:')
    for i, num in enumerate(nums_sorteados):
        if i in (5, 10, 15) and concurso != 'megasena':
            print()
        print(f'{num}', end = ' ')
    print()
    print('=' * 30)
<commit_msg>Adicionadas novas consultas<commit_after>import requests
from bs4 import BeautifulSoup


def dados_concurso(bsObj):
    div = bsObj.find('div', {'class': 'title-bar clearfix'})
    concurso = str(div.h2).split()
    c_num = concurso[4]
    c_data = str(concurso[5])[1:11]
    return [c_num, c_data]


def parse_concurso(concurso):

    url = 'http://loterias.caixa.gov.br/wps/portal/loterias/landing/' + concurso
    html = requests.request('GET', url)
    bsObj = BeautifulSoup(html.text, 'html.parser')

    return bsObj


def result_loterias(concurso, bsObj):

    nums_sorteados = []
    cells = []

    if concurso == 'lotomania' or concurso == 'lotofacil':

        if concurso == 'lotofacil':
            table = bsObj.find('table', {'class': 'simple-table lotofacil'})
        else:
            table = bsObj.table

        for row in table.findAll('tr'):
            cells.append(row.findAll('td'))

        for cell in cells:
            for i in range(len(cell)):
                nums_sorteados.append(str(cell[i])[4:6])

    elif concurso == 'megasena' or concurso == 'quina' or 'timemania':

        if concurso == 'megasena':
            classe = 'mega-sena'
        else:
            classe = concurso

        ul = bsObj.find('ul', {'class': classe})
        for i, li in enumerate(ul.findAll('li')):
            cells.append(li)
            nums_sorteados.append(str(cells[i])[4:6])

    return [concurso, nums_sorteados]


if __name__ == '__main__':
    lista_concursos = ['megasena', 'lotomania', 'quina', 'timemania', 'lotofacil']
    print('==== Resultado de Loteria ====')
    print('=' * 30)
    print(' ' * 13 + 'Menu' + ' ' * 13)
    print('=' * 30)
    for i, c in enumerate(lista_concursos):
        print(f'{i + 1} - {str(lista_concursos[i]).title()}')

    ent_concurso = int(input('Qual sua opção: '))
    if ent_concurso == 1:
        concurso = 'megasena'
    elif ent_concurso == 2:
        concurso = 'lotomania'
    elif ent_concurso == 3:
        concurso = 'quina'
    elif ent_concurso == 4:
        concurso = 'timemania'
    elif ent_concurso == 5:
        concurso = 'lotofacil'

    bsObj = parse_concurso(concurso)
    concurso_num, concurso_data = dados_concurso(bsObj)

    concurso, nums_sorteados = result_loterias(concurso, bsObj)
    print('=' * 30)
    print(concurso.upper())
    print(f'Concurso: {concurso_num} de {concurso_data}')
    print(f'Os números sorteados foram:')
    for i, num in enumerate(nums_sorteados):
        if i in (5, 10, 15) and concurso not in ('megasena', 'timemania'):
            print()
        print(f'{num}', end = ' ')
    print()
    print('=' * 30)

{'fraction_non_alphanumeric': 0.14756570443774236, 'fraction_numerical': 0.012710038776389487, 'mean_word_length': 2.7443548387096772, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5732121', 'n_tokens_mistral': 1811, 'n_tokens_neox': 1697, 'n_words': 354}