YuiBot/config.py at edge · staticaron/YuiBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
from discord import Color
from os import environ
import json
import traceback
import sys
from dotenv import load_dotenv
import logging

load_dotenv()

DISCORD_TOKEN = None
DISCORD_TEST_TOKEN = None
ANILIST_ID = None
ANILIST_TOKEN = None
MONGO_SRV = None
DB = "youranimedb"
INVITE = None
SUPPORT_SERVER_LINK = "https://discord.gg/AbfZPyx5MH"
VOTE_LINK = "https://top.gg/bot/991739924250362047"

# Loggers
SERVER_LOG_CHANNEL = 1011530425023344722

# URL Bases
ANILIST_BASE = "https://graphql.anilist.co"
ANILIST_LOGIN = "https://anilist.co/api/v2/oauth/authorize?client_id={client_id}&response_type=token"

CHIAKI_BASE = "https://chiaki.vercel.app/get?group_id={}"

# Colors
NORMAL_COLOR = Color.dark_theme()
INFO_COLOR = Color.gold()
ERROR_COLOR = Color.red()

# Emojis
YUI_SHY_EMOTE = "<a:yui_shy:991749482763014184>"
ADULT_CONTENT_EMOTE = "<:18_up:991930483309023352>"
BULLET_EMOTE = "<:bullet:992560047978729512>"
DOT_EMOTE = "<:dots:993535564835979357>"
LOADING_EMOTE = "<a:LOADING:994182649100894318>"
NEXT_EMOTE = "<:next:995484808207683604>"
PREV_EMOTE = "<:prev:995484847139209238>"
FIRST_EMOTE = "<:first:996104181515571201>"
LAST_EMOTE = "<:last:996104225459277854>"

YOUTUBE_VIDEO_EMOTE = "<:youtube:1371941562358960128>"
YOUTUBE_MUSIC_EMOTE = "<:youtubemusic:1371941660673708033>"

FILLER_DATA = None

ALL_GENRE = [
    "action",
    "adventure",
    "comedy",
    "drama",
    "ecchi",
    "fantasy",
    "hentai",
    "horror",
    "mecha",
    "music",
    "mystery",
    "psychological",
    "romance",
    "sci-fi",
    "sol",
    "sports",
    "supernatural",
    "thriller",
]

ALL_GENRE_ALTS = {
    "action": "Action",
    "adventure": "Adventure",
    "comedy": "Comedy",
    "drama": "Drama",
    "ecchi": "Ecchi",
    "fantasy": "Fantasy",
    "hentai": "Hentai",
    "horror": "Horror",
    "mecha": "Mecha",
    "music": "Music",
    "mystery": "Mystery",
    "psychological": "Psychological",
    "romance": "Romance",
    "sci-fi": "Sci-Fi",
    "sol": "Slice of Life",
    "sports": "Sports",
    "supernatural": "Supernatural",
    "thriller": "Thriller",
}

SECRET_KEY = ""

# spotify creds
SPOTIFY_CLIENT_ID = ""
SPOTIFY_CLIENT_SECRET = ""

YOUTUBE_API_KEY = ""
YOUTUBE_TRACK_BASE = "http://youtube.com/watch?v="
YOUTUBE_MUSIC_BASE = "https://music.youtube.com/watch?v="


def initialize_config_vars() -> str:
    global \
        DISCORD_TOKEN, \
        DISCORD_TEST_TOKEN, \
        ANILIST_ID, \
        ANILIST_TOKEN, \
        INVITE, \
        MONGO_SRV, \
        FILLER_DATA, \
        SECRET_KEY, \
        SPOTIFY_CLIENT_ID, \
        SPOTIFY_CLIENT_SECRET, \
        YOUTUBE_API_KEY

    load_dotenv()

    try:
        DISCORD_TOKEN = environ["TOKEN"]
        DISCORD_TEST_TOKEN = environ["TEST_TOKEN"]
        ANILIST_TOKEN = environ["ANILIST_TOKEN"]
        ANILIST_ID = environ["ANILIST_ID"]
        MONGO_SRV = environ["MONGO_SRV"]
        INVITE = environ["INVITE"]
        SECRET_KEY = environ["SECRET_KEY"]
        SPOTIFY_CLIENT_ID = environ["SPOTIFY_CLIENT_ID"]
        SPOTIFY_CLIENT_SECRET = environ["SPOTIFY_CLIENT_SECRET"]
        YOUTUBE_API_KEY = environ["YOUTUBE_API_KEY"]

        with open("data/filler_data.json", "r", encoding="utf8") as filler_data:
            FILLER_DATA = json.load(filler_data)

    except Exception as e:
        logging.error(f"Error occurred while trying to cache Config Vars! \n{e}")
        traceback.print_exception(type(e), e, e.__traceback__, file=sys.stderr)
    else:
        logging.info("Config LOADED!")