telegram_bots/feeds/modules/tapas.py

74 lines
2.6 KiB
Python

import json
import os
import re
from tempfile import TemporaryDirectory
import sentry_sdk
from django.db import models
from djconfig import config
from telebot import TeleBot
from telebot.types import InputMediaPhoto, InputMediaVideo
from bs4 import BeautifulSoup
import requests
from PIL import Image
from feeds.models import FeedModuleConfig
class TapasFeedModuleConfig(FeedModuleConfig):
display_name = models.CharField(max_length=256)
MODULE_NAME = 'Tapas.io comic'
def execute(self, bot: TeleBot, chat_id, last_id):
config._reload_maybe()
if last_id is None:
last_id = 0
series_page = requests.get(
f'https://tapas.io/series/{self.display_name}',
headers={'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:67.0) Gecko/20100101 Firefox/67.0'},
).text
episode_list = json.loads(re.search(r'episodeList : (\[.*\]),', series_page).group(1))
for episode in episode_list:
eid = episode['id']
title = episode['title']
if eid <= last_id:
continue
if not episode['free']:
continue
caption = f'{title}\nhttps://tapas.io/episode/{eid}'
esoup = BeautifulSoup(requests.get(
f'https://tapas.io/episode/{eid}',
headers={'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:67.0) Gecko/20100101 Firefox/67.0'},
).text, 'html.parser')
imgs = []
for img in esoup.select(f'#episode-{eid} .art-image'):
imgs.append(img['src'])
if len(imgs) == 1:
bot.send_photo(chat_id, imgs[0], caption=caption)
if len(imgs) > 1:
with TemporaryDirectory() as d:
ims = []
for i, img in enumerate(imgs):
img_path = os.path.join(d, f'{i}.jpg')
with open(img_path, 'wb') as f:
f.write(requests.get(img).content)
ims.append(Image.open(img_path))
widths, heights = zip(*(i.size for i in ims))
max_width = max(widths)
total_height = max(heights)
new_im = Image.new('RGB', (max_width, total_height))
offset = 0
for im in ims:
new_im.paste(im, (offset, 0))
offset += im.size[0]
new_im_path = os.path.join(d, 'combined.jpg')
new_im.save(new_im_path)
bot.send_document(chat_id, new_im_path, caption=caption)
yield eid