1
0
mirror of https://github.com/mbirth/tcl_ota_check.git synced 2024-11-10 06:16:46 +00:00
tcl_ota_check/tclcheck_findprd.py

63 lines
1.8 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# pylint: disable=C0111,C0326,C0103
import collections
import sys
import tcllib
from requests.exceptions import RequestException, Timeout
tcllib.make_escapes_work()
fc = tcllib.FotaCheck()
fc.serid = "3531510"
fc.fv = "AAA000"
fc.mode = fc.MODE_FULL
# CLTP = 10 (only show actual updates or HTTP 206) / 2010 (always show latest version for MODE_FULL)
#fc.cltp = 10
fc.cltp = 2010
print("Valid PRDs not already in database:")
with open("prds.txt", "r") as afile:
prddict = collections.defaultdict(list)
prda = afile.readlines()
prds = [x.split(" ")[0].replace("PRD-", "").split("-") for x in prda]
prdx = list({x[0]: x[1]} for x in prds)
for prdc in prdx:
for key, value in prdc.items():
prddict[key].append(value)
if len(sys.argv) > 1:
prdkeys = list(prddict.keys())
for k in prdkeys:
if k != sys.argv[1]:
del prddict[k]
for center in sorted(prddict.keys()):
tails = [int(i) for i in prddict[center]]
total_count = 1000 - len(tails)
done_count = 0
print("Checking {} variant codes for model {}.".format(total_count, center))
for j in range(0, 1000):
if j in tails:
continue
curef = "PRD-{}-{:03}".format(center, j)
done_count += 1
print("Checking {} ({}/{})".format(curef, done_count, total_count))
print(tcllib.ANSI_UP_DEL, end="")
try:
fc.reset_session()
fc.curef = curef
check_xml = fc.do_check(https=False, max_tries=3)
curef, fv, tv, fw_id, fileid, fn, fsize, fhash = fc.parse_check(check_xml)
txt_tv = tv
print("{}: {} {}".format(curef, txt_tv, fhash))
except (SystemExit, RequestException, Timeout) as e:
continue
print("Scan complete.")