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

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

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

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

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

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

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

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

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

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

Бот неказистый и простой. По событию создания файла на сервере проверяются его расширение и домашняя директория, и в случае успеха, вызывается питоновский скрипт, который делает реквест в 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

Источник