from requests_ntlm import HttpNtlmAuth
from requests.auth import HTTPBasicAuth
from requests.auth import HTTPDigestAuth
from requests.sessions import Session

#########################################################################################
# URL calls
#########################################################################################

def auth_call(
        session,
        authtype,
        user,
        passw
    ):
    """Authentication for url request. Requires request.sessions.Session() object.

    Args:
    - session (object): request Session() object.
    - authtype (str): 'ntlm', 'basic' or 'digest'.
    - user (str): auth user.
    - passw (str): auth password.
    """
    if authtype == 'ntlm':
        session.auth = HttpNtlmAuth(user, passw)
    elif authtype == 'basic':
        session.auth = HTTPBasicAuth(user, passw)
    elif authtype == 'digest':
        session.auth = HTTPDigestAuth(user, passw)

def call_url(
        url,
        authtype,
        user,
        passw,
        time_out
    ):
    """Call URL. Uses request module to get url contents.

    Args:
    - url (str): URL
    - authtype (str): ntlm', 'basic', 'digest'. Optional.
    - user (str): auth user. Optional.
    - passw (str): auth password. Optional.

    Returns:
    - str: call output
    """
    # using with so we make sure the session is closed even when exceptions are encountered
    with Session() as session:
        if authtype != None:
            auth_call(session, authtype, user, passw)
        try:
            output = session.get(url, timeout=time_out, verify=False)
        except ValueError:
            exit("Error: URL format not valid (example http://myserver/page.php)")
        except Exception as e:
            exit(f"{type(e).__name__}:\t{str(e)}")
        else:
            return output