xync-client 0.0.57.dev9__tar.gz → 0.0.57.dev10__tar.gz

{xync_client-0.0.57.dev9/xync_client.egg-info → xync_client-0.0.57.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xync-client
-Version: 0.0.57.dev9
+Version: 0.0.57.dev10
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 Project-URL: Homepage, https://gitlab.com/XyncNet/client
 Project-URL: Repository, https://gitlab.com/XyncNet/client

{xync_client-0.0.57.dev9 → xync_client-0.0.57.dev10}/xync_client/Bybit/agent.py

@@ -11,7 +11,7 @@ import pyotp
 from asyncpg import ConnectionDoesNotExistError
 from bybit_p2p import P2P
 from bybit_p2p._exceptions import FailedRequestError
-from tortoise.expressions import F
+from tortoise.expressions import F, Q
 from urllib3.exceptions import ReadTimeoutError
 from x_model import init_db
 from xync_schema import models
@@ -634,43 +634,46 @@ class AgentClient(BaseAgentClient):  # Bybit client
 
     async def cond_upsert(self, ad: Ad, rname: str = None, dr: Direction = None) -> tuple[models.Ad, bool]:
         sim = None
-        # если точно такого условия еще нет в бд
+        # если точно такое условие уже есть в бд, ниче делать не надо
         if not (cleaned := clean(ad.remark)) or cleaned in {oc[0] for oc in self.all_conds.values()}:
             return None, False
-        # если такая объява уже есть, просто заменяем текст ее условия
-        if ad_db := await models.Ad.get_or_none(exid=ad.id).prefetch_related("cond__ads__maker"):
+        # если эта объява уже есть в бд
+        if ad_db := await models.Ad.get_or_none(exid=ad.id).prefetch_related("cond__ads__maker", "maker"):
             # у измененного условия этой объявы есть другие объявы?
             if rest_ads := set(ad_db.cond.ads) - {ad_db}:
+                # другие объявы этого условия принадлежат другим юзерам
                 if rest_uids := {ra.maker_id for ra in rest_ads} - {ad_db.maker_id}:
-                    # todo: дописать проверку на измененность этого условия в других ads
-                    logging.exception("у измененного cond этой ad есть другие ads где оно не изменилось", rest_uids)
-            if cond := await Cond.get_or_none(raw_txt=cleaned):
-                if not rest_ads:
-                    self.all_conds.pop(ad_db.cond_id)
-                    await ad_db.cond.delete()
-                ad_db.cond_id = cond.id
-                await ad_db.save()
-            else:
-                if rest_ads and rest_uids:
-                    logging.error(rest_uids)
-                # щас переписывается условие во всех объявах даже если они разных юзеров
-                ad_db.cond.last_ver = ad_db.cond.raw_txt
-                ad_db.cond.raw_txt = cleaned
-                uids = {ra.maker.exid for ra in ad_db.cond.ads}
-                self.all_conds[ad_db.cond_id] = cleaned, uids
-                await ad_db.cond.save()
+                    # создадим новое условие и присвоим его только текущей объяве
+                    new_cond = await Cond.create(raw_txt=cleaned)
+                    await self.sim_new(new_cond.id, get_sim(cleaned, ad_db.cond.raw_txt), ad_db.cond_id)
+                    ad_db.cond_id = new_cond.id
+                    self.all_conds[ad_db.cond_id] = cleaned, {ad_db.maker.exid}
+                    await ad_db.save()
+                    return ad_db, True
+            # проверка на всякий что точно нет такого условия
+            if new_cond := await Cond.get_or_none(raw_txt=cleaned):
+                logging.exception("!!!!!!!!!", rest_uids)
+            # если других объяв со старым условием этой обявы нет, либо они все этого же юзера
+            # обновляем условие (в тч во всех ЕГО объявах)
+            ad_db.cond.last_ver = ad_db.cond.raw_txt
+            ad_db.cond.raw_txt = cleaned
+            if sim_db := await CondSim.get_or_none(Q(join_type="OR", cond=ad_db.cond, cond_rel=ad_db.cond)):
+                sim_db.similarity = get_sim(cleaned, ad_db.cond.last_ver)
+                await sim_db.save()
+            self.all_conds[ad_db.cond_id] = cleaned, {ad_db.maker.exid}
+            await ad_db.cond.save()
+
+            await self.sim_upd(ad_db.cond_id, cleaned)
             return ad_db, False
 
         # находим все старые тексты похожие на 90% и более
-        if sim_connds := {
+        if _sims := {
             old_cid: (txt, sim)
             for old_cid, (txt, uids) in self.all_conds.items()
-            if len(cleaned) > 15
-            and ad.userId not in uids
-            and (sim := int((SequenceMatcher(None, cleaned, txt).ratio() - 0.9) * 10_000)) > 0
+            if len(cleaned) > 15 and ad.userId not in uids and (sim := get_sim(cleaned, txt))
         }:
             # если есть, берем самый похожий из них
-            old_cid, (txt, sim) = max(sim_connds.items(), key=lambda x: x[1])
+            old_cid, (txt, sim) = max(_sims.items(), key=lambda x: x[1][1])
             old_ads = await models.Ad.filter(cond_id=old_cid, maker__exid=int(ad.userId)).prefetch_related("cond")
             for old_ad in old_ads:
                 # и у этого чела есть объява с почти таким же текстом
@@ -682,8 +685,8 @@ class AgentClient(BaseAgentClient):  # Bybit client
                 # но это не она, значит у него есть другая объява с похожим, но чуть отличающимся текстом
                 logging.warning(f"ad#{ad.id}-cond#{old_cid} txt updated:\n{txt}\n|\n|\nV\n{cleaned}")
 
-        cond = await Cond.create(raw_txt=cleaned)
-        self.all_conds[cond.id] = cond.raw_txt, {ad.userId}
+        new_cond = await Cond.create(raw_txt=cleaned)
+        self.all_conds[new_cond.id] = new_cond.raw_txt, {ad.userId}
         act_df = {"name": ad.nickName}
         if rname:
             act_df |= {"person": await Person.get_or_create(name=rname)}
@@ -693,7 +696,7 @@ class AgentClient(BaseAgentClient):  # Bybit client
             amount=float(ad.quantity) * float(ad.price),
             min_fiat=ad.minAmount,
             max_fiat=ad.maxAmount,
-            cond=cond,
+            cond=new_cond,
             exid=int(ad.id),
             direction=dr
             or await Direction.get(
@@ -705,14 +708,25 @@ class AgentClient(BaseAgentClient):  # Bybit client
             maker=actor,
         )
 
-        if sim and sim_connds:  # если нашелся похожий текст у другого юзера, добавим связь с % похожести
-            await CondSim.update_or_create({"similarity": sim, "cond_rel_id": old_cid}, cond_id=cond.id)
-            self.cond_sims[cond.id] = old_cid, sim
-            self.sim_conds[old_cid].add(cond.id)
+        # если нашелся похожий текст у другого юзера, добавим связь с % похожести
+        if sim:
+            await self.sim_new(new_cond.id, sim, old_cid)
 
         await ad_db.fetch_related("cond")
         return ad_db, True
 
+    async def sim_new(self, new_cid: int, sim: int, old_cid: int):
+        if not sim:
+            return None
+        return await CondSim.create(cond_id=new_cid, similarity=sim, cond_rel_id=old_cid)
+
+    async def sim_upd(self, cid: int, new_txt: int):
+        for sim_db in (_sims := await CondSim.filter(Q(join_type="OR", cond_id=cid, cond_rel_id=cid))):
+            (op,) = {sim_db.cond_id, sim_db.cond_rel_id} - {cid}
+            op_cond = await Cond[op]
+            sim_db.similarity = get_sim(new_txt, op_cond.raw_txt)
+            await sim_db.save()
+
     #
     # async def cond_upsert(
     #     self, ad: Ad, rname: str = None, dr: Direction = None, cid: int = 0
@@ -806,6 +820,11 @@ class AgentClient(BaseAgentClient):  # Bybit client
             await self.cond_upsert(ad, dr=dr, cid=ad_db.cond_id)
 
 
+def get_sim(s1, s2) -> int:
+    sim = int((SequenceMatcher(None, s1, s2).ratio() - 0.9) * 10_000)
+    return sim if sim > 0 else 0
+
+
 def detailed_diff(str1, str2):
     matcher = SequenceMatcher(None, str1, str2)
     result = []

{xync_client-0.0.57.dev9 → xync_client-0.0.57.dev10/xync_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xync-client
-Version: 0.0.57.dev9
+Version: 0.0.57.dev10
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 Project-URL: Homepage, https://gitlab.com/XyncNet/client
 Project-URL: Repository, https://gitlab.com/XyncNet/client