From a54bff144b65ec4e3d55b3fb7d0487076412ff28 Mon Sep 17 00:00:00 2001
From: djib <me@djib.fr>
Date: Wed, 9 Oct 2019 00:01:29 +0200
Subject: [PATCH] Filter movies on available channels

---
 FreeboxMoviePlanner.py | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/FreeboxMoviePlanner.py b/FreeboxMoviePlanner.py
index e0b39c0..e2422a9 100755
--- a/FreeboxMoviePlanner.py
+++ b/FreeboxMoviePlanner.py
@@ -43,11 +43,15 @@ class Movie:
         )
 
     def __repr__(self):
-        return 'Movie <{}({})>'.format(self.title, self.rating)
+        return "Movie <{} (Ch:{} – R:{})>".format(
+            self.title,
+            self.channel,
+            self.rating
+        )
 
 
 class TVGuideScraper:
-    TV_GUIDE_URL = 'https://www.programme-television.org/{}?bouquet=tnt'
+    TV_GUIDE_URL = 'https://www.programme-television.org/{}?bouquet=free'
 
     @staticmethod
     def getMovies(day=''):
@@ -100,6 +104,10 @@ class FreeboxMoviePlanner:
             token=self.config['freebox-session-token']
         )
         self.getListOfAvailableChannels()
+        self.scapeAllMovies()
+        # self.findMoviesOnTMDB()
+        # self.filterBadRatings()
+        self.filterUnavailableChannels()
 
     def __repr__(self):
         result = 'FreeboxMoviePlanner <Movies:\n'
@@ -151,7 +159,18 @@ class FreeboxMoviePlanner:
                     .format(movie.tmdb_id, self.config['tmdb-language'])
 
     def filterBadRatings(self):
-        self.movies = [movie for movie in self.movies if movie.good]
+        logging.info('Dropping bad ratings: {}'.format(
+            [m for m in self.movies if not m.good]
+        ))
+        self.movies = [m for m in self.movies if m.good]
+        logging.info('Kept {}'.format(self.movies))
+
+    def filterUnavailableChannels(self):
+        logging.info('Dropping unavailable channels: {}'.format(
+            [m for m in self.movies if m.channel not in self.channels]
+        ))
+        self.movies = [m for m in self.movies if m.channel in self.channels]
+        logging.info('Kept {}'.format(self.movies))
 
     def _findMovieOnTMDB(self, movie):
         logging.info("Searching for '{}' on TMDB".format(movie))
@@ -173,7 +192,4 @@ if __name__ == '__main__':
         format=' %(asctime)s - %(levelname)s - %(message)s'
     )
     fmp = FreeboxMoviePlanner()
-    # fmp.scapeAllMovies()
-    # fmp.findMoviesOnTMDB()
-    # fmp.filterBadRatings()
     # fmp.printAllMovies()