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

Fixed -dbadd bug

Browse Source
This commit is contained in:
Sleepingpirates
2020-06-11 13:57:14 -04:00
committed by GitHub
parent d205248c9a
commit 720dc42588
1 changed files with 28 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
Invitarr.py
View File

@@ -118,31 +118,34 @@ class MyClient(discord.Client):
if plexremove(mgs): if plexremove(mgs):
await message.channel.send('The email has been removed! {0.author.mention}'.format(message)) await message.channel.send('The email has been removed! {0.author.mention}'.format(message))
else: else:
message.channel.send('Error Check Logs! {0.author.mention}'.format(message)) message.channel.send('Error Check Logs! {0.author.mention}'.format(message))
if message.content.startswith('-dbadd'): if message.content.startswith('-dbadd'):
mgs = message.content.replace('-dbadd ','') mgs = message.content.replace('-dbadd ','')
try: try:
mgs = mgs.split(' ') mgs = mgs.split(' ')
email = mgs[0] email = mgs[0]
user_id = mgs[1] bad_chars = ['<','>','@','!']
db.save_user(user_id, email) user_id = mgs[1]
await message.channel.send('The user {} has been added to db!'.format(mgs[0])) for i in bad_chars:
except: user_id = user_id.replace(i, '')
print("Cannot add this user to db.") db.save_user(user_id, email)
await message.channel.send('The user {} has been added to db!'.format(mgs[0]))
async def on_member_remove(self, member): except:
if auto_remove_user: print("Cannot add this user to db.")
try:
user_id = member.id ## not there async def on_member_remove(self, member):
email = db.get_useremail(user_id) if auto_remove_user:
plexremove(email) try:
deleted = db.delete_user(user_id) user_id = member.id ## not there
if deleted: email = db.get_useremail(user_id)
print("Removed {} from db".format(email)) plexremove(email)
else: deleted = db.delete_user(user_id)
print("Cannot remove this user from db.") if deleted:
except: print("Removed {} from db".format(email))
else:
print("Cannot remove this user from db.")
except:
print("Cannot remove this user from plex.") print("Cannot remove this user from plex.")
client = MyClient() client = MyClient()