rewrote the bot to be self-hosted and use official API

Author: dartungar

NotionService.py

@@ -0,0 +1,66 @@
+import os
+import requests
+import re
+import json
+
+
+class NotionService:
+    BASE_API_URL = 'https://api.notion.com/v1/pages'
+
+    PAGE_TITLE_MAX_LENGTH = 40
+
+    def __init__(self) -> None:
+        pass
+
+    def setup_settings(self) -> None:
+        self.api_key: str = os.getenv('NOTION_API_KEY')
+        self.page_url: str = os.getenv('NOTION_PAGE_URL')
+        self.page_id: str = self.get_page_id_from_url(self.page_url)
+        self.URL_HEADERS = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+            "Notion-Version": "2021-08-16"
+        }
+
+    def create_page(self, text: str) -> None:
+        headers = self.URL_HEADERS
+
+        payload = self.generate_create_page_request_body(text)
+        response = requests.post(
+            self.BASE_API_URL, headers=headers, data=json.dumps(payload))
+
+        if not response.ok:
+            raise NotionServiceException(response.json())
+
+    def generate_create_page_request_body(self, text) -> object:
+        title = text if len(
+            text) < self.PAGE_TITLE_MAX_LENGTH else text[:self.PAGE_TITLE_MAX_LENGTH]+'...'
+        return {
+            "parent": {
+                "page_id": self.page_id
+            },
+            "properties": {
+                "title": {"title": [{"type": "text", "text": {"content": title}}]},
+
+            },
+            "children": [
+                {
+                    "object": "block",
+                    "type": "paragraph",
+                    "paragraph": {
+                        "text": [{"type": "text", "text": {"content": text}}]
+                    }
+                }
+            ]
+        }
+
+    def get_page_id_from_url(self, url: str) -> str:
+        id_regex = re.compile(r'([\w|\d]{32}$)')
+        id_raw = id_regex.findall(url)[0]
+        id_processed = '-'.join([
+            id_raw[0:8], id_raw[8:12], id_raw[12:16], id_raw[16:20], id_raw[20:]])
+        return id_processed
+
+
+class NotionServiceException(Exception):
+    pass

Pipfile

@@ -5,15 +5,15 @@ verify_ssl = true
 
 [dev-packages]
 flake8 = "*"
+pylint = "*"
+autopep8 = "*"
 
 [packages]
 pytelegrambotapi = "*"
 notion = "*"
 requests = "==2.22.0"
 boto = "*"
 python-telegram-bot = "==12.0.0b1"
-psycopg2 = "*"
-sqlalchemy = "*"
 
 [requires]
 python_version = "3.6"

Pipfile.lock

@@ -1,13 +1,16 @@
 # text-to-notion-bot
 
-A (really) simple Telegram bot for sending text into Notion, built with python-telegram-bot and notion.py.
-Here's how it works:
-1. Get your 'token_v2' from cookies while being on any of your Notion.so pages.
-2. Use /setclient command and pass token_v2's value to bot.
-3. Use /setpage command and pass URL of your Notion.so page to bot.
-4. You're set! Any text you send to bot afterwards will be appended to said Notion.so page.
+Really simple self-hosted single-user telegram bot for sending text into Notion, using the official API.
 
