Viren070/MediaFusion
1
0
Fork 0
mirror of https://github.com/Viren070/MediaFusion.git synced 2025-12-01 23:21:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactoring crud function to apply parental guide filters

Browse Source
This commit is contained in:
mhdzumair
2025-02-01 17:11:12 +05:30
parent 6e9ac6fc69
commit 722a9ddadb
2 changed files with 32 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
db/crud.py
+31 -44
View File
@@ -56,6 +56,35 @@ from utils.validation_helper import (
)
def apply_parental_guide_filters(
user_data: schemas.UserData, match_filter: dict
) -> None:
"""
Helper function to apply parental guide filters to a MongoDB match filter.
"""
# Handle nudity filter
if "Disable" not in user_data.nudity_filter:
if "Unknown" in user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {"$exists": True}
elif user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {
"$nin": user_data.nudity_filter
}
# Handle certification filter
if "Disable" not in user_data.certification_filter:
cert_filters = []
if "Unknown" in user_data.certification_filter:
cert_filters.append(
{"parent_guide_certificates": {"$exists": True, "$ne": []}}
)
filter_values = get_filter_certification_values(user_data)
if filter_values:
cert_filters.append({"parent_guide_certificates": {"$nin": filter_values}})
if cert_filters:
match_filter["$or"] = cert_filters
async def get_meta_list(
user_data: schemas.UserData,
catalog_type: str,
@@ -113,27 +142,7 @@ async def get_meta_list(
# Add genre filter to ignore 'Adult' genre
match_filter["genres"] = {"$nin": ["Adult"]}
# Handle nudity filter
if "Disable" not in user_data.nudity_filter:
if "Unknown" in user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {"$exists": True}
elif user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {
"$nin": user_data.nudity_filter
}
# Handle certification filter
if "Disable" not in user_data.certification_filter:
cert_filters = []
if "Unknown" in user_data.certification_filter:
cert_filters.append(
{"parent_guide_certificates": {"$exists": True, "$ne": []}}
)
filter_values = get_filter_certification_values(user_data)
if filter_values:
cert_filters.append({"parent_guide_certificates": {"$nin": filter_values}})
if cert_filters:
match_filter["$or"] = cert_filters
apply_parental_guide_filters(user_data, match_filter)
# Define the pipeline
pipeline = [
@@ -204,29 +213,7 @@ async def get_mdblist_meta_list(
"total_streams": {"$gt": 0},
}
# Handle nudity filter
if "Disable" not in user_data.nudity_filter:
if "Unknown" in user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {"$exists": True}
elif user_data.nudity_filter:
match_filter["parent_guide_nudity_status"] = {
"$nin": user_data.nudity_filter
}
# Handle certification filter
if "Disable" not in user_data.certification_filter:
cert_filters = []
if "Unknown" in user_data.certification_filter:
cert_filters.append(
{"parent_guide_certificates": {"$exists": True, "$ne": []}}
)
filter_values = get_filter_certification_values(user_data)
if filter_values:
cert_filters.append(
{"parent_guide_certificates": {"$nin": filter_values}}
)
if cert_filters:
match_filter["$or"] = cert_filters
apply_parental_guide_filters(user_data, match_filter)
# Get filtered results with pagination
poster_path = f"{settings.poster_host_url}/poster/{catalog_type}/"
scrapers/mdblist.py
+1 -1
View File
@@ -139,7 +139,7 @@ class MDBListScraper:
limit: int = 25,
genre: Optional[str] = None,
use_filters: bool = True,
) -> List[schemas.Meta] | List[str]:
) -> List[str] | List[schemas.Meta]:
"""
Get items from a MDBList list.
For filtered results (use_filters=True), fetches all available items.