def get(self, request, *args, **kwargs):
"""Trigger regeneration of machine-related/service files."""
fqdn = request.GET.get('fqdn', None)
option = request.GET.get('option', None)
service = request.GET.get('service', None)
if service and (service.lower()
in {Regenerate.COBBLER, Regenerate.SERIALCONSOLE}):
if isinstance(request.user, AnonymousUser) or not request.auth:
return AuthRequiredSerializer().as_json
if not request.user.is_superuser:
return ErrorMessage(
"Only superusers are allowed to perform this action!"
).as_json
# regenerate Cobbler entries
if service.lower() == Regenerate.COBBLER:
signal_cobbler_regenerate.send(sender=None, domain_id=None)
return Message("Regenerate Cobbler entries...").as_json
# regenerate serial console entries iterating over all cscreen servers
elif service.lower() == Regenerate.SERIALCONSOLE:
machines = SerialConsole.objects.all().values_list(
'cscreen_server__fqdn', flat=True)
for fqdn in machines.distinct():
signal_serialconsole_regenerate.send(
sender=None, cscreen_server_fqdn=fqdn)
return Message("Regenerate serial console entries...").as_json
elif (fqdn is not None) and (option is not None):
try:
result = get_machine(fqdn,
redirect_to='api:regenerate',
data=request.GET)
if isinstance(result, Serializer):
return result.as_json
elif isinstance(result, HttpResponseRedirect):
return result
machine = result
except Exception as e:
return ErrorMessage(str(e)).as_json
if option.lower() != Regenerate.MOTD:
return ErrorMessage(
"Unknown option '{}'!".format(option)).as_json
if isinstance(request.user, AnonymousUser) or not request.auth:
return AuthRequiredSerializer().as_json
try:
if option.lower() == Regenerate.MOTD:
machine.update_motd(user=request.user)
return Message("OK.").as_json
except Exception as e:
return ErrorMessage(str(e)).as_json
return ErrorMessage("Unknown service '{}'!".format(service)).as_json
def regenerate_serialconsole_record(self, user=None):
"""Call respective signal."""
from orthos2.data.signals import signal_serialconsole_regenerate
if self.has_serialconsole():
signal_serialconsole_regenerate.send(
sender=self.__class__,
cscreen_server_fqdn=self.fqdn_domain.cscreen_server.fqdn)
def get(self, request, *args, **kwargs):
"""Trigger regeneration of machine-related/service files."""
service = request.GET.get('service', None)
fqdn = request.GET.get('fqdn', None)
machine = None
if isinstance(request.user, AnonymousUser) or not request.auth:
return AuthRequiredSerializer().as_json
if not request.user.is_superuser:
return ErrorMessage(
"Only superusers are allowed to perform this action!").as_json
if not service:
return ErrorMessage("Service not set").as_json
if fqdn:
try:
machine = get_machine(fqdn,
redirect_to='api:regenerate',
data=request.GET)
if not machine:
return ErrorMessage(
"machine {} not found".format(fqdn)).as_json
if isinstance(machine, Serializer):
return machine.as_json
elif isinstance(machine, HttpResponseRedirect):
return machine
except Exception as e:
return ErrorMessage(str(e)).as_json
# regenerate Cobbler entries
if service.lower() == RegenerateCommand.COBBLER:
domain_id = None
if fqdn:
domain = get_domain(fqdn)
if not domain:
return ErrorMessage("No domain found for machine: " +
fqdn).as_json
o_domain = Domain.objects.get(name=domain)
if not o_domain:
return ErrorMessage("No orthos domain found for domain: " +
domain).as_json
domain_id = getattr(o_domain, 'id', None)
if not domain_id:
return ErrorMessage(
"Could not find id for orthos domain: " +
domain).as_json
msg = 'domain ' + domain
else:
domain = None
msg = 'all domains'
signal_cobbler_regenerate.send(sender=None, domain_id=domain_id)
return Message("Regenerate Cobbler entries for " + msg).as_json
# regenerate serial console entries iterating over all cscreen servers
elif service.lower() == RegenerateCommand.SERIALCONSOLE:
machines = SerialConsole.objects.all().values_list(
'cscreen_server__fqdn', flat=True)
if fqdn:
if fqdn in machines.distinct():
signal_serialconsole_regenerate.send(
sender=None, cscreen_server_fqdn=fqdn)
msg = fqdn
else:
return ErrorMessage("Not a serial console server: " +
fqdn).as_json
else:
msg = ''
for fqdn in machines.distinct():
signal_serialconsole_regenerate.send(
sender=None, cscreen_server_fqdn=fqdn)
msg += ' ' + fqdn
return Message(
"Regenerated serial console entries for serial console servers: "
+ msg).as_json
# regenerate MOTD (only works per machine atm)
elif service.lower() == RegenerateCommand.MOTD:
if not fqdn:
return Message("regenerte motd needs fqdn parameter").as_json
machine.update_motd(user=request.user)
return Message("OK.").as_json
else:
return ErrorMessage("Unknown service {}".format(service)).as_json
return ErrorMessage("Unknown error - params: {} - {}".format(
service, fqdn)).as_json