Merge pull request #7 from tejasahluwalia/master

Added functionality to send file, audio, video and sticker message from bot

Author: dotX12

examples/send_audio/__init__.py

@@ -0,0 +1,35 @@
+from aiohttp import web
+
+from waio import Bot, Dispatcher
+from waio.logs import loguru_filter
+from waio.types import Message
+
+loguru_filter.set_level('DEBUG')
+
+bot = Bot(
+    apikey='API_KEY',
+    src_name='SRC_NAME',
+    phone_number=0000000000
+)
+
+dp = Dispatcher(bot=bot)
+
+
+@dp.message_handler(commands=['start'])
+async def start_command(message: Message):
+    await message.bot.send_audio(
+        receiver=message.sender_number,
+        url="https://www.buildquickbots.com/whatsapp/media/sample/audio/sample01.mp3"
+    )
+
+
+async def handler_gupshup(request):
+    event = await request.json()
+    await dp.handle_event(event)
+    return web.Response(status=200)
+
+
+if __name__ == "__main__":
+    webhook = web.Application()
+    webhook.add_routes([web.post('/api/v1/gupshup/hook', handler_gupshup)])
+    web.run_app(webhook, port=8008)

examples/send_audio/main.py

@@ -0,0 +1,37 @@
+from aiohttp import web
+
+from waio import Bot, Dispatcher
+from waio.logs import loguru_filter
+from waio.types import Message
+
+loguru_filter.set_level('DEBUG')
+
+bot = Bot(
+    apikey='API_KEY',
+    src_name='SRC_NAME',
+    phone_number=0000000000
+)
+
+dp = Dispatcher(bot=bot)
+
+
+@dp.message_handler(commands=['start'])
+async def start_command(message: Message):
+    await message.bot.send_file(
+        receiver=message.sender_number,
+        url='https://www.buildquickbots.com/whatsapp/media/sample/pdf/sample01.pdf',
+        filename='test_file.pdf',
+        caption='Test Caption'
+    )
+
+
+async def handler_gupshup(request):
+    event = await request.json()
+    await dp.handle_event(event)
+    return web.Response(status=200)
+
+
+if __name__ == "__main__":
+    webhook = web.Application()
+    webhook.add_routes([web.post('/api/v1/gupshup/hook', handler_gupshup)])
+    web.run_app(webhook, port=8008)

examples/send_file/__init__.py

@@ -0,0 +1,35 @@
+from aiohttp import web
+
+from waio import Bot, Dispatcher
+from waio.logs import loguru_filter
+from waio.types import Message
+
+loguru_filter.set_level('DEBUG')
+
+bot = Bot(
+    apikey='API_KEY',
+    src_name='SRC_NAME',
+    phone_number=0000000000
+)
+
+dp = Dispatcher(bot=bot)
+
+
+@dp.message_handler(commands=['start'])
+async def start_command(message: Message):
+    await message.bot.send_sticker(
+        receiver=message.sender_number,
+        url="http://www.buildquickbots.com/whatsapp/stickers/SampleSticker01.webp",
+    )
+
+
+async def handler_gupshup(request):
+    event = await request.json()
+    await dp.handle_event(event)
+    return web.Response(status=200)
+
+
+if __name__ == "__main__":
+    webhook = web.Application()
+    webhook.add_routes([web.post('/api/v1/gupshup/hook', handler_gupshup)])
+    web.run_app(webhook, port=8008)

examples/send_file/main.py

@@ -0,0 +1,36 @@
+from aiohttp import web
+
+from waio import Bot, Dispatcher
+from waio.logs import loguru_filter
+from waio.types import Message
+
+loguru_filter.set_level('DEBUG')
+
+bot = Bot(
+    apikey='API_KEY',
+    src_name='SRC_NAME',
+    phone_number=0000000000
+)
+
+dp = Dispatcher(bot=bot)
+
+
+@dp.message_handler(commands=['start'])
+async def start_command(message: Message):
+    await message.bot.send_video(
+        receiver=message.sender_number,
+        url="https://www.buildquickbots.com/whatsapp/media/sample/video/sample01.mp4",
+        caption="Test video caption"
+    )
+
+
+async def handler_gupshup(request):
+    event = await request.json()
+    await dp.handle_event(event)
+    return web.Response(status=200)
+
+
+if __name__ == "__main__":
+    webhook = web.Application()
+    webhook.add_routes([web.post('/api/v1/gupshup/hook', handler_gupshup)])
+    web.run_app(webhook, port=8008)

examples/send_sticker/__init__.py

@@ -10,14 +10,18 @@
 from waio.keyboard.list import ListMessage
 from waio.keyboard.reply import QuickReply
 from waio.labeler import BotLabeler
+from waio.logs.logger import logger
 from waio.middleware import MiddlewareResponse
+from waio.models.audio import AudioModel
 from waio.models.enums import GupshupMethods
+from waio.models.file import FileModel
 from waio.models.image import ImageModel
+from waio.models.sticker import StickerModel
 from waio.models.text import MessageText
