def test_datetime_formats(self): f = [(datetime.datetime(2013, 1, 2, 3, 4, 0), "2013-01-02 03:04:00", "03:04", "02.01.2013"), (datetime.datetime(2013, 5, 2), "2013-05-02 00:00:00", "00:00", "02.05.2013"), (datetime.datetime(2013, 2, 2, 3, 4, 30), "2013-02-02 03:04:30", "03:04", "02.02.2013")] for dt, sf, st, sd in f: self.assertEqual(fullformat(dt), sf) self.assertEqual(dateformat(dt), sd) self.assertEqual(timeformat(dt), st)
def call(self): params = { 'input': self.station_enc, 'time': timeformat(datetime.datetime.now()), 'date': dateformat(datetime.datetime.now()), 'productsFilter': self.vehicles, 'maxJourneys': self.limit, 'start': 'yes' } response = requests.get(SCHEDULED_API_ENDPOINT, params=params) if response.ok: soup = BeautifulSoup(response.text) if soup.find('span', 'error'): # The station we are looking for is ambiguous or does not exist stations = soup.find('span', 'select').find_all('a') if stations: # The station is ambiguous stationlist = [s.text.strip() for s in stations] return Response(False, stationlist) else: # The station does not exist return Response(False, []) else: # The station seems to exist tbody = soup.find('tbody') if tbody is None: return Response(True, self.station, []) rows = tbody.find_all('tr') departures = [] for row in rows: tds = row.find_all('td') dep = Departure(start=self.station, end=tds[2].text.strip(), when=tds[0].text.strip(), line=tds[1].text.strip()) departures.append(dep) return Response(True, self.station, departures) else: try: response.raise_for_status() except requests.RequestException as e: return Response(False, error=e) else: return Response(False, error=Exception("An unknown error occured"))
def call(self): params = {'input': self.station_enc, 'time': timeformat(datetime.datetime.now()), 'date': dateformat(datetime.datetime.now()), 'productsFilter': self.vehicles, 'maxJourneys': self.limit, 'start': 'yes'} response = requests.get(SCHEDULED_API_ENDPOINT, params=params) if response.ok: soup = BeautifulSoup(response.text) if soup.find('span', 'error'): # The station we are looking for is ambiguous or does not exist stations = soup.find('span', 'select').find_all('a') if stations: # The station is ambiguous stationlist = [s.text.strip() for s in stations] return Response(False, stationlist) else: # The station does not exist return Response(False, []) else: # The station seems to exist tbody = soup.find('tbody') if tbody is None: return Response(True, self.station, []) rows = tbody.find_all('tr') departures = [] for row in rows: tds = row.find_all('td') dep = Departure(start=self.station, end=tds[2].text.strip(), when=tds[0].text.strip(), line=tds[1].text.strip()) departures.append(dep) return Response(True, self.station, departures) else: try: response.raise_for_status() except requests.RequestException as e: return Response(False, error=e) else: return Response(False, error=Exception("An unknown error occured"))