asyncio導入

dislocker_slash.py

@@ -3,7 +3,7 @@ import os
 import json
 import psycopg2
 from psycopg2 import sql
-from datetime import datetime
+from datetime import datetime, timedelta
 import asyncio
 import string
 import random
@@ -761,6 +761,92 @@ class ReasonModal(discord.ui.Modal):
         else:
             await interaction.response.send_message("# :skull_crossbones: 登録できませんでした。\n内部エラーが発生しています。", ephemeral=True)
 
+
+class Monitor():
+    def __init__(self, **kwargs) -> None:
+        self.search_frequency = kwargs["search_frequency"]
+        self.allowable_time = kwargs["allowable_time"]
+        self.fstop_time = kwargs["fstop_time"]
+        self.init_wait_time = 10
+
+    async def search(self):
+        try:
+            asyncio.sleep(self.init_wait_time)
+            while True:
+                cursor = dislocker.db.cursor()
+                cursor.execute("SELECT * FROM pc_list WHERE password_hash IS NOT NULL")
+                pc_list = cursor.fetchall()
+                current_datetime = datetime.now()
+                fstop_time = self.fstop_time
+                if current_datetime.time().strftime("%H:%M:%S") == fstop_time:
+                    dislocker.log(title=f"[INFO] 定期のPCの使用停止処理を開始します。", flag=0)
+                    for i in dislocker.pc_list:
+                        stop = dislocker.force_stop(pc_number=i, bot_about="使用停止忘れによるBotによる強制停止。")
+                    result = {"result": "FSTOP"}
+                    dislocker.log(title=f"[SUCCESS] 定期のPCの使用停止処理は完了しました。", flag=0)
+                else:
+                    if pc_list:
+                        if len(pc_list) == 1:
+                            member_id = pc_list[0][1]
+                            cursor.execute("SELECT * FROM pc_usage_history WHERE member_id= %s AND end_use_time IS NULL ORDER BY id DESC LIMIT 1", (member_id,))
+                            pc_usage = cursor.fetchall()
+                            start_time = pc_usage[0][5]
+                            time_difference = current_datetime - start_time
+                            dislocker.log(title=f"[INFO] 現在確認されているパスワード未使用のユーザー", message=f"レコード | {str(pc_usage)}, 経過時間(Sec) | {time_difference.seconds}/{timedelta(seconds=self.allowable_time).seconds}", flag=0)
+                            if time_difference.seconds >= timedelta(seconds=self.allowable_time).seconds:
+                                cursor.execute("SELECT * FROM club_member WHERE member_id = %s", (member_id,))
+                                user_info = cursor.fetchall()
+                                stop = dislocker.stop(discord_user_id=user_info[0][3], bot_about="パスワードのタイムアウトでBotによる強制停止。")
+                                
+                                #bot.timeout_notify(pc_number=pc_list[0][0], discord_display_name=user_info[0][1])
+                                dislocker.log(title=f"[INFO] パスワードのタイムアウト時間に達したため、強制停止されました。", flag=0)
+                                result = {"result": "STOP", "details": str(pc_usage)}
+                            else:
+                                result = {"result": "BUT SAFE", "details": str(pc_usage)}
+
+
+                        elif len(pc_list) >= 2:
+                            for i in pc_list:
+                                member_id = i[1]
+                                cursor.execute("SELECT * FROM pc_usage_history WHERE member_id= %s AND end_use_time IS NULL ORDER BY id DESC LIMIT 1", (member_id,))
+                                pc_usage = cursor.fetchall()
+                                start_time = pc_usage[0][5]
+                                time_difference = current_datetime - start_time
+                                dislocker.log(title=f"[INFO] 現在確認されているパスワード未使用のユーザー", message=f"レコード | {str(pc_usage)}, 経過時間(Sec) | {time_difference.seconds}/{timedelta(seconds=self.allowable_time).seconds}", flag=0)
+                                if time_difference.seconds >= timedelta(seconds=self.allowable_time).seconds:
+                                    cursor.execute("SELECT * FROM club_member WHERE member_id = %s", (member_id,))
+                                    user_info = cursor.fetchall()
+                                    stop = dislocker.stop(discord_user_id=user_info[0][3], bot_about="タイムアウトでBotによる強制停止。")
+                                    
+                                    #bot.timeout_notify(pc_number=i[0], discord_display_name=user_info[0][1])
+                                    dislocker.log(title=f"[INFO] パスワードのタイムアウト時間に達したため、強制停止されました。", flag=0)
+                                    result = {"result": "STOP", "details": str(pc_usage)}
+                                else:
+                                    result = {"result": "BUT SAFE", "details": str(pc_usage)}
+
+                        else:
+                            result = {"result": "NONE"}
+                    else:
+                        result = {"result": "NONE"}
+
+                if result["result"] == "NONE":
+                    pass
+                else:
+                    pass
+                asyncio.sleep(self.search_frequency)
+
+
+        except Exception as error:
+            dislocker.log(title=f"[ERROR] 自動停止処理中にエラーが発生しました。 {str(error.__class__.__name__)}", message=str(error.args), flag=1)
+            result = {"result": "error"}
+            dislocker.db.rollback()
+
+        finally:
+            if cursor:
+                cursor.close()
+            return result
+
+
 dislocker = DL()
 
 intents = discord.Intents.default()
@@ -1064,5 +1150,15 @@ async def button_init(interaction: discord.Interaction, text_channel: discord.Te
     await interaction.response.send_message(f":white_check_mark: ボタンを送信しました！", ephemeral=True)
 
 
+async def run():
+    print("Botを起動します...")
+    monitor = Monitor(search_frequency=dislocker.server_config["bot"]["monitor"]["search_frequency"], allowable_time=dislocker.server_config["bot"]["monitor"]["allowable_time"], fstop_time=dislocker.server_config["bot"]["monitor"]["fstop_time"])
+    monitor_task = asyncio.create_task(monitor.start())
+    client_task = asyncio.create_task(client.run(dislocker.server_config["bot"]["token"]))
+    await asyncio.gather(monitor_task, client_task)
+    
 
-client.run(dislocker.server_config["bot"]["token"])
+if dislocker.init_result == "ok":
+    asyncio.run(run())
+else:
+    pass