25 lines
565 B
Python
25 lines
565 B
Python
#encoding = utf8
|
|
import re
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
class NLPSplit(ABC):
|
|
@abstractmethod
|
|
def handle(self, message: str):
|
|
pass
|
|
|
|
|
|
class PunctuationSplit(NLPSplit):
|
|
def __init__(self):
|
|
self._pattern = r'[，。、；？！,.!?]'
|
|
|
|
def handle(self, message: str):
|
|
match = re.search(self._pattern, message)
|
|
if match:
|
|
pos = match.start() + 1
|
|
msg = message[:pos]
|
|
msg = msg.strip()
|
|
message = message[pos:]
|
|
return message, msg
|
|
return message, ''
|