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

Update and refactor policy-related logic

Browse source
This commit is contained in:
codeking 2025-11-27 08:46:59 +01:00
parent 8945ead9bd
commit 219a4d4dc1
1 changed files with 13 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
cli/__main__.py
View file

@ -2,13 +2,12 @@ from core.Constants import Constants
from core.Errors import MissingSubscriptionError, InvalidSubscriptionError, UnknownConnectionTypeError, ConnectionUnprotectedError, EndpointVerificationError, ProfileStateConflictError from core.Errors import MissingSubscriptionError, InvalidSubscriptionError, UnknownConnectionTypeError, ConnectionUnprotectedError, EndpointVerificationError, ProfileStateConflictError
from core.controllers.ApplicationController import ApplicationController from core.controllers.ApplicationController import ApplicationController
from core.controllers.ApplicationVersionController import ApplicationVersionController from core.controllers.ApplicationVersionController import ApplicationVersionController
from core.controllers.CapabilityPolicyController import CapabilityPolicyController
from core.controllers.ClientController import ClientController from core.controllers.ClientController import ClientController
from core.controllers.ClientVersionController import ClientVersionController from core.controllers.ClientVersionController import ClientVersionController
from core.controllers.ConfigurationController import ConfigurationController from core.controllers.ConfigurationController import ConfigurationController
from core.controllers.InvoiceController import InvoiceController from core.controllers.InvoiceController import InvoiceController
from core.controllers.LocationController import LocationController from core.controllers.LocationController import LocationController
from core.controllers.PrivilegePolicyController import PrivilegePolicyController from core.controllers.PolicyController import PolicyController
from core.controllers.ProfileController import ProfileController from core.controllers.ProfileController import ProfileController
from core.controllers.SubscriptionController import SubscriptionController from core.controllers.SubscriptionController import SubscriptionController
from core.controllers.SubscriptionPlanController import SubscriptionPlanController from core.controllers.SubscriptionPlanController import SubscriptionPlanController
@ -403,34 +402,32 @@ if __name__ == '__main__':
elif arguments.command == 'policy': elif arguments.command == 'policy':
policy_controller = None
if arguments.subcommand is None: if arguments.subcommand is None:
policy_parser.print_help() policy_parser.print_help()
elif arguments.policy == 'capability': policy = PolicyController.get(arguments.policy)
policy_controller = CapabilityPolicyController
elif arguments.policy == 'privilege': if policy is not None:
policy_controller = PrivilegePolicyController
if policy_controller is not None:
if arguments.subcommand == 'preview': if arguments.subcommand == 'preview':
print(policy_controller.preview()) print(PolicyController.preview(policy))
elif arguments.subcommand == 'instate': elif arguments.subcommand == 'instate':
policy_controller.instate() PolicyController.instate(policy)
elif arguments.subcommand == 'inspect': elif arguments.subcommand == 'inspect':
if policy_controller.is_instated(): if PolicyController.is_instated(policy):
pprint.pp({'status': 'Active'}) pprint.pp({'status': 'Instated'})
elif PolicyController.is_suggestible(policy):
pprint.pp({'status': 'Suggested'})
else: else:
pprint.pp({'status': 'Inactive'}) pprint.pp({'status': 'Uninstated'})
elif arguments.subcommand == 'revoke': elif arguments.subcommand == 'revoke':
policy_controller.revoke() PolicyController.revoke(policy)
elif arguments.command == 'get': elif arguments.command == 'get':