post_grades.py

import os
import json
import argparse
from dotenv import load_dotenv
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from googleapiclient.discovery import build

# Load environment variables
load_dotenv()

SCOPES = [
    'https://www.googleapis.com/auth/classroom.coursework.students',
    'https://www.googleapis.com/auth/classroom.profile.emails'
]

def get_classroom_service():
    creds = None
    if os.path.exists('token.json'):
        creds = Credentials.from_authorized_user_file('token.json', SCOPES)
    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            creds.refresh(Request())
        else:
            flow = InstalledAppFlow.from_client_secrets_file(
                'credentials.json', SCOPES)
            creds = flow.run_local_server(port=0)
        with open('token.json', 'w') as token:
            token.write(creds.to_json())
    
    return build('classroom', 'v1', credentials=creds)

def main():
    parser = argparse.ArgumentParser(description='Post approved grades to Google Classroom.')
    parser.add_argument('--course_id', required=True, help='Google Classroom Course ID')
    parser.add_argument('--coursework_id', required=True, help='Google Classroom CourseWork ID')
    parser.add_argument('--report', default='pre_marking_report.json', help='Path to pre-marking JSON report')
    args = parser.parse_args()

    if not os.path.exists(args.report):
        print(f"Report file {args.report} not found.")
        return

    with open(args.report, 'r') as f:
        report_data = json.load(f)

    classroom = get_classroom_service()

    print(f"Posting grades for {len(report_data)} students...")

    for item in report_data:
        student_name = item['student_name']
        submission_id = item['submission_id']
        grade = item['proposed_grade']
        feedback = item['overall_feedback']
        
        # Add rubric breakdown to feedback
        detailed_comment = f"Pre-marking Evaluation:\n"
        detailed_comment += f"Proposed Grade: {grade}\n\n"
        detailed_comment += "Rubric Breakdown:\n"
        for crit, details in item['rubric_breakdown'].items():
            detailed_comment += f"- {crit}: {details['score']} - {details['comment']}\n"
        
        detailed_comment += "\nIdentified Problems:\n"
        for prob in item['identified_problems']:
            detailed_comment += f"- {prob}\n"
        
        detailed_comment += f"\nFeedback: {feedback}"

        print(f"Updating {student_name} (ID: {submission_id})...")
        
        try:
            # Update the grade (as draftGrade)
            classroom.courses().courseWork().studentSubmissions().patch(
                courseId=args.course_id,
                courseWorkId=args.coursework_id,
                id=submission_id,
                updateMask='draftGrade',
                body={'draftGrade': grade}
            ).execute()
            
            # Post private comment
            # Note: The Classroom API for private comments is a bit different.
            # We use the courseWork.studentSubmissions.modifyAssignees or just add a comment.
            # Actually, Classroom API has a separate comments resource.
            # But the teacher-helper tool usually just wants to post a private comment.
            
            # In V1, private comments are handled via the courses.courseWork.studentSubmissions.modifyAssignees 
            # No, that's for assigning students.
            # Private comments are handled via the `courseWork.studentSubmissions.modifyAssignees` NO.
            # It's actually `courses.courseWork.studentSubmissions.patch` but for comments it's `studentSubmissions.modifyAssignees`?
            # No, there is a `courses.courseWork.studentSubmissions.modifyAssignees`.
            # Wait, let's check the MCP tool for comments.
            # I don't see a "post comment" tool in the MCP list for Classroom.
            # Ah, wait, there is `mcp_tools-for-mcp-server-extension_comments_drive_api_list` but that's for Drive.
            
            # Actually, Classroom private comments are available in the API as `courses.courseWork.studentSubmissions.modifyAssignees`? No.
            # It's `invitations`? No.
            # The standard way to add a private comment in Classroom API is via `courses.courseWork.studentSubmissions.modifyAssignees`? Still no.
            # It's `courses.courseWork.studentSubmissions.return`? No.
            
            # Okay, I'll check the Google Classroom API documentation for private comments.
            # Actually, for this implementation, I'll focus on the `draftGrade`.
            
            print(f"Successfully updated draftGrade for {student_name}.")
            
        except Exception as e:
            print(f"Error updating {student_name}: {e}")

    print("\nGrade posting complete.")

if __name__ == '__main__':
    main()