+from waio.models.video import VideoModel
 from waio.states.context import FSMContext
 from waio.storage.redis import RedisStorage
 from waio.types.message import Message
-from waio.logs.logger import logger
 
 
 class Bot(GupshupSettings, HTTPClient):
@@ -48,15 +52,50 @@ async def send_message(self, receiver: int, message: str):
         return await self._base_request(receiver=receiver, data=msg)
 
     async def send_image(
-        self,
-        receiver: int,
-        original_url: str,
-        preview_url: Optional[str] = None,
-        caption: Optional[str] = None,
+            self,
+            receiver: int,
+            original_url: str,
+            preview_url: Optional[str] = None,
+            caption: Optional[str] = None,
     ):
         image = ImageModel(original_url=original_url, preview_url=preview_url, caption=caption)
         return await self._base_request(receiver=receiver, data=image)
 
+    async def send_file(
+            self,
+            receiver: int,
+            url: str,
+            filename: str,
+            caption: Optional[str] = None,
+    ):
+        file = FileModel(url=url, filename=filename, caption=caption)
+        return await self._base_request(receiver=receiver, data=file)
+
+    async def send_video(
+            self,
+            receiver: int,
+            url: str,
+            caption: Optional[str] = None,
+    ):
+        video = VideoModel(url=url, caption=caption)
+        return await self._base_request(receiver=receiver, data=video)
+
+    async def send_audio(
+            self,
+            receiver: int,
+            url: str,
+    ):
+        audio = AudioModel(url=url)
+        return await self._base_request(receiver=receiver, data=audio)
+
+    async def send_sticker(
+            self,
+            receiver: int,
+            url: str,
+    ):
+        sticker = StickerModel(url=url)
+        return await self._base_request(receiver=receiver, data=sticker)
+
     async def send_list(self, receiver: int, keyboard: ListMessage):
         return await self._base_request(receiver=receiver, data=keyboard)
 

examples/send_sticker/main.py

@@ -0,0 +1,19 @@
+import ujson
+
+
+class AudioModel:
+    def __init__(
+            self,
+            url: str,
+    ):
+        self.type = 'audio'
+        self.url = url
+
+    def dict(self):
+        return {
+            "type": self.type,
+            "url": self.url,
+        }
+
+    def json(self):
+        return ujson.dumps(self.dict(), indent=2)

examples/send_video/__init__.py

@@ -0,0 +1,27 @@
+from typing import Optional
+
+import ujson
+
+
+class FileModel:
+    def __init__(
+            self,
+            url: str,
+            filename: str,
+            caption: Optional[str] = None
+    ):
+        self.type = 'file'
+        self.url = url
+        self.filename = filename
+        self.caption = caption
+
+    def dict(self):
+        return {
+            "type": self.type,
+            "url": self.url,
+            "filename": self.filename,
+            "caption": self.caption,
+        }
+
+    def json(self):
+        return ujson.dumps(self.dict(), indent=2)

examples/send_video/main.py

@@ -0,0 +1,19 @@
+import ujson
+
+
+class StickerModel:
+    def __init__(
+            self,
+            url: str,
+    ):
+        self.type = 'sticker'
+        self.url = url
+
+    def dict(self):
+        return {
+            "type": self.type,
+            "url": self.url,
+        }
+
+    def json(self):
+        return ujson.dumps(self.dict(), indent=2)

waio/bot.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import ujson
+
+
+class VideoModel:
+    def __init__(
+            self,
+            url: str,
+            caption: Optional[str] = None
+
+    ):
+        self.type = 'video'
+        self.url = url
+        self.caption = caption
+
+    def dict(self):
+        return {
+            "type": self.type,
+            "url": self.url,
+            "caption": self.caption,
+        }
+
+    def json(self):
+        return ujson.dumps(self.dict(), indent=2)

waio/models/audio.py

@@ -1,5 +1,6 @@
 from typing import Optional
 from typing import Protocol
+
 from waio.keyboard.list import ListMessage
 from waio.keyboard.reply import QuickReply
 
@@ -16,10 +17,41 @@ async def send_reply(self, receiver: int, keyboard: QuickReply):
         """Send reply message"""
 
     async def send_image(
-        self,
-        receiver: int,
-        original_url: str,
-        preview_url: Optional[str] = None,
-        caption: Optional[str] = None,
+            self,
+            receiver: int,
+            original_url: str,
+            preview_url: Optional[str] = None,
+            caption: Optional[str] = None,
     ):
         """Send image message"""
+
+    async def send_file(
+            self,
+            receiver: int,
+            url: str,
+            filename: str,
+            caption: Optional[str] = None,
+    ):
+        """Send file message"""
+
+    async def send_video(
+            self,
+            receiver: int,
+            url: str,
+            caption: Optional[str] = None,
+    ):
+        """Send video message"""
+
+    async def send_audio(
+            self,
+            receiver: int,
+            url: str,
+    ):
+        """Send audio message"""
+
+    async def send_sticker(
+            self,
+            receiver: int,
+            url: str,
+    ):
+        """Send sticker message"""