From c7de60ee03947f89287c5492e689813daae21762 Mon Sep 17 00:00:00 2001
From: 283375 <bilibili283375@163.com>
Date: Thu, 12 Oct 2023 17:05:33 +0800
Subject: [PATCH] feat(utils): Kanae day/night util

---
 src/arcaea_offline/utils/partner.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 src/arcaea_offline/utils/partner.py

diff --git a/src/arcaea_offline/utils/partner.py b/src/arcaea_offline/utils/partner.py
new file mode 100644
index 0000000..3ec4a61
--- /dev/null
+++ b/src/arcaea_offline/utils/partner.py
@@ -0,0 +1,15 @@
+from datetime import datetime
+from enum import IntEnum
+
+
+class KanaeDayNight(IntEnum):
+    Day = 0
+    Night = 1
+
+
+def kanae_day_night(timestamp: int) -> KanaeDayNight:
+    """
+    :param timestamp: POSIX timestamp, which is passed to `datetime.fromtimestamp(timestamp)`.
+    """
+    dt = datetime.fromtimestamp(timestamp)
+    return KanaeDayNight.Day if 6 <= dt.hour <= 19 else KanaeDayNight.Night