From 9a22a8543bb5e1aff8dc9d97cb9ae0ace4ca094e Mon Sep 17 00:00:00 2001 From: Dan Cojocaru Date: Tue, 31 Aug 2021 17:40:30 +0300 Subject: [PATCH] Added support for specifying date for train data request --- server/server/v2/v2.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/server/server/v2/v2.py b/server/server/v2/v2.py index 6559ced..fd1d53a 100644 --- a/server/server/v2/v2.py +++ b/server/server/v2/v2.py @@ -1,3 +1,4 @@ +from datetime import datetime import json from flask import Blueprint, jsonify, request from flask.helpers import url_for @@ -41,10 +42,16 @@ def get_train_info_schema(): @bp.route('/train/') def get_train_info(train_no: str): use_yesterday = check_yes_no(request.args.get('use_yesterday', ''), default=False) + date_override = request.args.get('date', default=None) + try: + date_override = datetime.fromisoformat(date_override) + except ValueError: + date_override = None + @filtered_data def get_data(): from ..scraper.scraper import scrape_train - result = scrape_train(train_no, use_yesterday=use_yesterday) + result = scrape_train(train_no, use_yesterday=use_yesterday, date_override=date_override) db.on_train_data(result) return result if (train_no, use_yesterday) not in train_data_cache: