diff --git a/MusicBot/database/base.py b/MusicBot/database/base.py index 943ac30..d374e88 100644 --- a/MusicBot/database/base.py +++ b/MusicBot/database/base.py @@ -13,8 +13,7 @@ users: AsyncCollection[ExplicitUser] = db.users guilds: AsyncCollection[ExplicitGuild] = db.guilds class BaseUsersDatabase: - DEFAULT_USER = ExplicitUser( - _id=0, + DEFAULT_USER = User( ym_token=None, playlists=[], playlists_page=0, @@ -61,8 +60,7 @@ class BaseUsersDatabase: class BaseGuildsDatabase: - DEFAULT_GUILD = ExplicitGuild( - _id=0, + DEFAULT_GUILD = Guild( next_tracks=[], previous_tracks=[], current_track=None, diff --git a/MusicBot/main.py b/MusicBot/main.py index 0268949..cf0248b 100644 --- a/MusicBot/main.py +++ b/MusicBot/main.py @@ -28,10 +28,16 @@ if __name__ == '__main__': except ImportError: pass - logging.getLogger().setLevel(logging.DEBUG) - logging.getLogger('discord').setLevel(logging.INFO) - logging.getLogger('pymongo').setLevel(logging.INFO) - logging.getLogger('yandex_music').setLevel(logging.WARNING) + if os.getenv('DEBUG') == 'True': + logging.getLogger().setLevel(logging.DEBUG) + logging.getLogger('discord').setLevel(logging.INFO) + logging.getLogger('pymongo').setLevel(logging.INFO) + logging.getLogger('yandex_music').setLevel(logging.WARNING) + else: + logging.getLogger().setLevel(logging.INFO) + logging.getLogger('discord').setLevel(logging.WARNING) + logging.getLogger('pymongo').setLevel(logging.WARNING) + logging.getLogger('yandex_music').setLevel(logging.WARNING) if not os.path.exists('music'): os.mkdir('music')