1
0
mirror of https://github.com/dkmstr/openuds.git synced 2024-12-22 13:34:04 +03:00

Removed unused (since years ago) "must_assign_manually"

This commit is contained in:
Adolfo Gómez García 2024-04-28 16:33:24 +02:00
parent 53e0cefc21
commit 628f43a2e7
No known key found for this signature in database
GPG Key ID: DD1ABF20724CDA23
16 changed files with 90 additions and 75 deletions

View File

@ -79,10 +79,7 @@ class Services(DetailHandler): # pylint: disable=too-many-public-methods
'cache_tooltip_l2': _(info.cache_tooltip_l2), 'cache_tooltip_l2': _(info.cache_tooltip_l2),
'needs_osmanager': info.needs_osmanager, 'needs_osmanager': info.needs_osmanager,
'allowed_protocols': info.allowed_protocols, 'allowed_protocols': info.allowed_protocols,
'services_type_provided': [ 'services_type_provided': info.services_type_provided,
info.services_type_provided
], # As a list for compatibility, to be removed TODO: Remove
'must_assign_manually': info.must_assign_manually,
'can_reset': info.can_reset, 'can_reset': info.can_reset,
'can_list_assignables': info.can_assign(), 'can_list_assignables': info.can_assign(),
} }

View File

@ -55,7 +55,6 @@ class DynamicService(services.Service, abc.ABC): # pylint: disable=too-many-pub
uses_cache = False # Cache are running machine awaiting to be assigned uses_cache = False # Cache are running machine awaiting to be assigned
uses_cache_l2 = False # L2 Cache are running machines in suspended state uses_cache_l2 = False # L2 Cache are running machines in suspended state
needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services, i.e. virtual machines with agent) needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services, i.e. virtual machines with agent)
must_assign_manually = False # If true, the system can't do an automatic assignation of a deployed user service from this service
# can_reset = True # can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -58,7 +58,6 @@ class FixedService(services.Service, abc.ABC): # pylint: disable=too-many-publi
uses_cache = False # Cache are running machine awaiting to be assigned uses_cache = False # Cache are running machine awaiting to be assigned
uses_cache_l2 = False # L2 Cache are running machines in suspended state uses_cache_l2 = False # L2 Cache are running machines in suspended state
needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services, i.e. virtual machines with agent) needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services, i.e. virtual machines with agent)
must_assign_manually = False # If true, the system can't do an automatic assignation of a deployed user service from this service
# can_reset = True # can_reset = True
# If machine has an alternate field with it, it will be used instead of "machines" field # If machine has an alternate field with it, it will be used instead of "machines" field

View File

@ -96,12 +96,5 @@ class ServiceProviderFactory(factory.ModuleFactory[ServiceProvider]):
s s
for p in self.providers().values() for p in self.providers().values()
for s in p.offers for s in p.offers
if s.publication_type is None and s.must_assign_manually is False if s.publication_type is None
] ]
# old code :-)
# res = []
# for p in self.providers().values():
# for s in p.offers:
# if s.publication_type is None and s.must_assign_manually is False:
# res.append(s)
# return res

View File

@ -160,15 +160,6 @@ class Service(Module):
# : If the service needs a o.s. manager (see os managers section) # : If the service needs a o.s. manager (see os managers section)
needs_osmanager: bool = False needs_osmanager: bool = False
# : If the service can be autoassigned or needs to be assigned by administrator
# : Not all services are for assigning it. Thing, i.e., a Service that manages
# : a number of Server. The desired behavior will be to let administrator
# : the service to a user in the administration interface, an not the system
# : to assign the service automatically. If this is true, the core will not
# : assign the service automatically, so if the user do not have a consumable
# : assigned, the user will never get one (of this kind, of course)
must_assign_manually: typing.ClassVar[bool] = False
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)
# : If you provide this, UDS will assume that the service needs a preparation. # : If you provide this, UDS will assume that the service needs a preparation.
# : If not provided (it is None), UDS will assume that service do not needs # : If not provided (it is None), UDS will assume that service do not needs
@ -307,18 +298,6 @@ class Service(Module):
# Keep untouched if services_limit is not present # Keep untouched if services_limit is not present
def user_services_for_assignation(self, **kwargs: typing.Any) -> collections.abc.Iterable['UserService']:
"""
override this if mustAssignManualy is True
@params kwargs: Named arguments
@return an iterable with the services that we can assign manually (they must be of type UserDeployment)
We will access the returned iterable in "name" basis. This means that the service will be assigned by "name", so be care that every single service
name returned is unique :-)
"""
raise Exception(
f'The class {self.__class__.__name__} has been marked as manually asignable but no requestServicesForAssignetion provided!!!'
)
def mac_generator(self) -> 'UniqueMacGenerator': def mac_generator(self) -> 'UniqueMacGenerator':
""" """
Utility method to access provided macs generator (inside environment) Utility method to access provided macs generator (inside environment)

