dylan/crosspost
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

refactor: move query parameter extraction function out of flask app

Browse Source
This commit is contained in:
Dylan De Faoite
2026-03-02 18:29:09 +00:00
parent 37cb2c9ff4
commit 5ea71023b5
3 changed files with 69 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server/stat_gen.py
View File

@@ -55,13 +55,15 @@ class StatGen:
if search_query:
mask = (
filtered_df["content"].str.contains(search_query, case=False, na=False)
| filtered_df["author"]
.str.contains(search_query, case=False, na=False)
.fillna(False)
| filtered_df["title"]
.str.contains(search_query, case=False, na=False, regex=False)
.fillna(False)
| filtered_df["author"].str.contains(search_query, case=False, na=False)
)
# Only include title if the column exists
if "title" in filtered_df.columns:
mask = mask | filtered_df["title"].str.contains(
search_query, case=False, na=False, regex=False
)
filtered_df = filtered_df[mask]
if start_date_filter:
@@ -76,6 +78,8 @@ class StatGen:
return filtered_df
## Public Methods
def filter_dataset(self, df: pd.DataFrame, filters: dict | None = None) -> dict:
return self._prepare_filtered_df(df, filters).to_dict(orient="records")
def get_time_analysis(self, df: pd.DataFrame, filters: dict | None = None) -> dict:
filtered_df = self._prepare_filtered_df(df, filters)