From f99339760cbcd1b818359cea4a27b8e596b68762 Mon Sep 17 00:00:00 2001 From: rlbr Date: Mon, 1 Oct 2018 13:13:24 -0500 Subject: [PATCH] errors log to file and using edlib instead of editdistance --- .gitignore | 3 ++- main.py | 3 +++ requirements.txt | 2 +- search.py | 4 +++- 4 files changed, 9 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index 647df45..8efb089 100644 --- a/.gitignore +++ b/.gitignore @@ -7,4 +7,5 @@ __pycache__ *.pyc *_out.json *.pdf -*.bat \ No newline at end of file +*.bat +stderr.log \ No newline at end of file diff --git a/main.py b/main.py index 651ea01..001ef13 100644 --- a/main.py +++ b/main.py @@ -4,6 +4,8 @@ import re import ctabus from dateutil.parser import parse as date_parse import datetime +import sys +import os.path as osp from search import Search,StopSearch def numb_sort(str): n = 40 @@ -92,6 +94,7 @@ if __name__ == "__main__": parser.add_argument('-d','--direction',default = None) parser.add_argument('-l','--lucky',action='store_true',help = 'picks first result') args = parser.parse_args() + sys.stderr = open(osp.join(osp.dirname(__file__),'stderr.log'),'w') args.arg = ' '.join(args.arg) if not args.arg.isdecimal(): #routes diff --git a/requirements.txt b/requirements.txt index 231acec..1dec4c5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -editdistance +edlib requests python-dateutil \ No newline at end of file diff --git a/search.py b/search.py index 70bc920..4485405 100644 --- a/search.py +++ b/search.py @@ -1,4 +1,6 @@ -from editdistance import eval as editdistance +import edlib +def editdistance(a,b): + return edlib.align(a,b)['editDistance'] import re import json class Search: