utg-base 1.6.1__tar.gz → 1.8.0__tar.gz

{utg_base-1.6.1 → utg_base-1.8.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: utg-base
-Version: 1.6.1
+Version: 1.8.0
 Summary: UTG Base Package
-Author: Rovshen
-Author-email: rovshenashirov1619@gmail.com
+Author: Olimboy
+Author-email: shavkatov.olimboy@mail.ru
 Requires-Python: >=3.14,<4.0
 Classifier: Programming Language :: Python :: 3
 Requires-Dist: celery (>=5.5.3,<6.0.0)

{utg_base-1.6.1 → utg_base-1.8.0}/pyproject.toml

@@ -1,8 +1,8 @@
 [tool.poetry]
 name = "utg-base"
-version = "1.6.1"
+version = "1.8.0"
 description = "UTG Base Package"
-authors = ["Rovshen <rovshenashirov1619@gmail.com>"]
+authors = ["Olimboy <shavkatov.olimboy@mail.ru>", "Rovshen <rovshenashirov1619@gmail.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]

{utg_base-1.6.1 → utg_base-1.8.0}/src/utg_base/celery/apps.py

@@ -1,6 +1,6 @@
 from django.apps import AppConfig
 
 
-class QApiConfig(AppConfig):
+class CeleryApiConfig(AppConfig):
     default_auto_field = "django.db.models.BigAutoField"
     name = "utg_base.celery"

utg_base-1.8.0/src/utg_base/migration/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class MigrationApiConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "utg_base.migration"

utg_base-1.8.0/src/utg_base/migration/management/commands/makemigrations.py

@@ -0,0 +1,91 @@
+import os
+
+from django.core.management.commands.makemigrations import Command as BaseMakeMigrations
+from django.core.management.utils import run_formatters
+from django.db.migrations.writer import MigrationWriter
+
+
+class Command(BaseMakeMigrations):
+
+    def write_migration_files(self, changes, update_previous_migration_paths=None):
+        """
+        Take a changes dict and write them out as migration files.
+        """
+        directory_created = {}
+        for app_label, app_migrations in changes.items():
+            if self.verbosity >= 1:
+                self.log(self.style.MIGRATE_HEADING("Migrations for '%s':" % app_label))
+            for migration in app_migrations:
+                # Describe the migration
+                writer = MigrationWriter(migration, self.include_header)
+
+                # BEGIN Customizing
+                # fix 'created_at', 'updated_at' fields order
+                for operation in migration.operations:
+                    _fields = []
+                    _end_fields = []
+                    for field_name, field in operation.fields:
+                        if field_name in ['created_at', 'updated_at']:
+                            _end_fields.append((field_name, field))
+                        else:
+                            _fields.append((field_name, field))
+
+                    operation.fields = _fields + _end_fields
+
+                # END Customizing
+
+                if self.verbosity >= 1:
+                    # Display a relative path if it's below the current working
+                    # directory, or an absolute path otherwise.
+                    migration_string = self.get_relative_path(writer.path)
+                    self.log("  %s\n" % self.style.MIGRATE_LABEL(migration_string))
+                    for operation in migration.operations:
+                        self.log("    %s" % operation.formatted_description())
+                    if self.scriptable:
+                        self.stdout.write(migration_string)
+                if not self.dry_run:
+                    # Write the migrations file to the disk.
+                    migrations_directory = os.path.dirname(writer.path)
+                    if not directory_created.get(app_label):
+                        os.makedirs(migrations_directory, exist_ok=True)
+                        init_path = os.path.join(migrations_directory, "__init__.py")
+                        if not os.path.isfile(init_path):
+                            open(init_path, "w").close()
+                        # We just do this once per app
+                        directory_created[app_label] = True
+                    migration_string = writer.as_string()
+                    with open(writer.path, "w", encoding="utf-8") as fh:
+                        fh.write(migration_string)
+                        self.written_files.append(writer.path)
+                    if update_previous_migration_paths:
+                        prev_path = update_previous_migration_paths[app_label]
+                        rel_prev_path = self.get_relative_path(prev_path)
+                        if writer.needs_manual_porting:
+                            migration_path = self.get_relative_path(writer.path)
+                            self.log(
+                                self.style.WARNING(
+                                    f"Updated migration {migration_path} requires "
+                                    f"manual porting.\n"
+                                    f"Previous migration {rel_prev_path} was kept and "
+                                    f"must be deleted after porting functions manually."
+                                )
+                            )
+                        else:
+                            os.remove(prev_path)
+                            self.log(f"Deleted {rel_prev_path}")
+                elif self.verbosity == 3:
+                    # Alternatively, makemigrations --dry-run --verbosity 3
+                    # will log the migrations rather than saving the file to
+                    # the disk.
+                    self.log(
+                        self.style.MIGRATE_HEADING(
+                            "Full migrations file '%s':" % writer.filename
+                        )
+                    )
+                    self.log(writer.as_string())
+        run_formatters(self.written_files, stderr=self.stderr)
+
+    def handle(self, *args, **options):
+        self.written_files = []
+        result = super().handle(*args, **options)
+        return result

utg_base-1.8.0/src/utg_base/utils/thread.py

@@ -0,0 +1,67 @@
+import multiprocessing
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import Callable, Any, Iterable, Tuple, Dict, Union
+
+
+class ThreadPoolException(Exception):
+    """Exception to wrap the error in each task."""
+
+    def __init__(self, func: str, exception: Exception):
+        self.func = func
+        self.exception = exception
+        super().__init__(f"Exception in {func}: {exception}")
+
+
+def parallel_execute(
+        *tasks: Union[
+            Callable,  # func
+            Tuple[Callable, Iterable],  # (func, args)
+            Tuple[Callable, Iterable, Dict[str, Any]]  # (func, args, kwargs)
+        ],
+        max_workers: int = None
+) -> list[Any]:
+    """
+    Executes multiple functions in parallel.
+    Each can be without arguments or with args/kwargs.
+
+    Example:
+        parallel_execute(func1, func2)
+        parallel_execute((func1, (1, 2)), (func2, (), {"x": 5}))
+
+    The sequence is preserved.
+
+    :return: Result list or ThreadPoolException objects
+    """
+    if not tasks:
+        return []
+
+    cpu_count = multiprocessing.cpu_count()
+    max_workers = max_workers or min(cpu_count * 10, len(tasks))
+    results = [None] * len(tasks)
+
+    with ThreadPoolExecutor(max_workers=max_workers) as executor:
+        futures = {}
+
+        for i, task in enumerate(tasks):
+            # Task unpacking
+            if callable(task):
+                func, args, kwargs = task, (), {}
+            elif isinstance(task, tuple):
+                func = task[0]
+                args = task[1] if len(task) > 1 else ()
+                kwargs = task[2] if len(task) > 2 else {}
+            else:
+                raise ValueError(f"Invalid task format: {task}")
+
+            futures[executor.submit(func, *args, **kwargs)] = i
+
+        for future in as_completed(futures):
+            idx = futures[future]
+            func = tasks[idx][0] if isinstance(tasks[idx], tuple) else tasks[idx]
+            func_name = getattr(func, "__name__", str(func))
+            try:
+                results[idx] = future.result()
+            except Exception as e:
+                results[idx] = ThreadPoolException(func=func_name, exception=e)
+
+    return results