xync-db 0.2.0__tar.gz → 0.2.1__tar.gz

{xync_db-0.2.0 → xync_db-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xync-db
-Version: 0.2.0
+Version: 0.2.1
 Summary: Xync DB schema: Tortoise ORM models, enums, pydantic DTOs.
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 License: LicenseRef-EULA

{xync_db-0.2.0 → xync_db-0.2.1}/src/xync_db/models/__init__.py

@@ -321,25 +321,37 @@ class Person(Model, TsTrait):
     # pm_agents: BackwardFKRelation[PmAgent]
 
     async def merge(self, new_person: Person):
-        """Вмерджить new_person в self: перенести акторов и креды, удалить new_person."""
-        await new_person.fetch_related("creds", "actors")
-        for cred in new_person.creds:
-            cred.person = self
-            await cred.save(update_fields=["person_id"])
-        for actor in new_person.actors:
-            actor.person = self
-            await actor.save(update_fields=["person_id"])
-        self.name = self.name or new_person.name
-        await self.save(update_fields=["name"])
-        # после переноса у new_person не должно остаться связей  # noqa: RUF003 - intentional Russian
-        await new_person.fetch_related("creds", "actors")
-        if not len(new_person.creds) + len(new_person.actors) and new_person.user_id in (self.user_id, None):
+        """Вмерджить new_person в self (перенести акторов и креды) и удалить new_person — АТОМАРНО,
+        одной транзакцией: сбой посреди переноса не должен оставлять пол-мерджа (актор на одной
+        персоне, кред на другой) и травить окружающую транзакцию запроса.
+
+        Коллизия uniq(person,pmcur,ovr_pm)/(person,pmcur,detail): у self уже есть «тот же» кред
+        (типично XyncPay по той же pmcur, забутстрапленный на другой бирже) — простой re-point
+        person_id упал бы IntegrityError. Молча сливать финансовые креды нельзя, поэтому громко
+        падаем ValueError (аномалию данных разруливает человек); транзакция всё откатывает чисто."""
+        if new_person.user_id not in (self.user_id, None):
+            raise ValueError(f"refuse to merge user-bound person#{new_person.id} into person#{self.id}")
+        async with in_transaction():
+            await new_person.fetch_related("creds", "actors")
+            for cred in new_person.creds:
+                if dup := await Cred.get_or_none(
+                    person=self, pmcur_id=cred.pmcur_id, ovr_pm_id=cred.ovr_pm_id
+                ) or await Cred.get_or_none(person=self, pmcur_id=cred.pmcur_id, detail=cred.detail):
+                    txt = (
+                        f"У старого person#{self.id}:{self.name}:{self.note} уже есть cred#{dup.id}:{dup.detail} "  # noqa: RUF001
+                        f"c pmcur:{dup.pmcur_id}:{dup.ovr_pm_id}, а мы на него пытаем перевесить "  # noqa: RUF001
+                        f"новый cred#{cred.id}:{cred.detail} c pmcur:{cred.pmcur_id}:{cred.ovr_pm_id} "
+                        f"нового person#{new_person.id}:{new_person.name}:{new_person.note}"
+                    )
+                    raise ValueError(txt)
+                cred.person = self
+                await cred.save(update_fields=["person_id"])
+            for actor in new_person.actors:
+                actor.person = self
+                await actor.save(update_fields=["person_id"])
+            self.name = self.name or new_person.name
+            await self.save(update_fields=["name"])
             await new_person.delete()
-        else:
-            raise ValueError(
-                f"Merge to person#{self.id} failed: new person#{new_person.id} still has "
-                f"{len(new_person.creds)} creds, {len(new_person.actors)} actors, user_id={new_person.user_id}"
-            )
 
 
 class User(TgUser, TsTrait):
@@ -902,27 +914,25 @@ class Actor(Model):
             logging.error(txt)
             return 3, txt, user
 
-        merged = False
-        for i, person in enumerate(user_persons):
-            if [a for a in person.actors if a.ex_id]:
-                # персона юзера уже занята актором(ами) — мерджим только если имена совпадают
-                if self.person.name == person.name:
-                    await person.merge(self.person)
-                    merged = True
-            elif i:
-                # у юзера >1 персоны без акторов — аномалия  # noqa: RUF003 - intentional Russian
-                txt = (
-                    f"У {i + 1}-й персоны #{person.id} нет акторов (юзер #{username_id}), но персон без акторов уже > 1"  # noqa: RUF001
-                )
-                logging.error(txt)
-                return 2, txt, user
-            else:
-                # первая персона юзера без акторов (базовая от триггера) — мерджим
-                await person.merge(self.person)
-                merged = True
-
-        if not merged:
-            # все персоны юзера заняты акторами с другими именами — оставляем новую, привязываем к юзеру # noqa: RUF003
+        # Цель мерджа выбираем БЕЗ зависимости от порядка выборки (старый цикл с `elif i` ошибочно
+        # ловил «аномалию», если пустая базовая персона оказывалась не первой):
+        #   1) занятая персона с тем же именем — тот же человек на другой бирже;
+        #   2) иначе базовая пустая персона юзера (её единственный актор — внутренний ex0).
+        # Пустых (без биржевого актора) персон у юзера норм быть не больше одной — базовая от триггера.
+        empties = [p for p in user_persons if not any(a.ex_id for a in p.actors)]
+        if len(empties) > 1:
+            txt = f"У юзера #{username_id} больше одной персоны без биржевых акторов — аномалия"  # noqa: RUF001
+            logging.error(txt)
+            return 2, txt, user
+        target = next(
+            (p for p in user_persons if any(a.ex_id for a in p.actors) and p.name == self.person.name), None
+        ) or (empties[0] if empties else None)
+        if target:
+            await target.merge(self.person)
+        else:
+            # все персоны юзера заняты акторами с другими именами — это другой человек под тем же  # noqa: RUF003
+            # ТГ-аккаунтом: оставляем новую персону актора и привязываем к юзеру. Осиротевшей пустой
+            # базы тут не остаётся — будь она, стала бы target выше и была бы поглощена.
             self.person.user = user
             await self.person.save(update_fields=["user_id"])