mirror of
https://github.com/nichkara/InfinitumBotty.git
synced 2026-06-11 06:36:24 +02:00
Context 77
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
import csv
|
|
import re
|
|
|
|
from FaustBot.Communication.Connection import Connection
|
|
from FaustBot.Modules.PrivMsgObserverPrototype import PrivMsgObserverPrototype
|
|
|
|
|
|
class ICDObserver(PrivMsgObserverPrototype):
|
|
with open("care_icd10_de.csv", "r", encoding="utf8") as icd10_codes:
|
|
icd10_dict = {
|
|
row[0]: row[1]
|
|
for row in csv.reader(icd10_codes, delimiter=";", quotechar='"')
|
|
}
|
|
|
|
@staticmethod
|
|
def cmd():
|
|
return None
|
|
|
|
@staticmethod
|
|
def help():
|
|
return None
|
|
|
|
def update_on_priv_msg(self, data, connection: Connection):
|
|
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)
|