plano/model_server/app/employee_data_generator.py

import pandas as pd
import random
import datetime


def generate_employee_data(conn):
    # List of possible names, positions, departments, and locations
    names = [
        "Alice",
        "Bob",
        "Charlie",
        "David",
        "Eve",
        "Frank",
        "Grace",
        "Hank",
        "Ivy",
        "Jack",
    ]
    positions = [
        "Manager",
        "Engineer",
        "Salesperson",
        "HR Specialist",
        "Marketing Analyst",
    ]
    departments = ["Engineering", "Marketing", "HR", "Sales", "Finance"]
    locations = ["New York", "San Francisco", "Austin", "Boston", "Chicago"]

    # Function to generate random hire date
    def random_hire_date():
        start_date = datetime.date(2000, 1, 1)
        end_date = datetime.date(2023, 12, 31)
        time_between_dates = end_date - start_date
        days_between_dates = time_between_dates.days
        random_number_of_days = random.randrange(days_between_dates)
        hire_date = start_date + datetime.timedelta(days=random_number_of_days)
        return hire_date

    # Function to generate random employee data
    def generate_employee_records(count):
        employees = []

        for _ in range(count):
            name = random.choice(names)
            position = random.choice(positions)
            salary = round(
                random.uniform(50000, 150000), 2
            )  # Salary between 50,000 and 150,000
            department = random.choice(departments)
            location = random.choice(locations)
            hire_date = random_hire_date()
            performance_score = round(
                random.uniform(1, 5), 2
            )  # Performance score between 1.0 and 5.0
            years_of_experience = random.randint(
                1, 30
            )  # Years of experience between 1 and 30

            employee = {
                "position": position,
                "name": name,
                "salary": salary,
                "department": department,
                "location": location,
                "hire_date": hire_date,
                "performance_score": performance_score,
                "years_of_experience": years_of_experience,
            }

            employees.append(employee)

        return employees

    # Generate 10 random employee records
    employee_records = generate_employee_records(200)

    # Convert the list of dictionaries to a DataFrame
    df = pd.DataFrame(employee_records)

    df.to_sql("employees", conn, index=False)

    return