-Plans
-* ✅ Persistent user data via DB
-* Multiple Notion pages
-* Some kind of interface to choose pages
+1. Register telegram bot and get bot token by communicationg with https://t.me/botfather
+2. Create Notion Integration and get API key (https://developers.notion.com/docs/getting-started#step-1-create-an-integration)
+3. 'Share' any Notion page with your Notion Integration (https://developers.notion.com/docs/getting-started#step-2-share-a-database-with-your-integration)
+4. Set up environment variables:
+   - BOT_TOKEN=your telegram bot token
+   - TELEGRAM_USERNAME=your telegram username (will be used for authorization)
+   - NOTION_API_KEY=Notion API key
+   - NOTION_PAGE_URL=URL of the page you shared with the integration (bot will parse page ID from it)
+5. Run the bot.
+   - Text sent to the bot will be added to Notion page specified in environment variable as child page.
+   - Title and content will be equal to the text sent.
+   - Long titles (>50 characters) will be truncated automatically

Procfile

@@ -0,0 +1,4 @@
+from Bot import Bot
+
+bot = Bot()
+bot.run()

README.md

@@ -1,70 +1,73 @@
-import logging
 import os
-import db
-from helpers import start, help_msg,  done
-from helpers import ask_notion_api_key, set_notion_api_key, setclient, check_client
-from helpers import askpage, set_page_address, connect_to_page, check_page, send_text_to_notion
-from helpers import TYPING_NOTION_API_KEY, TYPING_NOTION_PAGE_ADDRESS, keyboard
+import logging
 from telegram.ext import Updater, CommandHandler, MessageHandler, ConversationHandler, Filters
-
+from NotionService import NotionService
+from typing import Type
 
 logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                     level=logging.INFO)
-
 logger = logging.getLogger(__name__)
 
 
-BOT_TOKEN = os.environ.get('BOT_TOKEN_NOTION')
-
-def error(update, context):
-    """Log Errors caused by Updates."""
-    logger.warning('Update "%s" caused error "%s"', update, error)
-
-
-def main():
-    # pp = PicklePersistence(filename='notionbot')
-    logger.info(f'bot token: {BOT_TOKEN}')
-    updater = Updater(BOT_TOKEN, use_context=True)
-
-    dp = updater.dispatcher
-
-    convhandler = ConversationHandler(
-        entry_points=[
-                    CommandHandler('start', start),
-                    CommandHandler('setclient', ask_notion_api_key),
-                    CommandHandler('setpage', askpage),
-                    ],
-
-        states={
-            TYPING_NOTION_API_KEY: [MessageHandler(Filters.text, set_notion_api_key)],
-            TYPING_NOTION_PAGE_ADDRESS: [MessageHandler(Filters.text, set_page_address)],
-        },
-
-        fallbacks=[CommandHandler('done', done)],
-        name='my_conversation',
-        persistent=False
-    )
-
-    dp.add_handler(convhandler)
-
-    help_handler = CommandHandler('help', help_msg)
-    dp.add_handler(help_handler)
-
-    check_client_handler = CommandHandler('check_client', check_client)
-    dp.add_handler(check_client_handler)
-
-    check_page_handler = CommandHandler('check_page', check_page)
-    dp.add_handler(check_page_handler)
-
-    send_text_to_notion_handler = MessageHandler(Filters.text, send_text_to_notion)
-    dp.add_handler(send_text_to_notion_handler)
-
-    dp.add_error_handler(error)
-
-    updater.start_polling()
-
-    updater.idle()
-
-
-if __name__ == '__main__':
-    main()
+class Bot:
+    def __init__(self) -> None:
+        pass
+
+    def run(self) -> None:
+        self.setup()
+        self.updater.start_polling()
+        self.updater.idle()
+
+    def setup(self) -> None:
+        try:
+            self.BOT_TOKEN: str = os.getenv('BOT_TOKEN')
+            self.USERNAME: str = os.getenv('TELEGRAM_USERNAME')
+            self.notion: NotionService = NotionService()
+            self.updater: Type[Updater] = Updater(
+                self.BOT_TOKEN, use_context=True)
+            self.dispatcher = self.updater.dispatcher
+            self.notion.setup_settings()
+            self.register_handlers()
+        except Exception as e:
+            raise BotException(e)
+
+    def register_handlers(self) -> None:
+        self.dispatcher.add_handler(self.text_handler())
+        self.dispatcher.add_handler(self.start_handler())
+        self.dispatcher.add_handler(self.help_handler())
+        self.dispatcher.add_error_handler(self.error)
+
+    def text_handler(self) -> Type[MessageHandler]:
+        return MessageHandler(Filters.text, self.send_to_notion)
+
+    def send_to_notion(self, update, context):
+        sender_username = update.message.from_user.username
+        if sender_username == self.USERNAME:
+            self.notion.create_page(update.message.text)
+            logger.info('Successfully sent message to Notion')
+            update.message.reply_text("Text sent to Notion.")
+        else:
+            update.message.reply_text(
+                "⛔ Sorry, you are not authorized for this bot.")
+
+    def start_handler(self) -> Type[CommandHandler]:
+        return CommandHandler('start', help)
+
+    def help_handler(self) -> Type[CommandHandler]:
+        return CommandHandler('help', help)
+
+    def help(self, update, context):
+        reply_text = '''
+        1. Set up environment variables for BOT_TOKEN, USERNAME, NOTION_API_KEY, OTION_PAGE_URL https://github.com/dartungar/text-to-notion-bot
+        2. Send text here
+        3. Have a nice day!
+        '''
+        update.message.reply_text(reply_text)
+
+    def error(self, update, context):
+        """Log Errors caused by Updates."""
+        logger.warning('Update "%s" caused error "%s"', update, error)
+
+
+class BotException(Exception):
+    pass