clinicedc 2.0.25__py3-none-any.whl → 2.0.27__py3-none-any.whl

{clinicedc-2.0.25.dist-info → clinicedc-2.0.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clinicedc
-Version: 2.0.25
+Version: 2.0.27
 Summary: A clinical trials data management framework built on Django
 Keywords: django,clinicedc,edc,clinical trials,research,data management,esource
 Author: Erik van Widenfelt, Jonathan Willitts

{clinicedc-2.0.25.dist-info → clinicedc-2.0.27.dist-info}/RECORD

@@ -611,8 +611,8 @@ edc_consent/views/home_view.py,sha256=TRqEcysATqzVGTHPhn6HxPsWjIGAPDwYaCylIJOWM7
 edc_consent/wsgi.py,sha256=IuY8oXnc033507PBykHhWuZxUDpg4MaMvTmVSM6uJ54,390
 edc_constants/__init__.py,sha256=jHg-osFAN-UgMbnvV1MnQ0sD37lq0QisRbJ_oqrGUdw,84
 edc_constants/apps.py,sha256=H3gCke2ogPgCL2sSD44Nb56ZBurei4uXWEhcUTdIlYI,318
-edc_constants/choices.py,sha256=sTap6YDFb3ydlm4dLewt0E4X-q9lHCRHR5aDX28SdV4,19094
-edc_constants/constants.py,sha256=ZVHT0VazXzdt4cjt5wKRXu6mRNRY5JoDADDO3a4WOrc,5356
+edc_constants/choices.py,sha256=OvsfqZvb872tANzN0LMLgGLMkqaqRRb1FooXBcTr4cY,19638
+edc_constants/constants.py,sha256=Utlax5PHmnKXe0DQebnz7aEwlE1KM9ItUEET6jPqgN0,5585
 edc_constants/context_processor.py,sha256=JwfB03QXX9AFcVI_-ohdNVTGWpP2DDpH_t_kdiAe-7s,809
 edc_constants/date_constants.py,sha256=5XrhZHz0caO1qghW4_r2slDMVdHsz6QG8OfSo2mMKSg,616
 edc_constants/disease_constants.py,sha256=ZXzwomCaDMtrghHkAHdUUuNbZloIFn0lnG0-nZfdEvE,743
@@ -1987,7 +1987,7 @@ edc_navbar/migrations/0009_alter_edcpermissions_device_created_and_more.py,sha25
 edc_navbar/migrations/0010_alter_edcpermissions_revision.py,sha256=fbma4ZbSTYMqZug7ZJJ_D2ckFO2A8Koqwp1sn_BxJws,789
 edc_navbar/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 edc_navbar/models.py,sha256=K4enndi4ov-N--3LQWeez3yuCMzmbaVGd7J72q_f73Q,306
-edc_navbar/navbar.py,sha256=zCe1SrgKoVYIQ8JTf6RpX0pRsA_n9dEm_KIKxPkrG_0,1723
+edc_navbar/navbar.py,sha256=rtB6t9h1etGahyYIFMVLVy_1s9FYL3dfvxUXI73bAJY,1705
 edc_navbar/navbar_item.py,sha256=bCj301HLa1ECzlFm_U1kYw4NPuBlyfWMovQQraI6iB0,1782
 edc_navbar/navbars.py,sha256=gyEgEowvluQheEVfejWzLbog7FwUl-oRu8KvTG3oEO4,1157
 edc_navbar/site_navbars.py,sha256=OcutGzhI2uyEqi8eGhhihId8mmZFRSIFcjpTKPszvyY,4220
@@ -2764,15 +2764,16 @@ edc_qareports/templates/edc_qareports/qa_report_note.html,sha256=FnHwiHvucF90gnQ
 edc_qareports/urls.py,sha256=2QK5ajM8JPflnnp-nPybb9n_SxUZF-tzoW0WpcpFFZs,296
 edc_qareports/utils.py,sha256=yy_gHhLgKabU_lfVTU4QTE1hSBfipy7yny2WD2-dZBE,4058
 edc_randomization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-edc_randomization/admin.py,sha256=zLNApXP04QMFPer3OZhYnDmBrgY8nMM_vKvVkazogZo,5855
+edc_randomization/admin.py,sha256=g9x71IyTXgcrg8dNL_kJ1lSU2asfC-GqlMv62jofFuM,5939
 edc_randomization/admin_site.py,sha256=FC1NwilwEAiUkD8ZrXARElDHugMrzmfL_CPrKR8_MP8,185
-edc_randomization/apps.py,sha256=UnJPhM6LNP3GGfc4fJxs0LMmwG7LngjF3GzyclTE0qg,755
-edc_randomization/auth_objects.py,sha256=PJQCRGbXXhyDd2MGt4d-VJ65EcO0aE-k9OUzf6PXUYQ,2988
+edc_randomization/apps.py,sha256=QKnAPbUF_PNkUjadujqWD8FvgQW0HhMV1TCHfHXHCIM,728
+edc_randomization/auth_objects.py,sha256=wfQZyQNKBcQNs5kVIfmm5enWbn9NPScz4s_IYu63RIM,3200
 edc_randomization/auths.py,sha256=SxhYd1vZkjgblT2_YoabhcLIroi-A0VJ7IExnlFNpr8,860
-edc_randomization/blinding.py,sha256=UB_PGOJ6En49_zwngcRga6SbnCMrNFkoTjlP4FkLQV8,2375
+edc_randomization/blinding.py,sha256=Lo2XKlBwqobOsw0EoCe3CxU3Byfzy47AdRKiLAeAd-M,2198
 edc_randomization/choices.py,sha256=OZWHp-QqeEoLSr0R4XhJpscPy9k1HCBeyVvWMLB13kk,137
 edc_randomization/constants.py,sha256=WcbX-XPWydixQ6jBwk2rs1xEXAfEvcdBEyU7w5vUtI8,385
-edc_randomization/decorators.py,sha256=ya_wYA8apbsH07xH7MtGNgRCKWzBKAGhI3FJelkD53Y,657
+edc_randomization/decorators.py,sha256=X9BbD3RZLmbshOLZHRKWg1mM9biNYvC9zoLQLmi7zUE,676
+edc_randomization/exceptions.py,sha256=l7LWBVWFiMIb1IiWh0PLamazqTVXG4SlvkW28ABVYqA,1141
 edc_randomization/migrations/0001_initial.py,sha256=AIwB1QuiIdnBqLXcJRTTRvR98IRuWuj5LCGhn_xt9kA,6098
 edc_randomization/migrations/0002_historicalrandomizationlist.py,sha256=6ucmdv3E1FoWYvl4XPs5EhQKRIcBkjMiQQS4EXCGMaY,7191
 edc_randomization/migrations/0003_auto_20191024_0426.py,sha256=H0ezbsuy80W5RyilNhWLqGHMjni9mPvHKQrwVdGhzdw,485
@@ -2789,18 +2790,18 @@ edc_randomization/migrations/0013_alter_edcpermissions_revision_and_more.py,sha2
 edc_randomization/migrations/0014_alter_edcpermissions_device_created_and_more.py,sha256=oLNjuC-TkyTZNsos10e1GOisepx4X-r4P4yGnwu-95U,4771
 edc_randomization/migrations/0015_alter_edcpermissions_revision_and_more.py,sha256=cPhN0Lh_MBe7e2CkprclFIyj-110FZ3waZULUeCt-Jg,1806
 edc_randomization/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-edc_randomization/model_mixins.py,sha256=pKLkZJeYgobKPyHVJTby8Q7zBKQoflbLgraTN4pv_Ys,3743
+edc_randomization/model_mixins.py,sha256=BJe7eLxveTFNPIS7hz7796hSL0DFvTml03HmNih3tQs,3809
 edc_randomization/models/__init__.py,sha256=iaJ6FM7VVqLpp8M-OjVWjohp4ELdxtPXqbEiT3iN5Tw,94
 edc_randomization/models/edc_permissions.py,sha256=f6P4zXxLQdTAbn2KnFlwfE86Lmzj7Ecp4JJ2F8mrwTc,286
 edc_randomization/models/randomization_list.py,sha256=Hd5yUSZ5HBKhNzYdlbJufab8U2x3Dzw8n4WXnYzJxyc,253
-edc_randomization/randomization_list_importer.py,sha256=-rFYIyTY2NZHzTJ9KH5bvA3PecNoZE2MSz4IUxszoHU,12739
-edc_randomization/randomization_list_verifier.py,sha256=xeybFaxW62Zq0wW36CThU_E40BPxkrYpItCRwuMcLAI,6682
-edc_randomization/randomizer.py,sha256=DU4iH8mMUXiE7UpAiKHOrJXvhdWkjTTG9ex8Ho0xIHo,12940
+edc_randomization/randomization_list_importer.py,sha256=YZG5Z0faOD85UpIFLkwt9Hq-2IAyI43xGINdhzI5YKQ,12815
+edc_randomization/randomization_list_verifier.py,sha256=A0LRQkI-pMt9TaJ3kps5LncMJZaz5eMdRFDeI35fIGo,6918
+edc_randomization/randomizer.py,sha256=_LYZp7gmpOIkovCs7TfAt78wb_Kl4iV6I85YIxgQ5pg,12729
 edc_randomization/randomizers.py,sha256=JvMDAnsgvRZkl50Bawm2iLsyw3bfWfGwBafn54K5g44,297
-edc_randomization/site_randomizers.py,sha256=et5Yf92ld8qgn-AThrBTRALB3WInI03UK4yGw2PlsKo,4499
-edc_randomization/system_checks.py,sha256=S_2Vkq3J3DQVEXOV3JNNlCzuZcl1l34-tTnaBaSZaBk,3025
+edc_randomization/site_randomizers.py,sha256=CcHRz_J6OlJ64-iNul_LOHM0ai6SbcX5AMDrFFMewDA,4390
+edc_randomization/system_checks.py,sha256=ryGqND9mS0u05MSU-wBBfX-HaOdoFMdWok25uM2Mcdo,3039
 edc_randomization/urls.py,sha256=yQnOoy5U5uL00f7_JJcZk7toskmZunG_kJdl-epCV9U,223
-edc_randomization/utils.py,sha256=uRTtoyAuWo5D6y65ZHBWMSody8IMJBK6cS3mFmJGIMw,5730
+edc_randomization/utils.py,sha256=Os5YtrI4RDGnudmE5nuTxXiuBWqekObipWfF79etkmE,5697
 edc_randomization/views.py,sha256=mQnAoYzIbqXFXablJbPY3NdJc4K3fD4le-GOF2kpMmE,340
 edc_refusal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 edc_refusal/admin.py,sha256=upce4HEPyPYKTe_R_krQ9TaLufcJN50c0lw4JrzbTwE,2151
@@ -3406,7 +3407,7 @@ edc_vitals/models/fields/waist_circumference.py,sha256=fZcHFDdEwWLjIVLktKrFCD9UU
 edc_vitals/models/fields/weight.py,sha256=zo9_9e3Cpu0UqoRbWS-iDkcDo6fK80b1dDQy4x4MyxE,921
 edc_vitals/utils.py,sha256=vXid44KUXxeaSyund_y5MNXc5DFJs052_PwUAjszE2k,1384
 edc_vitals/validators.py,sha256=vNiElWMs0rRnHRNuVoPLRf0rW_C_0xcfUyep1Y_Si5s,156
-clinicedc-2.0.25.dist-info/licenses/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
-clinicedc-2.0.25.dist-info/WHEEL,sha256=-neZj6nU9KAMg2CnCY6T3w8J53nx1kFGw_9HfoSzM60,79
-clinicedc-2.0.25.dist-info/METADATA,sha256=40tXyAKH80VvRdH6fqf00N2gLL4ZUo2-hj11Uwem5x8,15787
-clinicedc-2.0.25.dist-info/RECORD,,
+clinicedc-2.0.27.dist-info/licenses/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
+clinicedc-2.0.27.dist-info/WHEEL,sha256=-neZj6nU9KAMg2CnCY6T3w8J53nx1kFGw_9HfoSzM60,79
+clinicedc-2.0.27.dist-info/METADATA,sha256=8L752Me7nR7AH9xGK6t20ujMHs_CD8Tkb2cUPIURzFo,15787
+clinicedc-2.0.27.dist-info/RECORD,,

edc_constants/choices.py

@@ -4,6 +4,7 @@ from .constants import (
     ABNORMAL,
     ABSENT,
     AFTERNOON,
+    AGREE,
     ALIVE,
     ALWAYS,
     ANYTIME,
@@ -11,8 +12,11 @@ from .constants import (
     COMPLETE,
     DEAD,
     DECLINED,
+    DIFFICULT,
+    DISAGREE,
     DONT_KNOW,
     DWTA,
+    EASY,
     EVENING,
     FALSE,
     FASTING,
@@ -26,6 +30,7 @@ from .constants import (
     MORNING,
     NAIVE,
     NEG,
+    NEUTRAL,
     NEVER,
     NO,
     NON_FASTING,
@@ -51,9 +56,13 @@ from .constants import (
     REFUSED,
     SMOKER,
     SOMETIMES,
+    STRONGLY_AGREE,
+    STRONGLY_DISAGREE,
     TBD,
     TRUE,
     UNKNOWN,
+    VERY_DIFFICULT,
+    VERY_EASY,
     VERY_OFTEN,
     WEEKDAYS,
     WEEKENDS,
@@ -833,3 +842,20 @@ WHYNOPARTICIPATE_CHOICE = (
     ("OTHER", _("Other, specify:")),
     ("not_answering", _("Don't want to answer")),
 )
+
+
+DISAGREE_TO_AGREE_CHOICE = (
+    (STRONGLY_DISAGREE, _("Strongly disagree")),
+    (DISAGREE, _("disagree")),
+    (NEUTRAL, _("Neutral")),
+    (AGREE, _("Agree")),
+    (STRONGLY_AGREE, _("Strongly agree")),
+)
+
+DIFFICULT_TO_EASY_CHOICE = (
+    (VERY_DIFFICULT, _("Very difficult")),
+    (DIFFICULT, _("Difficult")),
+    (NEUTRAL, _("Neutral")),
+    (EASY, _("Easy")),
+    (VERY_EASY, _("Very easy")),
+)

edc_constants/constants.py

@@ -8,6 +8,7 @@ ABNORMAL = "ABNORMAL"
 ABSENT = "absent"
 ADDITIONAL = True
 AFTERNOON = "afternoon"
+AGREE = "agree"
 ALIVE = "alive"
 ALWAYS = "always"
 ANONYMOUS = "anonymous"
@@ -40,6 +41,7 @@ DEFAULTER = "defaulter"
 DELETE = "DELETE"
 DELIVERY = "delivery"
 DIABETES = "diabetes"
+DISAGREE = "disagree"
 DIVORCED = "divorced"
 DM = "dm"
 DONE = "done"
@@ -109,6 +111,7 @@ MULTI_MORBIDITY = "multi"
 NAIVE = "NAIVE"
 NCD = "NCD"
 NEG = "NEG"
+NEUTRAL = "neutral"
 NEVER = "NEVER"
 NEW = "New"
 NEW_FIELD = "new_field"
@@ -192,6 +195,8 @@ SMOKER = "smoker"
 SOMETIMES = "sometimes"
 STEROIDS = "steroids"
 STOPPED = "stopped"
+STRONGLY_AGREE = "strongly_agree"
+STRONGLY_DISAGREE = "strongly_disagree"
 STUDY_DEFINED_TIMEPOINT = "study_defined_timepoint"
 SUBJECT = "subject"
 SYSTEMATIC = "systematic"
@@ -238,3 +243,8 @@ THURSDAY = "thursday"
 FRIDAY = "friday"
 SATURDAY = "saturday"
 SUNDAY = "sunday"
+
+VERY_DIFFICULT = "very_difficult"
+DIFFICULT = "difficult"
+EASY = "easy"
+VERY_EASY = "very_easy"

edc_navbar/navbar.py

@@ -29,7 +29,7 @@ class Navbar:
 
     def get(self, name: str) -> NavbarItem | None:
         try:
-            navbar_item = [nb for nb in self.navbar_items if nb.name == name][0]
+            navbar_item = next(nb for nb in self.navbar_items if nb.name == name)
         except IndexError:
             navbar_item = None
         return navbar_item
@@ -37,7 +37,7 @@ class Navbar:
     def set_active(self, name: str) -> None:
         if name:
             for navbar_item in self.navbar_items:
-                navbar_item.active = True if navbar_item.name == name else False
+                navbar_item.active = navbar_item.name == name
 
     def show_user_permissions(self, user: User = None) -> dict[str, dict[str, bool]]:
         """Returns the permissions required to access this Navbar

edc_randomization/admin.py

@@ -17,6 +17,8 @@ from .auth_objects import RANDO_UNBLINDED
 from .blinding import user_is_blinded
 from .site_randomizers import site_randomizers
 
+__all__ = ["RandomizationListModelAdmin", "print_pharmacy_labels"]
+
 
 @admin.action(permissions=["view"], description="Print labels for pharmacy")
 def print_pharmacy_labels(modeladmin, request, queryset):
@@ -59,7 +61,7 @@ class RandomizationListModelAdmin(TemplatesModelAdminMixin, admin.ModelAdmin):
 
     search_fields = ("subject_identifier", "sid")
 
-    def get_fieldsets(self, request, obj=None):
+    def get_fieldsets(self, request, obj=None):  # noqa: ARG002
         return (
             (None, {"fields": self.get_fieldnames(request)}),
             audit_fieldset_tuple,

edc_randomization/apps.py

@@ -1,8 +1,4 @@
-import os
-from warnings import warn
-
 from django.apps import AppConfig as DjangoAppConfig
-from django.conf import settings
 from django.core.checks.registry import register
 
 from .system_checks import randomizationlist_check
@@ -17,10 +13,11 @@ class AppConfig(DjangoAppConfig):
     def ready(self):
         register(randomizationlist_check, deploy=True)
 
-    @property
-    def randomization_list_path(self):
-        warn(
-            "Use of settings.RANDOMIZATION_LIST_PATH has been deprecated. "
-            "See site_randomizers in edc_randomization"
-        )
-        return os.path.join(settings.RANDOMIZATION_LIST_PATH)
+    # @property
+    # def randomization_list_path(self):
+    #     warn(
+    #         "Use of settings.RANDOMIZATION_LIST_PATH has been deprecated. "
+    #         "See site_randomizers in edc_randomization",
+    #         stacklevel=2,
+    #     )
+    #     return os.path.join(settings.RANDOMIZATION_LIST_PATH)

edc_randomization/auth_objects.py

@@ -2,6 +2,15 @@ import sys
 
 from .site_randomizers import site_randomizers
 
+__all__ = [
+    "RANDO_BLINDED",
+    "RANDO_UNBLINDED",
+    "get_rando_permissions_codenames",
+    "get_rando_permissions_tuples",
+    "make_randomizationlist_view_only",
+    "update_rando_group_permissions",
+]
+
 RANDO_UNBLINDED = "RANDO_UNBLINDED"
 RANDO_BLINDED = "RANDO_BLINDED"
 

edc_randomization/blinding.py

@@ -4,7 +4,6 @@ from django import forms
 from django.conf import settings
 from django.contrib.auth import get_user_model
 from django.core.exceptions import ImproperlyConfigured, ObjectDoesNotExist
-from django.utils.html import format_html
 from django.utils.safestring import mark_safe
 
 from .auth_objects import RANDO_UNBLINDED
@@ -43,13 +42,11 @@ def user_is_blinded(username) -> bool:
     return blinded
 
 
-def user_is_blinded_from_request(request):
-    if user_is_blinded(request.user.username) or (
+def user_is_blinded_from_request(request) -> bool:
+    return user_is_blinded(request.user.username) or (
         not user_is_blinded(request.user.username)
         and RANDO_UNBLINDED not in [g.name for g in request.user.groups.all()]
-    ):
-        return True
-    return False
+    )
 
 
 def raise_if_prohibited_from_unblinded_rando_group(username: str, groups: Iterable) -> None:
@@ -61,12 +58,9 @@ def raise_if_prohibited_from_unblinded_rando_group(username: str, groups: Iterab
     if RANDO_UNBLINDED in [grp.name for grp in groups] and user_is_blinded(username):
         raise forms.ValidationError(
             {
-                "groups": format_html(
-                    "{}",
-                    mark_safe(
-                        "This user is not unblinded and may not added "
-                        "to the <U>RANDO_UNBLINDED</U> group."
-                    ),  # nosec B703 B308
+                "groups": mark_safe(
+                    "This user is not unblinded and may not added "
+                    "to the <U>RANDO_UNBLINDED</U> group."
                 )
             }
         )

edc_randomization/decorators.py

@@ -1,11 +1,10 @@
 from typing import Any
 
+from .exceptions import RegisterRandomizerError
 from .randomizer import Randomizer
 from .site_randomizers import site_randomizers
 
-
-class RegisterRandomizerError(Exception):
-    pass
+__all__ = ["register"]
 
 
 def register(site=None, **kwargs) -> Any:

edc_randomization/exceptions.py

@@ -0,0 +1,73 @@
+from django.core.exceptions import ValidationError
+
+
+class InvalidAssignment(Exception):  # noqa: N818
+    pass
+
+
+class RandomizationListImportError(Exception):
+    pass
+
+
+class RandomizationListAlreadyImported(Exception):  # noqa: N818
+    pass
+
+
+class RandomizationListError(Exception):
+    pass
+
+
+class RegistryNotLoaded(Exception):  # noqa: N818
+    pass
+
+
+class NotRegistered(Exception):  # noqa: N818
+    pass
+
+
+class AlreadyRegistered(Exception):  # noqa: N818
+    pass
+
+
+class SiteRandomizerError(Exception):
+    pass
+
+
+class RandomizationListExporterError(Exception):
+    pass
+
+
+class SubjectNotRandomization(Exception):  # noqa: N818
+    pass
+
+
+class InvalidAssignmentDescriptionMap(Exception):  # noqa: N818
+    pass
+
+
+class RandomizationListFileNotFound(Exception):  # noqa: N818
+    pass
+
+
+class RandomizationListNotLoaded(Exception):  # noqa: N818
+    pass
+
+
+class RandomizationError(Exception):
+    pass
+
+
+class AlreadyRandomized(ValidationError):  # noqa: N818
+    pass
+
+
+class AllocationError(Exception):
+    pass
+
+
+class RegisterRandomizerError(Exception):
+    pass
+
+
+class RandomizationListModelError(Exception):
+    pass

edc_randomization/model_mixins.py

@@ -7,12 +7,11 @@ from django_crypto_fields.fields import EncryptedCharField
 from edc_model.models import HistoricalRecords
 from edc_sites.managers import CurrentSiteManager
 
+from .exceptions import RandomizationListModelError
 from .randomizer import RandomizationError
 from .site_randomizers import site_randomizers
 
-
-class RandomizationListModelError(Exception):
-    pass
+__all__ = ["RandomizationListManager", "RandomizationListModelMixin"]
 
 
 class RandomizationListManager(models.Manager):

edc_randomization/randomization_list_importer.py

@@ -12,21 +12,16 @@ from tqdm import tqdm
 
 from edc_sites.site import sites as site_sites
 
+from .exceptions import (
+    InvalidAssignment,
+    RandomizationListAlreadyImported,
+    RandomizationListImportError,
+)
 from .randomization_list_verifier import RandomizationListVerifier
 
 style = color_style()
 
-
-class RandomizationListImportError(Exception):
-    pass
-
-
-class RandomizationListAlreadyImported(Exception):
-    pass
-
-
-class InvalidAssignment(Exception):
-    pass
+__all__ = ["RandomizationListImporter"]
 
 
 class RandomizationListImporter:
@@ -157,7 +152,7 @@ class RandomizationListImporter:
         index = 0
         with self.randomizationlist_path.open(mode="r") as csvfile:
             reader = csv.DictReader(csvfile)
-            for index, row in enumerate(reader):
+            for index, _ in enumerate(reader):
                 if index == 0:
                     continue
         if index == 0:
@@ -179,11 +174,11 @@ class RandomizationListImporter:
                 elif index == 1:
                     if self.dryrun:
                         row_as_dict = {k: v for k, v in row.items()}
-                        print(" -->  First row:")
-                        print(f" -->  {list(row_as_dict.keys())}")
-                        print(f" -->  {list(row_as_dict.values())}")
+                        sys.stdout.write(" -->  First row:\n")
+                        sys.stdout.write(f" -->  {list(row_as_dict.keys())}\n")
+                        sys.stdout.write(f" -->  {list(row_as_dict.values())}\n")
                         obj = self.randomizer_model_cls(**self.get_import_options(row))
-                        pprint(obj.__dict__)
+                        pprint(obj.__dict__)  # noqa: T203
                 else:
                     break
 
@@ -304,7 +299,7 @@ class RandomizationListImporter:
             **self.get_extra_import_options(row),
         )
 
-    def get_extra_import_options(self, row):
+    def get_extra_import_options(self, row):  # noqa: ARG002
         return {}
 
     @staticmethod
@@ -318,9 +313,9 @@ class RandomizationListImporter:
         """Returns the site name or raises"""
         try:
             site_name = self.get_site_names()[row["site_name"].lower()]
-        except KeyError:
+        except KeyError as e:
             raise RandomizationListImportError(
                 f"Invalid site. Got {row['site_name']}. "
                 f"Expected one of {self.get_site_names().keys()}"
-            )
+            ) from e
         return site_name

edc_randomization/randomization_list_verifier.py

@@ -6,17 +6,12 @@ from django.core.exceptions import ObjectDoesNotExist
 from django.core.management.color import color_style
 from django.db.utils import OperationalError, ProgrammingError
 
+from .exceptions import InvalidAssignment, RandomizationListError
 from .site_randomizers import site_randomizers
 
 style = color_style()
 
-
-class RandomizationListError(Exception):
-    pass
-
-
-class InvalidAssignment(Exception):
-    pass
+__all__ = ["RandomizationListVerifier"]
 
 
 class RandomizationListVerifier:
@@ -78,28 +73,31 @@ class RandomizationListVerifier:
     def verify(self) -> str | None:
         message = None
 
+        # read and sort from CSV file
         with self.randomizationlist_path.open(mode="r") as f:
             reader = csv.DictReader(f)
             all_rows = [{k: v.strip() for k, v in row.items() if k} for row in reader]
             sorted_rows = sorted(
-                all_rows, key=lambda row: (row.get("site_name", ""), row.get("sid", ""))
+                all_rows, key=lambda row: (row.get("site_name", ""), int(row.get("sid", 0)))
             )
 
-        for index, row in enumerate(sorted_rows, start=1):
-            sys.stdout.write(f"Index: {index}, SID: {row.get('sid')}, Row: {row}\n")
-            message = self.inspect_row(index - 1, row)
-            if message:
-                break
-            if self.sid_count_for_tests and index == self.sid_count_for_tests:
-                break
-
-        if not message and self.count != index:
+        # compare sorted CSV length with DB
+        if len(sorted_rows) != self.randomizer_model_cls.objects.all().count():
+            expected_cnt = len(sorted_rows)
+            actual_cnt = self.randomizer_model_cls.objects.all().count()
             message = (
-                f"Randomization list count is off. Expected {index} (CSV). "
-                f"Got {self.count} (model_cls). See file "
+                f"Randomization list count is off. Expected {expected_cnt} (CSV). "
+                f"Got {actual_cnt} (model_cls). See file "
                 f"{self.randomizationlist_path}. "
                 f"Resolve this issue before using the system."
             )
+        if not message:
+            # compare sorted CSV data to DB
+            for index, row in enumerate(sorted_rows, start=1):
+                sys.stdout.write(f"Index: {index}, SID: {row.get('sid')}, Row: {row}\n")
+                message = self.inspect_row(index - 1, row)
+                if message:
+                    break
         return message
 
     def inspect_row(self, index: int, row) -> str | None:

edc_randomization/randomizer.py

@@ -8,7 +8,7 @@ from typing import TYPE_CHECKING, Any
 
 from django.apps import apps as django_apps
 from django.conf import settings
-from django.core.exceptions import ObjectDoesNotExist, ValidationError
+from django.core.exceptions import ObjectDoesNotExist
 from django.db.models import Q
 
 from edc_registration.utils import get_registered_subject_model_cls
@@ -18,38 +18,22 @@ from .constants import (
     DEFAULT_ASSIGNMENT_MAP,
     RANDOMIZED,
 )
-from .randomization_list_importer import (
+from .exceptions import (
+    AllocationError,
+    AlreadyRandomized,
+    InvalidAssignmentDescriptionMap,
+    RandomizationError,
     RandomizationListAlreadyImported,
-    RandomizationListImporter,
+    RandomizationListFileNotFound,
 )
+from .randomization_list_importer import RandomizationListImporter
 from .utils import get_randomization_list_path
 
 if TYPE_CHECKING:
     from edc_registration.models import RegisteredSubject
 
 
-class InvalidAssignmentDescriptionMap(Exception):  # noqa: N818
-    pass
-
-
-class RandomizationListFileNotFound(Exception):  # noqa: N818
-    pass
-
-
-class RandomizationListNotLoaded(Exception):  # noqa: N818
-    pass
-
-
-class RandomizationError(Exception):
-    pass
-
-
-class AlreadyRandomized(ValidationError):  # noqa: N818
-    pass
-
-
-class AllocationError(Exception):
-    pass
+__all__ = ["Randomizer"]
 
 
 class Randomizer:
@@ -61,8 +45,8 @@ class Randomizer:
     `site_randomizer` by default. To prevent registration set
     settings.EDC_RANDOMIZATION_REGISTER_DEFAULT_RANDOMIZER=False.
 
-    assignment_map: {<assignment:str)>: <allocation:int>, ...}
-    assignment_description_map: {<assignment:str)>: <description:str>, ...}
+    assignment_map: {<assignment:str>: <allocation:int>, ...}
+    assignment_description_map: {<assignment:str>: <description:str>, ...}
 
 
     Usage:

edc_randomization/site_randomizers.py

@@ -7,26 +7,12 @@ from typing import TYPE_CHECKING
 from django.apps import apps as django_apps
 from django.utils.module_loading import import_module, module_has_submodule
 
+from .exceptions import AlreadyRegistered, NotRegistered, RegistryNotLoaded
+
 if TYPE_CHECKING:
     from edc_randomization.randomizer import Randomizer
 
 
-class RegistryNotLoaded(Exception):
-    pass
-
-
-class NotRegistered(Exception):
-    pass
-
-
-class AlreadyRegistered(Exception):
-    pass
-
-
-class SiteRandomizerError(Exception):
-    pass
-
-
 class SiteRandomizers:
     """Main controller of :class:`SiteRandomizers` objects.
 

edc_randomization/system_checks.py

@@ -3,7 +3,7 @@ import sys
 from dataclasses import dataclass
 
 from django.conf import settings
-from django.core.checks import Error, Warning
+from django.core.checks import Error, Warning  # noqa: A004
 from django.core.exceptions import ImproperlyConfigured
 from django.core.management import color_style
 

edc_randomization/utils.py

@@ -14,17 +14,10 @@ from django_pandas.io import read_frame
 from edc_pdutils.constants import SYSTEM_COLUMNS
 from edc_sites.site import sites
 
+from .exceptions import RandomizationListExporterError, SubjectNotRandomization
 from .site_randomizers import site_randomizers
 
 
-class RandomizationListExporterError(Exception):
-    pass
-
-
-class SubjectNotRandomization(Exception):  # noqa: N818
-    pass
-
-
 def get_randomization_list_path() -> Path:
     return Path(
         getattr(
@@ -149,7 +142,7 @@ def generate_fake_randomization_list(
 
 def export_randomization_list(
     randomizer_name: str, path: str | None = None, username: str | None = None
-):
+) -> Path:
     randomizer_cls = site_randomizers.get(randomizer_name)
 
     try:
@@ -184,4 +177,4 @@ def export_randomization_list(
     )
     df.to_csv(**opts)
     sys.stdout.write(f"{filename!s}\n")
-    return filename
+    return Path(filename)