jl94x4 / ColleXions

# --- Main Function (UPDATED) ---

def main():

    """Main execution loop."""

    logging.info("Starting Collexions Script")

    while True:

        run_start = time.time()

        config = load_config()

        if not all(k in config for k in ['plex_url', 'plex_token', 'pinning_interval']):

             logging.critical("Config essentials missing. Exit."); sys.exit(1)

        pin_interval = config.get('pinning_interval', 60);

        if not isinstance(pin_interval, (int, float)) or pin_interval <= 0: pin_interval = 60

        sleep_sec = pin_interval * 60

        plex = connect_to_plex(config)

        if not plex:

            logging.error(f"Plex connection failed. Retrying in {pin_interval} min.")

        else:

            exclusion_list = config.get('exclusion_list', []);

            if not isinstance(exclusion_list, list): exclusion_list = []

            library_names = config.get('library_names', [])

            if not isinstance(library_names, list): library_names = []

            collections_per_library_config = config.get('number_of_collections_to_pin', {})

            if not isinstance(collections_per_library_config, dict): collections_per_library_config = {}

            selected_collections = load_selected_collections() # Load current history

            current_timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

            # Don't pre-initialize timestamp entry, add only if non-specials are pinned

            newly_pinned_titles_this_run = [] # Track all pins for this run

            # --- Get all special titles ONCE per cycle ---

            all_special_titles = get_all_special_collection_names(config)

            # -------------------------------------------

            for library_name in library_names:

                library_process_start = time.time()

                if not isinstance(library_name, str): logging.warning(f"Skipping invalid library name: {library_name}"); continue

                pin_limit = collections_per_library_config.get(library_name, 0);

                if not isinstance(pin_limit, int) or pin_limit < 0: pin_limit = 0

                if pin_limit == 0: logging.info(f"Skip '{library_name}': 0 limit."); continue

                logging.info(f"Processing '{library_name}' (Limit: {pin_limit})")

                unpin_collections(plex, [library_name], exclusion_list)

                active_specials = get_active_special_collections(config)

                all_colls = get_collections_from_all_libraries(plex, [library_name])

                if not all_colls: logging.info(f"No collections in '{library_name}'."); continue

                colls_to_pin = filter_collections(config, all_colls, active_specials, pin_limit, library_name, selected_collections)

                if colls_to_pin:

                    pin_collections(colls_to_pin, config)

                    # Add ALL pinned titles to the list for THIS run's tracking

                    newly_pinned_titles_this_run.extend([c.title for c in colls_to_pin if hasattr(c, 'title')])

                else: logging.info(f"No collections selected for '{library_name}'.")

                logging.info(f"Finished '{library_name}' in {time.time() - library_process_start:.2f}s.")

            # --- Modified History Update Logic ---

            if newly_pinned_titles_this_run:

                 unique_new_pins_all = set(newly_pinned_titles_this_run)

                 # Filter out special collections before saving to history

                 non_special_pins_for_history = {

                     title for title in unique_new_pins_all

                     if title not in all_special_titles # Use the set fetched earlier

                 }

                 # Only update history file if there were non-special items pinned

                 if non_special_pins_for_history:

                      history_entry = sorted(list(non_special_pins_for_history))

                      selected_collections[current_timestamp] = history_entry

                      save_selected_collections(selected_collections)

                      logging.info(f"Updated history for {current_timestamp} with {len(history_entry)} non-special items.")

                      if len(unique_new_pins_all) > len(non_special_pins_for_history):

                          logging.info(f"Note: {len(unique_new_pins_all) - len(non_special_pins_for_history)} special collection(s) were pinned but not added to recency history.")

                 else:

                      logging.info("Only special collections were pinned this cycle. History not updated for recency blocking.")

            else:

                 logging.info("Nothing pinned this cycle, history not updated.")

            # --- End Modified History Update ---

        run_end = time.time()

        logging.info(f"Cycle finished in {run_end - run_start:.2f} seconds.")

        logging.info(f"Sleeping for {pin_interval} minutes...")

        try: time.sleep(sleep_sec)

        except KeyboardInterrupt: logging.info("Script interrupted. Exiting."); break

# --- Script Entry Point ---

