DTT1 - Iteration 2 - Allocator module

[QU3B1M]

Epic: #4495

Description

This issue aims to englobe all the subtasks required for the completion of the DTT1 Allocator module

Subtasks

• DTT1 - Allocator module - Code refactor #4747

[QU3B1M]

Update report

• Allocation module refactored and changes merged on:
  • Upload allocator improvement to PoC #4800
• Implemented pydantic models for data validation
• Allocation launcher Improved

  import argparse
  import os
  import sys
  
  project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
  sys.path.append(project_root)
  
  from modules.allocation.generic.models import InputPayload
  from modules.allocation import Allocator
  
  
  def parse_arguments():
      parser = argparse.ArgumentParser(description="Infrastructure providing tool")
      parser.add_argument("--provider", choices=['vagrant', 'aws', None], required=False,  default=None)
      parser.add_argument("--size", choices=['micro', 'small', 'medium', 'large', None], required=False, default=None)
      parser.add_argument("--composite-name", required=False, default=None)
      parser.add_argument("--action", choices=['create', 'delete'], required=False, default='create')
      parser.add_argument("--custom-credentials", required=False, default=None)
      parser.add_argument("--track-output", required=False, default='/tmp/wazuh-qa/track.yml')
      parser.add_argument("--inventory-output", required=False, default='/tmp/wazuh-qa/inventory.yml')
      parser.add_argument("--working-dir", required=False, default='/tmp/wazuh-qa')
      return parser.parse_args()
  
  
  def main():
      Allocator.run(InputPayload(**vars(parse_arguments())))
  
  
  if __name__ == "__main__":
      main()

[jnasselle]

LGTM

[fcaffieri]

The init test will be executed in PoC Iteration 2 #4751
LGTM