From 34817697574c61a24d7aee13e397da79bbe7653e Mon Sep 17 00:00:00 2001
From: bohd4nx <bohd4n@proton.me>
Date: Fri, 5 Jun 2026 02:04:44 +0300
Subject: [PATCH] feat: add Highload wallet versions and update mypy exclusion
 patterns

---
 CHANGELOG.md               | 1 +
 pyfragment/models/enums.py | 6 +++++-
 pyproject.toml             | 2 +-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5987e73..07e5604 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -13,6 +13,7 @@ and this project uses [Calendar Versioning](https://calver.org/) (`YYYY.MINOR.MI
 
 - New `AlreadySubscribedError` exception for Premium purchase flows when Fragment returns: `This account is already subscribed to Telegram Premium.`
 - New `UserNotFoundError.NOT_A_USER` message for when Fragment returns: `Please enter a username assigned to a user.` (e.g. when the username belongs to a channel or bot).
+- Added `WalletVersion.HighloadV2` and `WalletVersion.HighloadV3R1` to `WalletVersion`
 
 ### Changed
 
diff --git a/pyfragment/models/enums.py b/pyfragment/models/enums.py
index 02b21ab..5b0a4a7 100644
--- a/pyfragment/models/enums.py
+++ b/pyfragment/models/enums.py
@@ -4,7 +4,7 @@ import sys
 from enum import Enum
 from typing import Any
 
-from tonutils.contracts.wallet import WalletV4R2, WalletV5R1
+from tonutils.contracts.wallet import WalletHighloadV2, WalletHighloadV3R1, WalletV4R2, WalletV5R1
 
 if sys.version_info >= (3, 11):
     from enum import StrEnum
@@ -29,11 +29,15 @@ class PaymentMethod(StrEnum):
 class WalletVersion(StrEnum):
     V4R2 = "V4R2"
     V5R1 = "V5R1"
+    HighloadV2 = "HighloadV2"
+    HighloadV3R1 = "HighloadV3R1"
 
 
 WALLET_CLASSES: dict[WalletVersion, Any] = {
     WalletVersion.V4R2: WalletV4R2,
     WalletVersion.V5R1: WalletV5R1,
+    WalletVersion.HighloadV2: WalletHighloadV2,
+    WalletVersion.HighloadV3R1: WalletHighloadV3R1,
 }
 
 
diff --git a/pyproject.toml b/pyproject.toml
index 40d0f7d..82974a4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -91,7 +91,7 @@ ignore = ["E501", "UP017"]
 [tool.mypy]
 python_version = "3.10"
 strict = true
-exclude = ["^examples/"]
+exclude = ["^examples/", "^logs/"]
 
 [[tool.mypy.overrides]]
 module = "rookiepy"