47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
from UserActType import UserActType
|
|
from UserAct import UserAct
|
|
import re
|
|
|
|
|
|
class NLU:
|
|
|
|
def __init__(self):
|
|
self.__actParsePatternList = [
|
|
(
|
|
r"(.*)(cze(ś|s)(ć|c)|witaj|dzie(ń|n) dobry)(.*)",
|
|
UserActType.WELCOME_MSG,
|
|
[]
|
|
),
|
|
(
|
|
r"(.*)(Jak (masz na imi(ę|e))|(si(ę|e) nazywasz))(.*)",
|
|
UserActType.REQUEST,
|
|
["name"]
|
|
),
|
|
(
|
|
r"(.*)((ż|z)egnaj)|(do widzenia)|(na razie)(.*)",
|
|
UserActType.BYE,
|
|
[]
|
|
)
|
|
]
|
|
|
|
def parseUserInput(self, text):
|
|
for pattern, actType, actParams in self.__actParsePatternList:
|
|
regex = re.compile(pattern, re.IGNORECASE)
|
|
match = regex.match(text)
|
|
if match:
|
|
return UserAct(actType, actParams)
|
|
return UserAct(UserActType.INVALID)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
nlu = NLU()
|
|
|
|
a = nlu.parseUserInput("czesc")
|
|
b = nlu.parseUserInput("jak masz na imie")
|
|
c = nlu.parseUserInput("zegnaj")
|
|
|
|
print(a)
|
|
print(b)
|
|
print(c)
|