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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: xync-db
-Version: 0.5.5.dev23
+Version: 0.5.5.dev26
 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.dev26}/src/xync_db/login_store.py

@@ -88,7 +88,7 @@ async def login_fulfil(token: str, payload: dict[str, Any]) -> bool:
     return True
 
 
-async def login_take(token: str) -> dict[str, Any] | None | object:
+async def login_take(token: str) -> dict[str, Any] | object | None:
     """Poll a token. Returns `MISSING` if it never existed / expired, `None` if it exists but
     Start wasn't tapped yet, or the payload dict once ready (consumed: the token is deleted)."""
     if _redis is not None:

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

@@ -490,6 +490,16 @@ class User(TgUser, TsTrait):
     #     self.pm_clients = {pma.pm_id: pma.client() for pma in self.pm_agents}
     #     self.agent_clients = {a.id: a.client() for a in self.person.actor.agents}
 
+    @property
+    def handle(self) -> str | None:
+        """«@username» из Telegram либо None (ждёт prefetched `username`)."""
+        return self.username.username and "@" + self.username.username
+
+    @property
+    def title(self) -> str:
+        """Отображаемое имя: first_name → @handle → id:N (ждёт prefetched `username`)."""
+        return self.first_name or self.handle or f"id:{self.id}"
+
     @classmethod
     async def mark_membership(
         cls, username_id: int, *, in_channel: bool | None = None, in_chat: bool | None = None
@@ -687,13 +697,7 @@ class User(TgUser, TsTrait):
             )
             c["amount"] += a["free"] / 10**scale
         users = await cls.filter(id__in=[uid for uid in ss if uid]).prefetch_related("username")
-        names = {
-            u.id: (
-                u.first_name or (u.username and u.username.username and "@" + u.username.username) or f"id:{u.id}",
-                u.username and u.username.username and "@" + u.username.username,
-            )
-            for u in users
-        }
+        names = {u.id: (u.title, u.handle) for u in users}
         return cls._assemble_rich(ss, cur_meta, usdt_rate, coins_by_user, names)
 
     async def balance(self, cur_id: int, ts: int | None = None) -> int:
@@ -1245,9 +1249,25 @@ 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 | None,
+        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` не трогаем. У OAuth-ключей
+        и имени нет (`rname=None`, `nick=""`) — их дозаполнит `Ex.person_save`/`actor_save` по данным биржи.
+
+        `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 +1285,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 +1308,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: