data_smart.py 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270
  1. # ex:ts=4:sw=4:sts=4:et
  2. # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
  3. """
  4. BitBake Smart Dictionary Implementation
  5. Functions for interacting with the data structure used by the
  6. BitBake build tools.
  7. """
  8. # Copyright (C) 2003, 2004 Chris Larson
  9. # Copyright (C) 2004, 2005 Seb Frankengul
  10. # Copyright (C) 2005, 2006 Holger Hans Peter Freyther
  11. # Copyright (C) 2005 Uli Luckas
  12. # Copyright (C) 2005 ROAD GmbH
  13. #
  14. # This program is free software; you can redistribute it and/or modify
  15. # it under the terms of the GNU General Public License version 2 as
  16. # published by the Free Software Foundation.
  17. #
  18. # This program is distributed in the hope that it will be useful,
  19. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  20. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  21. # GNU General Public License for more details.
  22. #
  23. # You should have received a copy of the GNU General Public License along
  24. # with this program; if not, write to the Free Software Foundation, Inc.,
  25. # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  26. # Based on functions from the base bb module, Copyright 2003 Holger Schurig
  27. import copy, os, re, sys, time, types
  28. import bb
  29. from bb import utils, methodpool
  30. from COW import COWDictBase
  31. from sets import Set
  32. from new import classobj
  33. __setvar_keyword__ = ["_append","_prepend"]
  34. __setvar_regexp__ = re.compile('(?P<base>.*?)(?P<keyword>_append|_prepend)(_(?P<add>.*))?')
  35. __expand_var_regexp__ = re.compile(r"\${[^{}]+}")
  36. __expand_python_regexp__ = re.compile(r"\${@.+?}")
  37. class DataSmart:
  38. def __init__(self, special = COWDictBase.copy(), seen = COWDictBase.copy() ):
  39. self.dict = {}
  40. # cookie monster tribute
  41. self._special_values = special
  42. self._seen_overrides = seen
  43. self.expand_cache = {}
  44. def expand(self,s, varname):
  45. def var_sub(match):
  46. key = match.group()[2:-1]
  47. if varname and key:
  48. if varname == key:
  49. raise Exception("variable %s references itself!" % varname)
  50. var = self.getVar(key, 1)
  51. if var is not None:
  52. return var
  53. else:
  54. return match.group()
  55. def python_sub(match):
  56. import bb
  57. code = match.group()[3:-1]
  58. locals()['d'] = self
  59. s = eval(code)
  60. if type(s) == types.IntType: s = str(s)
  61. return s
  62. if type(s) is not types.StringType: # sanity check
  63. return s
  64. if varname and varname in self.expand_cache:
  65. return self.expand_cache[varname]
  66. while s.find('${') != -1:
  67. olds = s
  68. try:
  69. s = __expand_var_regexp__.sub(var_sub, s)
  70. s = __expand_python_regexp__.sub(python_sub, s)
  71. if s == olds: break
  72. if type(s) is not types.StringType: # sanity check
  73. bb.msg.error(bb.msg.domain.Data, 'expansion of %s returned non-string %s' % (olds, s))
  74. except KeyboardInterrupt:
  75. raise
  76. except:
  77. bb.msg.note(1, bb.msg.domain.Data, "%s:%s while evaluating:\n%s" % (sys.exc_info()[0], sys.exc_info()[1], s))
  78. raise
  79. if varname:
  80. self.expand_cache[varname] = s
  81. return s
  82. def initVar(self, var):
  83. self.expand_cache = {}
  84. if not var in self.dict:
  85. self.dict[var] = {}
  86. def _findVar(self,var):
  87. _dest = self.dict
  88. while (_dest and var not in _dest):
  89. if not "_data" in _dest:
  90. _dest = None
  91. break
  92. _dest = _dest["_data"]
  93. if _dest and var in _dest:
  94. return _dest[var]
  95. return None
  96. def _makeShadowCopy(self, var):
  97. if var in self.dict:
  98. return
  99. local_var = self._findVar(var)
  100. if local_var:
  101. self.dict[var] = copy.copy(local_var)
  102. else:
  103. self.initVar(var)
  104. def setVar(self,var,value):
  105. self.expand_cache = {}
  106. match = __setvar_regexp__.match(var)
  107. if match and match.group("keyword") in __setvar_keyword__:
  108. base = match.group('base')
  109. keyword = match.group("keyword")
  110. override = match.group('add')
  111. l = self.getVarFlag(base, keyword) or []
  112. l.append([value, override])
  113. self.setVarFlag(base, keyword, l)
  114. # todo make sure keyword is not __doc__ or __module__
  115. # pay the cookie monster
  116. try:
  117. self._special_values[keyword].add( base )
  118. except:
  119. self._special_values[keyword] = Set()
  120. self._special_values[keyword].add( base )
  121. return
  122. if not var in self.dict:
  123. self._makeShadowCopy(var)
  124. if self.getVarFlag(var, 'matchesenv'):
  125. self.delVarFlag(var, 'matchesenv')
  126. self.setVarFlag(var, 'export', 1)
  127. # more cookies for the cookie monster
  128. if '_' in var:
  129. override = var[var.rfind('_')+1:]
  130. if not self._seen_overrides.has_key(override):
  131. self._seen_overrides[override] = Set()
  132. self._seen_overrides[override].add( var )
  133. # setting var
  134. self.dict[var]["content"] = value
  135. def getVar(self,var,exp):
  136. value = self.getVarFlag(var,"content")
  137. if exp and value:
  138. return self.expand(value,var)
  139. return value
  140. def delVar(self,var):
  141. self.expand_cache = {}
  142. self.dict[var] = {}
  143. def setVarFlag(self,var,flag,flagvalue):
  144. if not var in self.dict:
  145. self._makeShadowCopy(var)
  146. self.dict[var][flag] = flagvalue
  147. def getVarFlag(self,var,flag):
  148. local_var = self._findVar(var)
  149. if local_var:
  150. if flag in local_var:
  151. return copy.copy(local_var[flag])
  152. return None
  153. def delVarFlag(self,var,flag):
  154. local_var = self._findVar(var)
  155. if not local_var:
  156. return
  157. if not var in self.dict:
  158. self._makeShadowCopy(var)
  159. if var in self.dict and flag in self.dict[var]:
  160. del self.dict[var][flag]
  161. def setVarFlags(self,var,flags):
  162. if not var in self.dict:
  163. self._makeShadowCopy(var)
  164. for i in flags.keys():
  165. if i == "content":
  166. continue
  167. self.dict[var][i] = flags[i]
  168. def getVarFlags(self,var):
  169. local_var = self._findVar(var)
  170. flags = {}
  171. if local_var:
  172. for i in self.dict[var].keys():
  173. if i == "content":
  174. continue
  175. flags[i] = self.dict[var][i]
  176. if len(flags) == 0:
  177. return None
  178. return flags
  179. def delVarFlags(self,var):
  180. if not var in self.dict:
  181. self._makeShadowCopy(var)
  182. if var in self.dict:
  183. content = None
  184. # try to save the content
  185. if "content" in self.dict[var]:
  186. content = self.dict[var]["content"]
  187. self.dict[var] = {}
  188. self.dict[var]["content"] = content
  189. else:
  190. del self.dict[var]
  191. def createCopy(self):
  192. """
  193. Create a copy of self by setting _data to self
  194. """
  195. # we really want this to be a DataSmart...
  196. data = DataSmart(seen=self._seen_overrides.copy(), special=self._special_values.copy())
  197. data.dict["_data"] = self.dict
  198. return data
  199. # Dictionary Methods
  200. def keys(self):
  201. def _keys(d, mykey):
  202. if "_data" in d:
  203. _keys(d["_data"],mykey)
  204. for key in d.keys():
  205. if key != "_data":
  206. mykey[key] = None
  207. keytab = {}
  208. _keys(self.dict,keytab)
  209. return keytab.keys()
  210. def __getitem__(self,item):
  211. #print "Warning deprecated"
  212. return self.getVar(item, False)
  213. def __setitem__(self,var,data):
  214. #print "Warning deprecated"
  215. self.setVar(var,data)