nichkara/ninachloe
1
0
Fork 0
mirror of https://github.com/nichkara/InfinitumBotty.git synced 2026-06-10 22:26:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

require .icd to resolve ICD10 codes and limit answer to at most 5

Browse Source
This commit is contained in:
Context 77
2024-12-14 13:43:03 +01:00
parent f8f83e442f
commit 0abca6fb95
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
FaustBot/Modules/CustomUserModules/ICDObserver.py
View File

@@ -21,17 +21,18 @@ class ICDObserver(PrivMsgObserverPrototype):
return None
def update_on_priv_msg(self, data, connection: Connection):
if data["channel"] == connection.details.get_channel():
regex = r"\b(\w\d{2}\.?\d?\d?)\b"
codes = re.findall(regex, data["message"])
for code in codes:
code = code.capitalize()
text = self.icd10_dict.get(code, False)
if text == False:
if "." in code:
code += "-"
else:
code += ".-"
text = self.icd10_dict.get(code, False)
if text:
connection.send_back(f"{code} - {text}", data)
if data["message"].startswith(".icd "):
if data["channel"] == connection.details.get_channel():
regex = r"\b(\w\d{2}\.?\d?\d?)\b"
codes = re.findall(regex, data["message"])
for code in codes[:5]:
code = code.capitalize()
text = self.icd10_dict.get(code, False)
if text == False:
if "." in code:
code += "-"
else:
code += ".-"
text = self.icd10_dict.get(code, False)
if text:
connection.send_back(f"{code} - {text}", data)