Ejemplo n.º 1
0
class Category:
    def __init__(self):
        self.eventbrite = Eventbrite(settings.EVENTBRITE_API_KEY)

    def all(self):
        """
        This returns all categories from the eventbrite API

        :return:
        :return: List of categories
        """
        eventbrite_categories = self.eventbrite.get_categories()
        if eventbrite_categories.status_code == 200:
            return eventbrite_categories["categories"]
        else:
            return None

    def get(self, category_id):
        """
        Given the id of a category, returns that eventbrite category object

        :param category_id:
        :return:
        """
        data = self.eventbrite.get_category(category_id)
        if data.status_code == 200:
            return data
        else:
            return None

    def get_multiple(self, category_ids):
        """
        Returns category objects for category ids given

        :param category_ids:
        :return:
        """
        categories = []
        for category_id in category_ids:
            categories.append(self.get(category_id))

        return categories