github.com/rita33cool1/iot-system-gateway@v0.0.0-20200911033302-e65bde238cc5/gateway/Implementation/Gateway/PubAudios/SeeImage.py (about)

     1  import os
     2  import sys
     3  import cv2
     4  import wget
     5  import time
     6  import paho.mqtt.client as mqtt
     7  import paho.mqtt.publish as publish
     8  
     9  def on_connect(client, userdata, flags, rc):
    10      print("Connected with result code "+str(rc))
    11      client.subscribe("iot/iscc19/image/yolo5/image")
    12  
    13  def on_message(client, userdata, msg):
    14      f = open('yolo5.jpg','w')
    15      f.write(msg.payload)
    16      f.close()
    17      print 'img received'
    18      yolo()
    19      time.sleep(2)
    20  
    21  if __name__ == '__main__':
    22      # Debug and See the output
    23      client = mqtt.Client()
    24      client.on_connect = on_connect
    25      client.on_message = on_message
    26      client.connect(os.environ['BROKER'], 1883, 60)
    27      client.loop_forever()