edu-rdm-integration 3.9.2__py3-none-any.whl → 3.10.1__py3-none-any.whl

edu_rdm_integration/{utils.py → core/utils.py}

@@ -1,129 +1,75 @@
 import os
 from datetime import (
-    date,
     datetime,
     time,
     timedelta,
 )
 from typing import (
     TYPE_CHECKING,
-    Any,
+    Iterable,
     Optional,
     Union,
 )
 
+from _operator import (
+    attrgetter,
+)
 from django.conf import (
     settings,
 )
 from django.db.models import (
-    DateField,
     DateTimeField,
     Expression,
     F,
-)
-from django.db.models.expressions import (
     Func,
 )
 from django.db.models.functions import (
-    Cast,
     Now,
 )
 
-from edu_rdm_integration.apps import (
-    EduRDMIntegrationConfig,
+from m3_db_utils.mixins import (
+    BaseEnumRegisterMixin,
 )
-from edu_rdm_integration.consts import (
+
+from edu_rdm_integration.core.consts import (
     ACADEMIC_YEAR,
 )
 
 
 if TYPE_CHECKING:
-    from typing import (
-        NamedTuple,
-    )
-
     from django.db.models import (
         Model,
     )
 
-    from edu_rdm_integration.collect_and_export_data.models import (
-        EduRdmCollectDataCommandProgress,
-        EduRdmExportDataCommandProgress,
-    )
-    from edu_rdm_integration.helpers import (
+    from edu_rdm_integration.core.helpers import (
         Graph,
     )
-    from edu_rdm_integration.models import (
-        BaseEntityModel,
+    from edu_rdm_integration.stages.collect_data.models import (
+        EduRdmCollectDataCommandProgress,
     )
-
-
-def get_exporting_data_stage_attachment_path(instance, filename):
-    """Возвращает путь до файла-вложения в этап выгрузки данных сущности.
-
-    Args:
-        instance: объект ExportingDataStage
-        filename: имя загружаемого файла
-
-    Returns:
-        Строковое представление пути
-    """
-    datetime_now = datetime.now()
-
-    return os.path.join(
-        settings.UPLOADS,
-        EduRDMIntegrationConfig.label,
-        datetime_now.strftime('%Y/%m/%d'),
-        instance.exporting_data_sub_stage.__class__.__name__.lower(),
-        str(instance.exporting_data_sub_stage_id),
-        str(instance.operation),
-        filename,
+    from edu_rdm_integration.stages.export_data.models import (
+        EduRdmExportDataCommandProgress,
     )
 
 
-def update_fields(
-    entity: 'BaseEntityModel',
-    field_values: dict[str, Any],
-    mapping: dict[str, str]
-) -> None:
-    """Обновление значений полей сущности по измененным полям модели.
-
-    :param entity: Выгружаемая сущность
-    :param field_values: Словарь с измененными данными модели
-    :param mapping: Словарь маппинга полей модели к полям сущности
-    """
-    for model_field, entity_field in mapping.items():
-        if model_field in field_values:
-            setattr(entity, entity_field, field_values[model_field])
-
-
-def set_entity_field_by_model_object(
-    entity: 'BaseEntityModel',
-    model_object: Union['Model', 'NamedTuple'],
-    mapping: dict[str, str]
-) -> None:
-    """Обновление значений полей сущности по измененным полям модели.
+def register_classes(classes: Iterable[type[BaseEnumRegisterMixin]]) -> None:
+    """Вызывает метод регистрации в модель-перечисление у переданных классов.
 
     Args:
-        entity: Выгружаемая сущность;
-        model_object: Объект модели с измененными полями;
-        mapping: Словарь маппинга полей модели к полям сущности
+        classes: Классы, поддерживающие интерфейс регистрации.
     """
-    for model_field, entity_field in mapping.items():
-        if hasattr(model_object, model_field):
-            setattr(entity, entity_field, getattr(model_object, model_field))
+    classes = sorted(classes, key=attrgetter('order_number'))
 
-
-def get_isoformat_timezone():
-    """Возвращает временную зонну в ISO представлении."""
-    return datetime.now().astimezone().isoformat()[-6:]
+    for enum_class in classes:
+        enum_class.register()
 
 
 def split_by_academic_years(start_date: datetime, end_date: datetime) -> list[tuple]:
     """Разбивает исходный интервал из команды по учебным годам."""
     academic_year_end_date = datetime(
-        start_date.year, ACADEMIC_YEAR['end_month'], ACADEMIC_YEAR['end_day'],
-
+        start_date.year,
+        ACADEMIC_YEAR['end_month'],
+        ACADEMIC_YEAR['end_day'],
     )
     if academic_year_end_date <= start_date:
         academic_year_end_date = datetime(
@@ -193,8 +139,8 @@ def split_interval_by_delta(start_date: datetime, end_date: datetime, days_delta
 
 def camel_to_underscore(name: str, upper=False) -> str:
     """Переводит строку из верблюжьего в змеиный регистр.
-    По умолчанию строка приводится к нижнему регистру.
-    Несколько заглавных букв не будут разъединены, кроме последней.
+
+    По умолчанию строка приводится к нижнему регистру. Несколько заглавных букв не будут разъединены, кроме последней.
 
     Args:
         name: форматируемая строка;
@@ -221,7 +167,7 @@ def camel_to_underscore(name: str, upper=False) -> str:
             if c.isupper() and not pc.isupper() and pc != '_':
                 output.append('_')
             elif i > 3 and not c.isupper():
-                previous = name[i - 3:i]
+                previous = name[i - 3 : i]
                 if previous.isalpha() and previous.isupper():
                     output.insert(len(output) - 1, '_')
 
@@ -275,8 +221,7 @@ def build_related_model_graph(
 
 
 def get_data_command_progress_attachment_path(
-    instance: Union['EduRdmCollectDataCommandProgress', 'EduRdmExportDataCommandProgress'],
-    filename: str
+    instance: Union['EduRdmCollectDataCommandProgress', 'EduRdmExportDataCommandProgress'], filename: str
 ) -> str:
     """Возвращает путь до файла-вложения при формировании команды сборки или выгрузки данных.
 
@@ -294,7 +239,7 @@ def get_data_command_progress_attachment_path(
         date_now.strftime('%Y/%m/%d'),
         instance.__class__.__name__.lower(),
         str(instance.type),
-        filename
+        filename,
     )
 
 
@@ -304,7 +249,8 @@ class MakeInterval(Func):
     template = 'make_interval(%(expressions)s)'
 
     def __init__(
-        self, *,
+        self,
+        *,
         years: Union[int, F] = 0,
         months: Union[int, F] = 0,
         weeks: Union[int, F] = 0,
@@ -344,11 +290,3 @@ def make_passed_datetime_from_today(
     now_expr = Now()
 
     return now_expr - MakeInterval(years=years, months=months, days=days)
-
-
-TODAY_EXPR: Union[date, Expression, F] = Cast(make_passed_datetime_from_today(), output_field=DateField())
-
-THREE_YEARS_AGO_DATE_EXPR: Union[date, Expression, F] = Cast(
-    make_passed_datetime_from_today(years=3),
-    output_field=DateField()
-)