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-11 14:22:39 -04:00
committed by GitHub
parent cb06a626fe
commit 6e561e1f57
1 changed files with 33 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
Docker/Invitarr-docker.py
View File

@@ -78,7 +78,7 @@ class MyClient(discord.Client):
plexname = str(email.content)
if plexadd(plexname):
if auto_remove_user:
db.save_user(after.display_name, email.content)
db.save_user(str(after.id), email.content)
await asyncio.sleep(20)
await after.send('You have Been Added To Plex!')
secure = client.get_channel(chan)
@@ -88,10 +88,10 @@ class MyClient(discord.Client):
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)
user_id = after.id
email = db.get_useremail(user_id)
plexremove(email)
deleted = db.delete_user(username)
deleted = db.delete_user(user_id)
if deleted:
print("Removed {} from db".format(email))
else:
@@ -118,5 +118,33 @@ class MyClient(discord.Client):
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]
bad_chars = ['<','>','@','!']
user_id = mgs[1]
for i in bad_chars:
user_id = user_id.replace(i, '')
db.save_user(user_id, email)
await message.channel.send('The user {} has been added to db!'.format(mgs[0]))
except:
print("Cannot add this user to db.")
async def on_member_remove(self, member):
if auto_remove_user:
try:
user_id = member.id ## not there
email = db.get_useremail(user_id)
plexremove(email)
deleted = db.delete_user(user_id)
if deleted:
print("Removed {} from db".format(email))
else:
print("Cannot remove this user from db.")
except:
print("Cannot remove this user from plex.")
client = MyClient()
client.run(Discord_bot_token)