View File

@ -91,9 +91,6 @@ class OVirtLinkedService(services.Service): # pylint: disable=too-many-public-m
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = True needs_osmanager = True
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
can_reset = True can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -80,9 +80,6 @@ class OGService(services.Service):
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = False needs_osmanager = False
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)
# : In our case, we do no need a publication, so this is None # : In our case, we do no need a publication, so this is None

View File

@ -86,9 +86,6 @@ class OpenNebulaLiveService(services.Service):
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = True needs_osmanager = True
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
can_reset = True can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -88,9 +88,6 @@ class OpenStackLiveService(services.Service):
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = True needs_osmanager = True
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
can_reset = True can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -132,7 +132,6 @@ class IPMachinesService(services.Service):
uses_cache = False # Cache are running machine awaiting to be assigned uses_cache = False # Cache are running machine awaiting to be assigned
uses_cache_l2 = False # L2 Cache are running machines in suspended state uses_cache_l2 = False # L2 Cache are running machines in suspended state
needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services itselfs, i.e. virtual machines with agent) needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services itselfs, i.e. virtual machines with agent)
must_assign_manually = False # If true, the system can't do an automatic assignation of a deployed user service from this service
user_service_type = IPMachinesUserService user_service_type = IPMachinesUserService
@ -149,7 +148,7 @@ class IPMachinesService(services.Service):
def enumerate_assignables(self) -> collections.abc.Iterable[types.ui.ChoiceItem]: def enumerate_assignables(self) -> collections.abc.Iterable[types.ui.ChoiceItem]:
now = sql_now() now = sql_now()
return [ return [
gui.choice_item(f'{server.host}|{server.mac}', server.uuid) gui.choice_item(server.uuid, f'{server.host}|{server.mac}')
for server in fields.get_server_group_from_field(self.server_group).servers.all() for server in fields.get_server_group_from_field(self.server_group).servers.all()
if server.locked_until is None or server.locked_until < now if server.locked_until is None or server.locked_until < now
] ]
@ -163,12 +162,10 @@ class IPMachinesService(services.Service):
server: 'models.Server' = models.Server.objects.get(uuid=assignable_id) server: 'models.Server' = models.Server.objects.get(uuid=assignable_id)
ipmachine_instance: IPMachinesUserService = typing.cast(IPMachinesUserService, userservice_instance) ipmachine_instance: IPMachinesUserService = typing.cast(IPMachinesUserService, userservice_instance)
if server.locked_until is None or server.locked_until < sql_now(): if server.locked_until is None or server.locked_until < sql_now():
# Lock the server for 10 year right now... self.lock_server(server.uuid)
server.locked_until = sql_now() + datetime.timedelta(days=365) return ipmachine_instance.assign(server.uuid)
return ipmachine_instance.assign(server.host) return ipmachine_instance._error('Host already assigned')
return ipmachine_instance._error('IP already assigned')
def get_unassigned(self) -> str: def get_unassigned(self) -> str:
''' '''
@ -177,9 +174,9 @@ class IPMachinesService(services.Service):
list_of_servers = list(fields.get_server_group_from_field(self.server_group).servers.all()) list_of_servers = list(fields.get_server_group_from_field(self.server_group).servers.all())
if self.randomize_host.as_bool() is True: if self.randomize_host.as_bool() is True:
random.shuffle(list_of_servers) # Reorder the list randomly if required random.shuffle(list_of_servers) # Reorder the list randomly if required
for server in list_of_servers: for server in list_of_servers:
if server.locked_until is None or server.locked_until < sql_now(): if server.locked_until is None or server.locked_until < sql_now():
return server.uuid return server.uuid
raise exceptions.services.InsufficientResourcesException() raise exceptions.services.InsufficientResourcesException()
def get_host_mac(self, server_uuid: str) -> typing.Tuple[str, str]: def get_host_mac(self, server_uuid: str) -> typing.Tuple[str, str]:
@ -208,7 +205,10 @@ class IPMachinesService(services.Service):
# Maybe, an user has logged in on an unassigned machine # Maybe, an user has logged in on an unassigned machine
# if lockForExternalAccess is enabled, we must lock it # if lockForExternalAccess is enabled, we must lock it
if self.lock_on_external_access.as_bool() is True: if self.lock_on_external_access.as_bool() is True:
self.do_log(types.log.LogLevel.DEBUG, f'External login detected for {id}, locking machine for {self.get_max_lock_time()} or until logout') self.do_log(
types.log.LogLevel.DEBUG,
f'External login detected for {id}, locking machine for {self.get_max_lock_time()} or until logout',
)
self.lock_server(id) self.lock_server(id)
def process_logout(self, id: str, remote_login: bool) -> None: def process_logout(self, id: str, remote_login: bool) -> None:
@ -255,4 +255,3 @@ class IPMachinesService(services.Service):
# always is available # always is available
def is_avaliable(self) -> bool: def is_avaliable(self) -> bool:
return True return True

