more filesHEADmaster

1 files changed, 151 insertions, 0 deletions

diff --git a/src/automationstatus/crons.py b/src/automationstatus/crons.py
new file mode 100644
index 0000000..da1d59d
--- /dev/null
+++ b/src/automationstatus/crons.py
@@ -0,0 +1,151 @@
+from .models import DeploymentWorkflow, Summary
+from .models import OrchestrationWorkflow
+from datetime import datetime
+from django.apps import apps
+from django.db import connection
+from django.core import serializers
+from django.utils import timezone
+
+
+def count():
+  queryset = DeploymentWorkflow.objects.all()
+
+  in_progress_clusters = 0
+  failed_clusters = 0
+  completed_clusters = 0
+  ingested = 0
+  online = 0
+  bmc = 0
+  firmware_scheduled = 0
+  firmware_upgraded = 0
+  wr_scheduled = 0
+  wr_installed = 0
+
+  for workflow in queryset:
+    if workflow.wr_installed:
+      completed_clusters = completed_clusters + 1
+    else:
+      in_progress_clusters = in_progress_clusters + 1
+    # summary.failed = failed_clusters
+
+    if workflow.ingested:
+      ingested = ingested + 1
+
+    if workflow.online:
+      online = online + 1
+
+    if workflow.bmc:
+      bmc = bmc + 1
+
+    if workflow.firmware_scheduled and not workflow.firmware_upgraded:
+      firmware_scheduled = firmware_scheduled + 1
+
+    if workflow.firmware_upgraded:
+      firmware_upgraded = firmware_upgraded + 1
+
+    if workflow.wr_scheduled and not workflow.wr_installed:
+      wr_scheduled = wr_scheduled + 1
+
+    if workflow.wr_installed:
+      wr_installed = wr_installed + 1
+
+
+  summary = Summary()
+  summary.clusters = len(queryset)
+  summary.in_progress = in_progress_clusters
+  summary.completed = completed_clusters
+  summary.ingested = ingested
+  summary.online = online
+  summary.bmc = bmc
+  summary.firmware_scheduled = firmware_scheduled
+  summary.firmware_upgraded = firmware_upgraded
+  summary.wr_scheduled = wr_scheduled
+  summary.wr_installed = wr_installed
+  summary.created_at = timezone.now()
+  print(summary)
+  try:
+    summary.save()
+  except Exception as e:
+    print(e)
+
+
+def summary():
+  print("**** running cron: summary **** : {}".format(datetime.now()))
+  servers = apps.get_model('caas', 'Server')
+  query_set = servers.objects.select_related('cluster').select_related('blade')
+  for server in query_set:
+      # Set server id and date
+      status = DeploymentWorkflow()
+      status.cluster = server.cluster
+      status.created_at = timezone.now()
+      # CIQ ingestion
+      status.ingested = True
+      # Server online check
+      status.online = True if server.blade_id else False
+      #BMC done check
+      if server.pxe_mac_address and server.intel_nic_firmware_version:
+          status.bmc = True
+      # Check Firmware schedule and installation status
+      # Get server list from
+      Server = apps.get_model('caas', 'Server')
+      server = Server.objects.get(cluster=server.cluster)
+      FirmwareUpgradeSchedule = apps.get_model('caas', 'FirmwareUpgradeSchedule')
+      try:
+          firmware_schedule = FirmwareUpgradeSchedule.objects.get(server=server)
+          # Firmware installation Scheduled and completiong check
+          status.firmware_scheduled = True if firmware_schedule.date_scheduled else False
+          status.firmware_upgraded = True if firmware_schedule.date_completed else False
+      except FirmwareUpgradeSchedule.DoesNotExist:
+          status.firmware_schedule = False
+          status.firmware_upgraded = False
+      except Exception as e:
+        print(e)
+      
+      # Check WindRiver schedule and installation status
+      WrInstallSchedule = apps.get_model('caas', 'WrInstallSchedule')
+      try:
+          wr_schedule = WrInstallSchedule.objects.select_related('cluster').get(cluster=server.cluster)
+          # WindRiver installation Scheduled and completiong check
+          status.wr_scheduled = True if wr_schedule.date_scheduled else False
+          status.wr_installed = True if wr_schedule.date_completed else False
+      except WrInstallSchedule.DoesNotExist:
+          status.wr_scheduled = False
+          status.wr_installed = False
+      except Exception as e:
+        print(e)
+      # Save status object
+      status.save()
+  count()
+
+def orchestration_summary():
+  print("**** running cron: orchestration_summary **** : {}".format(datetime.now()))
+  regions = apps.get_model('orchestration', 'remoteregionsetup')
+  status = OrchestrationWorkflow()
+
+  # Namespace count and Timestamp
+  status.namespaces = regions.objects.count()
+  status.created_at = timezone.now()
+
+  # Service accounts
+  status.orch_service_account = regions.objects.filter(serviceaccount__in=['orchestration-sa', 'SVC-FE-Atlas']).count()
+  status.edge_eng_service_account = regions.objects.filter(serviceaccount='SVC-Edge-Eng').count() + regions.objects.filter(serviceaccount__contains='application-sa-').count()
+  status.samsung_service_account = regions.objects.filter(serviceaccount='samsung-sa').count()
+
+  # Orchestration Kubeconfigs
+  status.orch_kubeconfig = regions.objects.filter(serviceaccount__in=['orchestration-sa', 'SVC-FE-Atlas']).exclude(kubeconfig='').count()
+
+  # Edge engineering Kubeconfigs
+  status.edge_eng_kubeconfig = regions.objects.filter(serviceaccount='SVC-Edge-Eng').exclude(kubeconfig='').count() 
+  status.edge_eng_kubeconfig += regions.objects.filter(serviceaccount__contains='application-sa-').exclude(kubeconfig='').count()
+
+  # Samsung Kubeconfigs
+  status.samsung_kubeconfig= regions.objects.filter(serviceaccount__in=['orchestration-sa', 'SVC-FE-Atlas']).exclude(kubeconfig='').count()
+
+  # Uncomment and use once Dev completes tracking on his side
+  # Setting to 0 for now
+  status.data_network_setup = 0
+
+  status.save()
+
+    
+