Fixed #6. Support filtering by platform

humblebundle_downloader/cli.py

@@ -48,6 +48,12 @@ def cli():
         help="Folder to download all content to",
         required=True,
     )
+    parser_download.add_argument(
+        '-p', '--platform',
+        type=str, nargs='*',
+        help=("Only get content in a platform. Values can be seen in your "
+              "humble bundle's library dropdown. Ex: -p ebook video"),
+    )
     parser_download.add_argument(
         '--progress',
         action='store_true',
@@ -57,12 +63,13 @@ def cli():
     filter_ext.add_argument(
         '-e', '--exclude',
         type=str, nargs='*',
-        help="File extensions to ignore when downloading files. Ex: -e pdf mobi"
+        help=("File extensions to ignore when downloading files. "
+              "Ex: -e pdf mobi"),
     )
     filter_ext.add_argument(
         '-i', '--include',
         type=str, nargs='*',
-        help="Only download files with these extensions. Ex: -i pdf mobi"
+        help="Only download files with these extensions. Ex: -i pdf mobi",
     )
 
     cli_args = parser.parse_args()
@@ -78,5 +85,6 @@ def cli():
             cli_args.library_path,
             progress_bar=cli_args.progress,
             ext_include=cli_args.include,
-            ext_exclude=cli_args.exclude
+            ext_exclude=cli_args.exclude,
+            platform_include=cli_args.platform,
         )

humblebundle_downloader/download_library.py

@@ -18,11 +18,19 @@ def _clean_name(dirty_str):
 
 
 def download_library(cookie_path, library_path, progress_bar=False,
-                     ext_include=None, ext_exclude=None):
+                     ext_include=None, ext_exclude=None, platform_include=None):
     if ext_include is None:
         ext_include = []
+    ext_include = list(map(str.lower, ext_include))
+
     if ext_exclude is None:
         ext_exclude = []
+    ext_exclude = list(map(str.lower, ext_exclude))
+
+    if platform_include is None or 'all' in platform_include:
+        # if all then we do not even need to use this feature
+        platform_include = []
+    platform_include = list(map(str.lower, platform_include))
 
     # Load cookies
     with open(cookie_path, 'r') as f:
@@ -56,7 +64,13 @@ def download_library(cookie_path, library_path, progress_bar=False,
             # Get all types of download for a product
             for download_type in item['downloads']:
                 # Type of product, ebook, videos, etc...
-                # platform = download_type['platform']
+
+                platform = download_type['platform'].lower()
+                if (platform_include and platform not in platform_include):
+                    logger.info("Do not want " + platform
+                                + ", Skipping " + item_title)
+                    continue
+
                 item_folder = os.path.join(
                     library_path, bundle_title, item_title
                 )
@@ -70,14 +84,15 @@ def download_library(cookie_path, library_path, progress_bar=False,
                     try:
                         url = file_type['url']['web']
                     except KeyError:
-                        logger.info("No url found for " + item_title)
+                        logger.info("No url found for " + bundle_title
+                                    + "/" + item_title)
                         continue
 
                     ext = url.split('?')[0].split('.')[-1]
                     file_title = item_title + "." + ext
                     # Only get the file types we care about
-                    if ((ext_include and ext not in ext_include)
-                            or (ext_exclude and ext in ext_exclude)):
+                    if ((ext_include and ext.lower() not in ext_include)
+                            or (ext_exclude and ext.lower() in ext_exclude)):
                         logger.info("Skipping the file " + file_title)
                         continue