Working on Discord

app/__init__.py

@@ -6,7 +6,6 @@ Initialize the modules needed for the website
 import os
 
 import discord
-from discord.ext import commands
 from flask import Flask
 from flask_sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
@@ -21,6 +20,7 @@ from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
 
 load_dotenv()
 
+
 class Config(object):
     """Flask configuration"""
     UPLOAD_FOLDER = 'app/static/uploads'
@@ -56,10 +56,29 @@ argon2 = Argon2(app)
 
 scheduler = APScheduler()
 scheduler.init_app(app)
-scheduler.start()
 
 # Login
 login_manager = LoginManager()
 login_manager.init_app(app)
 login_manager.login_view = "auth.login"
 login_manager.login_message_category = "warning"
+
+
+class DiscordClient(discord.Client):
+    """Class for Discord bot"""
+    async def on_ready(self):
+        """Send message after login"""
+        print('Logged on as', self.user)
+
+    async def on_message(self, message):
+        """Read message"""
+        # don't respond to ourselves
+        if message.author == self.user:
+            return
+
+        if message.content == 'ping':
+            await message.channel.send('pong')
+
+
+discord_client = DiscordClient()
+discord_client.run(os.environ["DISCORD_TOKEN"])

app/modules/backend/app.py

@@ -10,6 +10,7 @@ from flask_login import login_required
 from flask_menu import register_menu
 from flask import render_template, request, redirect, url_for, flash, Blueprint
 from app.models import Page, File, User
+from app import discord_client
 
 
 BLUEPRINT = Blueprint(
@@ -41,6 +42,8 @@ def index():
 @login_required
 def render():
     """Render pages to file"""
+    channel = discord_client.get_channel(373000514608300034)
+    await channel.send('hello')
     pages = Page.query.filter(Page.parent_id == None).all()
     menu = []
     for page in pages: