import abc import os from urllib.parse import urlencode import requests from cms.models import Article class PromoteError(Exception): pass class Promoter(abc.ABC): def __init__(self, article: Article): self.article = article def promote(self): raise NotImplementedError class TelegramPromoter(Promoter): def promote(self): from crossposting_backend.settings import BOT_TOKEN, CHANNEL_ID long_text = f'{self.article.body}\n{self.article.link}' querystring = (('chat_id', CHANNEL_ID), ('text', long_text)) encoded_querystring = urlencode(querystring) send_message_url = f'https://api.telegram.org/bot{BOT_TOKEN}/sendMessage?{encoded_querystring}' response = requests.get(send_message_url) result = response.json() if not result['ok']: raise PromoteError('Посылка сообщения в телеграм провалилась. Подробности %s' % str(result)) class VkontaktePromoter(Promoter): def promote(self): vk_login = os.getenv('VK_LOGIN') vk_password = os.getenv('VK_PASSWORD') vk_owner_id = os.getenv('VK_OWNER_ID') import vk_api session = vk_api.VkApi(login=vk_login, password=vk_password) session.auth() api = session.get_api() try: api.wall.post(owner_id=vk_owner_id, message=self.article.body, attachments=self.article.link) except vk_api.VkApiError as exc: raise PromoteError(exc) class OdnoklassnikiPromoter(Promoter): def promote(self): from json import JSONEncoder import ok_api ok_access_token = os.getenv('OK_ACCESS_TOKEN') ok_application_key = os.getenv('OK_APPLICATION_KEY') ok_application_secret_key = os.getenv('OK_APPLICATION_SECRET_KEY') session = ok_api.OkApi(access_token=ok_access_token, application_key=ok_application_key, application_secret_key=ok_application_secret_key) attachments = { 'media': [ { 'type': 'text', 'text': self.article.body, }, { 'type': 'link', 'url': self.article.link, }, ] } encoded_attachments = JSONEncoder().encode(attachments) try: session.mediatopic.post(type='GROUP_THEME', gid='70000001426867', attachment=encoded_attachments) except ok_api.OkApiException as exc: raise PromoteError(exc) class Marketer: def __init__(self, article: Article): self.promoters = [ TelegramPromoter(article), VkontaktePromoter(article), OdnoklassnikiPromoter(article), ] def promote(self): for promoter in self.promoters: promoter.promote()