import pytest
from label_studio_sdk.client import LabelStudio
from tasks.models import Annotation

from label_studio.tests.sdk.common import LABEL_CONFIG_AND_TASKS

pytestmark = pytest.mark.django_db


def test_project_annotators_sdk(django_live_url, business_client):
    ls = LabelStudio(base_url=django_live_url, api_key=business_client.api_key)

    # Create project via SDK
    proj = ls.projects.create(title='Annotators Project', label_config=LABEL_CONFIG_AND_TASKS['label_config'])

    # Import two tasks
    ls.projects.import_tasks(
        id=proj.id,
        request=[
            {'data': {'my_text': 't1'}},
            {'data': {'my_text': 't2'}},
        ],
    )

    # Get created tasks
    tasks = list(ls.tasks.list(project=proj.id))

    # Create two users via SDK
    u2 = ls.users.create(email='a2@example.com', username='annotator2', first_name='A', last_name='Two')
    u3 = ls.users.create(email='a3@example.com', username='annotator3', first_name='A', last_name='Three')

    # Add annotations directly (SDK doesn't expose annotation create easily with arbitrary user)
    # Use ORM for completed_by set to the two users
    Annotation.objects.create(
        task_id=tasks[0].id, project_id=proj.id, completed_by_id=business_client.user.id, result=[{'r': 1}]
    )
    Annotation.objects.create(task_id=tasks[1].id, project_id=proj.id, completed_by_id=u2.id, result=[{'r': 2}])

    # Call annotators API via SDK wrapper
    resp = list(ls.projects.list_unique_annotators(id=proj.id))

    returned_ids = [u.id for u in resp]
    assert sorted(returned_ids) == sorted([business_client.user.id, u2.id])
    assert returned_ids == sorted(returned_ids)
    assert u3.id not in returned_ids   # no annotations created for this user