page.click('button[data-testid="modal-close-button"]')
У нас получилось авторизоваться и закрыть рекламную модалку. Таким образом мы и будем взаимодействовать с Affine. Этим и прекрасен Playwright: его API удобный и понятный, и его можно использовать даже без глубоких знаний по теме.
Полный код взаимодействия с Affine
Подобным образом я нашел все необходимые элементы и в конечном счете заполнил текстовое поле для строки в заметке:
def create_affine_note(text: str):
with sync_playwright() as p:
browser = p.chromium.launch(headless=False)
page = browser.new_page()
page.goto('адрес воркспейса в Affine')
page.fill('input[placeholder="Enter your email address"]', 'ваш email')
page.click('[data-testid="continue-login-button"]')
page.fill('input[data-testid="password-input"]', 'ваш пароль')
page.click('button[data-testid="sign-in-button"]')
page.click('button[data-testid="modal-close-button"]')
page.click('div[data-testid="slider-bar-journals-button"]')
page.click('div[data-testid="page-editor-blank"]')
locator = page.locator('div[data-testid="page-editor-blank"]')
locator.focus()
locator.press_sequentially(text)
locator.press('Enter')
browser.close()
Телеграм-бот
Написать телеграм-бота на Python с одной командой добавления заметки — это довольно тривиальная задача. На просторах интернета есть очень много статей на тему «hello world в телеграм-боте».
Своего бота можно создать через телеграм-бота @BotFather. Нам понадобится только токен, который мы укажем в конфигурации.
Мы воспользуемся библиотекой pyTelegramBotAPI:
pip install pyTelegramBotAPI
Определим требования к телеграм-боту:
- Какая-то аутентификация: мы не хотим, чтобы Affine заддосили люди, случайно нашедшие этого бота. Будем проверять свой userId из Телеграма.
- При отправке сообщения с каким-то текстом заходим в ежедневную заметку Affine и добавляем к ней указанный текст.
Конфигурация
Добавим в наш проект файл ENV и заполним его следующим образом:
AFFINE_WORKSPACE_URL=''
AFFINE_USER_EMAIL=''
AFFINE_USER_PASSWORD=''
IS_DEBUG='True'
TELEGRAM_API_KEY=''
USER_ID=''
Эти параметры позволят нам более гибко управлять нашим ботом. А чтобы загрузить данные из ENV-файла в приложение, воспользуемся библиотекой python-dotenv:
pip install python-dotenv
Теперь получим параметры из переменных окружения — добавим файл confg.py со следующим содержимым:
import os
TELEGRAM_API_KEY = os.getenv('TELEGRAM_API_KEY')
AFFINE_WORKSPACE_URL = os.getenv('AFFINE_WORKSPACE_URL')
AFFINE_USER_EMAIL = os.getenv('AFFINE_USER_EMAIL')
AFFINE_USER_PASSWORD = os.getenv('AFFINE_USER_PASSWORD')
IS_DEBUG = os.getenv('IS_DEBUG') == 'True'
USER_ID = os.getenv('USER_ID')
Приводим в порядок Affine
Теперь добавим файл affine.py (или поправим существующий), чтобы использовать конфиг вместо «прибитых гвоздями» логина, пароля и адреса пространства:
from playwright.sync_api import sync_playwright
import config
def create_affine_note(text: str):
with sync_playwright() as p:
browser = p.chromium.launch(headless=config.IS_DEBUG is False)
page = browser.new_page()
page.goto(config.AFFINE_WORKSPACE_URL)
page.fill('input[placeholder="Enter your email address"]', config.AFFINE_USER_EMAIL)
page.click('[data-testid="continue-login-button"]')
page.fill('input[data-testid="password-input"]', config.AFFINE_USER_PASSWORD)
page.click('button[data-testid="sign-in-button"]')
page.click('button[data-testid="modal-close-button"]')
page.click('div[data-testid="slider-bar-journals-button"]')
page.click('div[data-testid="page-editor-blank"]')
locator = page.locator('div[data-testid="page-editor-blank"]')
locator.focus()
locator.press_sequentially(text)
locator.press('Enter')
browser.close()
Основная логика бота
И, наконец, добавим файл main.py с основной логикой приложения:
import telebot
import config
from telebot import types
from affine import create_affine_note
import dotenv
#загрузим ENV-файл
dotenv.load_dotenv()
# создадим объект бота для взаимодействия с Telegram API
bot = telebot.TeleBot(config.TELEGRAM_API_KEY)
# обработчик команды start
@bot.message_handler(commands=['start'])
def start_bot(message: types.Message):
if str(message.from_user.id) != config.USER_ID:
bot.send_message(message.chat.id, 'Некорректный пользователь!')
return
first_mess = f"Привет! Я бот для создания заметок в Affine."
bot.send_message(message.chat.id, first_mess)
# обработчик остальных команд
@bot.message_handler()
def create_note(message: types.Message):
if str(message.from_user.id) != config.USER_ID:
bot.send_message(message.chat.id, 'Некорректный пользователь!')
return
create_affine_note(message.text)
bot.send_message(message.chat.id, 'Заметка успешно создана!')
if __name__ == '__main__':
# включаем поллинг, чтобы получать обновления от Telegram API
bot.infinity_polling()
Как узнать свой userId? Это можно сделать двумя способами:
- Во время отладки, когда придет первое сообщение от бота.
- Через других специальных телеграм-ботов, например https://t.me/username_to_id_bot (не знаю, насколько это безопасно).
Запускаемся
Заполняем файл ENV и запускаем файл main.py. Теперь наша программа «слушает» обновления от бота.
Переходим в чат с ботом и нажимаем на Start:
