Update Invitarr.py

Invitarr.py

@@ -14,24 +14,37 @@ PLEXUSER = ''           # Plex Username
 PLEXPASS = ''           # plex password
 PLEX_SERVER_NAME = ''   # Name of plex server 
 Plex_LIBS = ["Movies","TV Shows"] #name of the libraries you want the user to have access to.
-Webhookurl = '' # For logging the user repiles, create a webhook to the discord channel you want to log this in.
+chan = 
-
-
 
 account = MyPlexAccount(PLEXUSER, PLEXPASS)
 plex = account.resource(PLEX_SERVER_NAME).connect()  # returns a PlexServer instance
 
+def plexadd(plexname):
+    try:
+        plex.myPlexAccount().inviteFriend(user=plexname, server=plex, sections=Plex_LIBS, allowSync=False,
+                                              allowCameraUpload=False, allowChannels=False, filterMovies=None,
+                                              filterTelevision=None, filterMusic=None)
+
+    except Exception as e:
+        print(e)
+        return False
+    else:
+        print(plexname +' has been added to plex (☞ຈل͜ຈ)☞')
+        return True
+
+
 class MyClient(discord.Client):
     async def on_ready(self):
+        print('Made by Sleepingpirate https://github.com/Sleepingpirates/')
         print('Logged in as')
         print(self.user.name)
         print(self.user.id)
         print('------')
 
     async def on_member_update(self, before, after):
-        role = after.guild.get_role(roleid) 
+        role = after.guild.get_role(roleid)
         if (role in after.roles and role not in before.roles):
-            await after.send('Welcome To The Plex. Just reply with your email so we can add you to Plex!')
+            await after.send('Welcome To '+ PLEX_SERVER_NAME +'. Just reply with your email so we can add you to Plex!')
             await after.send('I will wait 10 minutes for your message, if you do not send it by then I will cancel the command.')
             def check(m):
                 return m.author == after and not m.guild
@@ -44,19 +57,26 @@ class MyClient(discord.Client):
                 await after.send('Got it we will be processing your email shortly')
             print(email.content) #make it go to a log channel
             plexname = str(email.content)
-            try:
+            if plexadd(plexname):
-                plex.myPlexAccount().inviteFriend(user=plexname, server=plex, sections=Plex_LIBS, allowSync=False, 
-                                                      allowCameraUpload=False, allowChannels=False, filterMovies=None,
-                                                      filterTelevision=None, filterMusic=None)
                 await asyncio.sleep(20)
-
-            except Exception as e:
-                print(e)
-            else:
                 await after.send('You have Been Added To Plex!')
-            async with aiohttp.ClientSession() as session:
+                secure = client.get_channel(chan)
-                webhook = Webhook.from_url(Webhookurl, adapter=AsyncWebhookAdapter(session)) 
+                await secure.send(plexname + ' ' + after.mention + ' was added to plex')
-                await webhook.send(plexname + ' ' + after.mention + ' was added to plex', username='Logger')
+            else:
+                await after.send('There was an error adding this email address. Message Server Admin.')
+
+    async def on_message(self, message):
+        secure = client.get_channel(chan)
+        if message.author.id == self.user.id:
+            return
+
+        if str(message.channel) == str(secure):
+            if message.content.startswith('-plexadd'):
+                mgs = message.content.replace('-plexadd ','')
+                if plexadd(mgs):
+                    await message.channel.send('The email has been added! {0.author.mention}'.format(message))
+                else:
+                    message.channel.send('Error Check Logs! {0.author.mention}'.format(message))
 
 client = MyClient()
 client.run(Discord_bot_token)