#!/usr/bin/env python3
"""Track employer drug testing policies and compliance."""

import json
from datetime import datetime
from pathlib import Path

class DrugTestingRegistry:
    def __init__(self, registry_file="drug_testing_record.json"):
        self.registry_file = registry_file
        self.load_registry()

    def load_registry(self):
        if Path(self.registry_file).exists():
            with open(self.registry_file, 'r') as f:
                self.data = json.load(f)
        else:
            self.data = {
                "employer": "",
                "state": "",
                "testing_policy": {},
                "test_history": [],
                "positive_results": [],
                "medications": []
            }

    def record_test(self, test_type, date, result):
        """Log a drug test in your personal registry."""
        self.data["test_history"].append({
            "type": test_type,
            "date": date,
            "result": result,
            "recorded_at": datetime.now().isoformat()
        })
        self.save()

    def register_medication(self, medication_name, prescriber, start_date):
        """Log prescription medications for test disputes."""
        self.data["medications"].append({
            "medication": medication_name,
            "prescriber": prescriber,
            "start_date": start_date,
            "documented_at": datetime.now().isoformat()
        })
        self.save()

    def generate_compliance_report(self):
        """Generate report showing employer compliance with state laws."""
        report = {
            "employer": self.data["employer"],
            "state": self.data["state"],
            "employer_policy": self.data["testing_policy"],
            "state_requirements": self.get_state_requirements(),
            "compliance_issues": self.identify_violations()
        }
        return report

    def get_state_requirements(self):
        """Return state-specific drug testing requirements."""
        state_reqs = {
            "CA": {
                "pre_employment": False,
                "random": False,
                "notice_required": "Reasonable",
                "marijuana_exception": True
            },
            "TX": {
                "pre_employment": True,
                "random": True,
                "notice_required": "Minimal",
                "marijuana_exception": False
            },
            "NY": {
                "pre_employment": True,
                "random": False,
                "notice_required": "24-48 hours",
                "marijuana_exception": True
            }
        }
        return state_reqs.get(self.data["state"], {})

    def identify_violations(self):
        """Compare employer policy against state law."""
        state_reqs = self.get_state_requirements()
        violations = []

        employer_policy = self.data["testing_policy"]

        # Check for illegal pre-employment testing
        if not state_reqs.get("pre_employment") and employer_policy.get("pre_employment"):
            violations.append({
                "violation": "Illegal pre-employment testing",
                "state_law": "State prohibits pre-employment testing",
                "employer_policy": "Requires pre-employment test"
            })

        return violations

    def save(self):
        with open(self.registry_file, 'w') as f:
            json.dump(self.data, f, indent=2)

Usage
registry = DrugTestingRegistry()
registry.record_test("pre-employment", "2026-01-15", "negative")
registry.register_medication("Adderall", "Dr. Smith", "2025-06-01")
report = registry.generate_compliance_report()
print(json.dumps(report, indent=2))

"""New York-specific drug test dispute procedures."""

new_york_procedures = {
    "MRO_Review": {
        "requirement": "Mandatory for positive results",
        "timeline": "3 business days",
        "action": "Contact MRO with prescription proof",
        "resource": "NY Department of Health"
    },
    "Demand_Letter": {
        "requirement": "Send certified mail",
        "content": [
            "Employer name and address",
            "Description of illegal testing or handling",
            "Citation of NY law violated",
            "Demand for remedy (reinstatement, damages)",
            "Deadline for response (14 days)"
        ]
    },
    "Regulatory_Complaint": {
        "agency": "New York State Department of Health",
        "form": "Drug Testing Complaint Form",
        "filing_deadline": "No statutory limit (within reason)"
    },
    "Legal_Action": {
        "claim_type": "Wrongful termination",
        "damages": "Back pay, benefits, punitive damages",
        "statute_of_limitations": "3 years from violation",
        "venue": "State court (not EEOC jurisdiction)"
    }
}

def get_healthcare_testing_requirements(state: str, profession: str) -> dict:
    """Healthcare workers may face stricter testing rules."""

    healthcare_reqs = {
        "CA": {
            "nurse": {"pre_employment": True, "random": True, "marijuana": False},
            "pharmacist": {"pre_employment": True, "random": True, "marijuana": False},
            "dental": {"pre_employment": True, "random": False, "marijuana": True}
        },
        "TX": {
            "nurse": {"pre_employment": True, "random": True, "marijuana": True},
            "pharmacist": {"pre_employment": True, "random": True, "marijuana": True},
            "dental": {"pre_employment": True, "random": False, "marijuana": True}
        }
    }

    return healthcare_reqs.get(state, {}).get(profession, {
        "pre_employment": True,
        "random": True,
        "marijuana": False
    })