Update app.py

app.py

@@ -157,11 +157,15 @@ The --setup command enables Lyre to communicate within a designated channel.
 
 ## Example
 
---setup <#1194161958505676851>
+```arduino
+--setup <#1234567890123456789>
+```
 
 or
 
+```arduino
 --setup #general
+```
 
 """
 
@@ -204,36 +208,43 @@ async def on_message(message):
                 embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in this guild. Use `--help` for command help.")
                 await message.reply(embed=embed)
                 return 0
-        elif args[0].startswith("#"):
+        else:
+            embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms. Use `--help` for command help.")
+            await message.reply(embed=embed)
+            return 0
+    if message.content.startswith("--revoke"):
+        args = message.content.split()
+        del args[0]
+        if len(args) == 0:
+            embed = discord.Embed(title="Error", description="No args provided. Use `--help` for command help.")
+            await message.reply(embed=embed)
+            return 0
+        if args[0] == "":
+            embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
+            await message.reply(embed=embed)
+            return 0
+        if args[0].startswith("<#") and args[0].endswith(">"):
             try:
-                cname = args[0][1:]
+                cid = int(args[0][2:-1])
             except:
                 embed = discord.Embed(title="Error", description="Parameter `[channel]` invalid. Use `--help` for command help.")
                 await message.reply(embed=embed)
                 return 0
-            try:
-                c = discord.utils.get(client.get_all_channels(), guild__id=message.guild.id, name=cname)
-                if c != None:
-                    with open(f"{guild_name}.guild", "a") as f:
-                        f.write(f"{c.id}\n")
-                    embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{c.id}>")
-                    message.reply(embed=embed)
-                    return 0
-                else:
-                    embed = discord.Embed(title="Error", description="Parameter `[channel]` is not a valid channel in this guild. Use `--help` for command help.")
-                    message.reply(embed=embed)
-                    return 0
-            except:
-                embed = discord.Embed(title="Error", description="Parameter `[channel]` is not a valid channel in this guild. Use `--help` for command help.")
-                message.reply(embed=embed)
+            with open(f"{guild_name}.guild", "r") as f:
+                k = f.read().split("\n")
+            if str(cid) in k:
+                k.remove(str(cid))
+                with open(f"{guild_name}.guild", "w") as f:
+                    f.write("\n".join(k))
+                embed = discord.Embed(title="Success", description=f"Lyre is now unable to chat in {args[0]}")
+                await message.reply(embed=embed)
                 return 0
         else:
-            embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms. Use `--help` for command help.")
+            embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms, or does not exist. Use `--help` for command help.")
             await message.reply(embed=embed)
             return 0
         
         
-        
     if message.content == "--help":
         def check(reaction, user):
             return reaction.message.id == msg.id and user == message.author