codeking/sp-hydra-veil-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve presentation of model member variables

Browse source
This commit is contained in:
codeking 2025-11-18 02:44:16 +01:00
parent b2e260475b
commit b887b8d2ff
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
core/models/Location.py
View file

@ -67,6 +67,12 @@ class Location(Model):
self.operator = Operator.find_by_id(self.operator_id) self.operator = Operator.find_by_id(self.operator_id)
self.available = self.exists(self.country_code, self.code) self.available = self.exists(self.country_code, self.code)
if isinstance(self.is_proxy_capable, int):
self.is_proxy_capable = bool(self.is_proxy_capable)
if isinstance(self.is_wireguard_capable, int):
self.is_wireguard_capable = bool(self.is_wireguard_capable)
def is_available(self): def is_available(self):
return self.exists(self.country_code, self.code) return self.exists(self.country_code, self.code)

8
core/models/SubscriptionPlan.py
View file

@ -27,6 +27,14 @@ class SubscriptionPlan(Model):
features_proxy: bool features_proxy: bool
features_wireguard: bool features_wireguard: bool
def __post_init__(self):
if isinstance(self.features_proxy, int):
self.features_proxy = bool(self.features_proxy)
if isinstance(self.features_wireguard, int):
self.features_wireguard = bool(self.features_wireguard)
@staticmethod @staticmethod
def find_by_id(id: int): def find_by_id(id: int):
Model._create_table_if_not_exists(table_name=_table_name, table_definition=_table_definition) Model._create_table_if_not_exists(table_name=_table_name, table_definition=_table_definition)