nothere990 commited on
Commit
b4a0a6a
·
1 Parent(s): 949d75b
Files changed (1) hide show
  1. main.py +39 -15
main.py CHANGED
@@ -793,35 +793,59 @@ async def tiktokdl_(link: str):
793
  @app.get("/akeno/xnxxsearch", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
794
  async def xnxx_search(
795
  query: str,
796
- quality: str,
797
- length: Optional[search_filters.Length] = None,
798
- upload_time: Optional[search_filters.UploadTime] = None,
799
- mode: Optional[search_filters.Mode] = None,
800
  limit: Optional[int] = None,
801
  ):
802
  data_dict = {
803
- "720p": search_filters.SearchingQuality.X_720p,
804
- "1080p": search_filters.SearchingQuality.X_1080p_plus
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
805
  }
 
806
  try:
807
- if quality not in data_dict:
808
- return SuccessResponse(
809
- status="False",
810
- randydev={"error": "Invalid quality"}
811
- )
812
- SearchingQuality = data_dict[quality]
 
 
 
 
 
 
813
  search_kwargs = {
814
  "query": query,
815
  "searching_quality": SearchingQuality,
816
  }
817
  if length is not None:
818
- search_kwargs["length"] = length
819
 
820
  if upload_time is not None:
821
- search_kwargs["upload_time"] = upload_time
822
 
823
  if mode is not None:
824
- search_kwargs["mode"] = mode
825
 
826
  if limit is not None:
827
  search_kwargs["limit"] = limit
 
793
  @app.get("/akeno/xnxxsearch", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
794
  async def xnxx_search(
795
  query: str,
796
+ quality: Optional[str] = None,
797
+ upload_time: Optional[str] = None,
798
+ length: Optional[str] = None,
799
+ mode: Optional[str] = None,
800
  limit: Optional[int] = None,
801
  ):
802
  data_dict = {
803
+ "quality": {
804
+ "720p": search_filters.SearchingQuality.X_720p,
805
+ "1080p": search_filters.SearchingQuality.X_1080p_plus
806
+ },
807
+ "upload_time": {
808
+ "year": search_filters.UploadTime.year,
809
+ "month": search_filters.UploadTime.month
810
+ },
811
+ "length": {
812
+ "0-10min": search_filters.Length.X_0_10min,
813
+ "10min+": search_filters.Length.X_10min_plus,
814
+ "10-20min": search_filters.Length.X_10_20min,
815
+ "20min+": search_filters.Length.X_20min_plus
816
+ },
817
+ "mode": {
818
+ "default": search_filters.Mode.default,
819
+ "hits": search_filters.Mode.hits,
820
+ "random": search_filters.Mode.random
821
+ }
822
  }
823
+
824
  try:
825
+ # Validate quality if provided
826
+ if quality is not None:
827
+ if quality not in data_dict["quality"]:
828
+ return SuccessResponse(
829
+ status="False",
830
+ randydev={"error": "Invalid quality"}
831
+ )
832
+ SearchingQuality = data_dict["quality"][quality]
833
+ else:
834
+ SearchingQuality = None
835
+
836
+ # Prepare search parameters
837
  search_kwargs = {
838
  "query": query,
839
  "searching_quality": SearchingQuality,
840
  }
841
  if length is not None:
842
+ search_kwargs["length"] = data_dict["length"][length]
843
 
844
  if upload_time is not None:
845
+ search_kwargs["upload_time"] = data_dict["upload_time"][upload_time]
846
 
847
  if mode is not None:
848
+ search_kwargs["mode"] = data_dict["mode"][mode]
849
 
850
  if limit is not None:
851
  search_kwargs["limit"] = limit