Module pipelines.rj_rioaguas.utils

Utils Gerais da Rio-Águas.

Functions

def login(url, user=None, password=None)

Expand source code

def login(url, user=None, password=None):
    """
    Função para fazer login no website.

    Args:
    url (str): URL da página inicial com os campos de usuário e senha.
    user (str): Usuário do login
    password (str): Senha de acesso.
    """

    client = requests.Session()

    # Retrieve the CSRF token first
    client.get(url)  # sets cookie
    if password:
        if "csrftoken" in client.cookies:
            # Django 1.6 and up
            csrftoken = client.cookies["csrftoken"]
        else:
            # older versions
            csrftoken = client.cookies["csrf"]

        payload = {
            "username": user,
            "password": password,
            "csrfmiddlewaretoken": csrftoken,
            "next": "/",
        }
    else:
        payload = {}

    client.post(url, data=payload, headers=dict(Referer=url))

    return client

Função para fazer login no website.

Args: url (str): URL da página inicial com os campos de usuário e senha. user (str): Usuário do login password (str): Senha de acesso.