Added Screenshot command and fixed network monitor

Author: mochiron-desu

.gitignore

@@ -9,4 +9,5 @@ config.json
 files/*
 data/data.json
 data/screenshot.png
+data/mouse-ss.png
 test.py

bot.py

@@ -9,31 +9,31 @@
 KEY_PATH = "D:\Coding\Discord bots\python-windows-bot\key.txt"
 CONFIG_PATH = "D:\Coding\Discord bots\python-windows-bot\config.json"
 
-
 def is_connected():
     try:
-        # Try to resolve a common domain to check if network is available
+        # Try to resolve a common domain to check if the network is available
         socket.create_connection(("www.google.com", 80))
         return True
     except OSError:
         pass
     return False
 
-
 def load_config():
     print("Loading configuration file...")
     with open(CONFIG_PATH, "r") as config_file:
         config = json.load(config_file)
         return config
 
-
-async def main():
+async def network_monitor():
     while True:
-        if is_connected():
-            break
-        print("Waiting for network...")
-        await asyncio.sleep(5)  # Wait for 5 seconds before checking again
+        if not is_connected():
+            print("Network connection lost. Waiting for network...")
+            while not is_connected():
+                await asyncio.sleep(5)  # Wait for 5 seconds before checking again
+            print("Network has been restored.")
+        await asyncio.sleep(60)  # Check network status every 60 seconds
 
+async def main():
     if os.path.exists(KEY_PATH):
         with open(KEY_PATH, "r") as f:
             TOKEN = f.read()
@@ -46,6 +46,9 @@ async def main():
 
     client = discord.Client(intents=intents)
 
+    # Start the network monitoring task in the background
+    asyncio.create_task(network_monitor())
+
     @client.event
     async def on_ready():
         print(f"We have logged in as {client.user}")
@@ -56,7 +59,8 @@ async def on_ready():
 
     @client.event
     async def on_message(message):
-        msg = str(message.content).lower()
+        msg = str(message.content).lower().split(' ')[0]
+        args = message.content.split(" ")[1:]
 
         if (
             message.author == client.user
@@ -66,7 +70,7 @@ async def on_message(message):
 
         if message.author.bot or message.author.id != config["author_id"]:
             return
-
+        
         if str(message.attachments) != "[]":
             from commands.file_save import execute_file_save
 
@@ -81,8 +85,11 @@ async def on_message(message):
             from commands.ping_command import execute_ping_command
 
             await execute_ping_command(client, message, config)
+        
+        if msg == "ss" or msg == "screenshot":
+            from commands.screenshot_command import execute_screenshot_command
+            await execute_screenshot_command(message, args)
 
     await client.start(TOKEN)
 
-
 asyncio.run(main())

commands/__pycache__/file_save.cpython-311.pyc

@@ -8,7 +8,7 @@
 from PIL import ImageGrab
 
 DATA_JSON_PATH = "D:\Coding\Discord bots\python-windows-bot\data\data.json"
-SCREENSHOT_PATH = "D:\Coding\Discord bots\python-windows-bot\data\screenshot.png"
+SCREENSHOT_PATH = "D:\Coding\Discord bots\python-windows-bot\data\mouse-ss.png"
 
 
 async def monitor_mouse_movement(client, config, starting_mouse_position):

commands/__pycache__/mouse_movement.cpython-311.pyc

@@ -0,0 +1,24 @@
+import discord
+import datetime
+from PIL import ImageGrab
+
+SCREENSHOT_PATH = "D:\Coding\Discord bots\python-windows-bot\data\screenshot.png"
+
+def get_ss():
+    screenshot = ImageGrab.grab()
+    screenshot.save(SCREENSHOT_PATH)
+    return SCREENSHOT_PATH
+
+async def execute_screenshot_command(message,args):
+    time = datetime.datetime.now().strftime("%d-%m-%Y %H:%M:%S")
+    MESSAGE = f"Screenshot at {time}"
+    PATH = get_ss()
+    ss_message = await message.channel.send(MESSAGE, file=discord.File(PATH))
+    await message.delete()
+    print(f"[screenshot_command_LOG] - Screenshot taken at {time}.")
+    if("-p" in args):
+        print("Persistent screenshot requested")
+    else:
+        time.sleep(5)
+        await ss_message.delete()
+        print("Screenshot Deleted")