61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
|
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
|