104 lines
3.1 KiB
Python
104 lines
3.1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# -*- coding: UTF8 -*-
|
||
|
|
|
||
|
|
# This program is free software: you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# This program is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
"""
|
||
|
|
Take a picture with the Pi Camera module and publish the path
|
||
|
|
and name of the file on a MQTT broker
|
||
|
|
|
||
|
|
@author: arofarn
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
from time import sleep
|
||
|
|
from datetime import datetime
|
||
|
|
import picamera
|
||
|
|
import paho.mqtt.publish as mqtt
|
||
|
|
from cameteo_conf import CONFIG_MQTT, CONFIG_PI_CAMERA, PICTURES_EXTENSIONS
|
||
|
|
|
||
|
|
MQTT_CLIENT_ID = "picamera"
|
||
|
|
NOW = datetime.utcNOW()
|
||
|
|
|
||
|
|
########
|
||
|
|
# Main #
|
||
|
|
########
|
||
|
|
|
||
|
|
if not os.path.exists(CONFIG_PI_CAMERA['PICT_DIR']):
|
||
|
|
print("Create photos directory : " + CONFIG_PI_CAMERA['PICT_DIR'])
|
||
|
|
os.mkdir(CONFIG_PI_CAMERA['PICT_DIR'])
|
||
|
|
|
||
|
|
FILE_NAME = os.path.extsep.join((datetime.strftime(NOW, CONFIG_PI_CAMERA['PICT_NAME']),
|
||
|
|
PICTURES_EXTENSIONS[CONFIG_PI_CAMERA['PICT_FORMAT']]))
|
||
|
|
FILE_PATH = os.path.join(CONFIG_PI_CAMERA['PICT_DIR'], FILE_NAME)
|
||
|
|
|
||
|
|
print("Shooting " + FILE_PATH)
|
||
|
|
|
||
|
|
# Taking a photo
|
||
|
|
#################
|
||
|
|
|
||
|
|
# Tell the MQTT that we start shooting
|
||
|
|
mqtt.single(CONFIG_PI_CAMERA['MQTT_TOPIC'] + "/shooting",
|
||
|
|
payload=1,
|
||
|
|
qos=1,
|
||
|
|
retain=True,
|
||
|
|
hostname=CONFIG_MQTT['HOST'],
|
||
|
|
port=CONFIG_MQTT['PORT'],
|
||
|
|
client_id=MQTT_CLIENT_ID,
|
||
|
|
auth=CONFIG_MQTT['AUTH'])
|
||
|
|
|
||
|
|
# Camera settings
|
||
|
|
cam = picamera.PiCamera(resolution=CONFIG_PI_CAMERA['RESOLUTION'])
|
||
|
|
cam.iso = CONFIG_PI_CAMERA['ISO']
|
||
|
|
cam.awb_mode = CONFIG_PI_CAMERA['AWB']
|
||
|
|
cam.rotation = CONFIG_PI_CAMERA['ROTATION']
|
||
|
|
cam.contrast = CONFIG_PI_CAMERA['CONSTRAST']
|
||
|
|
cam.exposure_mode = CONFIG_PI_CAMERA['EXPO_MODE']
|
||
|
|
cam.image_effect = 'none'
|
||
|
|
|
||
|
|
#Add some EXIF tags
|
||
|
|
cam.exif_tags['EXIF.UserComment'] = 'Cameteo !'
|
||
|
|
|
||
|
|
# Warm-up
|
||
|
|
cam.start_preview()
|
||
|
|
sleep(CONFIG_PI_CAMERA['WARMUP_TIME'])
|
||
|
|
|
||
|
|
cam.capture(FILE_PATH, format=CONFIG_PI_CAMERA['PICT_FORMAT'],
|
||
|
|
quality=95, thumbnail=None)
|
||
|
|
|
||
|
|
cam.close()
|
||
|
|
|
||
|
|
# MQTT publication
|
||
|
|
###################
|
||
|
|
|
||
|
|
MQTT_DATA = [{'topic': CONFIG_PI_CAMERA['MQTT_TOPIC'] + "/shooting",
|
||
|
|
'payload': 0,
|
||
|
|
'qos': 2,
|
||
|
|
'retain': True},
|
||
|
|
{'topic': CONFIG_PI_CAMERA['MQTT_TOPIC'] + "/photos_path",
|
||
|
|
'payload': CONFIG_PI_CAMERA['PICT_DIR'],
|
||
|
|
'qos': 2,
|
||
|
|
'retain': True},
|
||
|
|
{'topic': CONFIG_PI_CAMERA['MQTT_TOPIC'] + "/last_photo",
|
||
|
|
'payload': CONFIG_PI_CAMERA['PICT_NAME'],
|
||
|
|
'qos': 2,
|
||
|
|
'retain': True},
|
||
|
|
]
|
||
|
|
|
||
|
|
mqtt.multiple(MQTT_DATA,
|
||
|
|
hostname=CONFIG_MQTT['HOST'],
|
||
|
|
port=CONFIG_MQTT['PORT'],
|
||
|
|
client_id=MQTT_CLIENT_ID,
|
||
|
|
auth=CONFIG_MQTT['AUTH'])
|