diff --git a/remnawave/models/users.py b/remnawave/models/users.py index 12c544b..2e7ba56 100644 --- a/remnawave/models/users.py +++ b/remnawave/models/users.py @@ -1,5 +1,5 @@ from datetime import datetime -from typing import Annotated, List, Optional +from typing import Annotated, List, Literal, Optional from uuid import UUID from pydantic import ( @@ -177,6 +177,13 @@ class UserResponseDto(BaseModel): """Generate Happ Crypto Link""" crypto_link = create_happ_crypto_link(self.subscription_url) return HappCrypto(cryptoLink=crypto_link) + + def happ_with_version(self, version: Literal["v3", "v4"] = "v4") -> HappCrypto: + return self._generate_happ(version=version) + + def _generate_happ(self, version): + crypto_link = create_happ_crypto_link(content=self.subscription_url, method=version) + return HappCrypto(cryptoLink=crypto_link) class RevokeUserRequestDto(BaseModel):