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

Update and refactor existing codebase

Browse source
This commit is contained in:
codeking 2025-04-23 14:31:30 +02:00
parent 4d56dd0a8c
commit 3ff73cd079
1 changed files with 14 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
core/controllers/ProfileController.py
View file

@ -143,7 +143,7 @@ class ProfileController:
from core.controllers.ConnectionController import ConnectionController from core.controllers.ConnectionController import ConnectionController
if profile.subscription is not None: if profile.has_subscription():
subscription = ConnectionController.with_preferred_connection(profile.subscription.billing_code, task=WebServiceApiService.get_subscription, connection_observer=connection_observer) subscription = ConnectionController.with_preferred_connection(profile.subscription.billing_code, task=WebServiceApiService.get_subscription, connection_observer=connection_observer)
@ -175,39 +175,34 @@ class ProfileController:
if system_state is not None and system_state.profile_id is profile.id: if system_state is not None and system_state.profile_id is profile.id:
return ConnectionController.system_uses_wireguard_interface() return ConnectionController.system_uses_wireguard_interface()
else: return False
return False
@staticmethod @staticmethod
def get_invoice(profile: Union[SessionProfile, SystemProfile]): def get_invoice(profile: Union[SessionProfile, SystemProfile]):
if profile.subscription is None: if profile.has_subscription():
return WebServiceApiService.get_invoice(profile.subscription.billing_code)
else:
return None return None
return WebServiceApiService.get_invoice(profile.subscription.billing_code)
@staticmethod @staticmethod
def attach_proxy_configuration(profile: Union[SessionProfile, SystemProfile]): def attach_proxy_configuration(profile: Union[SessionProfile, SystemProfile]):
if not profile.is_session_profile(): if profile.is_session_profile() and profile.has_subscription():
return None
if profile.subscription is None: proxy_configuration = WebServiceApiService.get_proxy_configuration(profile.subscription.billing_code)
return None
proxy_configuration = WebServiceApiService.get_proxy_configuration(profile.subscription.billing_code) if proxy_configuration is not None:
profile.attach_proxy_configuration(proxy_configuration)
if proxy_configuration is not None:
profile.attach_proxy_configuration(proxy_configuration)
@staticmethod @staticmethod
def get_proxy_configuration(profile: Union[SessionProfile, SystemProfile]): def get_proxy_configuration(profile: Union[SessionProfile, SystemProfile]):
if not profile.is_session_profile(): if profile.is_session_profile():
return profile.get_proxy_configuration()
else:
return None return None
profile.get_proxy_configuration()
@staticmethod @staticmethod
def has_proxy_configuration(profile: Union[SessionProfile, SystemProfile]): def has_proxy_configuration(profile: Union[SessionProfile, SystemProfile]):
profile.has_proxy_configuration() profile.has_proxy_configuration()
@ -217,10 +212,10 @@ class ProfileController:
from core.controllers.ConnectionController import ConnectionController from core.controllers.ConnectionController import ConnectionController
if profile.subscription is None: if not profile.has_subscription():
raise MissingSubscriptionError() raise MissingSubscriptionError()
if profile.location is None: if not profile.has_location():
raise MissingLocationError() raise MissingLocationError()
wireguard_keys = ProfileController.__generate_wireguard_keys() wireguard_keys = ProfileController.__generate_wireguard_keys()