P 태그 이외의 모든 HTML 태그를 제거하도록 개선

Author: g6123

.gitignore

@@ -1,4 +1,4 @@
 __pycache__
 build
 *.pyc
-*.old
+*.old

examples/smi2vtt.py

@@ -8,19 +8,8 @@ def throw(code):
 	print(error.msg, file=stderr)
 	exit(error.code)
 
-print('변환을 시작합니다.')
-
-try:
-	smi_filepath = argv[1].rsplit('.', 1)
-except IndexError:
-	throw(-1)
-
-if len(smi_filepath) < 2:
-	vtt_filepath = argv[1]+'.vtt'
-else:
-	vtt_filepath = smi_filepath[0]+'.vtt'
-
 smi_filepath = argv[1]
+vtt_filepath = smi_filepath.rsplit('.', 1)[0]+'.vtt'
 
 try:
 	smi = SmiFile(smi_filepath)
@@ -36,9 +25,6 @@ def throw(code):
 	vtt_file.write(vtt)
 	vtt_file.close()
 except:
-	print('변환한 자막 파일을 저장할 수 없습니다.', file=stderr)
 	exit(-5)
 
-print('성공적으로 변환했습니다.')
-print('  WebVTT 파일 : '+vtt_filepath)
-exit(1)
+exit(1)

src/__init__.py

@@ -53,7 +53,10 @@ def parse_p(item):
 			lang = search(item, '<p(.+)class=([a-z]+)').group(2)
 
 			content = item[search(item, '<p(.+)>').end():]
-			content = re.sub('<br ?/?>', '\n', content.replace('\n', ''), flags=re.I).strip()
+			content = content.replace('\n', '')
+			content = re.sub('<br ?/?>', '\n', content, flags=re.I)
+			content = re.sub('<.*?>','', content)
+			content = content.strip()
 
 			return [lang, content]
 
@@ -80,7 +83,7 @@ def convert(self, target, lang='ENCC'):
 		result = ''
 
 		if target == 'vtt':
-			result += 'WEBVTT FILE'
+			result += 'WEBVTT'
 
 			loop_index = 0
 			sub_index = 1
@@ -111,4 +114,4 @@ def convert(self, target, lang='ENCC'):
 		else:
 			raise ConversionError(-4)
 
-		return result
+		return result