MouseTools 2.1.2.dev2__py3-none-any.whl → 3.0.0.dev1__py3-none-any.whl

mousetools/attractions.py

@@ -1,370 +0,0 @@
-"""
-attractions module
-"""
-import json
-from datetime import datetime, timedelta
-import requests
-import pytz
-from .auth import get_headers
-from .ids import themeparkapi_ids, WDW_ID, DLR_ID
-
-
-
-class Attraction:
-
-    def __init__(self, id = None):
-        """
-        Constructor Function
-        Gets all attraction data available and stores various elements into variables.
-        """
-
-        error = True
-        self.__data = requests.get(f"https://api.wdpro.disney.go.com/global-pool-override-B/facility-service/attractions/{id}", headers=get_headers()).json()
-        try:
-            if self.__data['id'] is not None:
-                error = False
-        except:
-            pass
-
-        if error:
-            raise ValueError('That attraction is not available. id: ' + str(id))
-
-        self.__id = id
-        self.__name = self.__data['name']
-        self.__entityType = self.__data['type']
-        try:
-            self.__subType = self.__data['subType']
-        except:
-            self.__subType = None
-
-        try:
-            self.__anc_dest_id = self.__data['ancestorDestination']['id'].split(';')[0]
-        except:
-            self.__anc_dest_id = None
-
-        try:
-            self.__anc_park_id = self.__data['links']['ancestorThemePark']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_park_id = self.__data['links']['ancestorWaterPark']['href'].split('/')[-1].split('?')[0]
-            except:
-                try:
-                    self.__anc_park_id = self.__data['ancestorThemeParkId'].split(';')[0]
-                except:
-                    try:
-                        self.__anc_park_id = self.__data['ancestorWaterParkId'].split(';')[0]
-                    except:
-                        self.__anc_park_id = None
-
-        try:
-            self.__anc_resort_id = self.__data['links']['ancestorResort']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_resort_id = self.__data['ancestorResortId'].split(';')[0]
-            except:
-                self.__anc_resort_id = None
-
-        try:
-            self.__anc_land_id = self.__data['links']['ancestorLand']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_land_id = self.__data['ancestorLandId'].split(';')[0]
-            except:
-                self.__anc_land_id = None
-
-        try:
-            self.__anc_ra_id = self.__data['links']['ancestorResortArea']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_ra_id = self.__data['ancestorResortAreaId'].split(';')[0]
-            except:
-                self.__anc_ra_id = None
-
-        try:
-            self.__anc_ev_id = self.__data['links']['ancestorEntertainmentVenue']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_ev_id = self.__data['ancestorEntertainmentVenueId'].split(';')[0]
-            except:
-                self.__anc_ev_id = None
-
-        if self.__anc_dest_id == WDW_ID:
-            self.__time_zone = pytz.timezone('US/Eastern')
-        elif self.__anc_dest_id == DLR_ID:
-            self.__time_zone = pytz.timezone('US/Pacific')
-        else:
-            self.__time_zone = pytz.utc
-
-
-    def get_id(self):
-        """Return object id"""
-        return self.__id
-
-    def get_name(self):
-        """Return object name"""
-        return self.__name
-
-    def get_entityType(self):
-        """Return object entityType"""
-        return self.__entityType
-
-    def get_subType(self):
-        """Return object subType"""
-        return self.__subType
-
-    def get_ancestor_destination_id(self):
-        """Return object ancestor destination id"""
-        return self.__anc_dest_id
-
-    def get_ancestor_park_id(self):
-        """Return object ancestor theme or water park id"""
-        return self.__anc_park_id
-
-    def get_ancestor_resort_id(self):
-        """Return object ancestor resort id"""
-        return self.__anc_resort_id
-
-    def get_ancestor_land_id(self):
-        """Return object land id"""
-        return self.__anc_land_id
-
-    def get_ancestor_resort_area_id(self):
-        """Return object resort area id"""
-        return self.__anc_ra_id
-
-    def get_ancestor_entertainment_venue_id(self):
-        """Return object entertainment venue id"""
-        return self.__anc_ev_id
-
-    def get_links(self):
-        """Returns a dictionary of related links"""
-        return self.__data['links']
-
-    def get_time_zone(self):
-        """Returns pytz timezone object"""
-        return self.__time_zone
-
-    def get_raw_data(self):
-        """Returns the raw data from global-facility-service"""
-        return self.__data
-
-    def get_themeparkapi_data(self):
-        """Returns the dictionary from the themepark api for the given id"""
-        park = themeparkapi_ids[self.__anc_park_id]
-        themepark_id = f"{park}_{self.__id}"
-        all_data = requests.get(f"https://api.themeparks.wiki/preview/parks/{park}/waittime").json()
-        for i in all_data:
-            if i["id"] == themepark_id:
-                return i
-        return None
-
-    def get_wait_time(self):
-        """Return current wait time of the object. Returns None if object doesn't have a wait time or no wait currently exists (eg. closed)"""
-        data = self.get_themeparkapi_data()
-        if data is None:
-            return None
-
-        return data['waitTime']
-
-    def get_status(self):
-        """Return current status of the object."""
-        data = self.get_themeparkapi_data()
-        if data is None:
-            return None
-
-        return data['status']
-
-    def fastpass_available(self):
-        """Returns a boolean of whether this object has FastPass"""
-        data = self.get_themeparkapi_data()
-        if data is None:
-            return False
-
-        return data['fastPass']
-
-    def get_last_update_time(self):
-        """Returns facilities last update time as a datetime object"""
-        facility_data = self.get_themeparkapi_data()
-        if facility_data is None:
-            return None
-
-        update_time = datetime.strptime(facility_data['lastUpdate'], "%Y-%m-%dT%H:%M:%S.%fZ")
-        update_time = update_time.replace(tzinfo=pytz.utc)
-        update_time = update_time.astimezone(self.__time_zone)
-        return update_time
-
-    def get_coordinates(self):
-        """Returns the object's latitude and longitude"""
-        try:
-            return self.__data['coordinates']['Guest Entrance']['gps']
-        except:
-            return None
-
-    def get_description(self):
-        """Returns the object's description"""
-        facility_data = self.__data
-        if facility_data is None:
-            return None
-
-        try:
-            return facility_data['descriptions']['shortDescription']['sections']['body']
-        except:
-            return None
-
-    def get_media(self):
-        """Returns a dictionary of dictionaries of media relating to the entity"""
-        facility_data = self.__data
-        if facility_data is None:
-            return None
-
-        return facility_data['media']
-
-    def get_facets(self):
-        """Returns a list of  dictionaries of the object's facets"""
-        facility_data = self.__data
-        if facility_data is None:
-            return None
-
-        try:
-            return facility_data['facets']
-        except:
-            return None
-
-    def get_classifications(self):
-        """Returns a dictionary of lists of classifications related to the entity"""
-
-        classifications = {}
-
-        try:
-            for i in self.__data['classifications']:
-                id = i['id'].split("/")[-1]
-                if id not in classifications:
-                    classifications[id] = [i['text']]
-                else:
-                    classifications[id].append(i['text'])
-        except:
-            pass
-
-        return classifications
-
-    def admission_required(self):
-        """Returns boolean of admission required"""
-        return self.__data['admissionRequired']
-
-    def get_hours(self, date = ""):
-        """
-        Gets the object's hours on a specific day and returns them as a datetime object.
-        Returns the object's hours in the following order: operating open, operating close, Extra Magic open, Extra Magic close.
-        Extra Magic hours will return None if there are none for today.
-        If all hours are None then Disney has no hours for that day.
-        date = "yyyy-mm-dd"
-        If you don't pass a date, it will get today's hours
-        """
-
-        if date == "":
-            DATE = datetime.today()
-        else:
-            year, month, day = date.split('-')
-            DATE = datetime(int(year), int(month), int(day))
-
-        s = requests.get(f"https://api.wdpro.disney.go.com/facility-service/schedules/{self.__id}?date={DATE.year}-{self.__formatDate(str(DATE.month))}-{self.__formatDate(str(DATE.day))}", headers=get_headers())
-        data = json.loads(s.content)
-
-        operating_hours_start = None
-        operating_hours_end = None
-        extra_hours_start = None
-        extra_hours_end = None
-
-        try:
-            for i in range(len(data['schedules'])):
-                if data['schedules'][i]['type'] == 'Operating':
-                    operating_hours_start = datetime(DATE.year, DATE.month, DATE.day, int(data['schedules'][i]['startTime'][0:2]), int(data['schedules'][i]['startTime'][3:5]))
-                    if int(data['schedules'][i]['endTime'][0:2]) >= 0 and int(data['schedules'][i]['endTime'][0:2]) <= 7:
-                        DATETEMP = DATE + timedelta(days=1)
-                        operating_hours_end = datetime(DATETEMP.year, DATETEMP.month, DATETEMP.day, int(data['schedules'][i]['endTime'][0:2]), int(data['schedules'][i]['endTime'][3:5]))
-                    else:
-                        operating_hours_end = datetime(DATE.year, DATE.month, DATE.day, int(data['schedules'][i]['endTime'][0:2]), int(data['schedules'][i]['endTime'][3:5]))
-
-                if data['schedules'][i]['type'] == "Special Ticketed Event":
-                    extra_hours_start = datetime(DATE.year, DATE.month, DATE.day, int(data['schedules'][i]['startTime'][0:2]), int(data['schedules'][i]['startTime'][3:5]))
-                    if int(data['schedules'][i]['endTime'][0:2]) >= 0 and int(data['schedules'][i]['endTime'][0:2]) <= 7:
-                        DATETEMP = DATE + timedelta(days=1)
-                        extra_hours_end = datetime(DATETEMP.year, DATETEMP.month, DATETEMP.day, int(data['schedules'][i]['endTime'][0:2]), int(data['schedules'][i]['endTime'][3:5]))
-                    else:
-                        operating_hours_end = datetime(DATE.year, DATE.month, DATE.day, int(data['schedules'][i]['endTime'][0:2]), int(data['schedules'][i]['endTime'][3:5]))
-
-        except KeyError:
-            pass
-        return operating_hours_start, operating_hours_end, extra_hours_start, extra_hours_end
-
-    def check_associated_characters(self):
-        """
-        Checks if object has any associated characters
-        """
-        s = requests.get("https://api.wdpro.disney.go.com/global-pool-override-B/facility-service/associated-characters/{};entityType={}".format(self.__id, self.__entityType), headers=get_headers())
-        data = json.loads(s.content)
-
-        if data['total'] > 0:
-            return True
-        else:
-            return False
-
-    def get_number_associated_characters(self):
-        """
-        Gets the total number of characters associated with this object
-        """
-        s = requests.get("https://api.wdpro.disney.go.com/facility-service/associated-characters/{};entityType={}".format(self.__id, self.__entityType), headers=get_headers())
-        data = json.loads(s.content)
-
-        return data['total']
-
-    def get_associated_characters(self):
-        """
-        Returns a list of associated characters Character objects
-        """
-        from .characters import Character
-        chars = []
-
-        s = requests.get("https://api.wdpro.disney.go.com/global-pool-override-B/facility-service/associated-characters/{};entityType={}".format(self.__id, self.__entityType), headers=get_headers())
-        data = json.loads(s.content)
-
-        for i in range(len(data['entries'])):
-            try:
-                chars.append(Character(data['entries'][i]['links']['self']['href'].split('/')[-1]))
-            except:
-                pass
-        return chars
-
-    def get_associated_character_ids(self):
-        """
-        Returns a list of associated characters IDs
-        """
-        chars = []
-
-        s = requests.get("https://api.wdpro.disney.go.com/global-pool-override-B/facility-service/associated-characters/{};entityType={}".format(self.__id, self.__entityType), headers=get_headers())
-        data = json.loads(s.content)
-
-        for i in range(len(data['entries'])):
-            try:
-                chars.append(data['entries'][i]['links']['self']['href'].split('/')[-1])
-            except:
-                pass
-
-        return chars
-
-    def __formatDate(self, num):
-        """
-        Formats month and day into proper format
-        """
-        if len(num) < 2:
-            num = '0'+num
-        return num
-
-    def __eq__(self, other):
-        """
-        Checks if objects are equal
-        """
-        return self.__id == other.get_id()
-
-    def __str__(self):
-        return 'Attraction object for {}'.format(self.__name)

