basyx commited on
Commit
208e83d
·
verified ·
1 Parent(s): 7440751

Create validators.py

Browse files
Files changed (1) hide show
  1. utils/validators.py +22 -0
utils/validators.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import UploadFile, HTTPException
2
+ from utils.config import settings
3
+
4
+ ALLOWED_TYPES = [
5
+ "video/mp4",
6
+ "video/mpeg",
7
+ "video/quicktime",
8
+ "audio/mpeg",
9
+ "audio/wav"
10
+ ]
11
+
12
+
13
+ def validate_upload(file: UploadFile):
14
+
15
+ if file.content_type not in ALLOWED_TYPES:
16
+
17
+ raise HTTPException(
18
+ status_code=400,
19
+ detail="Unsupported file type"
20
+ )
21
+
22
+ return True