Commited backup

AlpacaGang/ftg-modules/sirius.py

@@ -0,0 +1,81 @@
+# requires: pymongo dnspython
+
+# Забейте
+import asyncio
+import pymongo
+
+import logging
+logger = logging.getLogger(__name__)
+from .. import loader, utils
+
+class Student:
+    def __init__(self, id: int, last_name: str, first_name: str, 
+                    patronymic: str, grade: int, region: str, academ: bool, approved: int=None):
+        self.last_name = last_name
+        self.first_name = first_name
+        self.patronymic = patronymic
+        self.grade = int(grade)
+        self.region = region
+        self.academ = bool(academ)
+        self.approved = approved
+        self.id = int(id)
+    
+    def __str__(self):
+        p = 'Академ' if self.academ else 'Отбор'
+        # a = f'Уже вроде бы добавлен в чат (tg://user?id={self.approved})' if self.approved else 'Еще нет в чате'
+        return f'[{p}.{self.id}] {self.last_name} {self.first_name} {self.patronymic}, '\
+        f'{self.grade} класс, из {self.region}'
+
+class SiriusMod(loader.Module):
+    """Ищем поступивших на ИЮ2020"""
+    strings = {"name": "Sirius"}
+    def __init__(self):
+        self.config = loader.ModuleConfig(
+            # name - default - description
+            "db_uri", None, "Database URI, if you dont know where to take it - nevermind",
+            "db_db", None, "database",
+            "db_coll", None, "collection",
+            "replace_ё", True, "replace ё with е in requests, incorrect usage may return incorrect result"
+        )
+        self.name = self.strings['name']
+        self.db = None
+    
+    def config_complete(self):
+        self.db = pymongo.MongoClient(self.config['db_uri'])\
+            .get_database(self.config['db_db']).get_collection(self.config['db_coll'])
+
+    async def findcmd(self, message):
+        arg = utils.get_args_raw(message).strip()
+        if self.config['replace_ё']:
+            arg = arg.replace('ё', 'е')
+            arg = arg.replace('Ё', 'Е')
+        logger.debug('Got: %s', arg)
+        if not arg:
+            await utils.answer(message, 'Только 1 аргумент - номер в списке или фамилия/имя')
+        if arg.isdigit():
+            add = f'людей с номером {arg}'
+            arg = int(arg)
+            users = list(self.db.find({"id": arg}))
+        elif ' ' in arg or arg.lower() == 'янао': # Костыли костыли
+            add = f'людей из региона {arg}'
+            _users = list(self.db.find())
+            users = []
+            for user in _users:
+                if user['region'].lower() == arg.lower():
+                    users.append(user)
+        else:
+            add = f'людей, которых зовут {arg}'
+            arg = arg.capitalize()
+            users = list(self.db.find({'$or': [{"last_name": arg}, {"first_name": arg}, {"patronymic": arg}]}))
+
+        msg = [f'{len(users)} всего {add}', '==']
+        for user in users:
+            del user['_id']
+            s = Student(**user)
+            msg += [str(s), '==']
+        msg = msg[:-1]
+        msg = '\n'.join(msg)
+        await utils.answer(message, msg)
+
+
+