Merge pull request #79 from LlmKira/dev

⬆️ feat: migrate from datetime to arrow for datetime handling

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "novelai-python"
-version = "0.5.2"
+version = "0.5.3"
 description = "NovelAI Python Binding With Pydantic"
 authors = [
     { name = "sudoskys", email = "[email protected]" },
@@ -29,6 +29,7 @@ dependencies = [
     "json-repair>=0.29.4",
     "robust-downloader>=0.0.2",
     "bchlib>=2.1.3",
+    "arrow>=1.3.0",
 ]
 requires-python = ">=3.9"
 readme = "README.md"

src/novelai_python/credential/ApiToken.py

@@ -4,8 +4,7 @@
 # @File    : ApiToken.py
 # @Software: PyCharm
 
-import datetime
-
+import arrow
 import shortuuid
 from curl_cffi.requests import AsyncSession
 from loguru import logger
@@ -31,7 +30,7 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
             "x-correlation-id": self._x_correlation_id,
-            "x-initiated-at": f"{datetime.datetime.now(datetime.UTC).isoformat()}Z",
+            "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 
         if update_headers:

src/novelai_python/credential/JwtToken.py

@@ -3,8 +3,8 @@
 # @Author  : sudoskys
 # @File    : JwtToken.py
 # @Software: PyCharm
-import datetime
 
+import arrow
 import shortuuid
 from curl_cffi.requests import AsyncSession
 from loguru import logger
@@ -30,7 +30,7 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
             "x-correlation-id": self._x_correlation_id,
-            "x-initiated-at": f"{datetime.datetime.now(datetime.UTC).isoformat()}Z",
+            "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 
         if update_headers:

src/novelai_python/credential/UserAuth.py

@@ -2,10 +2,10 @@
 # @Time    : 2024/2/7 下午12:14
 # @Author  : sudoskys
 # @File    : UserAuth.py
-import datetime
 import time
 from typing import Optional
 
+import arrow
 import shortuuid
 from curl_cffi.requests import AsyncSession
 from pydantic import SecretStr, Field
@@ -33,7 +33,7 @@ async def get_session(self, timeout: int = 180, update_headers: dict = None):
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
             "x-correlation-id": self._x_correlation_id,
-            "x-initiated-at": f"{datetime.datetime.now(datetime.UTC).isoformat()}Z",
+            "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 
         # 30 天有效期