deadcxap/Membarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Invitarr-docker.py

Browse Source
This commit is contained in:
Sleepingpirates
2020-06-10 12:54:30 -04:00
committed by GitHub
parent a53e39ea3a
commit 4c6600c09e
1 changed files with 47 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
Invitarr-docker.py
View File

@@ -1,4 +1,5 @@
#Copyright 2020 Sleepingpirate.
from os import environ
import discord
from discord.ext import commands
@@ -7,7 +8,6 @@ from plexapi.myplex import MyPlexAccount
from discord import Webhook, AsyncWebhookAdapter
import aiohttp
# settings
Discord_bot_token = environ.get('discord_bot_token')
roleid = int(environ.get('roleid')) # Role Id, right click the role and copy id.
@@ -16,10 +16,15 @@ PLEXPASS = environ.get('PLEXPASS') # plex password
PLEX_SERVER_NAME = environ.get('PLEX_SERVER_NAME') # Name of plex server
Plex_LIBS = environ.get('Plex_LIBS') #name of the libraries you want the user to have access to.
chan = int(environ.get('channelid'))
auto_remove_user = environ.get('autoremoveuser') if environ.get('autoremoveuser') else False # auto remove user from plex and db if removed from the role
li = list(Plex_LIBS.split(','))
Plex_LIBS = li
if auto_remove_user:
print("auto remove user = True")
import db as db
account = MyPlexAccount(PLEXUSER, PLEXPASS)
plex = account.resource(PLEX_SERVER_NAME).connect() # returns a PlexServer instance
@@ -37,6 +42,16 @@ def plexadd(plexname):
return True
def plexremove(plexname):
try:
plex.myPlexAccount().removeFriend(user=plexname)
except Exception as e:
print(e)
return False
else:
print(plexname +' has been removed from plex (☞ຈل͜ຈ)☞')
return True
class MyClient(discord.Client):
async def on_ready(self):
print('Made by Sleepingpirate https://github.com/Sleepingpirates/')
@@ -62,12 +77,27 @@ class MyClient(discord.Client):
print(email.content) #make it go to a log channel
plexname = str(email.content)
if plexadd(plexname):
if auto_remove_user:
db.save_user(after.display_name, email.content)
await asyncio.sleep(20)
await after.send('You have Been Added To Plex!')
secure = client.get_channel(chan)
await secure.send(plexname + ' ' + after.mention + ' was added to plex')
else:
await after.send('There was an error adding this email address. Message Server Admin.')
elif(role not in after.roles and role in before.roles):
if auto_remove_user:
try:
username = after.display_name
email = db.get_useremail(username)
plexremove(email)
deleted = db.delete_user(username)
if deleted:
print("Removed {} from db".format(email))
else:
print("Cannot remove this user from db.")
except:
print("Cannot remove this user from plex.")
async def on_message(self, message):
secure = client.get_channel(chan)
@@ -81,6 +111,22 @@ class MyClient(discord.Client):
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))
if message.content.startswith('-plexrm'):
mgs = message.content.replace('-plexrm ','')
if plexremove(mgs):
await message.channel.send('The email has been removed! {0.author.mention}'.format(message))
else:
message.channel.send('Error Check Logs! {0.author.mention}'.format(message))
if message.content.startswith('-dbadd'):
mgs = message.content.replace('-dbadd ','')
try:
mgs = mgs.split(' ')
email = mgs[0]
username = mgs[1].replace('@', '').split('#')[0]
db.save_user(username, email)
await message.channel.send('The user {} has been added to db!'.format(mgs[1]))
except:
print("Cannot add this user to db.")
client = MyClient()
client.run(Discord_bot_token)