]> gitweb.michael.orlitzky.com - dead/whatever-dl.git/commitdiff
Modified the default get_video_filename() so that it removes URL parameters.
authorMichael Orlitzky <michael@orlitzky.com>
Thu, 1 Apr 2010 22:29:57 +0000 (18:29 -0400)
committerMichael Orlitzky <michael@orlitzky.com>
Thu, 1 Apr 2010 22:29:57 +0000 (18:29 -0400)
src/website.rb

index 75f5aa8907d4c6c8e3fa237172a3e02a1f80f71c..bfa1b3d618a04163856fee297dd65ce4b0e380b8 100644 (file)
@@ -114,8 +114,15 @@ class Website
   # of the video URL, but in some cases, subclasses will want
   # to override this behavior.
   def get_video_filename()
-    # Use whatever comes after the final front slash.    
-    return get_video_url().split('/').pop()
+    # Use whatever comes after the final front slash.
+    file_and_params = get_video_url().split('/').pop()
+
+    # Unless it contains URL parameters. We don't want those.
+    return file_and_params unless file_and_params.include?('?')
+    
+    # There must be some parameters. Strip them off.
+    param_start_idx = file_and_params.index('?')
+    return file_and_params[0...(param_start_idx)]
   end
   
 end