заливка базы

medods/asterisk_ami.py

@@ -0,0 +1,94 @@
+import asyncio
+from datetime import datetime
+import os
+import socket
+import aiofiles
+import call_handler
+import config
+import logging
+
+
+def send_action(s, action):
+    s.send(action.encode())
+    response = ""
+    while True:
+        chunk = s.recv(4096)
+        try:
+            response += chunk.decode()
+        except UnicodeDecodeError:
+            pass
+        if "\r\n\r\n" in response:
+            break
+    response_parts = response.split("\r\n\r\n", 1)
+    body = response_parts[1]
+    body_dict = {
+        line.split(":")[0].strip(): line.split(":")[1].strip()
+        for line in body.split("\n")
+        if len(line.split(":")) > 1
+    }
+    return body_dict
+
+
+async def full_log(event):
+    file_name = f"log/{datetime.now().strftime('%Y-%m-%d')}.log"
+    async with aiofiles.open(file_name, "a") as f:
+        await f.write("\n\n" + datetime.now().strftime("%H:%M:%S") + "\n\n")
+        for key, value in event.items():
+            await f.write(f"{key}: {value}\n")
+
+
+async def ami_listening():
+    # процедура регулярной проверки соединения
+    async def check_connection():
+        while True:
+            try:
+                await asyncio.sleep(3600)
+                s.send(b"Action: Ping\r\n\r\n")
+                await asyncio.sleep(1)
+            except (ConnectionError, ConnectionResetError):
+                if config.DEBUG:
+                    logging.warning("Connection lost. Restarting...")
+                await asyncio.sleep(1)
+                return True  # возвращаемся в начало функции ami_listening
+
+    # проверка соединения каждые 5 секунд
+    conn_check_task = asyncio.create_task(check_connection())
+    conn_check_task.add_done_callback(lambda t: ami_listening() if t.result() else None)
+
+    while True:
+        try:
+            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            logging.info("Connecting to Asterisk...")
+            s.connect((config.AMI_HOST, config.AMI_PORT))
+            break
+        except ConnectionRefusedError:
+            logging.warning("Connection refused. Retrying...")
+            await asyncio.sleep(1)
+
+    logging.info("Connected to Asterisk")
+    logging.info("Logging in...")
+
+    login_action = f"Action: Login\r\nUsername: {config.AMI_USER}\r\nSecret: {config.AMI_PASSWORD}\r\n\r\n"
+    send_action(s, login_action)
+
+    logging.info("Logged in")
+    logging.info("Listening for events...")
+
+    events_action = "Action: Events\r\nEventMask: on\r\n\r\n"
+    send_action(s, events_action)
+
+    callhandler = call_handler.CallHandler()
+
+    try:
+        while True:
+            event = send_action(s, "")
+            if event:
+                await callhandler.handle_event(event)
+                if config.DEBUG:
+                    await full_log(event)
+    except (KeyboardInterrupt, SystemExit):
+        logging.info("Exiting...")
+        s.close()
+    except ConnectionResetError:
+        logging.warning("Connection reset. Restarting...")
+        await ami_listening()