Skip to content

Absence function is added. #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
altar91 wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Absence function is added. #10

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
237c36a
Absence function is added.
May 16, 2023
de164f6
Account replaced with self
May 16, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 31 additions & 0 deletions pystaffo/staffo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,37 @@ def get_shifts(self, location_id=None, loc_name=None, department_id=None, dep_na
'until': '{en_date}T23:59:59{en_tz}'.format(en_date=end_date, en_tz=end_tz)})
return get(auth=self.auth, url=self.base_url + extension, extras=params)

def get_absences(self, location_id=None, loc_name=None, start_date=None, end_date=None) -> dict:
"""
Gets the absence for a specified location where they may also be
filtered for a date range or department.
For filtering by location and department it is expected that
the user is either using names or IDs but not a
mixture of the two.
Input dates expected to be date strings in yyyy-mm-dd format.
"""
params = {}
if location_id or loc_name:
if not location_id:
location_id = self.locations[loc_name]
extension = 'locations/{loc_id}/absences.json'.format(loc_id=location_id)
else:
extension = 'absences.json'
if start_date:
start_tz = self.timezone.localize(datetime.strptime(start_date, '%Y-%m-%d'))
start_tz = datetime.strftime(start_tz, '%z')
start_tz = start_tz[:3] + ':' + start_tz[3:]
if not end_date:
end_tz = datetime.now(tz=self.timezone)
else:
end_tz = self.timezone.localize(datetime.strptime(end_date, '%Y-%m-%d'))
end_date = end_tz.strftime('%Y-%m-%d')
end_tz = end_tz.strftime('%z')
end_tz = end_tz[:3] + ':' + end_tz[3:]
params.update({'from': '{st_date}T00:00:00{st_tz}'.format(st_date=start_date, st_tz=start_tz),
'until': '{en_date}T23:59:59{en_tz}'.format(en_date=end_date, en_tz=end_tz)})
return get(auth=self.auth, url=self.base_url + extension, extras=params)

def add_users(self, department_id=None, loc_name=None, dep_name=None, users=None, remove=False):
"""
Add/Remove a list of user ids to/from a department identified either by its ID or by its location and
Expand Down