diff --git a/core/models/Location.py b/core/models/Location.py index 6fa3fa3..62c963b 100644 --- a/core/models/Location.py +++ b/core/models/Location.py @@ -67,6 +67,12 @@ class Location(Model): self.operator = Operator.find_by_id(self.operator_id) 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): return self.exists(self.country_code, self.code) diff --git a/core/models/SubscriptionPlan.py b/core/models/SubscriptionPlan.py index 2b6262b..5a55c8e 100644 --- a/core/models/SubscriptionPlan.py +++ b/core/models/SubscriptionPlan.py @@ -27,6 +27,14 @@ class SubscriptionPlan(Model): features_proxy: 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 def find_by_id(id: int): Model._create_table_if_not_exists(table_name=_table_name, table_definition=_table_definition)