Нейрокурятник часть 2: про бота, который постит фотографии

Простейшее работающее решение для информирования о событиях в курятнике в режиме реального времени.
И еще немного болтовни о том, почему надо браться за задачи и изучение нового, даже если у вас недостаточно знаний.

Статьи про нейрокурятник

Мама, мессенджер, бот, куры

Сначала немного про питоновский скрипт из прошлой статьи ^[4]. Скрипт претерпел изменения, например, мы отказались от контуров совсем, поправили некоторые баги, связанные со счетчиком движения, который игнорировался, и связанные со временем аплода. Но это все вещи простые и при желании легко правятся.

Ранее я упоминала про то, как я запускаю этот скрипт. Делала я это при помощи вечных ssh-сессий, но вот проблема: мне нужно уехать и взять с собой ноутбук, на это время я не могу поддерживать ssh-сессию. Очевидное решение — сделать демона в бекграунде, где он будет работать независимо.

Оказалось, что просто отправить процесс в бекграунд недостаточно, при смерти сессии процесс тоже погибает. Для того, чтобы сделать полноценного демона, процесс нужно отвязать от терминала-оунера. Для этого есть несколько способов и вот тут ^[5] есть подборка методов и дополнительная информация.

Вторая проблема заключалась в виртуальной среде. Мне был нужен питон из виртуальной среды для работы скрипта. Решение оказалось безумно простым, достаточно просто прописать путь к виртуальной среде при вызове. В итоге для превращения моего процесса в демона оказалось достаточным запустить его вот так:

/home/sshuser/.virtualenvs/cv/bin/python3 /home/sshuser/chickencoop/pi_surveillance_mod2.py --conf /home/sshuser/chickencoop/conf.json </dev/null &>/dev/null &

Теперь про бота ^[6], который спамит моей маме фотографии кур из курятника в режиме реального времени. Там пока нет никакой аналитики и распознаваний, и бот просто постит фотографии по детекции движения. И там все еще криво висит камера, и вместо куриных голов наблюдаются сплошные хвосты.

Бот неказистый и простой. По событию создания файла на сервере проверяются его расширение и домашняя директория, и в случае успеха, вызывается питоновский скрипт, который делает реквест в API мессенджера (постит фотографию на канал) и пишет логи в обычный текстовый файл (можно парсить ответ и складывать в базу, но нет надобности).

Чтобы сделать такого бота в Telegram, вам нужно:

1. Создать свой канал
2. Создать бота в телеграм при помощи BotFather (это тоже бот в телеграме), вам понадобится токен бота для того, чтобы постить сообщения и фотографии.
3. Сделать скрипт для бота и демонизировать его.

С первыми двумя пунктами все просто, третий тоже прост до скучноты с учетом того, что у телеграма хорошо документированное API и вам понадобится всего один метод.

Для отслеживания события создания файла я использую bash-скрипт, события слушает inotify-tools. Скрипт отправки вызывается не сразу, а после 10-ти секундного ожидания. Сделано это потому, что файл загружается еще некоторое время после создания, а с учетом прекрасной скорости интернета в курятнике, опытным путем установлено, что 10 секунд — самое оно.

#!/bin/sh
DIR="/your/directory"
inotifywait -m -r -e create --format '%w%f' "$DIR" | while read f

do
	PARDIR="$(dirname $f)"
	PARFOLD="$(basename $PARDIR)"
	FBASENAME="$(basename $f)"
	FILEEXT="${FBASENAME##*.}"
	EXT="jpg"
	FOLD="resized"
	if [ $FILEEXT = $EXT ]; then
		if [ $PARFOLD = $FOLD ]; then
			# sleep til file is uploaded
			sleep 10
			python3 /your/directory/bot_request.py --params /your/directory/params_public.json -f $f
		fi
	fi

done

bot_request.py:

import requests
import json
import datetime
import argparse
import warnings

# construct the argument parser and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-p", "--params", required=True,
        help="params json. chat, token")
ap.add_argument("-f", "--file", required=True,
        help="file path")
args = vars(ap.parse_args())

# filter warnings, load the configuration and check if we are going to use server
warnings.filterwarnings("ignore")
params = json.load(open(args["params"]))

# set url and post data
payload = {"chat_id": params["chat"],
	"caption": "motion detected {ts}".format(ts=datetime.datetime.now().strftime("%A-%d-%B-%Y-%I:%M:%S%p"))}
url = "https://api.telegram.org/bot" + params["token"] + "/sendPhoto"
files = {'photo': open(args["file"], "rb")}

# post request
request = requests.post(url, files=files, data=payload)

# log the output
with open("/path/to/your/botlogs/{dt}-{chat}.txt".format(dt=datetime.datetime.now().strftime("%d-%B-%Y"), chat=params["type"]), "a") as logfile:
    logfile.write("n" + request.text)

params_public.json:

{
	"chat": "@your_chat",
	"token": "your_tocken",
	"type": "public"
}

Вот, собственно, и все, максимально простое и работающее решение готово.

Автор: snakers4

Источник ^[3]