libervia-backend: tools/memory.py annotate

annotate tools/memory.py @ 20:fc8c202cda87

refactoring: using xml params part IV (default values) - File Transfert/IP is now set by default to the ip externally visible

author	Goffi <goffi@goffi.org>
date	Sun, 08 Nov 2009 00:11:00 +0100
parents	f2a745ca0fbc
children	633c5ed65701

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

Mercurial > libervia-backend

annotate tools/memory.py @ 20:fc8c202cda87