libervia-backend: tools/memory.py annotate

annotate tools/memory.py @ 46:18f6c9e60507

jid parsing fix: node is now empty for transports

author	Goffi <goffi@goffi.org>
date	Wed, 06 Jan 2010 23:43:54 +1100
parents	d24629c631fc
children	9aa2d9dd4045

rev	line source
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	1 #!/usr/bin/python
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	2 # -- coding: utf-8 --
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	3
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	4 """
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	5 SAT: a jabber client
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	6 Copyright (C) 2009 Jérôme Poisson (goffi@goffi.org)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	7
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	8 This program is free software: you can redistribute it and/or modify
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	9 it under the terms of the GNU General Public License as published by
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	10 the Free Software Foundation, either version 3 of the License, or
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	11 (at your option) any later version.
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	12
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	13 This program is distributed in the hope that it will be useful,
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	14 but WITHOUT ANY WARRANTY; without even the implied warranty of
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	16 GNU General Public License for more details.
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	17
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	18 You should have received a copy of the GNU General Public License
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	19 along with this program. If not, see <http://www.gnu.org/licenses/>.
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	20 """
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	21
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	22 from __future__ import with_statement
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	23
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	24 import os.path
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	25 import time
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	26 import pickle
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	27 from xml.dom import minidom
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	28 from logging import debug, info, error
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	29 import pdb
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	30 from twisted.internet import defer
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	31
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	32 SAVEFILE_PARAM="/param"
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	33 SAVEFILE_HISTORY="/history"
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	34
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	35 class Param():
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	36 """This class manage parameter with xml"""
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	37 ### TODO: add desciption in params
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	38
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	39 #TODO: mettre Watched dans un plugin
34 a544b376b6f0 use proper utf-8 encoding for parsing xml in parameters Goffi <goffi@goffi.org> parents: 22 diff changeset	40 default_xml = u"""
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	41 <params>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	42 <category name="Connection">
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	43 <param name="JabberID" value="goffi@necton2.int/TestScript" type="string" />
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	44 <param name="Password" value="toto" type="password" />
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	45 <param name="Server" value="necton2.int" type="string" />
21 633c5ed65701 parameters: new button type (not finished) Goffi <goffi@goffi.org> parents: 20 diff changeset	46 <param name="NewAccount" value="Register new account" type="button" callback="registerNewAccount"/>
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	47 </category>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	48 <category name="Misc">
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	49 <param name="Watched" value="test@Jabber.goffi.int" type="string" />
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	50 </category>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	51 </params>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	52 """
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	53
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	54 def load_default_params(self):
34 a544b376b6f0 use proper utf-8 encoding for parsing xml in parameters Goffi <goffi@goffi.org> parents: 22 diff changeset	55 self.dom = minidom.parseString(Param.default_xml.encode('utf-8'))
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	56
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	57 def load(self, file):
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	58 """Load parameters from file"""
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	59 self.dom = minidom.parse(file)
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	60
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	61 def save(self, file):
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	62 """Save parameters to xml file"""
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	63 with open(file, 'wb') as xml_file:
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	64 self.dom.writexml(xml_file)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	65
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	66 def __init__(self, host):
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	67 debug("Parameters init")
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	68 self.host = host
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	69 host.set_const('savefile_param', SAVEFILE_PARAM)
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	70 host.set_const('savefile_history', SAVEFILE_HISTORY)
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	71 host.registerGeneralCB("registerNewAccount", host.registerNewAccountCB)
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	72
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	73 def __get_unique_node(self, parent, tag, name):
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	74 """return node with given tag, create a new one if the node doesn't exist
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	75 @param parent: parent of nodes to check (e.g. documentElement)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	76 @param tag: tag to check (e.g. "category")
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	77 @param name: name to check (e.g. "JID")
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	78 @return: node if it exist or None
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	79 """
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	80 for node in parent.childNodes:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	81 if node.nodeName == tag and node.getAttribute("name") == name:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	82 #the node already exists
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	83 return node
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	84 #the node is new
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	85 return None
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	86
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	87 def importParams(self, xml):
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	88 """import xml in parameters, do nothing if the param already exist
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	89 @param xml: parameters in xml form"""
34 a544b376b6f0 use proper utf-8 encoding for parsing xml in parameters Goffi <goffi@goffi.org> parents: 22 diff changeset	90 src_dom = minidom.parseString(xml.encode('utf-8'))
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	91
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	92 def import_node(tgt_parent, src_parent):
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	93 for child in src_parent.childNodes:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	94 if child.nodeName == '#text':
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	95 continue
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	96 node = self.__get_unique_node(tgt_parent, child.nodeName, child.getAttribute("name"))
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	97 if not node: #The node is new
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	98 tgt_parent.appendChild(child)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	99 else:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	100 import_node(node, child)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	101
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	102 import_node(self.dom.documentElement, src_dom.documentElement)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	103
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	104 def __default_ok(self, value, name, category):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	105 self.setParam(name, value, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	106
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	107 def __default_ko(self, failure, name, category):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	108 error ("Can't determine default value for [%s/%s]: %s" % (category, name, str(failure.value)))
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	109
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	110 def setDefault(self, name, category, callback, errback=None):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	111 """Set default value of parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	112 'default_cb' attibute of parameter must be set to 'yes'
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	113 @param name: name of the parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	114 @param category: category of the parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	115 @param callback: must return a string with the value (use deferred if needed)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	116 @param errback: must manage the error with args failure, name, category
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	117 """
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	118 #TODO: send signal param update if value changed
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	119 node = self.__getParamNode(name, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	120 if not node:
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	121 error("Requested param [%s] in category [%s] doesn't exist !" , name, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	122 return
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	123 if node.getAttribute('default_cb') == 'yes':
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	124 del node.attributes['default_cb']
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	125 d = defer.maybeDeferred(callback)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	126 d.addCallback(self.__default_ok, name, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	127 d.addErrback(errback or self.__default_ko, name, category)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	128
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	129 def getParamA(self, name, category, attr="value"):
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	130 """Helper method to get a specific attribute
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	131 @param name: name of the parameter
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	132 @param category: category of the parameter
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	133 @param attr: name of the attribute (default: "value")
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	134
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	135 @return: attribute"""
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	136 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	137 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	138 error("Requested param [%s] in category [%s] doesn't exist !" , name, category)
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	139 return None
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	140 return node.getAttribute(attr)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	141
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	142 def getParams(self):
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	143 """Return the whole params XML"""
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	144 return self.dom.toxml()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	145
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	146 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	147 """Return node's xml for selected category"""
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	148 for node in self.dom.documentElement.childNodes:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	149 if node.nodeName == "category" and node.getAttribute("name") == category:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	150 return node.toxml()
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	151 return "<category />"
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	152
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	153 def __getParamNode(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	154 for node in self.dom.documentElement.childNodes:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	155 if node.nodeName == "category" and node.getAttribute("name") == category:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	156 params = node.getElementsByTagName("param")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	157 for param in params:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	158 if param.getAttribute("name") == name:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	159 return param
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	160 return None
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	161
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	162 def getParamsCategories(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	163 """return the categories availables"""
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	164 categories=[]
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	165 for cat in self.dom.getElementsByTagName("category"):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	166 categories.append(cat.getAttribute("name"))
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	167 return categories
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	168
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	169 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	170 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	171 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	172 return #TODO: throw an error
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	173 type = node.getAttribute("type")
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	174 if type=="button":
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	175 print "clique",node.toxml()
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	176 else:
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	177 node.setAttribute("value", value)
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	178 self.host.bridge.paramUpdate(name, value, category)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	179
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	180 class Memory:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	181 """This class manage all persistent informations"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	182
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	183 def __init__(self, host):
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	184 info ("Memory manager init")
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	185 self.host = host
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	186 self.contact={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	187 self.presenceStatus={}
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	188 self.params=Param(host)
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	189 self.history={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	190 self.disco={} #XXX: maybe best in a separate class
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	191 self.features={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	192 self.load()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	193
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	194 def load(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	195 """Load parameters and all memory things from file/db"""
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	196 param_file = os.path.expanduser(self.host.get_const('local_dir')+
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	197 self.host.get_const('savefile_param'))
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	198 history_file = os.path.expanduser(self.host.get_const('local_dir')+
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	199 self.host.get_const('savefile_history'))
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	200
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	201 #parameters
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	202 if os.path.exists(param_file):
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	203 try:
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	204 self.params.load(param_file)
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	205 debug("params loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	206 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	207 error ("Can't load params !")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	208 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	209 else:
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	210 error ("No params, using default parameters")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	211 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	212
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	213 #history
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	214 if os.path.exists(history_file):
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	215 try:
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	216 with open(history_file, 'r') as history_pickle:
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	217 self.history=pickle.load(history_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	218 debug("history loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	219 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	220 error ("Can't load history !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	221
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	222
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	223 def save(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	224 """Save parameters and all memory things to file/db"""
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	225 #TODO: need to encrypt files (at least passwords !) and set permissions
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	226 param_file = os.path.expanduser(self.host.get_const('local_dir')+
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	227 self.host.get_const('savefile_param'))
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	228 history_file = os.path.expanduser(self.host.get_const('local_dir')+
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	229 self.host.get_const('savefile_history'))
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	230
d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	231 self.params.save(param_file)
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	232 debug("params saved")
41 d24629c631fc SàT: new constant management, a local dir (~/.sat) is now used Goffi <goffi@goffi.org> parents: 38 diff changeset	233 with open(history_file, 'w') as history_pickle:
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	234 pickle.dump(self.history, history_pickle)
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	235 debug("history saved")
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	236
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	237 def addToHistory(self, me_jid, from_jid, to_jid, type, message):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	238 me_short=me_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	239 from_short=from_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	240 to_short=to_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	241
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	242 if from_jid==me_jid:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	243 key=to_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	244 else:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	245 key=from_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	246
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	247 if not self.history.has_key(me_short):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	248 self.history[me_short]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	249 if not self.history[me_short].has_key(key):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	250 self.history[me_short][key]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	251
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	252 self.history[me_short][key][int(time.time())] = (from_short, message)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	253
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	254 def getHistory(self, from_jid, to_jid, size):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	255 ret={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	256 if not self.history.has_key(from_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	257 error("source JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	258 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	259 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	260 if not self.history[from_jid].has_key(to_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	261 error("dest JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	262 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	263 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	264 stamps=self.history[from_jid][to_jid].keys()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	265 stamps.sort()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	266 for stamp in stamps[-size:]:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	267 ret[stamp]=self.history[from_jid][to_jid][stamp]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	268
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	269 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	270
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	271 def addContact(self, JID, attributes, groups):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	272 debug("Memory addContact: %s",JID)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	273 assert(isinstance(attributes,dict))
13 bd9e9997d540 wokkel integration (not finished yet) Goffi <goffi@goffi.org> parents: 0 diff changeset	274 assert(isinstance(groups,set))
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	275 self.contact[JID]=[attributes, groups]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	276
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	277 def addPresenceStatus(self, jid, type, show, status, priority):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	278 self.presenceStatus[jid]=[type, show, status, priority]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	279
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	280 def getContacts(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	281 debug ("Memory getContact OK (%s)", self.contact)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	282 ret=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	283 for contact in self.contact:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	284 ret.append([contact] + [self.contact[contact][0]] + [self.contact[contact][1]]) #very ugly I know !
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	285 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	286
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	287 def getPresenceStatus(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	288 status=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	289 for contact, contactStatus in self.presenceStatus.items():
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	290 status.append([contact]+contactStatus)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	291 debug ("Memory getPresenceStatus (%s)", status)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	292 return status
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	293
22 bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	294 def getParamA(self, name, category, attr="value"):
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button. Goffi <goffi@goffi.org> parents: 21 diff changeset	295 return self.params.getParamA(name, category, attr)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	296
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	297 def getParams(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	298 return self.params.getParams()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	299
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	300 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	301 return self.params.getParamsForCategory(category)
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	302
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	303 def getParamsCategories(self):
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	304 return self.params.getParamsCategories()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	305
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	306 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	307 return self.params.setParam(name, value, category)
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	308
38 3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	309 def importParams(self, xml):
3e24753b9e0b Fixed parameters loading/saving Goffi <goffi@goffi.org> parents: 34 diff changeset	310 return self.params.importParams(xml)
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	311
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	312 def setDefault(self, name, category, callback, errback=None):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	313 return self.params.setDefault(name, category, callback, errback)

Mercurial > libervia-backend

annotate tools/memory.py @ 46:18f6c9e60507