Updated. job_compassplus/mysite_compassplus_com/get_profile_info_from_web/db.py. Added "prev_name" field in "Person"

Author: gil9red

job_compassplus/mysite_compassplus_com/get_profile_info_from_web/db.py

@@ -117,6 +117,7 @@ class Person(BaseModel):
     create_date = DateField(default=date.today)
     last_check_date = DateField(default=date.today)
     is_active = BooleanField(default=None, null=True)
+    prev_name = TextField(default=None, null=True)
 
     class Meta:
         indexes = (
@@ -131,9 +132,23 @@ def get_last_by_name(cls, name: str) -> Optional["Person"]:
 
     @classmethod
     def get_all(cls, name: str) -> list["Person"]:
-        return list(
-            cls.select().where(cls.name == name).order_by(cls.id.desc())
-        )
+        prev_names: set[str] = set()
+
+        persons: set[Person] = set()
+        for p in cls.select().where(cls.name == name):
+            persons.add(p)
+
+            if p.prev_name:
+                prev_names.add(p.prev_name)
+
+        for prev_name in prev_names:
+            if prev_name in persons:
+                continue
+
+            for p in Person.get_all(prev_name):
+                persons.add(p)
+
+        return sorted(persons, key=lambda p: p.id, reverse=True)
 
     @classmethod
     def get_all_name(cls) -> list[str]:
@@ -161,3 +176,15 @@ def get_all_name(cls) -> list[str]:
     print(f"Total: {len(persons)}")
     print(f"Active: {len([p for p in persons if p.is_active])}")
     print(f"Non active: {len([p for p in persons if not p.is_active])}")
+
+    # Поиск людей с одинаковой картинкой в профиле
+    # from collections import defaultdict
+    # img_by_persons: defaultdict[bytes, set[str]] = defaultdict(set)
+    #
+    # for p in Person:
+    #     img_by_persons[p.img].add(p.name)
+    #
+    # for img, persons in sorted(img_by_persons.items(), key=lambda x: len(x[1]), reverse=True):
+    #     if len(persons) > 1:
+    #         print(len(persons), persons)
+    #

job_compassplus/mysite_compassplus_com/get_profile_info_from_web/migrations/001.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+__author__ = "ipetrash"
+
+
+# SOURCE: http://docs.peewee-orm.com/en/latest/peewee/playhouse.html#schema-migrations
+
+
+from playhouse.migrate import SqliteDatabase, SqliteMigrator, migrate
+from db import TextField, DB_FILE_NAME, Person
+
+
+db = SqliteDatabase(DB_FILE_NAME)
+migrator = SqliteMigrator(db)
+
+
+with db.atomic():
+    migrate(
+        migrator.add_column(
+            Person._meta.table_name,
+            "prev_name",
+            TextField(default=None, null=True),
+        ),
+    )