Browse Source

Make main_loop __init__.py

master
Raphael Roberts 5 years ago
parent
commit
16d636fa81
  1. 26
      panera_sync/__init__.py
  2. 26
      panera_sync/main_loop.py

26
panera_sync/__init__.py

@ -0,0 +1,26 @@
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)

26
panera_sync/main_loop.py

@ -1,26 +0,0 @@
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)
Loading…
Cancel
Save