from abc import ABC, abstractmethod
from enum import IntEnum
from telegram import Update
from telegram.ext import CallbackContext

class State(ABC):
    @abstractmethod
    async def handle(self, update: Update, context: CallbackContext) -> int:
        pass

class BotStates(IntEnum):
    CHOOSE_CHANNEL = 0
    CHOOSE_TYPE = 1
    ENTER_TEXT = 2
    ENTER_MEDIA = 3
    EDIT_KEYBOARD = 4
    CONFIRM_SEND = 5
    ENTER_SCHEDULE = 6