- Add a right click to a row to allow for faster and more targettted freshing now

DBEditor/DBEditor.py

@@ -1,13 +1,20 @@
 import sys
 from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QListWidget, QLineEdit, 
                              QLabel, QCheckBox, QPushButton, QFormLayout, QListWidgetItem, QSplitter, QTreeWidget, 
-                             QTreeWidgetItem, QDialog, QDialogButtonBox, QComboBox, QMessageBox, QSpinBox, QTabWidget)
+                             QTreeWidgetItem, QDialog, QDialogButtonBox, QComboBox, QMessageBox, QSpinBox, QMenu)
 from PyQt6.QtCore import Qt, QSize
-from PyQt6.QtGui import QPixmap, QFontMetrics, QColor
+from PyQt6.QtGui import QPixmap, QFontMetrics, QColor, QAction
 import sqlite3
 import json
 import os
 
+# Add the parent directory to the Python path
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+# Now try to import
+from DataGatherers.update_location_information import process_pokemon_for_location_data
+from DataGatherers.cache_manager import CacheManager
+
 class EvolutionEditDialog(QDialog):
     def __init__(self, parent=None, from_pfic=None, to_pfic=None, method=None):
         super().__init__(parent)
@@ -237,16 +244,7 @@ class DBEditor(QMainWindow):
     def init_ui(self):
         central_widget = QWidget()
         self.setCentralWidget(central_widget)
-        main_layout = QVBoxLayout(central_widget)
+        main_layout = QHBoxLayout(central_widget)
-
-        # Create a tab widget
-        self.tab_widget = QTabWidget()
-        main_layout.addWidget(self.tab_widget)
-
-        # Main tab
-        main_tab = QWidget()
-        main_tab_layout = QHBoxLayout(main_tab)
-        self.tab_widget.addTab(main_tab, "Main")
 
         # Left side: Search and List
         left_layout = QVBoxLayout()
@@ -259,13 +257,17 @@ class DBEditor(QMainWindow):
         left_layout.addLayout(search_layout)
 
         self.pokemon_list = QListWidget()
+        self.pokemon_list.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
+        self.pokemon_list.customContextMenuRequested.connect(self.show_pokemon_context_menu)
         self.pokemon_list.currentItemChanged.connect(self.load_pokemon_details)
         left_layout.addWidget(self.pokemon_list)
 
+        # Move the checkbox here, after the pokemon_list
         self.highlight_no_encounters = QCheckBox("Highlight Pokémon without encounters")
         self.highlight_no_encounters.stateChanged.connect(self.toggle_highlight_mode)
         left_layout.addWidget(self.highlight_no_encounters)
 
+            # Add the new checkbox for filtering Home-storable Pokémon
         self.filter_home_storable = QCheckBox("Show only Home-storable Pokémon")
         self.filter_home_storable.stateChanged.connect(self.filter_pokemon_list)
         left_layout.addWidget(self.filter_home_storable)
@@ -336,20 +338,8 @@ class DBEditor(QMainWindow):
         right_layout.addLayout(text_layout)
         right_layout.addLayout(image_layout)
 
-        main_tab_layout.addLayout(left_layout, 1)
+        main_layout.addLayout(left_layout, 1)
-        main_tab_layout.addLayout(right_layout, 1)
+        main_layout.addLayout(right_layout, 1)
-
-        # Advanced tab
-        advanced_tab = QWidget()
-        advanced_tab_layout = QVBoxLayout(advanced_tab)
-        self.tab_widget.addTab(advanced_tab, "Advanced")
-
-        # Add Manage Exclusive Groups button to the Advanced tab
-        manage_groups_button = QPushButton("Manage Exclusive Encounter Groups")
-        manage_groups_button.clicked.connect(self.manage_exclusive_groups)
-        advanced_tab_layout.addWidget(manage_groups_button)
-
-        # Add more advanced features here as needed
 
         self.load_pokemon_list()
 
@@ -1000,6 +990,63 @@ class DBEditor(QMainWindow):
                 self.conn.commit()
                 group_list.takeItem(group_list.row(item))
 
