id: python-generate-outputs
namespace: company.team
variables:
  file: orders.csv
tasks:
  - id: analyze_orders
    type: io.kestra.plugin.scripts.python.Script
    warningOnStdErr: false
    beforeCommands:
      - pip install faker kestra > /dev/null
    outputFiles:
      - "*.csv"
    script: >
      import csv

      import random

      import time

      from faker import Faker

      from kestra import Kestra


      start_time = time.time()

      fake = Faker()


      # list of columns for the CSV file

      columns = [
          "order_id",
          "customer_name",
          "customer_email",
          "product_id",
          "price",
          "quantity",
          "total",
      ]

      filename = "{{ vars.file }}"

      tags = {'file': filename}


      # Generate 100 random orders

      orders = []

      for i in range(100):
          order_id = i + 1
          customer_name = fake.name()
          customer_email = fake.email()
          product_id = random.randint(1, 20)
          price = round(random.uniform(10.0, 200.0), 2)
          quantity = random.randint(1, 10)
          total = round(price * quantity, 2)
          orders.append(
              [order_id, customer_name, customer_email, product_id, price, quantity, total]
          )

      # Write the orders to a CSV file

      with open(filename, "w", newline="") as file:
          writer = csv.writer(file)
          writer.writerow(columns)
          writer.writerows(orders)

      # Calculate and print the sum and average of the "total" column

      total_sum = sum(order[6] for order in orders)

      average_order = round(total_sum / len(orders), 2)

      print(f"Total sum: {total_sum}")

      print(f"Average Order value: {average_order}")


      Kestra.outputs({"total_sum": total_sum, "average_order": average_order})

      Kestra.counter('total_sum', total_sum, tags)

      Kestra.counter('average_order', average_order, tags)


      end_time = time.time()  

      processing_time = end_time - start_time

      Kestra.timer('processing_time', processing_time, tags)

      print(f"The script execution took: {processing_time} seconds")