xync-db 0.5.5.dev23__tar.gz → 0.5.5.dev25__tar.gz

{xync_db-0.5.5.dev23 → xync_db-0.5.5.dev25}/PKG-INFO

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

{xync_db-0.5.5.dev23 → xync_db-0.5.5.dev25}/src/xync_db/models/__init__.py

@@ -1245,9 +1245,24 @@ class Agent(Model, TsTrait):
 
     @classmethod
     async def upsert(
-        cls, exid: int, host: str, auth: dict, exp: int | None, rname: str, nick: str, email: str | None = None
+        cls,
+        exid: int,
+        host: str,
+        auth: dict,
+        exp: int | None,
+        rname: str,
+        nick: str,
+        email: str | None = None,
+        user: User | None = None,
     ) -> tuple[int, Agent, bool]:
-        """Единый писатель агента (за `/public/set-agent`). Bootstrap-персону нового актора резолвим
+        """Единый писатель агента (за `/public/set-agent` и OAuth «Connect Bybit»). `exp=None` — срок
+        web-сессии неизвестен (Htx/Mexc, голые API-ключи из OAuth) → `expire_at` не трогаем.
+
+        `user` — владелец УЖЕ известен (OAuth из приложения залогиненного юзера): актор сразу
+        привязывается к нему `Actor.set_user` (тот же мердж персон, что по ответу админа), без
+        USER_DEF-вопроса «чей это агент?». Актор чужого юзера → PermissionError, агента не трогаем.
+
+        Bootstrap-персону нового актора резолвим
         ПО `note` (`"{ex_id}:{exid}"`, уникальному по актору) — тем же ключом, что клиентский
         `Ex.person_save`, чтобы у Person был ОДИН путь создания (D-035). По НЕуникальному `name` НЕ
         ищем: именно `get_or_none(name=rname)` ронял прод `MultipleObjectsReturned` на тёзках без юзера.
@@ -1265,6 +1280,8 @@ class Agent(Model, TsTrait):
                 note = f"{ex.id}:{exid}"  # ключ bootstrap-персоны актора (как в Ex.person_save)
                 if person := await Person.get_or_none(note=note):
                     pass  # актор уже знаком — берём его bootstrap-персону
+                elif user:  # владелец известен — персона без USER_DEF (админа не зовём), к юзеру ниже set_user'ом
+                    person = await Person.create(name=rname, note=note)
                 else:
                     # gmail-автолинк: замаскированная биржей почта, однозначно (ровно 1 gmail по LIKE,
                     # где `*`→`_`) указывающая на юзера, сразу привязывает актора к его персоне — без
@@ -1286,9 +1303,16 @@ class Agent(Model, TsTrait):
                         person = await Person.create(name=rname, note=note, status=PersonStatus.USER_DEF)
                         err = 1  # ждём привязки к юзеру через def_user
                 actor = await Actor.create(ex=ex, exid=exid, name=nick, person=person)
+            if user:
+                await actor.fetch_related("person")
+                if actor.person.user_id not in (None, user.id):
+                    raise PermissionError(f"Actor#{actor.id} ({ex.name}:{exid}) belongs to user#{actor.person.user_id}")
+                if not actor.person.user_id:
+                    await actor.set_user(user.username_id)
             if ag := await Agent.get_or_none(actor=actor):
                 ag.auth |= auth
-                ag.expire_at = exp
+                if exp is not None:
+                    ag.expire_at = exp
                 await ag.save(update_fields=["auth", "expire_at"])
                 created = False
             else: