libervia-backend: tools/memory.py annotate

annotate tools/memory.py @ 19:f2a745ca0fbc

refactoring: using xml params part III (parameters import)

author	Goffi <goffi@goffi.org>
date	Sat, 07 Nov 2009 21:23:28 +0100
parents	6928e3cb73a8
children	fc8c202cda87

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
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	30
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	31 const_SAVEFILE_PARAM=os.path.expanduser("~/.sat_param.save")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	32 const_SAVEFILE_HISTORY=os.path.expanduser("~/.sat_history.save")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	33
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	34 class Param():
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	35 """This class manage parameter with xml"""
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	36 ### TODO: add desciption in params
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	37
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	38 #TODO: mettre Watched dans un plugin
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	39 default_xml = """
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	40 <params>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	41 <category name="Connection">
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	42 <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	43 <param name="Password" value="toto" type="password" />
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	44 <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	45 </category>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	46 <category name="Misc">
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	47 <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	48 </category>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	49 </params>
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	50 """
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 def load_default_params(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	53 self.dom = minidom.parseString(Param.default_xml)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	54
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	55 def __init__(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	56 debug("Parameters init")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	57 self.load_default_params()
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	58
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	59 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	60 """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	61 @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	62 @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	63 @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	64 @return: node if it exist or None
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	65 """
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	66 for node in parent.childNodes:
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	67 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	68 #the node already exists
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	69 return node
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	70 #the node is new
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	71 return None
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 import_params(self, parent, xml):
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	74 """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	75 @param parent: parent class (usefull for callbacks)
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	76 @param xml: parameters in xml form"""
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	77
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
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	93
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	94 def getParamV(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	95 """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	96 getParamV stands for GetParamValue"""
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	97 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	98 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	99 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	100 return None
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	101 return node.getAttribute("value")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	102
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	103 def getParams(self):
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	104 """Return the whole params XML"""
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	105 return self.dom.toxml()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	106
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	107 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	108 """Return node's xml for selected category"""
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	109 for node in self.dom.documentElement.childNodes:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	110 if node.nodeName == "category" and node.getAttribute("name") == category:
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	111 return node.toxml()
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	112 return "<category />"
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	113
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	114 def __getParamNode(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	115 for node in self.dom.documentElement.childNodes:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	116 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	117 params = node.getElementsByTagName("param")
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	118 for param in params:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	119 if param.getAttribute("name") == name:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	120 return param
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	121 return None
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	122
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	123 def getParamsCategories(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	124 """return the categories availables"""
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	125 categories=[]
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	126 for cat in self.dom.getElementsByTagName("category"):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	127 categories.append(cat.getAttribute("name"))
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	128 return categories
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	129
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	130 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	131 node = self.__getParamNode(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	132 if not node:
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	133 return #TODO: throw an error
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	134 node.setAttribute("value", value)
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	135
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	136 class Memory:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	137 """This class manage all persistent informations"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	138
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	139 def __init__(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	140 info ("Memory manager init")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	141 self.contact={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	142 self.presenceStatus={}
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	143 self.params=Param()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	144 self.history={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	145 self.disco={} #XXX: maybe best in a separate class
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	146 self.features={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	147 self.load()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	148
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	149 def load(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	150 """Load parameters and all memory things from file/db"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	151
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	152 #first parameters
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	153 if os.path.exists(const_SAVEFILE_PARAM):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	154 try:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	155 with open(const_SAVEFILE_PARAM, 'r') as params_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	156 self.params=pickle.load(params_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	157 debug("params loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	158 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	159 error ("Can't load params !")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	160 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	161 else:
19 f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	162 error ("No params, using default parameters")
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	163 self.params.load_default_params()
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	164
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	165 #history
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	166 if os.path.exists(const_SAVEFILE_HISTORY):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	167 try:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	168 with open(const_SAVEFILE_HISTORY, 'r') as history_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	169 self.history=pickle.load(history_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	170 debug("history loaded")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	171 except:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	172 error ("Can't load history !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	173
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	174
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	175 def save(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	176 """Save parameters and all memory things to file/db"""
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	177 with open(const_SAVEFILE_PARAM, 'w') as params_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	178 pickle.dump(self.params, params_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	179 with open(const_SAVEFILE_HISTORY, 'w') as history_pickle:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	180 pickle.dump(self.history, history_pickle)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	181
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	182 def addToHistory(self, me_jid, from_jid, to_jid, type, message):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	183 me_short=me_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	184 from_short=from_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	185 to_short=to_jid.userhost()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	186
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	187 if from_jid==me_jid:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	188 key=to_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	189 else:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	190 key=from_short
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	191
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	192 if not self.history.has_key(me_short):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	193 self.history[me_short]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	194 if not self.history[me_short].has_key(key):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	195 self.history[me_short][key]={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	196
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	197 self.history[me_short][key][int(time.time())] = (from_short, message)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	198
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	199 def getHistory(self, from_jid, to_jid, size):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	200 ret={}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	201 if not self.history.has_key(from_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	202 error("source JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	203 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	204 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	205 if not self.history[from_jid].has_key(to_jid):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	206 error("dest JID not found !")
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	207 #TODO: throw an error here
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	208 return {}
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	209 stamps=self.history[from_jid][to_jid].keys()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	210 stamps.sort()
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	211 for stamp in stamps[-size:]:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	212 ret[stamp]=self.history[from_jid][to_jid][stamp]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	213
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	214 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	215
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	216 def addContact(self, JID, attributes, groups):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	217 debug("Memory addContact: %s",JID)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	218 assert(isinstance(attributes,dict))
13 bd9e9997d540 wokkel integration (not finished yet) Goffi <goffi@goffi.org> parents: 0 diff changeset	219 assert(isinstance(groups,set))
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	220 self.contact[JID]=[attributes, groups]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	221
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	222 def addPresenceStatus(self, jid, type, show, status, priority):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	223 self.presenceStatus[jid]=[type, show, status, priority]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	224
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	225 def getContacts(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	226 debug ("Memory getContact OK (%s)", self.contact)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	227 ret=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	228 for contact in self.contact:
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	229 ret.append([contact] + [self.contact[contact][0]] + [self.contact[contact][1]]) #very ugly I know !
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	230 return ret
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	231
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	232 def getPresenceStatus(self):
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	233 status=[]
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	234 for contact, contactStatus in self.presenceStatus.items():
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	235 status.append([contact]+contactStatus)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	236 debug ("Memory getPresenceStatus (%s)", status)
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	237 return status
c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	238
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	239 def getParamV(self, name, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	240 return self.params.getParamV(name, category)
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	241
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	242 def getParams(self):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	243 return self.params.getParams()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	244
18 6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	245 def getParamsForCategory(self, category):
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	246 return self.params.getParamsForCategory(category)
6928e3cb73a8 refactoring: using xml params part II Goffi <goffi@goffi.org> parents: 17 diff changeset	247
0 c4bc297b82f0 sat: goffi@necton2 parents: diff changeset	248 def getParamsCategories(self):
17 74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	249 return self.params.getParamsCategories()
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	250
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	251 def setParam(self, name, value, category):
74a39f40eb6d refactoring: using xml params (not finished yet) Goffi <goffi@goffi.org> parents: 14 diff changeset	252 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	253
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	254 def import_params(self, parent, xml):
f2a745ca0fbc refactoring: using xml params part III (parameters import) Goffi <goffi@goffi.org> parents: 18 diff changeset	255 return self.params.import_params(parent, xml)

Mercurial > libervia-backend

annotate tools/memory.py @ 19:f2a745ca0fbc