Skip to content

Commit 5754876

Browse files
committed
feat: review request
1 parent c68859d commit 5754876

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

kiteconnect/connect.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from six.moves.urllib.parse import urljoin
1212
import csv
1313
import json
14-
from dateutil.parser import parse
14+
from dateutil.parser import datetimeparse
1515
from dateutil.tz import tzoffset
1616
from dateutil.utils import default_tzinfo
1717
import hashlib
@@ -266,7 +266,7 @@ def generate_session(self, request_token, api_secret):
266266
self.set_access_token(resp["access_token"])
267267

268268
if resp["login_time"] and self.is_timestamp(resp["login_time"]):
269-
resp["login_time"] = self.set_tz(resp["login_time"])
269+
resp["login_time"] = self.parseDateTime(resp["login_time"])
270270

271271
return resp
272272

@@ -401,7 +401,7 @@ def _format_response(self, data):
401401
# Convert date time string to datetime object
402402
for field in ["order_timestamp", "exchange_timestamp", "created", "last_instalment", "fill_timestamp", "timestamp", "last_trade_time"]:
403403
if item.get(field) and self.is_timestamp(item[field]):
404-
item[field] = self.set_tz(item[field])
404+
item[field] = self.parseDateTime(item[field])
405405

406406
return _list[0] if type(data) == dict else _list
407407

@@ -636,7 +636,7 @@ def _format_historical(self, data):
636636
records = []
637637
for d in data["candles"]:
638638
record = {
639-
"date": self.set_tz(d[0]),
639+
"date": self.parseDateTime(d[0]),
640640
"open": d[1],
641641
"high": d[2],
642642
"low": d[3],
@@ -798,7 +798,7 @@ def _parse_instruments(self, data):
798798

799799
# Parse date
800800
if self.is_timestamp(row["expiry"]):
801-
row["expiry"] = self.set_tz(row["expiry"]).date()
801+
row["expiry"] = self.parseDateTime(row["expiry"]).date()
802802

803803
records.append(row)
804804

@@ -825,7 +825,7 @@ def _parse_mf_instruments(self, data):
825825

826826
# Parse date
827827
if self.is_timestamp(row["last_price_date"]):
828-
row["last_price_date"] = self.set_tz(row["last_price_date"]).date()
828+
row["last_price_date"] = self.parseDateTime(row["last_price_date"]).date()
829829

830830
records.append(row)
831831

@@ -834,16 +834,16 @@ def _parse_mf_instruments(self, data):
834834
def is_timestamp(self, string):
835835
"""Checks if string is timestamp"""
836836
try:
837-
parse(string)
837+
datetimeparse(string)
838838
return True
839839
except ValueError:
840840
return False
841841

842-
def set_tz(self, string):
842+
def parseDateTime(self, string):
843843
"""Set default timezone to IST for naive time object"""
844844
# Default timezone for all datetime object
845845
default_tz = tzoffset("Asia/Kolkata", 19800)
846-
return default_tzinfo(parse(string), default_tz)
846+
return default_tzinfo(datetimeparse(string), default_tz)
847847

848848
def _user_agent(self):
849849
return (__title__ + "-python/").capitalize() + __version__

0 commit comments

Comments
 (0)