Habiticalib 0.1.0a2__py3-none-any.whl → 0.2.0__py3-none-any.whl

habiticalib/__init__.py

@@ -6,18 +6,26 @@ from .exceptions import (
     HabiticaException,
     NotAuthorizedError,
     NotFoundError,
+    TooManyRequestsError,
 )
+from .helpers import deserialize_task
 from .lib import Habitica
 from .types import (
     Attributes,
-    Class,
+    ChangeClassData,
+    ContentData,
     Direction,
     Frequency,
+    HabiticaClass,
     HabiticaClassSystemResponse,
+    HabiticaContentResponse,
     HabiticaErrorResponse,
+    HabiticaGroupMembersResponse,
     HabiticaLoginResponse,
+    HabiticaQuestResponse,
     HabiticaResponse,
     HabiticaScoreResponse,
+    HabiticaSleepResponse,
     HabiticaStatsResponse,
     HabiticaTagResponse,
     HabiticaTagsResponse,
@@ -27,10 +35,18 @@ from .types import (
     HabiticaUserExport,
     HabiticaUserResponse,
     Language,
+    LoginData,
+    QuestData,
+    ScoreData,
     Skill,
+    StatsUser,
+    TagsUser,
     Task,
+    TaskData,
     TaskFilter,
+    TaskPriority,
     TaskType,
+    UserData,
     UserStyles,
 )
 
@@ -39,17 +55,24 @@ __all__ = [
     "ASSETS_URL",
     "Attributes",
     "BadRequestError",
-    "Class",
+    "ChangeClassData",
+    "ContentData",
     "DEFAULT_URL",
+    "deserialize_task",
     "Direction",
     "Frequency",
     "Habitica",
+    "HabiticaClass",
     "HabiticaClassSystemResponse",
+    "HabiticaContentResponse",
     "HabiticaErrorResponse",
     "HabiticaException",
+    "HabiticaGroupMembersResponse",
     "HabiticaLoginResponse",
+    "HabiticaQuestResponse",
     "HabiticaResponse",
     "HabiticaScoreResponse",
+    "HabiticaSleepResponse",
     "HabiticaStatsResponse",
     "HabiticaTagResponse",
     "HabiticaTagsResponse",
@@ -59,11 +82,20 @@ __all__ = [
     "HabiticaUserExport",
     "HabiticaUserResponse",
     "Language",
+    "LoginData",
     "NotAuthorizedError",
     "NotFoundError",
+    "QuestData",
+    "ScoreData",
     "Skill",
+    "StatsUser",
+    "TagsUser",
     "Task",
+    "TaskData",
     "TaskFilter",
+    "TaskPriority",
     "TaskType",
+    "TooManyRequestsError",
+    "UserData",
     "UserStyles",
 ]

habiticalib/const.py

@@ -1,6 +1,6 @@
 """Constants for Habiticalib."""
 
-__version__ = "0.1.0a2"
+__version__ = "0.2.0"
 
 DEFAULT_URL = "https://habitica.com/"
 ASSETS_URL = "https://habitica-assets.s3.amazonaws.com/mobileApp/images/"
@@ -14,3 +14,5 @@ BACKER_ONLY_GEAR = {
     "shield_special_ks2019": "BackerOnly-Equip-MythicGryphonShield.gif",
     "weapon_special_ks2019": "BackerOnly-Equip-MythicGryphonGlaive.gif",
 }
+
+PAGE_LIMIT = 60

habiticalib/exceptions.py

@@ -1,5 +1,6 @@
 """Exceptions for Habiticalib."""
 
+from datetime import datetime
 from typing import Self
 
 from multidict import CIMultiDictProxy
@@ -17,10 +18,18 @@ class HabiticaException(Exception):  # noqa: N818
     ) -> None:
         """Initialize the Exception."""
         self.error = error
-        self.rate_limit = headers.get("x-ratelimit-limit")
-        self.rate_limit_remaining = headers.get("x-ratelimit-remaining")
-        self.rate_limit_reset = headers.get("x-ratelimit-reset")
-        self.retry_after = headers.get("retry-after")
+        self.rate_limit: int | None = (
+            int(r) if (r := headers.get("x-ratelimit-limit")) else None
+        )
+        self.rate_limit_remaining: int | None = (
+            int(r) if (r := headers.get("x-ratelimit-remaining")) else None
+        )
+        self.rate_limit_reset: datetime | None = (
+            datetime.strptime(r[:33], "%a %b %d %Y %H:%M:%S %Z%z")
+            if (r := headers.get("x-ratelimit-reset"))
+            else None
+        )
+        self.retry_after: int = round(float(headers.get("retry-after", 0)))
 
         super().__init__(error.message)
 

habiticalib/helpers.py

@@ -1,7 +1,10 @@
 """Helper functions for Habiticalib."""
 
-from dataclasses import asdict
+from dataclasses import asdict, is_dataclass
+from datetime import date, datetime
+from enum import Enum
 import platform
+from typing import Any
 import uuid
 
 import aiohttp
@@ -112,3 +115,24 @@ def extract_user_styles(user_data: HabiticaUserResponse) -> UserStyles:
     """Extract user styles from a user data object."""
     data: UserData = user_data.data
     return UserStyles.from_dict(asdict(data))
+
+
+def deserialize_task(value: Any) -> Any:  # noqa: PLR0911
+    """Recursively convert Enums to values, dates to ISO strings, UUIDs to strings."""
+
+    if is_dataclass(value) and not isinstance(value, type):
+        # Convert dataclass to dict and recursively deserialize
+        return deserialize_task(asdict(value))
+    if isinstance(value, Enum):
+        return value.value  # Convert Enum to its value
+    if isinstance(value, uuid.UUID):
+        return str(value)  # Convert UUID to string
+    if isinstance(value, datetime | date):
+        return value.isoformat()  # Convert datetime/date to ISO string
+    if isinstance(value, list):
+        # Recursively apply deserialization to each item in the list
+        return [deserialize_task(item) for item in value]
+    if isinstance(value, dict):
+        # Recursively apply deserialization to each key-value pair in the dictionary
+        return {k: deserialize_task(v) for k, v in value.items()}
+    return value  # Return other types unchanged