From e4943b1f6e7f1a2cc04fb4958597c5374462f51a Mon Sep 17 00:00:00 2001
From: Dan Cojocaru <dan.cojocaru@3pillarglobal.com>
Date: Sat, 28 Aug 2021 00:36:07 +0300
Subject: [PATCH] Allowed any digits in train number (ex: IRN 01641)

---
 scraper/scrape_train.py | 2 +-
 server/server/v2/v2.py  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/scraper/scrape_train.py b/scraper/scrape_train.py
index 9f6aa13..02c9a5a 100644
--- a/scraper/scrape_train.py
+++ b/scraper/scrape_train.py
@@ -35,7 +35,7 @@ STATION_DEPARR_STATUS_REGEX = re.compile(r'^(?:(la timp)|(?:((?:\+|-)[0-9]+) min
 
 # endregion
 
-def scrape(train_no: int, use_yesterday=False, date_override=None):
+def scrape(train_no: str, use_yesterday=False, date_override=None):
 	# Start scrapping session
 	s = requests.Session()
 
diff --git a/server/server/v2/v2.py b/server/server/v2/v2.py
index 2261fa4..6559ced 100644
--- a/server/server/v2/v2.py
+++ b/server/server/v2/v2.py
@@ -38,8 +38,8 @@ train_data_cache = {}
 def get_train_info_schema():
 	return jsonify(TRAIN_INFO_SCHEMA['v2'])
 
-@bp.route('/train/<int:train_no>')
-def get_train_info(train_no: int):
+@bp.route('/train/<train_no>')
+def get_train_info(train_no: str):
 	use_yesterday = check_yes_no(request.args.get('use_yesterday', ''), default=False)
 	@filtered_data
 	def get_data():