projectal.entities.location

 1from datetime import datetime
 2
 3import projectal
 4from projectal.entity import Entity
 5from projectal.linkers import *
 6
 7
 8class Location(
 9    Entity,
10    CompanyLinker,
11    CustomerLinker,
12    ProjectLinker,
13    StaffLinker,
14    NoteLinker,
15    CalendarLinker,
16    TagLinker,
17):
18    """
19    Implementation of the [Location](https://projectal.com/docs/latest/#tag/Location) API.
20    """
21
22    _path = "location"
23    _name = "location"
24    _links = [NoteLinker, CalendarLinker, TagLinker]
25    _links_reverse = [CompanyLinker, CustomerLinker, ProjectLinker, StaffLinker]
26
27    def calendar(self, begin=None, until=None):
28        """
29        Get the location's calendar.
30        `uuId`: The location `uuId`
31        `begin`: Start date in `yyyy-MM-dd` format
32        `until`: End date in `yyyy-MM-dd` format
33        """
34        if begin:
35            begin = datetime.strptime(begin, "%Y-%m-%d").date()
36        if until:
37            until = datetime.strptime(until, "%Y-%m-%d").date()
38
39        url = "/api/location/{}/calendar?".format(self["uuId"])
40        params = []
41        params.append("begin={}".format(begin)) if begin else None
42        params.append("until={}".format(until)) if until else None
43        url += "&".join(params)
44
45        cals = api.get(url)
46        cals = [projectal.Calendar(c) for c in cals]
47        return cals