From cc7246facd2860c3bf5050fcf1e1c9d86a43354f Mon Sep 17 00:00:00 2001 From: Dan Cojocaru Date: Tue, 31 Aug 2021 18:19:32 +0300 Subject: [PATCH] Fixed train info cache --- server/server/v2/v2.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/server/server/v2/v2.py b/server/server/v2/v2.py index fd1d53a..47583a2 100644 --- a/server/server/v2/v2.py +++ b/server/server/v2/v2.py @@ -1,4 +1,4 @@ -from datetime import datetime +from datetime import date, datetime import json from flask import Blueprint, jsonify, request from flask.helpers import url_for @@ -55,8 +55,8 @@ def get_train_info(train_no: str): db.on_train_data(result) return result if (train_no, use_yesterday) not in train_data_cache: - train_data_cache[(train_no, use_yesterday)] = CachedData(get_data, validity=1000 * 30) - data, fetch_time = train_data_cache[(train_no, use_yesterday)]() + train_data_cache[(train_no, use_yesterday or date_override)] = CachedData(get_data, validity=1000 * 30) + data, fetch_time = train_data_cache[(train_no, use_yesterday or date_override)]() data['$schema'] = get_hostname() + url_for('.get_train_info_schema') validate(data, schema=TRAIN_INFO_SCHEMA['v2']) resp = jsonify(data)