+    def show_pokemon_context_menu(self, position):
+        item = self.pokemon_list.itemAt(position)
+        if item is not None:
+            context_menu = QMenu(self)
+            refresh_action = QAction("Refresh Encounters", self)
+            refresh_action.triggered.connect(lambda: self.refresh_pokemon_encounters(item))
+            context_menu.addAction(refresh_action)
+            context_menu.exec(self.pokemon_list.viewport().mapToGlobal(position))
+
+    def refresh_pokemon_encounters(self, item):
+        pfic = item.data(Qt.ItemDataRole.UserRole)
+        self.cursor.execute('''
+            SELECT name, form_name, national_dex
+            FROM pokemon_forms
+            WHERE PFIC = ?
+        ''', (pfic,))
+        pokemon_data = self.cursor.fetchone()
+        if pokemon_data:
+            name, form, national_dex = pokemon_data
+            
+            # Import the necessary function and classes
+            #from DataGatherers.update_location_information import process_pokemon_for_location_data
+            
+            import json
+
+            # Create a temporary connection for this operation
+            temp_conn = sqlite3.connect('pokemon_forms.db')
+            
+            # Load default forms
+            try:
+                with open('./DataGatherers/DefaultForms.json', 'r') as f:
+                    default_forms = json.load(f)
+            except FileNotFoundError:
+                default_forms = []
+
+            # Create a cache manager instance
+            cache = CacheManager()
+
+            # Delete existing encounters for this Pokémon
+            self.cursor.execute('DELETE FROM encounters WHERE pfic = ?', (pfic,))
+            self.conn.commit()
+
+            # Process the Pokémon data
+            process_pokemon_for_location_data(pfic, name, form, national_dex, default_forms, cache, temp_conn)
+
+            # Close the temporary connection
+            temp_conn.close()
+
+            # Refresh the encounter locations in the UI
+            self.load_encounter_locations(pfic)
+
+            # Update the encounter cache and highlights
+            self.update_encounter_cache()
+            self.update_pokemon_list_highlights()
+
+            print(f"Refreshed encounters for {name} {form if form else ''}")
+
 if __name__ == '__main__':
     app = QApplication(sys.argv)
     editor = DBEditor()

DataGatherers/DetermineOriginGame.py

@@ -13,7 +13,13 @@ from fuzzywuzzy import fuzz
 from fuzzywuzzy import process
 from collections import defaultdict
 
-from cache_manager import CacheManager
+import os
+import sys
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+
+from DataGatherers.cache_manager import CacheManager
 
 # List of all main series Pokémon games in chronological order, with special games first in each generation
 all_games = [

DataGatherers/update_location_information.py

@@ -1,7 +1,12 @@
+import os
+import sys
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 import json
 import sqlite3
-from cache_manager import CacheManager
+from DataGatherers.cache_manager import CacheManager
-from DetermineOriginGame import get_locations_from_bulbapedia
+from DataGatherers.DetermineOriginGame import get_locations_from_bulbapedia
 from bs4 import BeautifulSoup, Tag
 import re
 import time
@@ -204,26 +209,77 @@ def extract_additional_information(s):
     
 def save_encounter(conn, pfic, game, location, days, times, dual_slot, static_encounter, static_encounter_count, extra_text, stars, rods, fishing, starter):
     cursor = conn.cursor()
+    
+    # Convert lists to strings for comparison, except days and times
+    extra_text_str = ' '.join(extra_text) if extra_text else None
+    stars_str = ','.join(sorted(stars)) if stars else None
+    rods_str = ','.join(sorted(rods)) if rods else None
+
     if len(days) > 0:
         for day in days:
+            # Check if an identical record already exists
             cursor.execute('''
-            INSERT OR REPLACE INTO encounters 
+            SELECT COUNT(*) FROM encounters
+            WHERE pfic = ? AND game = ? AND location = ? AND day = ? AND time IS NULL
+            AND dual_slot = ? AND static_encounter = ? AND static_encounter_count = ?
+            AND extra_text = ? AND stars = ? AND rods = ? AND fishing = ? AND starter = ?
+            ''', (pfic, game, location, day, dual_slot, static_encounter, 
+                  static_encounter_count, extra_text_str, stars_str, rods_str, fishing, starter))
+            
+            if cursor.fetchone()[0] == 0:
+                cursor.execute('''
+                INSERT INTO encounters 
                 (pfic, game, location, day, time, dual_slot, static_encounter_count, static_encounter, extra_text, stars, rods, fishing, starter)
                 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-            ''', (pfic, game, location, day, None, dual_slot, static_encounter_count, static_encounter, ' '.join(extra_text), ','.join(stars), ','.join(rods), fishing, starter))
+                ''', (pfic, game, location, day, None, dual_slot, static_encounter_count, 
+                      static_encounter, extra_text_str, stars_str, rods_str, fishing, starter))
+                print(f"New encounter added for {pfic} in {game} at {location} on {day}")
+            else:
+                print(f"Identical encounter already exists for {pfic} in {game} at {location} on {day}")
+
     elif len(times) > 0:
         for time in times:
+            # Check if an identical record already exists
             cursor.execute('''
-            INSERT OR REPLACE INTO encounters 
+            SELECT COUNT(*) FROM encounters
+            WHERE pfic = ? AND game = ? AND location = ? AND day IS NULL AND time = ?
+            AND dual_slot = ? AND static_encounter = ? AND static_encounter_count = ?
+            AND extra_text = ? AND stars = ? AND rods = ? AND fishing = ? AND starter = ?
+            ''', (pfic, game, location, time, dual_slot, static_encounter, 
+                  static_encounter_count, extra_text_str, stars_str, rods_str, fishing, starter))
+            
+            if cursor.fetchone()[0] == 0:
+                cursor.execute('''
+                INSERT INTO encounters 
                 (pfic, game, location, day, time, dual_slot, static_encounter_count, static_encounter, extra_text, stars, rods, fishing, starter)
                 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-            ''', (pfic, game, location, None, time, dual_slot, static_encounter_count, static_encounter, ' '.join(extra_text), ','.join(stars), ','.join(rods), fishing, starter))
+                ''', (pfic, game, location, None, time, dual_slot, static_encounter_count, 
+                      static_encounter, extra_text_str, stars_str, rods_str, fishing, starter))
+                print(f"New encounter added for {pfic} in {game} at {location} at {time}")
+            else:
+                print(f"Identical encounter already exists for {pfic} in {game} at {location} at {time}")
+
     else:
