Fix for #7

pytchat/tool/extract/duplcheck.py

@@ -70,7 +70,8 @@ def check_duplicate_offset(chatdata):
         if is_duplicate(i,i+1)]
 
 def remove_duplicate_head(blocks):
-    if len(blocks) == 1 : return blocks
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks
 
     def is_duplicate_head(index):
 
@@ -97,7 +98,8 @@ def remove_duplicate_head(blocks):
     return ret
 
 def remove_duplicate_tail(blocks):
-    if len(blocks) == 1 : return blocks    
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks    
 
     def is_duplicate_tail(index):
         if len(blocks[index].chat_data) == 0:
@@ -126,7 +128,8 @@ def remove_overlap(blocks):
     Align the last offset of each block to the first offset 
     of next block (equals `end` offset of each block).
     """
-    if len(blocks) == 1 : return blocks
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks    
 
     for block in blocks:
         if block.is_last:

pytchat/tool/extract/extractor.py

@@ -40,10 +40,11 @@ class Extractor:
         return self
 
     def _set_block_end(self):
-        for i in range(len(self.blocks)-1):
+        if len(self.blocks) > 0:
-            self.blocks[i].end = self.blocks[i+1].first
+            for i in range(len(self.blocks)-1):
-        self.blocks[-1].end = self.duration*1000
+                self.blocks[i].end = self.blocks[i+1].first
-        self.blocks[-1].is_last =True
+            self.blocks[-1].end = self.duration*1000
+            self.blocks[-1].is_last =True
         return self
 
     def _remove_overlap(self):