import os import sys from pathlib import Path from dotenv import load_dotenv def app_root_path(): app_path = Path(sys.argv[0]).resolve() try: if app_path.parent.name == "bin": # Installed in unixy hierachy return app_path.parents[1] except IndexError: pass return app_path.parent # None will let direnv do its' thing env_paths = [Path(".env"), app_root_path() / "etc/environment", None] for env_path in env_paths: print("Loading environment from " + str(env_path)) load_dotenv(dotenv_path=env_path) class EnvService: # To be expanded upon later! def __init__(self): self.env = {} @staticmethod def environment_path_with_fallback(env_name, relative_fallback=None): dir = os.getenv(env_name) if dir != None: return Path(dir).resolve() if relative_fallback: app_relative = (app_root_path() / relative_fallback).resolve() if app_relative.exists(): return app_relative return Path.cwd() @staticmethod def find_shared_file(file_name): share_file_paths = [] share_dir = os.getenv("SHARE_DIR") if share_dir != None: share_file_paths.append(share_dir) share_file_paths.extend( [ app_root_path() / "share" / file_name, app_root_path() / file_name, Path(file_name), ] ) for share_file_path in share_file_paths: if share_file_path.exists(): return share_file_path.resolve() raise ValueError(f"Unable to find shared data file {file_name}") @staticmethod def get_allowed_guilds(): # ALLOWED_GUILDS is a comma separated list of guild ids # It can also just be one guild ID # Read these allowed guilds and return as a list of ints try: allowed_guilds = os.getenv("ALLOWED_GUILDS") except: allowed_guilds = None if allowed_guilds is None: raise ValueError( "ALLOWED_GUILDS is not defined properly in the environment file!" "Please copy your server's guild ID and put it into ALLOWED_GUILDS in the .env file." 'For example a line should look like: `ALLOWED_GUILDS="971268468148166697"`' ) allowed_guilds = ( allowed_guilds.split(",") if "," in allowed_guilds else [allowed_guilds] ) allowed_guilds = [int(guild) for guild in allowed_guilds] return allowed_guilds @staticmethod def get_admin_roles(): # ADMIN_ROLES is a comma separated list of string roles # It can also just be one role # Read these allowed roles and return as a list of strings try: admin_roles = os.getenv("ADMIN_ROLES") except: admin_roles = None if admin_roles is None: print( "ADMIN_ROLES is not defined properly in the environment file!" "Please copy your server's role and put it into ADMIN_ROLES in the .env file." 'For example a line should look like: `ADMIN_ROLES="Admin"`' ) print("Defaulting to allowing all users to use admin commands...") return [None] admin_roles = ( admin_roles.lower().split(",") if "," in admin_roles else [admin_roles.lower()] ) return admin_roles @staticmethod def get_dalle_roles(): # DALLE_ROLES is a comma separated list of string roles # It can also just be one role # Read these allowed roles and return as a list of strings try: dalle_roles = os.getenv("DALLE_ROLES") except: dalle_roles = None if dalle_roles is None: print( "DALLE_ROLES is not defined properly in the environment file!" "Please copy your server's role and put it into DALLE_ROLES in the .env file." 'For example a line should look like: `DALLE_ROLES="Dalle"`' ) print("Defaulting to allowing all users to use Dalle commands...") return [None] dalle_roles = ( dalle_roles.lower().split(",") if "," in dalle_roles else [dalle_roles.lower()] ) return dalle_roles @staticmethod def get_gpt_roles(): # GPT_ROLES is a comma separated list of string roles # It can also just be one role # Read these allowed roles and return as a list of strings try: gpt_roles = os.getenv("GPT_ROLES") except: gpt_roles = None if gpt_roles is None: print( "GPT_ROLES is not defined properly in the environment file!" "Please copy your server's role and put it into GPT_ROLES in the .env file." 'For example a line should look like: `GPT_ROLES="Gpt"`' ) print("Defaulting to allowing all users to use GPT commands...") return [None] gpt_roles = ( gpt_roles.lower().strip().split(",") if "," in gpt_roles else [gpt_roles.lower()] ) return gpt_roles @staticmethod def get_welcome_message(): # WELCOME_MESSAGE is a default string used to welcome new members to the server if GPT3 is not available. # The string can be blank but this is not advised. If a string cannot be found in the .env file, the below string is used. # The string is DMd to the new server member as part of an embed. try: welcome_message = os.getenv("WELCOME_MESSAGE") except: welcome_message = "Hi there! Welcome to our Discord server!" return welcome_message @staticmethod def get_moderations_alert_channel(): # MODERATIONS_ALERT_CHANNEL is a channel id where moderation alerts are sent to # The string can be blank but this is not advised. If a string cannot be found in the .env file, the below string is used. try: moderations_alert_channel = os.getenv("MODERATIONS_ALERT_CHANNEL") except: moderations_alert_channel = None return moderations_alert_channel