mousetools/characters.py

@@ -1,241 +0,0 @@
-import requests
-import pytz
-from datetime import datetime, timedelta
-from .auth import get_headers
-from .attractions import Attraction
-from .entertainments import Entertainment
-from .facilities import Facility
-from .ids import WDW_ID, DLR_ID
-
-
-
-class Character(object):
-
-    def __init__(self, id = None):
-        """
-        Constructor Function
-        Gets all character data available and stores various elements into variables.
-        """
-
-        error = True
-        self.__data = requests.get("https://api.wdpro.disney.go.com/facility-service/characters/{}".format(id), headers=get_headers()).json()
-        try:
-            if self.__data['id'] is not None:
-                error = False
-        except:
-            pass
-
-        if error:
-            raise ValueError('That character is not available. id: ' + str(id))
-
-        self.__id = id
-        self.__name = self.__data['name']
-        self.__entityType = self.__data['type']
-        try:
-            self.__subType = self.__data['subType']
-        except:
-            self.__subType = None
-        try:
-            self.__anc_dest_id = self.__data['ancestorDestination']['id'].split(';')[0]
-        except:
-            self.__anc_dest_id = None
-
-        try:
-            self.__anc_park_id = self.__data['links']['ancestorThemePark']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_park_id = self.__data['links']['ancestorWaterPark']['href'].split('/')[-1].split('?')[0]
-            except:
-                try:
-                    self.__anc_park_id = self.__data['ancestorThemeParkId'].split(';')[0]
-                except:
-                    try:
-                        self.__anc_park_id = self.__data['ancestorWaterParkId'].split(';')[0]
-                    except:
-                        self.__anc_park_id = None
-
-        try:
-            self.__anc_resort_id = self.__data['links']['ancestorResort']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_resort_id = self.__data['ancestorResortId'].split(';')[0]
-            except:
-                self.__anc_resort_id = None
-
-        try:
-            self.__anc_land_id = self.__data['links']['ancestorLand']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_land_id = self.__data['ancestorLandId'].split(';')[0]
-            except:
-                self.__anc_land_id = None
-
-        try:
-            self.__anc_ra_id = self.__data['links']['ancestorResortArea']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_ra_id = self.__data['ancestorResortAreaId'].split(';')[0]
-            except:
-                self.__anc_ra_id = None
-
-        try:
-            self.__anc_ev_id = self.__data['links']['ancestorEntertainmentVenue']['href'].split('/')[-1].split('?')[0]
-        except:
-            try:
-                self.__anc_ev_id = self.__data['ancestorEntertainmentVenueId'].split(';')[0]
-            except:
-                self.__anc_ev_id = None
-
-        if self.__anc_dest_id == WDW_ID:
-            self.__time_zone = pytz.timezone('US/Eastern')
-        elif self.__anc_dest_id == DLR_ID:
-            self.__time_zone = pytz.timezone('US/Pacific')
-        else:
-            self.__time_zone = pytz.utc
-
-
-    def get_id(self):
-        """Return object id"""
-        return self.__id
-
-    def get_name(self):
-        """Return object name"""
-        return self.__name
-
-    def get_entityType(self):
-        """Return object entityType"""
-        return self.__entityType
-
-    def get_subType(self):
-        """Return object subType"""
-        return self.__subType
-
-    def get_ancestor_destination_id(self):
-        """Return object ancestor destination id"""
-        return self.__anc_dest_id
-
-    def get_ancestor_park_id(self):
-        """Return object ancestor theme or water park id"""
-        return self.__anc_park_id
-
-    def get_ancestor_resort_id(self):
-        """Return object ancestor resort id"""
-        return self.__anc_resort_id
-
-    def get_ancestor_land_id(self):
-        """Return object land id"""
-        return self.__anc_land_id
-
-    def get_ancestor_resort_area_id(self):
-        """Return object resort area id"""
-        return self.__anc_ra_id
-
-    def get_ancestor_entertainment_venue_id(self):
-        """Return object entertainment venue id"""
-        return self.__anc_ev_id
-
-    def get_time_zone(self):
-        """Returns pytz timezone object"""
-        return self.__time_zone
-
-    def get_raw_data(self):
-        """Returns the raw data from global-facility-service"""
-        return self.__data
-
-    def get_links(self):
-        """Returns a dictionary of related links"""
-        return self.__data['links']
-
-
-    def check_related_locations(self):
-        """
-        Returns true if it has related locations, false if none
-        """
-        try:
-            check = self.__data['relatedLocations']
-            return True
-        except:
-            return False
-
-    def get_related_locations(self):
-        """
-        Gets the locations the character is realted to.
-        """
-        locs = []
-        try:
-            if self.check_related_locations():
-                for loc in self.__data['relatedLocations']['primaryLocations']:
-                    type = loc['facilityType']
-                    loc_id = loc['links']['self']['href'].split('/')[-1].split('?')[0]
-
-                    if type == 'Attraction':
-                        locs.append(Attraction(loc_id))
-                    elif type == 'Facility':
-                        locs.append(Facility(loc_id))
-                    else:
-                        print('no class for {} at this time'.format(type))
-            return locs
-        except:
-            return locs
-
-    def get_related_location_ids(self):
-        """
-        Gets the locations the character is realted to as a list of tuples [(id, type)]
-        """
-        locs = []
-        try:
-            if self.check_related_locations():
-                for loc in self.__data['relatedLocations']['primaryLocations']:
-                    type = loc['facilityType']
-                    loc_id = loc['links']['self']['href'].split('/')[-1].split('?')[0]
-
-                    locs.append((loc_id, type))
-
-            return locs
-        except:
-            return locs
-
-    def get_associated_events(self):
-        """
-        Returns a list of Entertainment objects of events associated with the character.
-        Will print an error if the ID does not exist anymore. Unfortunately Disney, lists some events that don't exist, and thus will throw the error
-        when it tries to create the Entertainment object.
-        """
-        entertainments = []
-        try:
-            for entertainment in self.__data['associatedEvents']:
-                try:
-                    entertainments.append(Entertainment(entertainment['links']['self']['href'].split('/')[-1].split('?')[0]))
-                except:
-                    pass
-            return entertainments
-        except:
-            return entertainments
-
-    def get_associated_event_ids(self):
-        """
-        Returns a list of Entertainment ids of events associated with the character.
-        Will print an error if the ID does not exist anymore. Unfortunately Disney, lists some events that don't exist, and thus will throw the error
-        when it tries to create the Entertainment object.
-        """
-        entertainments = []
-        try:
-            for entertainment in self.__data['associatedEvents']:
-                try:
-                    entertainments.append(entertainment['links']['self']['href'].split('/')[-1].split('?')[0])
-                except:
-                    pass
-            return entertainments
-        except:
-            return entertainments
-
-    def __formatDate(self, num):
-        """
-        Formats month and day into proper format
-        """
-        if len(num) < 2:
-            num = '0'+num
-        return num
-
-    def __str__(self):
-        return 'Character object for {}'.format(self.__name)