fix: apply timezone

zeroam · Nov 5, 2024 · be495ee · be495ee
1 parent 62fd984
commit be495ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -23,6 +23,6 @@ repos:
     rev: v1.13.0
     hooks:
       - id: mypy
-        additional_dependencies: [types-requests]
+        additional_dependencies: [types-requests, types-pytz]
         args: [--config-file=mypy.ini]
         exclude: ^tests/
diff --git a/kispy/auth.py b/kispy/auth.py
@@ -4,6 +4,7 @@
 import tempfile
 from datetime import datetime
 
+import pytz
 import requests
 from pydantic import BaseModel, Field
 
@@ -20,7 +21,8 @@ class Token(BaseModel):
     access_token_token_expired: datetime = Field(description="액세스 토큰 만료일시")
 
     def is_expired(self) -> bool:
-        return datetime.now() > self.access_token_token_expired
+        kst = pytz.timezone("Asia/Seoul")
+        return datetime.now(kst) > self.access_token_token_expired
 
 
 class KisAuth:
@@ -56,10 +58,11 @@ def _get_token(self) -> Token:
                 "appsecret": self.app_secret,
             },
         )
+        kst = pytz.timezone("Asia/Seoul")
         access_token_token_expired = datetime.strptime(
             resp.json["access_token_token_expired"],
             "%Y-%m-%d %H:%M:%S",
-        )
+        ).replace(tzinfo=kst)
         return Token(
             token_type=resp.json["token_type"],
             access_token=resp.json["access_token"],