+        # Check if an identical record already exists
+        cursor.execute('''
+        SELECT COUNT(*) FROM encounters
+        WHERE pfic = ? AND game = ? AND location = ? AND day IS NULL AND time IS NULL
+        AND dual_slot = ? AND static_encounter = ? AND static_encounter_count = ?
+        AND extra_text = ? AND stars = ? AND rods = ? AND fishing = ? AND starter = ?
+        ''', (pfic, game, location, dual_slot, static_encounter, 
+              static_encounter_count, extra_text_str, stars_str, rods_str, fishing, starter))
+        thing = cursor.fetchone()
+        if thing[0] == 0:
             cursor.execute('''
-            INSERT OR REPLACE INTO encounters 
+            INSERT INTO encounters 
             (pfic, game, location, day, time, dual_slot, static_encounter_count, static_encounter, extra_text, stars, rods, fishing, starter)
             VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-            ''', (pfic, game, location, None, None, dual_slot, static_encounter_count, static_encounter, ' '.join(extra_text), ','.join(stars), ','.join(rods), fishing, starter))
+            ''', (pfic, game, location, None, None, dual_slot, static_encounter_count, 
+                  static_encounter, extra_text_str, stars_str, rods_str, fishing, starter))
+            print(f"New encounter added for {pfic} in {game} at {location}")
+        else:
+            print(f"Identical encounter already exists for {pfic} in {game} at {location}")
+
     conn.commit()
 
 def compare_forms(a, b):
@@ -241,25 +297,7 @@ def compare_forms(a, b):
 
     return False
 
-if __name__ == "__main__":
+def process_pokemon_for_location_data(pfic, name, form, national_dex, default_forms, cache, conn):
-    cache = CacheManager()
-
-    conn = create_encounters_table()
-    cursor = conn.cursor()
-    cursor.execute('''
-            SELECT pf.PFIC, pf.name, pf.form_name, pf.national_dex
-            FROM pokemon_forms pf
-            ORDER BY pf.national_dex, pf.form_name
-        ''')
-    pokemon_forms = cursor.fetchall()
-
-    try:
-        with open('./DataGatherers/DefaultForms.json', 'r') as f:
-            default_forms = json.load(f)
-    except FileNotFoundError:
-        default_forms = []
-
-    for pfic, name, form, national_dex in pokemon_forms:
     print(f"Processing {name} {form if form else ''}")
 
     if form and name in form:
@@ -281,25 +319,22 @@ if __name__ == "__main__":
         form = None
 
     search_form = form
-        # unrecognized_forms = ["Unown", "Zacian", "Zamazenta"]
-        # if name in unrecognized_forms:
-        #     search_form = None
 
     encounters_to_ignore = ["trade", "time capsule", "unobtainable", "evolve", "tradeversion", "poké transfer", "friend safari", "unavailable", "pokémon home"]
 
     encounter_data = get_locations_from_bulbapedia(name, search_form, cache, default_forms)
     if encounter_data == None:
-            continue
+        return
 
     for encounter in encounter_data:
         if len(encounter_data[encounter]) == 0:
-                continue
+            return
 
         print_encounter = True
 
         for location in encounter_data[encounter]:
             if location == "":
-                    continue
+                return
             test_location = location["location"].strip().lower()    
             
             ignore_location = False
@@ -309,7 +344,7 @@ if __name__ == "__main__":
                     break
             
             if ignore_location:
-                    continue
+                return
             
             if print_encounter:
                 print(f"Found in {encounter}:")
@@ -332,4 +367,26 @@ if __name__ == "__main__":
                 remaining_locations = remaining.replace(" and ", ",").split(",")
                 for remaining_location in remaining_locations:
                     save_encounter(conn, pfic, encounter, remaining_location.strip(), details["days"], details["times"], details["dual_slot"], details["static_encounter"], details["static_encounter_count"], details["extra_text"], details["stars"], details["Rods"], details["Fishing"], details["starter"]  )
+
+if __name__ == "__main__":
+    cache = CacheManager()
+
+    conn = create_encounters_table()
+    cursor = conn.cursor()
+    cursor.execute('''
+            SELECT pf.PFIC, pf.name, pf.form_name, pf.national_dex
+            FROM pokemon_forms pf
+            ORDER BY pf.national_dex, pf.form_name
+        ''')
+    pokemon_forms = cursor.fetchall()
+
+    try:
+        with open('./DataGatherers/DefaultForms.json', 'r') as f:
+            default_forms = json.load(f)
+    except FileNotFoundError:
+        default_forms = []
+
+    for pfic, name, form, national_dex in pokemon_forms:
+        process_pokemon_for_location_data(pfic, name, form, national_dex, default_forms, cache, conn)
+
     conn.close()