You have already learned how to parse dates from strings and compare date components to automate HR notifications. When employee hire dates are stored as strings in the `YYYY-MM-DD` format, you can extract the **month** portion to check for work anniversaries. This is an essential skill for creating scripts that help HR teams celebrate employee milestones and foster engagement.

# Extracting the month from a date string in 'YYYY-MM-DD' format
hire_date = "2019-07-15"
month = hire_date[5:7]  # This gets the substring representing the month
print("Month:", month)

When creating notification messages, make sure they are **clear** and easy to understand. Include the employee's **name** and the number of **years** they have been with the company. This helps personalize the message and ensures the recipient understands the purpose of the notification.


import unittest
import user_code
import ast
import re   
import importlib
import csv
import unittest
import importlib
import io
from contextlib import redirect_stdout

class TestTask(unittest.TestCase):
    def test_anniversary_notification_prints_for_matching_month(self):
        import user_code
        importlib.reload(user_code)
        employees = [
            {"name": "Alice Smith", "hire_date": "2018-07-10"},
            {"name": "Bob Lee", "hire_date": "2020-03-22"},
            {"name": "Carla Gomez", "hire_date": "2019-07-15"}
        ]
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.anniversary_notifications(employees, "07", 2024)
        output = normalize_text(f.getvalue())
        _dynamic_test(
            self,
            "alice smith celebrates 6 year(s) with the company this month!" in output,
            "Correct notification for Alice Smith printed for July anniversary.",
            f"Expected notification for Alice Smith in output, got: {output}"
        )
        _dynamic_test(
            self,
            "carla gomez celebrates 5 year(s) with the company this month!" in output,
            "Correct notification for Carla Gomez printed for July anniversary.",
            f"Expected notification for Carla Gomez in output, got: {output}"
        )

    def test_no_notification_for_non_matching_month(self):
        import user_code
        importlib.reload(user_code)
        employees = [
            {"name": "Alice Smith", "hire_date": "2018-07-10"},
            {"name": "Bob Lee", "hire_date": "2020-03-22"},
            {"name": "Carla Gomez", "hire_date": "2019-07-15"}
        ]
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.anniversary_notifications(employees, "03", 2024)
        output = normalize_text(f.getvalue())
        _dynamic_test(
            self,
            "bob lee celebrates 4 year(s) with the company this month!" in output,
            "Correct notification for Bob Lee printed for March anniversary.",
            f"Expected notification for Bob Lee in output, got: {output}"
        )
        _dynamic_test(
            self,
            "alice smith" not in output and "carla gomez" not in output,
            "No notifications printed for employees not matching March anniversary.",
            f"Unexpected notification for Alice Smith or Carla Gomez in output: {output}"
        )

    def test_years_calculation(self):
        import user_code
        importlib.reload(user_code)
        employees = [
            {"name": "Dana Lee", "hire_date": "2015-11-01"}
        ]
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.anniversary_notifications(employees, "11", 2023)
        output = normalize_text(f.getvalue())
        _dynamic_test(
            self,
            "dana lee celebrates 8 year(s) with the company this month!" in output,
            "Correct years of service calculated and printed for Dana Lee.",
            f"Expected notification for Dana Lee with 8 years, got: {output}"
        )

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

def normalize_text(text):
    text = text.lower()
    text = re.sub(r"\\s{2,}", " ", text)
    text = re.sub(r"\\s*([,:?])\\s*", r"\\1 ", text)
    return text.strip()

def change_var(code: str, var_name: str, value: str) -> str:
    tree = ast.parse(code)
    lines = code.splitlines()
    changed = False
    # Collect all assignment nodes to modify
    assign_nodes = [
        (i, node)
        for i, node in enumerate(tree.body)
        if isinstance(node, ast.Assign)
        and any(isinstance(target, ast.Name) and target.id == var_name for target in node.targets)
    ]

    # If nothing to change, return unmodified code
    if not assign_nodes:
        return code

    # Perform replacements for all matching assignments (from last to first to not break line offsets)
    for i, node in reversed(assign_nodes):
        start_line = node.lineno - 1
        line = lines[start_line]
        indent = ' ' * (len(line) - len(line.lstrip()))
        lines[start_line] = f"{indent}{var_name} = {value}"
        next_line = len(lines)
        for next_node in tree.body[i+1:]:
            if hasattr(next_node, 'lineno'):
                next_line = next_node.lineno - 1
                break
        if next_line > start_line + 1:
            lines[start_line+1:next_line] = []
        changed = True

    return '\\n'.join(lines) if changed else code

if __name__ == "__main__":
    unittest.main()


test_main.py

A practical Python course tailored for HR professionals. Learn how to automate HR tasks, analyze employee data, and generate insightful reports using Python. Each section blends engaging theory with hands-on, real-world challenges relevant to HR workflows.

Learn how to use Python to automate common HR tasks, such as attendance tracking, onboarding, and document management.

Dive into data analysis techniques for HR, including calculating statistics, visualizing trends, and interpreting employee data.

Master the art of generating automated reports and actionable insights from HR data using Python.

Challenge: Anniversary Notification Script

Solution

Challenge: Anniversary Notification Script

Solution