提交脚本

Bin

2025-12-16 971a2a12c03b74dd2d7d668b9dbc599f5131bcaf

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
from datetime import timedelta
 
import factory
from core.utils.common import load_func
from django.conf import settings
from django.utils import timezone
from faker import Faker
from tasks.models import Annotation, AnnotationDraft, Prediction, Task, TaskLock
 
 
class TaskFactory(factory.django.DjangoModelFactory):
    data = factory.LazyFunction(
        lambda: {
            'text': Faker().sentence(),
        }
    )
    project = factory.SubFactory(load_func(settings.PROJECT_FACTORY))
 
    class Meta:
        model = Task
 
 
class AnnotationFactory(factory.django.DjangoModelFactory):
    task = factory.SubFactory(TaskFactory)
    project = factory.SelfAttribute('task.project')
    completed_by = factory.SubFactory(load_func(settings.USER_FACTORY))
 
    class Meta:
        model = Annotation
 
    @classmethod
    def create_hypertextlabels(cls, **kwargs):
        return cls.create(
            result=[
                {
                    'value': {
                        'htmllabels': ['Strong negative'],
                        'start': 1,
                        'end': 10,
                        'startOffset': '/text()[1]',
                        'endOffset': '/text()[2]',
                        'text': 'Test example phrase',
                    },
                    'id': 'wMmVN7k_47',
                    'from_name': 'label',
                    'to_name': 'text',
                    'type': 'hypertextlabels',
                }
            ],
            **kwargs,
        )
 
 
class AnnotationDraftFactory(factory.django.DjangoModelFactory):
    task = factory.SubFactory(TaskFactory)
    user = factory.SubFactory(load_func(settings.USER_FACTORY))
    result = [
        {
            'value': {'choices': ['neg']},
            'id': 'wMmVN7k_47',
            'from_name': 'sentiment',
            'to_name': 'text',
            'type': 'choices',
        }
    ]
 
    class Meta:
        model = AnnotationDraft
 
 
class PredictionFactory(factory.django.DjangoModelFactory):
    task = factory.SubFactory(TaskFactory)
    project = factory.SelfAttribute('task.project')
    result = [{}]
 
    class Meta:
        model = Prediction
 
 
class TaskLockFactory(factory.django.DjangoModelFactory):
    task = factory.SubFactory(TaskFactory)
    user = factory.SubFactory(load_func(settings.USER_FACTORY))
    expire_at = factory.LazyFunction(lambda: timezone.now() + timedelta(seconds=10))
 
    class Meta:
        model = TaskLock