You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.7 KiB
70 lines
1.7 KiB
import os |
|
import sys |
|
import asyncio |
|
import aiohttp |
|
import RPi.GPIO as GPIO |
|
import time |
|
from aiohttp import ClientSession, CookieJar |
|
|
|
import settings |
|
|
|
from pyunifiprotect.unifi_protect_server import UpvServer |
|
|
|
import logging |
|
logger = logging.getLogger(__name__) |
|
|
|
RELAY_ON = False |
|
RELAY_OFF = True |
|
|
|
def ring_bell(mac): |
|
try: |
|
doorbell = settings.DOORBELLS[mac] |
|
GPIO.output(doorbell['gpio'], RELAY_ON) |
|
time.sleep(0.5) |
|
GPIO.output(doorbell['gpio'], RELAY_OFF) |
|
except KeyError: |
|
logger.error('Doorbell %s not found!', mac) |
|
|
|
def subscriber(updated): |
|
logger.debug('Subscription: updated=%s', updated) |
|
|
|
for _, data in updated.items(): |
|
if data['event_type'] == 'ring' and data['event_ring_on']: |
|
logger.info('%s: %s is ringing!', data['mac'], data['name']) |
|
ring_bell(data['mac']) |
|
|
|
async def ws_listener(): |
|
session = ClientSession(cookie_jar=CookieJar(unsafe=True)) |
|
|
|
unifiprotect = UpvServer( |
|
session, |
|
settings.UFP_ADDRESS, |
|
settings.UFP_PORT, |
|
settings.UFP_USERNAME, |
|
settings.UFP_PASSWORD, |
|
) |
|
|
|
await unifiprotect.check_unifi_os() |
|
await unifiprotect.update() |
|
|
|
unsub = unifiprotect.subscribe_websocket(subscriber) |
|
|
|
for i in range(15000): |
|
await asyncio.sleep(1) |
|
|
|
await session.close() |
|
unsub() |
|
|
|
def init_relays(): |
|
GPIO.setmode(GPIO.BCM) |
|
GPIO.setwarnings(False) |
|
|
|
for _, doorbell in settings.DOORBELLS.items(): |
|
GPIO.setup(doorbell['gpio'], GPIO.OUT) |
|
GPIO.output(doorbell['gpio'], RELAY_OFF) |
|
|
|
if __name__ == '__main__': |
|
logging.basicConfig(level=logging.INFO) |
|
loop = asyncio.get_event_loop() |
|
loop.run_until_complete(ws_listener()) |
|
loop.close()
|
|
|