canvas 0.15.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

canvas_sdk/v1/data/protocol_override.py

@@ -1,6 +1,13 @@
+from typing import cast
+
 from django.db import models
 
-from canvas_sdk.v1.data.base import CommittableModelManager
+from canvas_sdk.v1.data.base import (
+    BaseModelManager,
+    BaseQuerySet,
+    CommittableQuerySetMixin,
+    ForPatientQuerySetMixin,
+)
 
 
 class IntervalUnit(models.TextChoices):
@@ -18,6 +25,15 @@ class Status(models.TextChoices):
     INACTIVE = "inactive", "inactive"
 
 
+class ProtocolOverrideQuerySet(BaseQuerySet, ForPatientQuerySetMixin, CommittableQuerySetMixin):
+    """ProtocolOverrideQuerySet."""
+
+    pass
+
+
+ProtocolOverrideManager = BaseModelManager.from_queryset(ProtocolOverrideQuerySet)
+
+
 class ProtocolOverride(models.Model):
     """ProtocolOverride."""
 
@@ -25,7 +41,7 @@ class ProtocolOverride(models.Model):
         managed = False
         db_table = "canvas_sdk_data_api_protocoloverride_001"
 
-    objects = CommittableModelManager()
+    objects = cast(ProtocolOverrideQuerySet, ProtocolOverrideManager())
 
     id = models.UUIDField()
     dbid = models.BigIntegerField(primary_key=True)

canvas_sdk/v1/data/questionnaire.py

@@ -1,10 +1,14 @@
 from collections.abc import Container
+from typing import cast
 
 from django.db import models
 from django.db.models import Q
 
 from canvas_sdk.v1.data.base import (
-    CommittableModelManager,
+    BaseModelManager,
+    BaseQuerySet,
+    CommittableQuerySetMixin,
+    ForPatientQuerySetMixin,
     ValueSetLookupByNameQuerySet,
 )
 
@@ -123,6 +127,15 @@ class QuestionnaireQuestionMap(models.Model):
     question = models.ForeignKey(Question, on_delete=models.DO_NOTHING, null=True)
 
 
+class InterviewQuerySet(BaseQuerySet, ForPatientQuerySetMixin, CommittableQuerySetMixin):
+    """InterviewQuerySet."""
+
+    pass
+
+
+InterviewManager = BaseModelManager.from_queryset(InterviewQuerySet)
+
+
 class Interview(models.Model):
     """Interview."""
 
@@ -130,7 +143,7 @@ class Interview(models.Model):
         managed = False
         db_table = "canvas_sdk_data_api_interview_001"
 
-    objects = CommittableModelManager()
+    objects = cast(InterviewQuerySet, InterviewManager())
 
     id = models.UUIDField()
     dbid = models.BigIntegerField(primary_key=True)