security

core/config.py

@@ -1,7 +1,12 @@
+import os
 from pathlib import Path
 
+from dotenv import load_dotenv
+
 BASE_DIR = Path(__file__).resolve().parent.parent
 
+load_dotenv(BASE_DIR / '.env')
+
 YT_DIR_NAME = "yt"
 YT_DIR = BASE_DIR / "media" / YT_DIR_NAME
 THUMBS = "_thumbs"
@@ -14,7 +19,7 @@ STATIC_DIR = BASE_DIR / "static"
 
 VIDEO_EXTENSIONS = {".mp4", ".mkv", ".webm", ".avi", ".mov", ".flv", ".m4v"}
 
-AUDIO_EXTENSIONS = {".mp3"}
+AUDIO_EXTENSIONS = {".mp3", ".flac", ".ogg"}
 AUDIO_DIR = BASE_DIR / "media" / "audio"
 COVERS = "_covers"
 COVERS_DIR = AUDIO_DIR / COVERS
@@ -23,7 +28,7 @@ from pydantic_settings import BaseSettings
 
 
 class Settings(BaseSettings):
-    SECRET_KEY: str = "CHANGE_ME_TO_RANDOM_STRING"
+    SECRET_KEY: str = os.getenv("SECRET_KEY")
     ALGORITHM: str = "HS256"
     ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
     REFRESH_TOKEN_EXPIRE_DAYS: int = 7