JoostSijm 6 rokov pred
rodič
commit
91a38b53f6
3 zmenil súbory, kde vykonal 34 pridanie a 10 odobranie
  1. 3 7
      app/__init__.py
  2. 2 3
      app/modules/backend/app.py
  3. 29 0
      app/util/job.py

+ 3 - 7
app/__init__.py

@@ -77,19 +77,15 @@ login_manager.login_message_category = "warning"
 
 class DiscordClient(discord.Client):
     """Class for Discord bot"""
-    async def on_ready(self):
+    def on_ready(self):
         """Send message after login"""
         print('Logged on as', self.user)
 
-    async def on_message(self, message):
+    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"])
+            message.channel.send('pong')

+ 2 - 3
app/modules/backend/app.py

@@ -10,7 +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
+from app.util import job
 
 
 BLUEPRINT = Blueprint(
@@ -42,8 +42,7 @@ def index():
 @login_required
 def render():
     """Render pages to file"""
-    channel = discord_client.get_channel(373000514608300034)
-    await channel.send('hello')
+    job.add_send_code()
     pages = Page.query.filter(Page.parent_id == None).all()
     menu = []
     for page in pages:

+ 29 - 0
app/util/job.py

@@ -0,0 +1,29 @@
+"""Scheduler class"""
+
+import os
+
+from app import scheduler, DiscordClient
+
+
+def add_send_code():
+    """Send code to user"""
+    job = scheduler.get_job("send_code")
+    print("hi")
+    if job is None:
+        job = scheduler.add_job(
+            id="send_code",
+            func=run_send_code,
+            trigger="date",
+        )
+
+
+def run_send_code():
+    """Upgrade to new version"""
+    discord_client = DiscordClient()
+    discord_client.run(os.environ["DISCORD_TOKEN"])
+    discord_channel = client.get_channel(373000514608300034)
+    discord_channel.send('hello')
+
+
+if __name__ == "__main__":
+    run_send_code.__module__ = "scheduler"