libervia-backend: tools/memory.py annotate

annotate tools/memory.py @ 21:633c5ed65701

parameters: new button type (not finished)

author	Goffi <goffi@goffi.org>
date	Sun, 08 Nov 2009 01:49:08 +0100
parents	fc8c202cda87
children	bb72c29f3432

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
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	32 const_SAVEFILE_PARAM=os.path.expanduser("~/.sat_param.save")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	33 const_SAVEFILE_HISTORY=os.path.expanduser("~/.sat_history.save")
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
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	40 default_xml = """
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):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	55 self.dom = minidom.parseString(Param.default_xml)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	56
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	57 def __init__(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	58 debug("Parameters init")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	59 self.load_default_params()
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	60
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	61 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	62 """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	63 @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	64 @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	65 @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	66 @return: node if it exist or None
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	67 """
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	68 for node in parent.childNodes:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	69 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	70 #the node already exists
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	71 return node
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	72 #the node is new
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	73 return None
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	74
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	75 def importParams(self, parent, xml):
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	76 """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	77 @param parent: parent class (usefull for callbacks)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	78 @param xml: parameters in xml form"""
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	79 src_dom = minidom.parseString(xml)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	80
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	81 def import_node(tgt_parent, src_parent):
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	82 print "import_node [%s, %s]" % (tgt_parent.nodeName, src_parent.nodeName)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	83 for child in src_parent.childNodes:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	84 if child.nodeName == '#text':
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	85 continue
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	86 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	87 if not node: #The node is new
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	88 tgt_parent.appendChild(child)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	89 else:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	90 import_node(node, child)
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 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	93
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	94 def __default_ok(self, value, name, category):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	95 self.setParam(name, value, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	96
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	97 def __default_ko(self, failure, name, category):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	98 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	99
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	100 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	101 """Set default value of parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	102 '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	103 @param name: name of the parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	104 @param category: category of the parameter
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	105 @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	106 @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	107 """
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	108 node = self.__getParamNode(name, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	109 if not node:
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	110 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	111 return
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	112 if node.getAttribute('default_cb') == 'yes':
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	113 del node.attributes['default_cb']
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	114 d = defer.maybeDeferred(callback)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	115 d.addCallback(self.__default_ok, name, category)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	116 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	117
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	118 def getParamV(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	119 """Helper method to get the value in the good type
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	120 getParamV stands for GetParamValue"""
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	121 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	122 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	123 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	124 return None
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	125 return node.getAttribute("value")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	126
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	127 def getParams(self):
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	128 """Return the whole params XML"""
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	129 return self.dom.toxml()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	130
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	131 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	132 """Return node's xml for selected category"""
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	133 for node in self.dom.documentElement.childNodes:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	134 if node.nodeName == "category" and node.getAttribute("name") == category:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	135 return node.toxml()
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	136 return "<category />"
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	137
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	138 def __getParamNode(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	139 for node in self.dom.documentElement.childNodes:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	140 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	141 params = node.getElementsByTagName("param")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	142 for param in params:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	143 if param.getAttribute("name") == name:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	144 return param
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	145 return None
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	146
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	147 def getParamsCategories(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	148 """return the categories availables"""
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	149 categories=[]
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	150 for cat in self.dom.getElementsByTagName("category"):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	151 categories.append(cat.getAttribute("name"))
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	152 return categories
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	153
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	154 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	155 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	156 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	157 return #TODO: throw an error
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	158 node.setAttribute("value", value)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	159
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	160 class Memory:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	161 """This class manage all persistent informations"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	162
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	163 def __init__(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	164 info ("Memory manager init")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	165 self.contact={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	166 self.presenceStatus={}
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	167 self.params=Param()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	168 self.history={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	169 self.disco={} #XXX: maybe best in a separate class
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	170 self.features={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	171 self.load()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	172
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	173 def load(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	174 """Load parameters and all memory things from file/db"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	175
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	176 #first parameters
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	177 if os.path.exists(const_SAVEFILE_PARAM):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	178 try:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	179 with open(const_SAVEFILE_PARAM, 'r') as params_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	180 self.params=pickle.load(params_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	181 debug("params loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	182 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	183 error ("Can't load params !")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	184 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	185 else:
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	186 error ("No params, using default parameters")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	187 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	188
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	189 #history
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	190 if os.path.exists(const_SAVEFILE_HISTORY):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	191 try:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	192 with open(const_SAVEFILE_HISTORY, 'r') as history_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	193 self.history=pickle.load(history_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	194 debug("history loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	195 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	196 error ("Can't load history !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	197
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	198
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	199 def save(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	200 """Save parameters and all memory things to file/db"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	201 with open(const_SAVEFILE_PARAM, 'w') as params_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	202 pickle.dump(self.params, params_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	203 with open(const_SAVEFILE_HISTORY, 'w') as history_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	204 pickle.dump(self.history, history_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	205
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	206 def addToHistory(self, me_jid, from_jid, to_jid, type, message):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	207 me_short=me_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	208 from_short=from_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	209 to_short=to_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	210
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	211 if from_jid==me_jid:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	212 key=to_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	213 else:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	214 key=from_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	215
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	216 if not self.history.has_key(me_short):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	217 self.history[me_short]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	218 if not self.history[me_short].has_key(key):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	219 self.history[me_short][key]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	220
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	221 self.history[me_short][key][int(time.time())] = (from_short, message)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	222
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	223 def getHistory(self, from_jid, to_jid, size):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	224 ret={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	225 if not self.history.has_key(from_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	226 error("source JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	227 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	228 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	229 if not self.history[from_jid].has_key(to_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	230 error("dest JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	231 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	232 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	233 stamps=self.history[from_jid][to_jid].keys()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	234 stamps.sort()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	235 for stamp in stamps[-size:]:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	236 ret[stamp]=self.history[from_jid][to_jid][stamp]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	237
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	238 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	239
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	240 def addContact(self, JID, attributes, groups):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	241 debug("Memory addContact: %s",JID)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	242 assert(isinstance(attributes,dict))
13 bd9e9997d540 wokkel integration (not finished yet) Goffi <goffi@goffi.org> parents: 0 diff changeset	243 assert(isinstance(groups,set))
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	244 self.contact[JID]=[attributes, groups]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	245
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	246 def addPresenceStatus(self, jid, type, show, status, priority):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	247 self.presenceStatus[jid]=[type, show, status, priority]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	248
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	249 def getContacts(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	250 debug ("Memory getContact OK (%s)", self.contact)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	251 ret=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	252 for contact in self.contact:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	253 ret.append([contact] + [self.contact[contact][0]] + [self.contact[contact][1]]) #very ugly I know !
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	254 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	255
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	256 def getPresenceStatus(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	257 status=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	258 for contact, contactStatus in self.presenceStatus.items():
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	259 status.append([contact]+contactStatus)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	260 debug ("Memory getPresenceStatus (%s)", status)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	261 return status
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	262
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	263 def getParamV(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	264 return self.params.getParamV(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	265
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	266 def getParams(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	267 return self.params.getParams()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	268
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	269 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	270 return self.params.getParamsForCategory(category)
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	271
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	272 def getParamsCategories(self):
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	273 return self.params.getParamsCategories()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	274
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	275 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	276 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	277
20 fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	278 def importParams(self, parent, xml):
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	279 return self.params.importParams(parent, xml)
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	280
fc8c202cda87 refactoring: using xml params part IV (default values) Goffi <goffi@goffi.org> parents: 19 diff changeset	281 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	282 return self.params.setDefault(name, category, callback, errback)

Mercurial > libervia-backend

annotate tools/memory.py @ 21:633c5ed65701