xync-db 0.4.0__tar.gz → 0.4.2.dev12__tar.gz

{xync_db-0.4.0 → xync_db-0.4.2.dev12}/PKG-INFO

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

xync_db-0.4.2.dev12/src/xync_db/migrations/models/0007_agent_auth_notify.py

@@ -0,0 +1,41 @@
+"""Agent `auth` change → `agents_upd_auth` channel (reactive session reload).
+
+The chrome-extension re-posts an agent's web session (cookies/headers) through `POST /public/set-agent`
+whenever the owner clicks it — but it fires even when nothing changed (the partner just re-opens the
+ext). So instead of reacting in the endpoint on every call, we react in the DB on a real change: the
+existing `agent_upd()` AFTER-UPDATE trigger gains a second, independent branch that emits
+`agents_upd_auth` ONLY when `auth` actually changed (`OLD.auth IS DISTINCT FROM NEW.auth`).
+
+service-api's pg-listener (`api.handlers.agents.agents_auth_upd`) then patches the cookies/headers into
+the LIVE http-session of the running agent client (`BaseAgentClient.reload_auth`) — so subsequent
+requests use the fresh session without a service restart — and notifies the agent's owner + the admin.
+
+Independent of the `agents_upd_status` branch (status and auth can change together or apart). Payload is
+the full `row_to_json(NEW)` (handler hydrates `Agent(**data)` for actor_id + auth), mirroring
+`actor_person_needs_def_user`.
+"""
+
+from typing import ClassVar
+
+from tortoise.migrations.migration import Migration as BaseMigration
+from tortoise.migrations.operations import RunSQL
+
+AGENT_AUTH_SQL = """
+CREATE OR REPLACE FUNCTION agent_upd() returns trigger as $agent_upd_trg$
+BEGIN
+    IF OLD.status != NEW.status THEN
+        PERFORM pg_notify('agents_upd_status', jsonb_build_array(row_to_json(NEW), OLD.status)::varchar);
+    END IF;
+    IF OLD.auth IS DISTINCT FROM NEW.auth THEN
+        PERFORM pg_notify('agents_upd_auth', row_to_json(NEW)::varchar);
+    END IF;
+    RETURN NULL;
+END
+$agent_upd_trg$ LANGUAGE plpgsql;
+CREATE OR REPLACE TRIGGER agent_upd AFTER UPDATE ON "agent" FOR EACH ROW EXECUTE FUNCTION agent_upd();
+"""
+
+
+class Migration(BaseMigration):
+    dependencies: ClassVar = [("models", "0006_credex_orphan")]
+    operations: ClassVar = [RunSQL(AGENT_AUTH_SQL)]

{xync_db-0.4.0 → xync_db-0.4.2.dev12}/src/xync_db/models/__init__.py

@@ -15,6 +15,7 @@ from cryptography.exceptions import InvalidSignature
 from cryptography.hazmat.primitives.asymmetric.ed25519 import Ed25519PrivateKey, Ed25519PublicKey
 from tortoise import BackwardFKRelation, BackwardOneToOneRelation, BaseDBAsyncClient, connections
 from tortoise import Model as BaseModel
+from tortoise.expressions import Q
 from tortoise.fields import (
     CASCADE,
     BigIntField,
@@ -685,9 +686,10 @@ class User(TgUser, TsTrait):
             # тейкеру прячем незашаренные объявы share-hex-бирж (hex IS NULL AND ex∈SHARE_HEX_EX_IDS): без
             # share-hex url-кнопку не построить. Сужено до этих бирж, чтобы не зацепить hot-биржи без share-hex.
             # Свои (mine, страница settings) оставляем — их видно для шеринга.
-            adq = adq.exclude(hex__isnull=True, ad__maker__ex_id__in=SHARE_HEX_EX_IDS)
+            # один Q: исключаем строго пересечение (Tortoise's exclude(**kwargs) отрицал бы условия порознь)
+            adq = adq.exclude(Q(hex__isnull=True, ad__maker__ex_id__in=SHARE_HEX_EX_IDS))
         bads = await adq.filter(ad__pair_side__is_sell=False)
-        # остаток каждой монеты (asset.free) у акторов персон этого юзера, по бирже и монете  # noqa: RUF003
+        # остаток каждой монеты (asset.free) у акторов персон этого юзера, по бирже и монете # noqa: RUF003
         coin_balances: dict[int, dict[int, int]] = {}
         for ex_id, coin_id, total in (
             await Asset.filter(agent__actor__person__user_id=self.id)
@@ -1016,7 +1018,7 @@ class Agent(Model, TsTrait):
 
     @classmethod
     async def upsert(
-        cls, exid: int, host: str, auth: dict, exp: int, rname: str, nick: str, email: str
+        cls, exid: int, host: str, auth: dict, exp: int | None, rname: str, nick: str, email: str = ""
     ) -> tuple[int, Agent, bool]:
         err = 0
         if host and (ex := await Ex.get_or_none(host=host)):
@@ -1317,18 +1319,18 @@ class MyAd(Model):  # Road
         )
 
     @classmethod
-    async def unshared_hot_ad_ids(cls, hex_ex_ids: list[int] = SHARE_HEX_EX_IDS) -> list[int]:
-        """ad_id активных hot-объяв share-hex-бирж без MyAd.hex, чей мейкер-агент ОНЛАЙН (listen-бит,
-        `status & 2` → статусы listen|race_listen). Их надо было зашарить (агент слушает), но hex нет —
-        для старт-алерта. Объява без hex у НЕактивного агента (off/race, бит 2 снят) — норма (он не шарит),
-        не включаем. Гонка (бит 1) для шеринга не нужна, поэтому условие `& 2`, а не `>=3` (hot_mads_query)."""
+    async def unshared_ad_ids(cls, agent_id: int, hex_ex_ids: list[int] = SHARE_HEX_EX_IDS) -> list[int]:
+        """ad_id активных hot-объяв ДАННОГО агента-мейкера на share-hex-бирже без MyAd.hex (не зашарены →
+        скрыты от тейкеров get_hot_ads'ом). Для адресного уведомления ВЛАДЕЛЬЦА (не админа), когда его
+        торговля на бирже выключена и он отказался её включать (handlers.agents.share_or_ask).
+        Раньше тут был глобальный старт-алибат `unshared_hot_ad_ids` всем — его слал админу бот на старте."""
         return (
             await cls.filter(
                 hex__isnull=True,
                 ad__status=AdStatus.active,
                 ad__maker__ex_id__in=hex_ex_ids,
+                ad__maker__agent__id=agent_id,
                 credexs__cred__ovr_pm_id=0,
-                ad__maker__agent__status__in=[AgentStatus.listen, AgentStatus.race_listen],  # listen-бит (& 2)
             )
             .distinct()
             .values_list("ad_id", flat=True)
@@ -1952,7 +1954,7 @@ class Order(Model, TsTrait):
 
             # партнерские отчисления:
             if self.ad.maker.person.user.status == UserStatus.HOT_PARTNER:
-                hspread = abs(self.ad.price - self.ad.pair_side.pair.cur.rate)
+                hspread = abs(self.ad.price - self.ad.pair_side.pair.rate)
                 earn = hspread * self.quantity * 0.0001
                 fee = int(earn * (0.7 - self.ad.maker.person.user.bonus * 0.0001))  # 1% = 100
                 b = await self.ad.maker.person.user.balance(self.ad.pair_side.pair.cur_id)