if __name__ == "__main__":

# --- Main Function (UPDATED) ---

def main():

    """Main execution loop."""

    logging.info("Starting Collexions Script")

    while True:

        run_start = time.time()

        config = load_config()

        if not all(k in config for k in ['plex_url', 'plex_token', 'pinning_interval']):

             logging.critical("Config essentials missing. Exit."); sys.exit(1)

        pin_interval = config.get('pinning_interval', 60);

        if not isinstance(pin_interval, (int, float)) or pin_interval <= 0: pin_interval = 60

        sleep_sec = pin_interval * 60

        plex = connect_to_plex(config)

        if not plex:

            logging.error(f"Plex connection failed. Retrying in {pin_interval} min.")

        else:

            exclusion_list = config.get('exclusion_list', []);

            if not isinstance(exclusion_list, list): exclusion_list = []

            library_names = config.get('library_names', [])

            if not isinstance(library_names, list): library_names = []

            collections_per_library_config = config.get('number_of_collections_to_pin', {})

            if not isinstance(collections_per_library_config, dict): collections_per_library_config = {}

            selected_collections = load_selected_collections() # Load current history

            current_timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

            # Don't pre-initialize timestamp entry, add only if non-specials are pinned

            newly_pinned_titles_this_run = [] # Track all pins for this run

            # --- Get all special titles ONCE per cycle ---

            all_special_titles = get_all_special_collection_names(config)

            # -------------------------------------------

            for library_name in library_names:

                library_process_start = time.time()

                if not isinstance(library_name, str): logging.warning(f"Skipping invalid library name: {library_name}"); continue

                pin_limit = collections_per_library_config.get(library_name, 0);

                if not isinstance(pin_limit, int) or pin_limit < 0: pin_limit = 0

                if pin_limit == 0: logging.info(f"Skip '{library_name}': 0 limit."); continue

                logging.info(f"Processing '{library_name}' (Limit: {pin_limit})")

                unpin_collections(plex, [library_name], exclusion_list)

                active_specials = get_active_special_collections(config)

                all_colls = get_collections_from_all_libraries(plex, [library_name])

                if not all_colls: logging.info(f"No collections in '{library_name}'."); continue

                colls_to_pin = filter_collections(config, all_colls, active_specials, pin_limit, library_name, selected_collections)

                if colls_to_pin:

                    pin_collections(colls_to_pin, config)

                    # Add ALL pinned titles to the list for THIS run's tracking

                    newly_pinned_titles_this_run.extend([c.title for c in colls_to_pin if hasattr(c, 'title')])

                else: logging.info(f"No collections selected for '{library_name}'.")

                logging.info(f"Finished '{library_name}' in {time.time() - library_process_start:.2f}s.")

            # --- Modified History Update Logic ---

            if newly_pinned_titles_this_run:

                 unique_new_pins_all = set(newly_pinned_titles_this_run)

                 # Filter out special collections before saving to history

                 non_special_pins_for_history = {

                     title for title in unique_new_pins_all

                     if title not in all_special_titles # Use the set fetched earlier

                 }

                 # Only update history file if there were non-special items pinned

                 if non_special_pins_for_history:

                      history_entry = sorted(list(non_special_pins_for_history))

                      selected_collections[current_timestamp] = history_entry

                      save_selected_collections(selected_collections)

                      logging.info(f"Updated history for {current_timestamp} with {len(history_entry)} non-special items.")

                      if len(unique_new_pins_all) > len(non_special_pins_for_history):

                          logging.info(f"Note: {len(unique_new_pins_all) - len(non_special_pins_for_history)} special collection(s) were pinned but not added to recency history.")

                 else:

                      logging.info("Only special collections were pinned this cycle. History not updated for recency blocking.")

            else:

                 logging.info("Nothing pinned this cycle, history not updated.")

            # --- End Modified History Update ---

        run_end = time.time()

        logging.info(f"Cycle finished in {run_end - run_start:.2f} seconds.")

        logging.info(f"Sleeping for {pin_interval} minutes...")

        try: time.sleep(sleep_sec)

        except KeyboardInterrupt: logging.info("Script interrupted. Exiting."); break

# --- Script Entry Point ---

if __name__ == "__main__":