import datetime

import lxml
import requests

from panera_sync.saml_check import saml_check
from panera_sync.login import login_user
from panera_sync.oauth_required import (
    get_bearer_token,
    get_cafe_id,
    get_iBOH_id,
    get_shifts,
)


def login(username, password):
    session = requests.Session()
    login_user(session, username, password)
    return session


def get_shifts_between(session, start_date: datetime.date, end_date: datetime.date):
    bearer_token = get_bearer_token(session)
    cafe_id = get_cafe_id(session)
    iBOH_id = get_iBOH_id(session, bearer_token)
    return get_shifts(session, bearer_token, iBOH_id, cafe_id, start_date, end_date)