View File

@ -59,7 +59,6 @@ class IPSingleMachineService(services.Service):
uses_cache = False # Cache are running machine awaiting to be assigned uses_cache = False # Cache are running machine awaiting to be assigned
uses_cache_l2 = False # L2 Cache are running machines in suspended state uses_cache_l2 = False # L2 Cache are running machines in suspended state
needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services itselfs, i.e. virtual machines with agent) needs_osmanager = False # If the service needs a s.o. manager (managers are related to agents provided by services itselfs, i.e. virtual machines with agent)
must_assign_manually = False # If true, the system can't do an automatic assignation of a deployed user service from this service
user_service_type = IPMachineUserService user_service_type = IPMachineUserService

View File

@ -95,9 +95,6 @@ class ProxmoxServiceLinked(DynamicService):
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = True needs_osmanager = True
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
can_reset = True can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -104,9 +104,6 @@ class ServiceOne(services.Service):
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = False needs_osmanager = False
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)
# : In our case, we do no need a publication, so this is None # : In our case, we do no need a publication, so this is None
@ -212,7 +209,6 @@ class ServiceTwo(services.Service):
cache_tooltip_l2 = _('L2 cache for dummy elements') cache_tooltip_l2 = _('L2 cache for dummy elements')
needs_osmanager = False needs_osmanager = False
must_assign_manually = False
# : Types of publications. In this case, we will include a publication # : Types of publications. In this case, we will include a publication
# : type for this one # : type for this one

View File

@ -65,7 +65,6 @@ class TestServiceNoCache(services.Service):
cache_tooltip_l2 = _('None') cache_tooltip_l2 = _('None')
needs_osmanager = False needs_osmanager = False
must_assign_manually = False
publication_type = None publication_type = None
user_service_type = TestUserService user_service_type = TestUserService
@ -97,7 +96,6 @@ class TestServiceCache(services.Service):
cache_tooltip_l2 = _('L2 cache for dummy elements') cache_tooltip_l2 = _('L2 cache for dummy elements')
needs_osmanager = False needs_osmanager = False
must_assign_manually = False
# : Types of publications. In this case, we will include a publication # : Types of publications. In this case, we will include a publication
# : type for this one # : type for this one

View File

@ -86,9 +86,6 @@ class XenLinkedService(services.Service): # pylint: disable=too-many-public-met
# : If the service needs a s.o. manager (managers are related to agents # : If the service needs a s.o. manager (managers are related to agents
# : provided by services itselfs, i.e. virtual machines with actors) # : provided by services itselfs, i.e. virtual machines with actors)
needs_osmanager = True needs_osmanager = True
# : If true, the system can't do an automatic assignation of a deployed user
# : service from this service
must_assign_manually = False
can_reset = True can_reset = True
# : Types of publications (preparated data for deploys) # : Types of publications (preparated data for deploys)

View File

