Support .part file names. Fixes #4 and #6

BenoitZugmeyer · Jul 16, 2013 · ec18e3c · ec18e3c
1 parent 56c4699
commit ec18e3c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/addic7ed/__init__.py b/addic7ed/__init__.py
@@ -239,8 +239,11 @@ def launch(self):
         args = self.args
         filename = self.filename
 
-        if os.path.isfile(filename) and not filename.endswith('.srt'):
-            filename = remove_extension(filename) + '.srt'
+        if os.path.isfile(filename):
+            if filename.endswith('.part'):
+                filename = remove_extension(filename)
+            if not filename.endswith('.srt'):
+                filename = remove_extension(filename) + '.srt'
 
         print 'Target SRT file:', filename
         ignore = False