minor

endpoints/constructors.py

@@ -1,6 +1,9 @@
 from typing import Optional
+from typing import Literal
 from pydantic import BaseModel
 
+Station = Literal["main", "rock", "rap", "electronic", "pop"]
+
 """
 LastFM
 """
@@ -96,25 +99,6 @@ class ValidYTSearchRequest(BaseModel):
     t: str = "rick astley - never gonna give you up"
 
 
-"""
-XC
-"""
-
-
-class ValidXCRequest(BaseModel):
-    """
-    - **key**: valid XC API key
-    - **bid**: bot id
-    - **cmd**: bot command
-    - **data**: command data
-    """
-
-    key: str
-    bid: int
-    cmd: str
-    data: Optional[dict]
-
-
 """
 Transcriptions
 """
@@ -211,7 +195,7 @@ class ValidRadioSongRequest(BaseModel):
     song: Optional[str] = None
     artistsong: Optional[str] = None
     alsoSkip: Optional[bool] = False
-    station: str = "main"
+    station: Station = "main"
 
 
 class ValidRadioTypeaheadRequest(BaseModel):
@@ -241,7 +225,7 @@ class ValidRadioNextRequest(BaseModel):
 
     key: str
     skipTo: Optional[str] = None
-    station: str = "main"
+    station: Station = "main"
 
 
 class ValidRadioReshuffleRequest(ValidRadioNextRequest):
@@ -262,7 +246,7 @@ class ValidRadioQueueRequest(BaseModel):
     draw: Optional[int] = 1
     start: Optional[int] = 0
     search: Optional[str] = None
-    station: str = "main"
+    station: Station = "main"
 
 
 class ValidRadioQueueShiftRequest(BaseModel):
@@ -276,7 +260,7 @@ class ValidRadioQueueShiftRequest(BaseModel):
     key: str
     uuid: str
     next: Optional[bool] = False
-    station: str = "main"
+    station: Station = "main"
 
 
 class ValidRadioQueueRemovalRequest(BaseModel):
@@ -288,4 +272,4 @@ class ValidRadioQueueRemovalRequest(BaseModel):
 
     key: str
     uuid: str
-    station: str = "main"
+    station: Station = "main"