mirror of
https://port.numenaute.org/aleajactaest/clientbot.git
synced 2024-09-19 11:10:34 +00:00
139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# module DecodeDatabase
|
|
#
|
|
# Copyright (C) 2019 AleaJactaEst
|
|
#
|
|
# 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 <http://www.gnu.org/licenses/>.
|
|
|
|
import logging
|
|
from tools import getPowerOf2
|
|
|
|
LOGGER='DecodeDatabase'
|
|
|
|
def show_dico(dico, level=1):
|
|
for ele in dico:
|
|
if isinstance(dico[ele], dict):
|
|
print("." * level, ele , ":")
|
|
if isinstance(dico[ele], dict):
|
|
show_dico(dico[ele], level+1)
|
|
else:
|
|
print("." * level, ele, ':', dico[ele])
|
|
|
|
def child(ele):
|
|
ret = {}
|
|
ret_branch = {}
|
|
ref_other = {}
|
|
id_branch = 0
|
|
id_other = 0
|
|
min_i = -1
|
|
max_i = -1
|
|
for k in ele.keys():
|
|
ret[k] = ele.get(k)
|
|
print(k, ele.get(k))
|
|
for _child in list(ele):
|
|
x = child(_child)
|
|
if x['name'] == 'branch':
|
|
ret_branch.setdefault(id_branch, x)
|
|
id_branch += 1
|
|
else:
|
|
ref_other.setdefault(id_other, x)
|
|
id_other += 1
|
|
if ret_branch or ref_other:
|
|
ret['child'] = {}
|
|
id = 0
|
|
for x in ret_branch:
|
|
min_i = max_i + 1
|
|
max_i = min_i
|
|
#show_dico( ret_branch[x])
|
|
if 'count' in ret_branch[x]:
|
|
max_i = min_i + int(ret_branch[x]['count']) - 1
|
|
ret_branch[x]['min'] = min_i
|
|
ret_branch[x]['max'] = max_i
|
|
ret['child'].setdefault(id, ret_branch[x])
|
|
id += 1
|
|
for x in ref_other:
|
|
min_i = max_i + 1
|
|
max_i = min_i
|
|
#show_dico( ref_other[x])
|
|
if 'count' in ref_other[x]:
|
|
max_i = min_i + int(ref_other[x]['count']) - 1
|
|
ref_other[x]['min'] = min_i
|
|
ref_other[x]['max'] = max_i
|
|
ret['child'].setdefault(id, ref_other[x])
|
|
id += 1
|
|
return ret
|
|
|
|
def count_elements(head):
|
|
try:
|
|
return head.items()[-1]['max'] + 1
|
|
except TypeError:
|
|
return len(head)
|
|
|
|
def get_element(head, id):
|
|
print("id:", id)
|
|
for ele in head:
|
|
if id <= head[ele]['max'] and id >= head[ele]['min']:
|
|
return head[ele]
|
|
return None
|
|
|
|
class DecodeDatabase():
|
|
def __init__(self):
|
|
self.databaseXml = None
|
|
self.databasePlr = None
|
|
|
|
def loadDatabase(self, databaseXml):
|
|
logging.getLogger(LOGGER).debug("loadDatabase")
|
|
self.databaseXml = databaseXml
|
|
id = 0
|
|
self.databasePlr = {}
|
|
for ele in self.databaseXml:
|
|
if ele.tag == 'branch':
|
|
if ele.get('bank') == "PLR":
|
|
self.databasePlr[id] = child(ele)
|
|
self.databasePlr[id]['min'] = id
|
|
self.databasePlr[id]['max'] = id
|
|
id += 1
|
|
print(dir(ele))
|
|
print("-" * 80)
|
|
show_dico(self.databasePlr)
|
|
print("-" * 80)
|
|
#raise "Decode"
|
|
|
|
def execute(self, msgin, world):
|
|
logging.getLogger(LOGGER).debug("execute")
|
|
head = self.databasePlr
|
|
listpath = []
|
|
while True:
|
|
logging.getLogger(LOGGER).debug("count_elements:" + str(count_elements(head)))
|
|
nbBit = getPowerOf2.getPowerOf2(count_elements(head))
|
|
logging.getLogger(LOGGER).debug("nbBit:" + str(nbBit))
|
|
id = msgin.readSerial(nbBit, name='DatabaseXML', typeName='Number', emulate=True)
|
|
|
|
logging.getLogger(LOGGER).debug("XML DECODE : %3d -> %s" % (nbBit, ':'.join(listpath)) )
|
|
ele = get_element(head, id)
|
|
print(ele)
|
|
show_dico(ele)
|
|
name = ele['name']
|
|
listpath.append(name)
|
|
fullname = ':'.join(listpath)
|
|
logging.getLogger(LOGGER).debug(fullname)
|
|
if 'type' in ele:
|
|
print("+"*80)
|
|
return True
|
|
head = ele
|
|
print("-"*80)
|
|
return False
|
|
|