Skip to content

E
evsuits
提交 1e0e5f44 authored 作者: blu's avatar blu
浏览文件
操作

ota updater revise

上级 8cd5cd32
隐藏空白字符变更
内嵌 并排
正在显示 包含 54 行增加 和 47 行删除
opencv-motion-detect/updater.py
浏览文件 @ 1e0e5f44
......@@ -22,7 +22,7 @@ MIN_SN_LEN = 8
DOWNLOAD_PATH = '/tmp/update_package/'
PACKAGE_NAME = 'update.zip'
EXTRACT_PATH = '/tmp/update_extract/'
BIN_DIR = '/tmp/work/opencv-pocs/opencv-motion-detect/'
BIN_DIR = '/root/work/opencv-pocs/opencv-motion-detect/'
LIB_DIR = BIN_DIR + 'vendor/lib/'
SKU = 'ILSA-HW-TM0401'
REPLY_TOPIC='/device/ota/update'
......@@ -46,10 +46,13 @@ class OTAClient:
continue
else:
return sn
else:
logger.error("updater sn file does not exist: {}".format(SN_FILE))
return ''
def _get_sn_from_evdaemon(self):
errmsg = 'failed connect to evdameon, please check network'
errmsg = 'updater failed connect to evdameon, please check network'
try:
r = requests.get(LOCAL_CFG_URL, timeout=2)
if r.status_code != 200:
......@@ -58,15 +61,15 @@ class OTAClient:
for k,v in r.json().items():
return k
except requests.exceptions.ConnectionError as e:
logger.error("Error Connecting: {}".format(e))
logger.error("updater _get_sn_from_evdaemon Error Connecting: {}".format(e))
except requests.exceptions.Timeout as e:
logger.error("Timeout Error: {}".format(e))
logger.error("updater Timeout Error: {}".format(e))
except requests.exceptions.RequestException as e:
logger.error("Request Exception: {}".format(e))
logger.error("updater Request Exception: {}".format(e))
except IOError as e:
logger.error("IOError Exception: {}".format(e))
logger.error("updater IOError Exception: {}".format(e))
except Exception as e:
logger.error("General Exception: {}".format(e))
logger.error("updater General Exception: {}".format(e))
return ''
......@@ -79,52 +82,51 @@ class OTAClient:
self.batchId = ver.get('batchId', '')
return self.version
except Exception as e:
logger.error("failed reading verisn file {}: {}", VER_FILE, str(e))
logger.error("updater failed reading verisn file {}: {}", VER_FILE, str(e))
return ''
def get_sn(self):
sn = self._get_sn_from_evdaemon()
if not sn:
sn = self._get_sn_from_file()
logger.info("updater sn1: {}".format(sn))
if not sn:
msg = "failed to get sn, please check network and host files"
msg = "updater failed to get sn, please check network and host files"
logger.error(msg)
raise Exception(msg)
if len(sn) > 2:
if(sn.startswith('"') and sn.endswith('"')):
sn = sn[1:len(sn)-2]
self.sn = sn
logger.info("updater sn2: {}".format(sn))
return sn
@staticmethod
def on_connect(client, ud, flags, rc):
topic = '/device/ota/init/' + ud.sn
client.subscribe(topic, qos=1)
logger.info('subscribed to {} '.format(topic))
if ud.version is '':
ud.reply(-1, '', 'fail', 'no local version')
return
logger.info('updater subscribed to {} '.format(topic))
# check system status
# 1. is evdaemon ok?
sn = ud._get_sn_from_evdaemon()
if sn == '':
ud.reply(-1, ud.version, 'failed', 'evdaemon is down')
sn = ud.get_sn()
if len(sn) == 0:
logger.error("updater no sn found")
return
ver = ud.get_version()
if ver == '':
ud.reply(-1, '', 'failed', 'no local version')
return
logger.error("updater no version found")
# report status
ud.reply(ud.batchId, ver, 'complete', 'OK')
@staticmethod
def on_message(client, ud, msg):
payload = msg.payload.decode('utf-8')
logger.info(msg.topic+" " + payload)
logger.info("updater "+msg.topic+" " + payload)
try:
jd = json.loads(payload)
ud.handle(jd)
except Exception as e:
logger.info('exception in process message: %s', e)
logger.info('updater exception in process message: %s', e)
extype, value, tb = sys.exc_info()
traceback.print_exc()
......@@ -132,7 +134,7 @@ class OTAClient:
self.version = self.get_version()
sn = self.get_sn()
if not sn:
logger.error("exiting since no sn config could be retrived")
logger.error("updater exiting since no sn config could be retrived")
exit(1)
self.client = mqtt.Client('EVB-'+sn, userdata=self) # , protocol=mqtt.MQTTv5)
......@@ -149,7 +151,7 @@ class OTAClient:
try:
self.client.loop_forever()
except Exception as e:
logger.error("network failure - failed to connect mqtt host: {}".format(str(e)))
logger.error("updater network failure - failed to connect mqtt host: {}".format(str(e)))
# NOTE: reset network and evdaemon
if retry < 2:
os.system("systemctl restart networking; systemctl restart evdaemon")
......@@ -158,43 +160,47 @@ class OTAClient:
time.sleep(10)
retry += 1
# NOTE: IT SHOULD NEVER REACH HERE unless network issues.
logger.error("exiting since network failure")
logger.error("updater exiting since network failure")
def run_nonblock(self):
self.client.loop_start()
def download(self, url):
logger.info("downloading package {}".format(url))
logger.info("updater downloading package {}".format(url))
os.system('mkdir -p ' + DOWNLOAD_PATH)
os.system('mkdir -p '+ EXTRACT_PATH)
msg = ''
try:
with requests.get(url, stream=True, timeout=5) as r:
r.raise_for_status()
logger.info("download {} -> {}".format(url, DOWNLOAD_PATH))
logger.info("updater downloading {} -> {}".format(url, DOWNLOAD_PATH))
dcnt = 0
with open(DOWNLOAD_PATH + PACKAGE_NAME, 'wb') as f:
for ch in r.iter_content(chunk_size=8000):
f.write(ch)
dcnt+=1
if dcnt %200 == 0:
logger.info("updater downloaded size: {}KB".format(8*dcnt))
except requests.exceptions.ConnectionError as e:
msg = "Error Connecting: {}".format(e)
msg = "updater download Error Connecting: {}".format(e)
logger.error(msg)
except requests.exceptions.Timeout as e:
msg = "Timeout Error: {}".format(e)
msg = "updater Timeout Error: {}".format(e)
logger.error(msg)
except requests.exceptions.RequestException as e:
msg = "Request Exception: {}".format(e)
msg = "updater Request Exception: {}".format(e)
logger.error(msg)
except IOError as e:
msg = "IOError Exception: {}".format(e)
msg = "updater IOError Exception: {}".format(e)
logger.error(msg)
except Exception as e:
msg = "General Exception: {}".format(e)
msg = "updater General Exception: {}".format(e)
logger.error(msg)
return msg
def reply(self, batchId, version, status, message):
if status not in ['confirm', 'refuse', 'fail', 'complete']:
logger.error("invalid reply status: %s", status)
logger.error("updater invalid reply status: %s", status)
return
rep = {}
rep['sn'] = self.sn
......@@ -204,16 +210,16 @@ class OTAClient:
rep['firmware'] = version
rep['status'] = status
try:
logger.info("reply with status {}".format(status))
logger.info("updater reply with status {}".format(status))
# mc = mqtt.Client('EVBRep-'+self.sn, userdata=self) # , protocol=mqtt.MQTTv5)
# if self.username and self.password:
# mc.username_pw_set(username=self.username, password=self.password)
# mc.connect_async(self.host, self.port, 10)
# mc.publish(REPLY_TOPIC, json.dumps(rep), 1)
self.client.publish(REPLY_TOPIC, json.dumps(rep), 1)
logger.info("replyed")
logger.info("updater replyed")
except Exception as e:
logger.info('exception in publish reply message: %s', e)
logger.info('updater exception in publish reply message: %s', e)
extype, value, tb = sys.exc_info()
traceback.print_exc()
......@@ -227,19 +233,19 @@ class OTAClient:
msg = 'OK'
if not sku or sku != SKU:
msg = "invalid sku {} in update request, expected {}. ignored".format(sku, SKU)
msg = "updater invalid sku {} in update request, expected {}. ignored".format(sku, SKU)
logger.error(msg);
elif not sn or sn != self.sn:
msg = "invalid sn {} in update request, expected {} ignored".format(sn, self.sn)
msg = "updater invalid sn {} in update request, expected {} ignored".format(sn, self.sn)
logger.error(msg);
elif not url or url[:7] != 'http://':
msg = "invalid url {} in update request, ignored".format(url)
msg = "updater invalid url {} in update request, ignored".format(url)
logger.error(msg);
elif not batchId or not str(batchId).isdigit():
msg = "invalid batchId {} in update request, ignored".format(batchId)
msg = "updater invalid batchId {} in update request, ignored".format(batchId)
logger.error(msg);
elif not self.version:
logger.warn("no local version file, update initialized")
logger.warn("updater no local version file, update initialized")
else:
pass
......@@ -257,16 +263,16 @@ class OTAClient:
with zipfile.ZipFile(DOWNLOAD_PATH + PACKAGE_NAME, 'r') as zip_ref:
zip_ref.extractall(EXTRACT_PATH)
except Exception as e:
msg = 'failed to extract pakage: {}'.format(str(e))
msg = 'updater failed to extract pakage: {}'.format(str(e))
logger.error(msg)
extype, value, tb = sys.exc_info()
traceback.print_exc()
self.reply(batchId, ver, 'fail', msg)
return
# copy files
os.system('mv -f ' + EXTRACT_PATH + 'bin/*', BIN_DIR)
os.system('mv -f ' + EXTRACT_PATH + 'lib/*', LIB_DIR)
os.system('mv -f ' + EXTRACT_PATH + 'bin/* ' + BIN_DIR)
os.system('mv -f ' + EXTRACT_PATH + 'lib/* ' + LIB_DIR)
os.system('chmod +x ' + BIN_DIR +"/*")
# run extra scripts
runsh = EXTRACT_PATH + 'run.sh'
if os.path.exists(runsh):
......@@ -275,7 +281,7 @@ class OTAClient:
ans = subprocess.check_output(['sh', runsh])
logger.info(ans)
except Exception as e:
msg = "failed to execute run.sh: " + str(e)
msg = "updater failed to execute run.sh: " + str(e)
self.reply(batchId, ver, 'fail', msg)
logger.error(msg)
extype, value, tb = sys.exc_info()
......@@ -287,13 +293,14 @@ class OTAClient:
# report status
# self.reply(batchId, ver, 'complete', 'OK')
os.system('systemctl restart evdaemon')
self.reply(batchId, ver, 'complete', "OK")
if __name__ == "__main__":
# avoid multiple instance
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(('localhost', 45432))
logger.setLevel(logging.INFO)
logger.info("connecting mqtt {}:{}: {} {}".format(MQTT_HOST, MQTT_PORT, MQTT_USER_NAME, MQTT_USER_PASSWORD))
logger.info("updater connecting mqtt {}:{}: {} {}".format(MQTT_HOST, MQTT_PORT, MQTT_USER_NAME, MQTT_USER_PASSWORD))
ota = OTAClient(MQTT_HOST, MQTT_PORT, MQTT_USER_NAME, MQTT_USER_PASSWORD)
ota.run()
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论