@ -30,6 +30,7 @@
""" """
Author: Adolfo Gómez, dkmaster at dkmon dot com Author: Adolfo Gómez, dkmaster at dkmon dot com
""" """
import random
from unittest import mock from unittest import mock
from uds import models from uds import models
@ -51,9 +52,12 @@ class TestServiceMulti(UDSTransactionTestCase):
self.assertTrue(server_group.servers.count() > 0) self.assertTrue(server_group.servers.count() > 0)
self.assertEqual(service.port.value, fixtures.SERVICE_MULTI_VALUES_DICT['port']) self.assertEqual(service.port.value, fixtures.SERVICE_MULTI_VALUES_DICT['port'])
self.assertEqual( self.assertEqual(
service.ignore_minutes_on_failure.value, fixtures.SERVICE_MULTI_VALUES_DICT['ignore_minutes_on_failure'] service.ignore_minutes_on_failure.value,
fixtures.SERVICE_MULTI_VALUES_DICT['ignore_minutes_on_failure'],
)
self.assertEqual(
service.max_session_hours.value, fixtures.SERVICE_MULTI_VALUES_DICT['max_session_hours']
) )
self.assertEqual(service.max_session_hours.value, fixtures.SERVICE_MULTI_VALUES_DICT['max_session_hours'])
self.assertEqual( self.assertEqual(
service.lock_on_external_access.value, fixtures.SERVICE_MULTI_VALUES_DICT['lock_on_external_access'] service.lock_on_external_access.value, fixtures.SERVICE_MULTI_VALUES_DICT['lock_on_external_access']
) )
@ -163,3 +167,73 @@ class TestServiceMulti(UDSTransactionTestCase):
service.unlock_server(server.uuid) service.unlock_server(server.uuid)
server.refresh_from_db() server.refresh_from_db()
self.assertIsNone(server.locked_until) self.assertIsNone(server.locked_until)
def test_get_unassigned(self) -> None:
service = fixtures.create_service_multi()
# Without random host
service.randomize_host.value = False
server_list = list(fields.get_server_group_from_field(service.server_group).servers.all())
for num, server in enumerate(server_list, 1):
unassigned_uuid = service.get_unassigned()
# Must be the first server
self.assertEqual(unassigned_uuid, server.uuid, f'Error on element {num}') # type: ignore # if first is None,raises error
# Lock it, so it's not returned again
service.lock_server(unassigned_uuid)
# Now, randomized, we must no receive same servers
service.randomize_host.value = True
# unlock all servers
for server in server_list:
service.unlock_server(server.uuid)
# Now, execute at most 128 iterations, if we get first server all times, it's a problem
count = 0
for count in range(128):
unassigned_uuid = service.get_unassigned()
self.assertNotEqual(unassigned_uuid, server_list[0].uuid)
if unassigned_uuid != server_list[0].uuid:
break
if count == 127:
self.fail('Randomized server selection failed')
# Now ensure we can lock all servers
for num in range(len(server_list)):
unassigned_uuid = service.get_unassigned()
self.assertIsNotNone(unassigned_uuid)
service.lock_server(unassigned_uuid)
def test_enumerate_assignables(self) -> None:
service = fixtures.create_service_multi()
server_group = fields.get_server_group_from_field(service.server_group)
server_list = list(server_group.servers.all())
# lock 4 random servers, and remove them from the list
locked_servers = random.sample(server_list, 4)
for server in locked_servers:
service.lock_server(server.uuid)
server_list.remove(server)
assignables = {i['id'] for i in service.enumerate_assignables()}
self.assertEqual(assignables, {i.uuid for i in server_list})
def test_assign_from_asignables(self) -> None:
service = fixtures.create_service_multi()
server_group = fields.get_server_group_from_field(service.server_group)
server_list = list(server_group.servers.all())
# lock 4 random servers, and remove them from the list
locked_servers = random.sample(server_list, 4)
for server in locked_servers:
service.lock_server(server.uuid)
server_list.remove(server)
for server in server_list:
# Assign a server
userservice_mock = mock.MagicMock()
service.assign_from_assignables(server.uuid, mock.Mock(), userservice_mock)
userservice_mock.assign.assert_called_once_with(server.uuid)
# Ensure is locked
server.refresh_from_db()
self.assertIsNotNone(server.locked_until)