Gyron 0.1.3__py3-none-any.whl

Gyron/Filters.py

@@ -0,0 +1,301 @@
+import re
+from typing import Pattern, Union, Callable, Optional, Dict
+from .models import Message
+from time import time
+class Filters:
+    __slots__ = ()   
+
+    @staticmethod
+    def text(pattern = None) -> Callable:
+        
+      
+        if pattern is None:
+         
+            return lambda message: message.text is not None
+
+        elif isinstance(pattern, str):
+            compiled = re.compile(pattern)
+        else:
+            compiled = pattern
+
+
+        return lambda message: bool(message.text and compiled.search(message.text))
+    
+    @staticmethod
+    def caption(pattern = None):
+
+        if pattern is None:
+
+            return lambda message: message.caption is not None
+        
+        elif isinstance(pattern, str):
+
+            compiled = re.compile(pattern)
+        
+        else:
+            compiled = pattern
+
+        return lambda message: bool(message.caption and compiled.search(message.caption))
+        
+    
+    @staticmethod
+    def private() -> Callable :
+
+
+        return lambda message: message.chat.type == 'private'
+    
+    @staticmethod
+    def group():
+
+
+        return lambda message: message.chat.type == 'group'
+    
+    @staticmethod
+    def channel():
+
+        return lambda message: message.chat.type == 'channel'
+
+
+
+    @staticmethod
+    def new_member():
+
+        return lambda message: message.new_chat_member.id is not None
+    
+    @staticmethod
+    def left_member():
+
+        return lambda message: message.left_chat_member.id is not None
+
+
+    @staticmethod
+    def is_referral():
+
+        def _filter(message):
+
+
+
+            result = message.text.split(maxsplit = 1)
+
+            if len(result) == 2 and result[0] == "/start" and result[1].isdigit() and len(result[1]) < 15:
+                return True
+            return False
+        return _filter
+    
+    @staticmethod
+    def is_joined(chat_id : Union[str, int]) -> Callable:
+
+
+        async def _filter( message: Message):
+
+            bot = message._bot
+
+            response = await bot.get_chat_member(chat_id, user_id = message.author.id)
+            result = response.get('result', {}).get('status')
+            return result in ('member', 'creator', 'administrator')
+
+
+        return _filter
+    
+    @staticmethod
+    def not_is_joined(chat_id : Union[str, int]) -> Callable:
+
+
+        async def _filter( message: Message):
+
+            bot = message._bot
+
+            response = await bot.get_chat_member(chat_id, user_id = message.author.id)
+            result = response.get('result', {}).get('status')
+            return result not in ('member', 'creator', 'administrator')
+
+
+        return _filter
+    
+    @staticmethod
+    def pre_checkout_query():
+
+        def _filter(message:Message):
+
+            if message.pre_checkout_query.id:
+
+                return True
+            return False
+        return _filter
+    
+    @staticmethod
+    def at_state(manager, state : str = None):
+
+
+        async def _filter(message: Message):
+
+
+            if message.author.id :
+
+                current_state = await manager.get_state_for(message)
+
+                if state is None:
+                    return current_state is not None
+                
+                return current_state == state
+            
+        return _filter
+    
+    @staticmethod
+    def callbhack_query(button_data : str = None):
+    
+
+
+        def _filter(message: Message):
+
+
+            if button_data is None:
+
+                return message.callback_query.data is not None
+            
+            return message.callback_query.data == button_data
+        
+        return _filter
+    
+    @staticmethod
+    def contact(phone_number : str = None):
+
+
+        def _filter(message:Message):
+
+            if phone_number is None:
+
+                return message.contact.phone_number is not None
+            return message.contact.phone_number == phone_number
+        
+        return _filter
+    
+    @staticmethod
+    def location():
+
+
+        def _filter(message:Message):
+
+            return message.location.latitude is not None
+        return _filter
+    
+
+    @staticmethod
+    def forward():
+
+
+        def _filter(message:Message):
+
+            return message.forward_date is not None
+        return _filter
+    
+    @staticmethod
+    def document():
+
+
+        def _filter(message:Message):
+
+            return message.document.id is not None
+        return _filter
+    
+    
+    
+    @staticmethod
+    def photo():
+
+        def _filter(message:Message):
+
+
+            return message.photo.id is not None
+        return _filter
+        
+    @staticmethod
+    def reply():
+
+
+        def _filter(message:Message):
+
+            return message.reply.message_id is not None
+        return _filter
+    
+
+    @staticmethod
+    def voice():
+
+
+        def _filter(message:Message):
+
+            return message.voice.id is not None
+        return _filter
+    
+
+
+class StateManager:
+
+    __slots__ = (
+        'auto_clear',
+        '_states'
+    )
+
+    def __init__(self, auto_clear: bool = True):
+
+      
+        self.auto_clear = auto_clear
+        self._states = {}
+       
+        
+
+    async def set_state_for(self, update, state, expire : float = 10):
+
+        expires_at = None
+
+        if expire is not None:
+            expires_at = time() + expire
+
+        self._states[update.author.id] = {
+            'state': state,
+            'expires_at': expires_at
+        }
+
+    async def clear_state_for(self, update):
+
+        self._states.pop(update.author.id, None)
+
+    async def get_state_for(self, update):
+
+        data = self._states.get(update.author.id)
+
+        if not data:
+            return None
+
+        expires_at = data['expires_at']
+
+        if expires_at is not None and time() > expires_at:
+
+            if self.auto_clear:
+                self._states.pop(update.author.id, None)
+
+            return None
+
+        return data['state']
+
+    async def __call__(self, update):
+
+        data = self._states.get(update.author.id)
+
+        if not data:
+            return False
+
+        expires_at = data['expires_at']
+
+        if expires_at is not None and time() > expires_at:
+
+            if self.auto_clear:
+                self._states.pop(update.author.id, None)
+
+            return False
+
+        return True
+
+
+

Gyron/__init__.py

@@ -0,0 +1 @@
+