data_models.py

#!/usr/bin/env python3
"""
Data Models for Attendance System
Contains dataclasses for AttendanceRecord and MembershipRecord.
"""

from dataclasses import dataclass
from datetime import datetime
from typing import List


@dataclass
class AttendanceRecord:
    """Represents an attendance record."""
    timestamp: datetime
    eid: str
    type: str
    activity: str
    subactivity: str
    
    @classmethod
    def from_row(cls, row: List[str]) -> 'AttendanceRecord':
        """Create an AttendanceRecord from a spreadsheet row."""
        return cls(
            timestamp=datetime.fromisoformat(row[0].replace('Z', '+00:00')),
            eid=row[1],
            type=row[2],
            activity=row[3],
            subactivity=row[4] if len(row) > 4 else ""
        )


@dataclass
class MembershipRecord:
    """Represents a membership record."""
    timestamp: datetime
    eid: str
    type: str
    firstname: str
    lastname: str
    email: str
    committees: str
    referral: str
    major: str
    joined_discord: str
    agree_to_waiver: str
    grad_semester: str
    gender: str
    hispanic: str
    race: str
    minor: str
    
    @classmethod
    def from_row(cls, row: List[str]) -> 'MembershipRecord':
        """Create a MembershipRecord from a spreadsheet row."""
        # Pad row with empty strings if needed
        padded_row = row + [''] * (16 - len(row))
        return cls(
            timestamp=datetime.fromisoformat(padded_row[0].replace('Z', '+00:00')),
            eid=padded_row[1],
            type=padded_row[2],
            firstname=padded_row[3],
            lastname=padded_row[4],
            email=padded_row[5],
            committees=padded_row[6],
            referral=padded_row[7],
            major=padded_row[8],
            joined_discord=padded_row[9],
            agree_to_waiver=padded_row[10],
            grad_semester=padded_row[11],
            gender=padded_row[12],
            hispanic=padded_row[13],
            race=padded_row[14],
            minor=padded_row[15]
        )