improving Search functionality
This commit is contained in:
parent
b5314497da
commit
bee297bf19
1
.gitignore
vendored
1
.gitignore
vendored
@ -4,3 +4,4 @@ __pycache__/
|
|||||||
anime_tracker.log
|
anime_tracker.log
|
||||||
todo.txt
|
todo.txt
|
||||||
*.csv
|
*.csv
|
||||||
|
anime_backlog.db.bk
|
||||||
|
41
backend.py
41
backend.py
@ -30,28 +30,10 @@ class AnimeBackend:
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Error creating table: {e}")
|
logging.error(f"Error creating table: {e}")
|
||||||
|
|
||||||
def get_pre_2010_entries(self, status_filter=None, type_filter=None, search=None, year_filter=None):
|
def get_pre_2010_entries(self):
|
||||||
try:
|
try:
|
||||||
cursor = self.db.cursor()
|
cursor = self.db.cursor()
|
||||||
sql = "SELECT * FROM anime WHERE year < 2010"
|
cursor.execute("SELECT * FROM anime WHERE year < 2010 ORDER BY year DESC, name")
|
||||||
add_where = []
|
|
||||||
params = []
|
|
||||||
if year_filter is not None:
|
|
||||||
add_where.append("year = ?")
|
|
||||||
params.append(year_filter)
|
|
||||||
if status_filter is not None:
|
|
||||||
add_where.append("status = ?")
|
|
||||||
params.append(status_filter)
|
|
||||||
if type_filter is not None:
|
|
||||||
add_where.append("type = ?")
|
|
||||||
params.append(type_filter)
|
|
||||||
if search is not None:
|
|
||||||
add_where.append("(name LIKE ? OR comment LIKE ?)")
|
|
||||||
params.append(f"%{search}%")
|
|
||||||
params.append(f"%{search}%")
|
|
||||||
where_clause = " AND " + " AND ".join(add_where) if add_where else ""
|
|
||||||
sql += where_clause + " ORDER BY year DESC, name"
|
|
||||||
cursor.execute(sql, params)
|
|
||||||
return cursor.fetchall()
|
return cursor.fetchall()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Error getting pre-2010 entries: {e}")
|
logging.error(f"Error getting pre-2010 entries: {e}")
|
||||||
@ -66,25 +48,10 @@ class AnimeBackend:
|
|||||||
logging.error(f"Error getting years: {e}")
|
logging.error(f"Error getting years: {e}")
|
||||||
return []
|
return []
|
||||||
|
|
||||||
def get_entries_for_season(self, year, season, status_filter=None, type_filter=None, search=None):
|
def get_entries_for_season(self, year, season):
|
||||||
try:
|
try:
|
||||||
cursor = self.db.cursor()
|
cursor = self.db.cursor()
|
||||||
sql = "SELECT * FROM anime WHERE year = ? AND season = ?"
|
cursor.execute("SELECT * FROM anime WHERE year = ? AND season = ? ORDER BY name", (year, season))
|
||||||
params = [year, season]
|
|
||||||
add_where = []
|
|
||||||
if status_filter is not None:
|
|
||||||
add_where.append("status = ?")
|
|
||||||
params.append(status_filter)
|
|
||||||
if type_filter is not None:
|
|
||||||
add_where.append("type = ?")
|
|
||||||
params.append(type_filter)
|
|
||||||
if search is not None:
|
|
||||||
add_where.append("(name LIKE ? OR comment LIKE ?)")
|
|
||||||
params.append(f"%{search}%")
|
|
||||||
params.append(f"%{search}%")
|
|
||||||
where_clause = " AND " + " AND ".join(add_where) if add_where else ""
|
|
||||||
sql += where_clause + " ORDER BY name"
|
|
||||||
cursor.execute(sql, params)
|
|
||||||
return cursor.fetchall()
|
return cursor.fetchall()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Error getting entries for season {season} in year {year}: {e}")
|
logging.error(f"Error getting entries for season {season} in year {year}: {e}")
|
||||||
|
80
frontend.py
80
frontend.py
@ -153,45 +153,18 @@ class AnimeTracker(QMainWindow):
|
|||||||
central_layout = QVBoxLayout(self.central_widget)
|
central_layout = QVBoxLayout(self.central_widget)
|
||||||
self.filter_bar = QWidget()
|
self.filter_bar = QWidget()
|
||||||
filter_layout = QHBoxLayout(self.filter_bar)
|
filter_layout = QHBoxLayout(self.filter_bar)
|
||||||
label_year = QLabel("Year:")
|
|
||||||
self.year_spin = QSpinBox()
|
|
||||||
self.year_spin.setRange(0, 2100)
|
|
||||||
self.year_spin.setValue(0)
|
|
||||||
self.year_spin.setSpecialValueText("All")
|
|
||||||
filter_layout.addWidget(label_year)
|
|
||||||
filter_layout.addWidget(self.year_spin)
|
|
||||||
label_season = QLabel("Season:")
|
|
||||||
self.season_combo = QComboBox()
|
|
||||||
self.season_combo.addItems(['All', 'winter', 'spring', 'summer', 'fall', 'Other'])
|
|
||||||
filter_layout.addWidget(label_season)
|
|
||||||
filter_layout.addWidget(self.season_combo)
|
|
||||||
label_status = QLabel("Status:")
|
|
||||||
self.status_combo = QComboBox()
|
|
||||||
self.status_combo.addItems(['All', 'unwatched', 'watching', 'completed'])
|
|
||||||
filter_layout.addWidget(label_status)
|
|
||||||
filter_layout.addWidget(self.status_combo)
|
|
||||||
label_type = QLabel("Type:")
|
|
||||||
self.type_combo = QComboBox()
|
|
||||||
self.type_combo.addItems(['All', 'TV', 'Movie', 'OVA', 'Special', 'Short TV', 'Other'])
|
|
||||||
filter_layout.addWidget(label_type)
|
|
||||||
filter_layout.addWidget(self.type_combo)
|
|
||||||
label_search = QLabel("Search:")
|
label_search = QLabel("Search:")
|
||||||
self.search_edit = QLineEdit()
|
self.search_edit = QLineEdit()
|
||||||
self.search_edit.setPlaceholderText("Search name or comment")
|
self.search_edit.setPlaceholderText("Search name")
|
||||||
|
self.search_edit.textChanged.connect(self.filter_tables)
|
||||||
filter_layout.addWidget(label_search)
|
filter_layout.addWidget(label_search)
|
||||||
filter_layout.addWidget(self.search_edit)
|
filter_layout.addWidget(self.search_edit)
|
||||||
apply_btn = QPushButton("Apply")
|
|
||||||
apply_btn.clicked.connect(self.apply_filters)
|
|
||||||
filter_layout.addWidget(apply_btn)
|
|
||||||
central_layout.addWidget(self.filter_bar)
|
central_layout.addWidget(self.filter_bar)
|
||||||
central_layout.addWidget(self.tab_widget)
|
central_layout.addWidget(self.tab_widget)
|
||||||
self.setCentralWidget(self.central_widget)
|
self.setCentralWidget(self.central_widget)
|
||||||
self.year_filter = None
|
self.search_text = ''
|
||||||
self.season_filter = None
|
|
||||||
self.status_filter = None
|
|
||||||
self.type_filter = None
|
|
||||||
self.search_text = None
|
|
||||||
self.table_scale = self.settings.value("tableScale", 1.0, type=float)
|
self.table_scale = self.settings.value("tableScale", 1.0, type=float)
|
||||||
|
self.tables = []
|
||||||
self.create_menu()
|
self.create_menu()
|
||||||
self.load_tabs()
|
self.load_tabs()
|
||||||
self.restoreGeometry(self.settings.value("geometry", self.saveGeometry()))
|
self.restoreGeometry(self.settings.value("geometry", self.saveGeometry()))
|
||||||
@ -200,18 +173,14 @@ class AnimeTracker(QMainWindow):
|
|||||||
self.set_current_tab_by_identifier(last_tab)
|
self.set_current_tab_by_identifier(last_tab)
|
||||||
self.tab_widget.setFocus()
|
self.tab_widget.setFocus()
|
||||||
|
|
||||||
def apply_filters(self):
|
def filter_tables(self, text):
|
||||||
year_val = self.year_spin.value()
|
self.search_text = text.strip().lower()
|
||||||
self.year_filter = year_val if year_val > 0 else None
|
for table in self.tables:
|
||||||
season_val = self.season_combo.currentText()
|
for row in range(table.rowCount()):
|
||||||
self.season_filter = None if season_val == 'All' else ('' if season_val == 'Other' else season_val)
|
name_col = 2 if table.is_pre else 1
|
||||||
status_val = self.status_combo.currentText()
|
name_text = table.cellWidget(row, name_col).text()
|
||||||
self.status_filter = None if status_val == 'All' else status_val
|
clean_name = re.sub(r'<[^>]+>', '', name_text).lower()
|
||||||
type_val = self.type_combo.currentText()
|
table.setRowHidden(row, self.search_text not in clean_name if self.search_text else False)
|
||||||
self.type_filter = None if type_val == 'All' else type_val
|
|
||||||
search_val = self.search_edit.text().strip()
|
|
||||||
self.search_text = search_val if search_val else None
|
|
||||||
self.load_tabs()
|
|
||||||
|
|
||||||
def closeEvent(self, event):
|
def closeEvent(self, event):
|
||||||
self.settings.setValue("geometry", self.saveGeometry())
|
self.settings.setValue("geometry", self.saveGeometry())
|
||||||
@ -281,19 +250,17 @@ class AnimeTracker(QMainWindow):
|
|||||||
|
|
||||||
def load_tabs(self):
|
def load_tabs(self):
|
||||||
self.tab_widget.clear()
|
self.tab_widget.clear()
|
||||||
show_pre = True
|
self.tables = []
|
||||||
if self.season_filter is not None and self.season_filter != '':
|
# Pre-2010 tab
|
||||||
show_pre = False
|
pre_entries = self.backend.get_pre_2010_entries()
|
||||||
if self.year_filter is not None and self.year_filter >= 2010:
|
|
||||||
show_pre = False
|
|
||||||
if show_pre:
|
|
||||||
pre_entries = self.backend.get_pre_2010_entries(self.status_filter, self.type_filter, self.search_text, self.year_filter if self.year_filter and self.year_filter < 2010 else None)
|
|
||||||
pre_tab = QScrollArea()
|
pre_tab = QScrollArea()
|
||||||
pre_tab.setWidgetResizable(True)
|
pre_tab.setWidgetResizable(True)
|
||||||
pre_content = QWidget()
|
pre_content = QWidget()
|
||||||
pre_layout = QVBoxLayout(pre_content)
|
pre_layout = QVBoxLayout(pre_content)
|
||||||
if pre_entries:
|
if pre_entries:
|
||||||
table = CustomTableWidget(self, is_pre=True)
|
table = CustomTableWidget(self, is_pre=True)
|
||||||
|
table.is_pre = True
|
||||||
|
self.tables.append(table)
|
||||||
table.setRowCount(len(pre_entries))
|
table.setRowCount(len(pre_entries))
|
||||||
table.setColumnCount(7)
|
table.setColumnCount(7)
|
||||||
headers = ['ID', 'Year', 'Name', 'Type', 'Status', 'Comment', 'Actions']
|
headers = ['ID', 'Year', 'Name', 'Type', 'Status', 'Comment', 'Actions']
|
||||||
@ -416,8 +383,6 @@ class AnimeTracker(QMainWindow):
|
|||||||
self.tab_widget.tabBar().setTabTextColor(index, QColor('gray'))
|
self.tab_widget.tabBar().setTabTextColor(index, QColor('gray'))
|
||||||
# Years >= 2010
|
# Years >= 2010
|
||||||
years = self.backend.get_years()
|
years = self.backend.get_years()
|
||||||
if self.year_filter is not None:
|
|
||||||
years = [self.year_filter] if self.year_filter in years else []
|
|
||||||
for year in years:
|
for year in years:
|
||||||
year_tab = QScrollArea()
|
year_tab = QScrollArea()
|
||||||
year_tab.setWidgetResizable(True)
|
year_tab.setWidgetResizable(True)
|
||||||
@ -426,12 +391,7 @@ class AnimeTracker(QMainWindow):
|
|||||||
total_entries = 0
|
total_entries = 0
|
||||||
comp_entries = 0
|
comp_entries = 0
|
||||||
for season in ['winter', 'spring', 'summer', 'fall', '']:
|
for season in ['winter', 'spring', 'summer', 'fall', '']:
|
||||||
show_section = True
|
entries = self.backend.get_entries_for_season(year, season)
|
||||||
if self.season_filter is not None:
|
|
||||||
if season != self.season_filter:
|
|
||||||
show_section = False
|
|
||||||
if show_section:
|
|
||||||
entries = self.backend.get_entries_for_season(year, season, self.status_filter, self.type_filter, self.search_text)
|
|
||||||
if entries:
|
if entries:
|
||||||
s_name = season.capitalize() if season else 'Other'
|
s_name = season.capitalize() if season else 'Other'
|
||||||
label = QLabel(s_name)
|
label = QLabel(s_name)
|
||||||
@ -441,6 +401,8 @@ class AnimeTracker(QMainWindow):
|
|||||||
label.setFont(season_font)
|
label.setFont(season_font)
|
||||||
layout.addWidget(label)
|
layout.addWidget(label)
|
||||||
table = CustomTableWidget(self, is_pre=False)
|
table = CustomTableWidget(self, is_pre=False)
|
||||||
|
table.is_pre = False
|
||||||
|
self.tables.append(table)
|
||||||
table.setRowCount(len(entries))
|
table.setRowCount(len(entries))
|
||||||
table.setColumnCount(6)
|
table.setColumnCount(6)
|
||||||
headers = ['ID', 'Name', 'Type', 'Status', 'Comment', 'Actions']
|
headers = ['ID', 'Name', 'Type', 'Status', 'Comment', 'Actions']
|
||||||
@ -545,7 +507,7 @@ class AnimeTracker(QMainWindow):
|
|||||||
total_entries += len(entries)
|
total_entries += len(entries)
|
||||||
comp_entries += sum(1 for e in entries if e[4] == 'completed')
|
comp_entries += sum(1 for e in entries if e[4] == 'completed')
|
||||||
year_tab.setWidget(content)
|
year_tab.setWidget(content)
|
||||||
if total_entries > 0 or (self.year_filter == year):
|
if total_entries > 0:
|
||||||
perc = (comp_entries / total_entries * 100) if total_entries > 0 else 0
|
perc = (comp_entries / total_entries * 100) if total_entries > 0 else 0
|
||||||
tab_text = f"{year} ({perc:.0f}%)"
|
tab_text = f"{year} ({perc:.0f}%)"
|
||||||
completed = (comp_entries == total_entries)
|
completed = (comp_entries == total_entries)
|
||||||
|
Loading…
Reference in New Issue
Block a user