Plex setup and roles setup now thru bot
This commit is contained in:
Faiz Ahmed
@@ -1,22 +1,48 @@
|
||||
import logging
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
import asyncio
|
||||
from plexapi.myplex import MyPlexAccount
|
||||
from discord import Webhook, AsyncWebhookAdapter
|
||||
from app.bot.helper.confighelper import roles, PLEXUSER, PLEXPASS, PLEX_SERVER_NAME, Plex_LIBS
|
||||
logging.basicConfig(filename="app/config/invitarr.log", filemode='a', level=logging.ERROR)
|
||||
import app.bot.helper.db as db
|
||||
import app.bot.helper.plexhelper as plexhelper
|
||||
import texttable
|
||||
import os
|
||||
from os import path
|
||||
import configparser
|
||||
CONFIG_PATH = 'app/config/config.ini'
|
||||
BOT_SECTION = 'bot_envs'
|
||||
|
||||
# settings
|
||||
roles = None
|
||||
PLEXUSER = ""
|
||||
PLEXPASS = ""
|
||||
PLEX_SERVER_NAME = ""
|
||||
Plex_LIBS = ["all"]
|
||||
|
||||
if(path.exists('app/config/config.ini')):
|
||||
try:
|
||||
config = configparser.ConfigParser()
|
||||
config.read(CONFIG_PATH)
|
||||
PLEXUSER = config.get(BOT_SECTION, 'plex_user')
|
||||
PLEXPASS = config.get(BOT_SECTION, 'plex_pass')
|
||||
PLEX_SERVER_NAME = config.get(BOT_SECTION, 'plex_server_name')
|
||||
except:
|
||||
pass
|
||||
if(path.exists('app/config/config.ini')):
|
||||
try:
|
||||
roles = config.get(BOT_SECTION, 'roles')
|
||||
except:
|
||||
pass
|
||||
|
||||
try:
|
||||
account = MyPlexAccount(PLEXUSER, PLEXPASS)
|
||||
plex = account.resource(PLEX_SERVER_NAME).connect() # returns a PlexServer instance
|
||||
logging.info('Logged into plex!')
|
||||
print('Logged into plex!')
|
||||
except:
|
||||
logging.error('Error with plex login. Please check username and password and Plex server name.')
|
||||
print('Error with plex login. Please check username and password and Plex server name or setup plex in the bot.')
|
||||
|
||||
if roles is not None:
|
||||
roles = list(roles.split(','))
|
||||
|
||||
class app(commands.Cog):
|
||||
|
||||
@@ -25,9 +51,11 @@ class app(commands.Cog):
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_ready(self):
|
||||
logging.info('Made by Sleepingpirate https://github.com/Sleepingpirates/')
|
||||
logging.info(f'Logged in as {self.bot.user} (ID: {self.bot.user.id})')
|
||||
logging.info('------')
|
||||
print('Made by Sleepingpirate https://github.com/Sleepingpirates/')
|
||||
print(f'Logged in as {self.bot.user} (ID: {self.bot.user.id})')
|
||||
print('------')
|
||||
if roles is None:
|
||||
print('Configure roles to enable auto invite after a role is assigned.')
|
||||
|
||||
async def embederror(self, author, message):
|
||||
embed1 = discord.Embed(title="ERROR",description=message, color=0xf50000)
|
||||
@@ -58,10 +86,11 @@ class app(commands.Cog):
|
||||
await self.embederror(after, message)
|
||||
return None
|
||||
|
||||
|
||||
async def addtoplex(self, email, channel):
|
||||
if(plexhelper.verifyemail(email)):
|
||||
if plexhelper.plexadd(plex,email):
|
||||
await self.embedinfo(channel, 'There was an error adding this email address. Message Server Admin.')
|
||||
await self.embedinfo(channel, 'This email address has been added to plex')
|
||||
return True
|
||||
else:
|
||||
await self.embederror(channel, 'There was an error adding this email address. Check logs.')
|
||||
@@ -73,7 +102,7 @@ class app(commands.Cog):
|
||||
async def removefromplex(self, email, channel):
|
||||
if(plexhelper.verifyemail(email)):
|
||||
if plexhelper.plexadd(plex,email):
|
||||
await self.embedinfo(channel, 'There was an error removing this email address. Message Server Admin.')
|
||||
await self.embedinfo(channel, 'This email address has been removed from plex.')
|
||||
return True
|
||||
else:
|
||||
await self.embederror(channel, 'There was an error removing this email address. Check logs.')
|
||||
@@ -81,11 +110,11 @@ class app(commands.Cog):
|
||||
else:
|
||||
await self.embederror(channel, 'Invalid email.')
|
||||
return False
|
||||
|
||||
#Auto add or remove user from plex if role is given or taken.
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_member_update(self, before, after):
|
||||
if roles is None:
|
||||
return
|
||||
roles_in_guild = after.guild.roles
|
||||
role = None
|
||||
for role_for_app in roles:
|
||||
@@ -112,25 +141,23 @@ class app(commands.Cog):
|
||||
plexremove(email)
|
||||
deleted = db.delete_user(user_id)
|
||||
if deleted:
|
||||
logging.info("Removed {} from db".format(email))
|
||||
print("Removed {} from db".format(email))
|
||||
#await secure.send(plexname + ' ' + after.mention + ' was removed from plex')
|
||||
else:
|
||||
logging.error("Cannot remove this user from db.")
|
||||
print("Cannot remove this user from db.")
|
||||
except:
|
||||
logging.error("Cannot remove this user from plex.")
|
||||
print("Cannot remove this user from plex.")
|
||||
return
|
||||
|
||||
|
||||
@commands.has_permissions(administrator=True)
|
||||
@commands.command()
|
||||
async def plexinvite(self, ctx):
|
||||
email = str(ctx.content)
|
||||
async def plexinvite(self, ctx, email):
|
||||
await self.addtoplex(email, ctx.channel)
|
||||
|
||||
@commands.has_permissions(administrator=True)
|
||||
@commands.command()
|
||||
async def plexremove(self, ctx):
|
||||
email = str(ctx.content)
|
||||
async def plexremove(self, ctx, email):
|
||||
await self.removefromplex(email, ctx.channel)
|
||||
|
||||
@commands.has_permissions(administrator=True)
|
||||
@@ -143,7 +170,7 @@ class app(commands.Cog):
|
||||
await self.embedinfo(ctx.channel,'email and user were added to the database.')
|
||||
except Exception as e:
|
||||
await self.embedinfo(ctx.channel, 'There was an error adding this email address to database.')
|
||||
logging.error(e)
|
||||
print(e)
|
||||
else:
|
||||
await self.embederror(ctx.channel, 'Invalid email.')
|
||||
|
||||
@@ -209,12 +236,12 @@ class app(commands.Cog):
|
||||
email = db.get_useremail(id)
|
||||
deleted = db.delete_user(id)
|
||||
if deleted:
|
||||
logging.info("Removed {} from db".format(email))
|
||||
print("Removed {} from db".format(email))
|
||||
await self.embedinfo(ctx.channel,"Removed {} from db".format(email))
|
||||
else:
|
||||
await self.embederror(ctx.channel,"Cannot remove this user from db.")
|
||||
except Exception as e:
|
||||
logging.error(e)
|
||||
print(e)
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(app(bot))
|
||||
@@ -1,14 +1,65 @@
|
||||
import configparser
|
||||
import os
|
||||
from os import environ, path
|
||||
from dotenv import load_dotenv
|
||||
config = configparser.ConfigParser()
|
||||
|
||||
CONFIG_PATH = 'app/config/config.ini'
|
||||
BOT_SECTION = 'bot_envs'
|
||||
config = configparser.ConfigParser()
|
||||
|
||||
CONFIG_KEYS = ['username', 'password', 'discord_bot_token', 'plex_user', 'plex_pass',
|
||||
'roles', 'plex_server_name', 'plex_libs', 'owner_id', 'channel_id',
|
||||
'auto_remove_user']
|
||||
|
||||
# settings
|
||||
Discord_bot_token = ""
|
||||
roles = None
|
||||
PLEXUSER = ""
|
||||
PLEXPASS = ""
|
||||
PLEX_SERVER_NAME = ""
|
||||
Plex_LIBS = ["all"]
|
||||
switch = 0
|
||||
|
||||
|
||||
if(path.exists('bot.env')):
|
||||
try:
|
||||
load_dotenv(dotenv_path='bot.env')
|
||||
# settings
|
||||
Discord_bot_token = environ.get('discord_bot_token')
|
||||
switch = 1
|
||||
|
||||
except Exception as e:
|
||||
pass
|
||||
|
||||
elif(path.exists('app/config/config.ini')):
|
||||
try:
|
||||
config = configparser.ConfigParser()
|
||||
config.read(CONFIG_PATH)
|
||||
Discord_bot_token = config.get(BOT_SECTION, 'discord_bot_token')
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
try:
|
||||
Discord_bot_token = str(os.environ['token'])
|
||||
switch = 1
|
||||
except Exception as e:
|
||||
print("ERROR. No config found.")
|
||||
|
||||
if(path.exists('app/config/config.ini')):
|
||||
try:
|
||||
config = configparser.ConfigParser()
|
||||
config.read(CONFIG_PATH)
|
||||
PLEXUSER = config.get(BOT_SECTION, 'plex_user')
|
||||
PLEXPASS = config.get(BOT_SECTION, 'plex_pass')
|
||||
PLEX_SERVER_NAME = config.get(BOT_SECTION, 'plex_server_name')
|
||||
except:
|
||||
pass
|
||||
|
||||
if(path.exists('app/config/config.ini')):
|
||||
try:
|
||||
roles = config.get(BOT_SECTION, 'roles')
|
||||
except:
|
||||
pass
|
||||
|
||||
def get_config():
|
||||
"""
|
||||
Function to return current config
|
||||
@@ -22,40 +73,26 @@ def get_config():
|
||||
return None
|
||||
|
||||
|
||||
CONFIG_PATH = 'app/config/config.ini'
|
||||
BOT_SECTION = 'bot_envs'
|
||||
def change_config(key, value):
|
||||
"""
|
||||
Function to change the key, value pair in config
|
||||
"""
|
||||
try:
|
||||
config = configparser.ConfigParser()
|
||||
config.read(CONFIG_PATH)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print("Cannot Read config.")
|
||||
|
||||
# settings
|
||||
Discord_bot_token = ""
|
||||
roles = ""
|
||||
PLEXUSER = ""
|
||||
PLEXPASS = ""
|
||||
PLEX_SERVER_NAME = ""
|
||||
Plex_LIBS = ""
|
||||
chan = 0
|
||||
ownerid = 0
|
||||
auto_remove_user = ""
|
||||
|
||||
switch = 0
|
||||
|
||||
try:
|
||||
load_dotenv(dotenv_path='bot.env')
|
||||
|
||||
# settings
|
||||
Discord_bot_token = environ.get('discord_bot_token')
|
||||
roles = (environ.get('roles')) # Role Id, right click the role and copy id.
|
||||
PLEXUSER = environ.get('plex_user') # Plex Username
|
||||
PLEXPASS = environ.get('plex_pass') # 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('channel_id'))
|
||||
ownerid = int(environ.get('owner_id'))
|
||||
auto_remove_user = environ.get('auto_remove_user') if environ.get('auto_remove_user') else False # auto remove user from plex and db if removed from the role
|
||||
switch = 1
|
||||
if switch == 1:
|
||||
Plex_LIBS = list(Plex_LIBS.split(','))
|
||||
roles = list(roles.split(','))
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
try:
|
||||
config.set(BOT_SECTION, key, str(value))
|
||||
except Exception as e:
|
||||
config.add_section(BOT_SECTION)
|
||||
config.set(BOT_SECTION, key, str(value))
|
||||
|
||||
try:
|
||||
with open(CONFIG_PATH, 'w') as configfile:
|
||||
config.write(configfile)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print("Cannot write to config.")
|
||||
|
||||
@@ -1,8 +1,6 @@
|
||||
from plexapi.myplex import MyPlexAccount
|
||||
import re
|
||||
from app.bot.helper.confighelper import Plex_LIBS
|
||||
import logging
|
||||
logging.basicConfig(filename="app/config/plex.log", filemode='a', level=logging.ERROR)
|
||||
|
||||
def plexadd(plex, plexname):
|
||||
global Plex_LIBS
|
||||
@@ -12,25 +10,25 @@ def plexadd(plex, plexname):
|
||||
plex.myPlexAccount().inviteFriend(user=plexname, server=plex, sections=Plex_LIBS, allowSync=False,
|
||||
allowCameraUpload=False, allowChannels=False, filterMovies=None,
|
||||
filterTelevision=None, filterMusic=None)
|
||||
logging.info(plexname +' has been added to plex')
|
||||
print(plexname +' has been added to plex')
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.error(e)
|
||||
print(e)
|
||||
return False
|
||||
|
||||
def plexremove(plex, plexname):
|
||||
try:
|
||||
plex.myPlexAccount().removeFriend(user=plexname)
|
||||
logging.info(plexname +' has been removed from plex')
|
||||
print(plexname +' has been removed from plex')
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.error(e)
|
||||
print(e)
|
||||
return False
|
||||
'''
|
||||
|
||||
plex python api has no tools to remove unaccepted invites...
|
||||
|
||||
logging.info("Trying to remove invite...")
|
||||
print("Trying to remove invite...")
|
||||
removeinvite = plexremoveinvite(plex, plexname)
|
||||
if removeinvite:
|
||||
return True
|
||||
@@ -40,10 +38,10 @@ def plexremove(plex, plexname):
|
||||
def plexremoveinvite(plex, plexname):
|
||||
try:
|
||||
plex.myPlexAccount().removeFriend(user=plexname)
|
||||
logging.info(plexname +' has been removed from plex')
|
||||
print(plexname +' has been removed from plex')
|
||||
return True
|
||||
except Exception as e:
|
||||
logging.error(e)
|
||||
print(e)
|
||||
return False
|
||||
'''
|
||||
def verifyemail(addressToVerify):
|
||||
|
||||
Reference in New Issue
Block a user