git-svn-id: http://svn.calculate.ru/calculate2/calculate-lib/trunk@504 c91db197-33c1-4113-bf15-f8a5c547ca64

pym/cl_profile.py

@@ -1851,7 +1851,7 @@ class profile(_file, _terms):
         varStart = "#-"
         varEnd = "-#"
         self._reVar = re.compile(("%s[a-zA-Z0-9_-]+%s")%(varStart,varEnd),re.M)
-        self._reFunc = re.compile(("%s[a-zA-Z0-9_\-\+\(\)\,]+%s")\
+        self._reFunc = re.compile(("%s[a-zA-Z0-9_\-\+\(\)\, \*\/]+%s")\
             %(varStart,varEnd),re.M)
         self._deltVarStart = len(varStart)
         self._deltVarEnd = len(varEnd)
@@ -1909,16 +1909,19 @@ class profile(_file, _terms):
             resS = self._reVar.search(textProfileTmp)
         return textProfileTmp
 
+
     def applyFuncProfile(self, textProfile, nameProfile):
         """ Применяет функции к тексту профиля
         """
-        def equalTerm(term, sNum, localVars):
+        def equalTerm(term, sNum, sMD, localVars):
             """Локальная функция для вычисления выражения"""
             terms = sNum.findall(term)
             if terms:
                 strNumers = []
                 for n in terms:
                     strNum = n.strip()
+                    if "*" in strNum or "/" in strNum:
+                        strNum = multAndDiv(strNum,sNum,sMD,localVars)
                     try:
                         num = int(strNum)
                     except:
@@ -1942,16 +1945,33 @@ class profile(_file, _terms):
             print _("error profile term %s, incorrect data")%str(term)
             exit(1)
 
+        def multAndDiv(term,sNum,sMD,localVars):
+            """локальная функция для умножения и деления"""
+            termTmp = term
+            varsLocal = sMD.findall(term)
+            for var in varsLocal:
+                flagVarTxt = True
+                try:
+                    int(var)
+                except:
+                    flagVarTxt = False
+                if flagVarTxt:
+                    continue
+                varReplace = str(equalTerm(var,sNum,sMD,localVars))
+                termTmp = termTmp.replace(var,varReplace)
+            ret = eval(termTmp)
+            return ret
+
         def funcSum(funTxt,resS,localVars,textProfileTmp):
             """локальная функция вычисляет первую функцию sum() в профиле"""
-            terms = funTxt[4:-1].split(",")
+            terms = funTxt[4:-1].replace(" ","").split(",")
             # Название локальной переменной
-            nameLocVar = terms[0].strip()
+            nameLocVar = terms[0]
             if not localVars.has_key(nameLocVar):
                 localVars[nameLocVar] = 0
             if len(terms) == 2:
                 if terms[1].strip():
-                    localVars[nameLocVar] = equalTerm(terms[1],sNum,
+                    localVars[nameLocVar] = equalTerm(terms[1],sNum,sMD,
                                                         localVars)
                     replace = str(localVars[nameLocVar])
                 else:
@@ -1960,14 +1980,14 @@ class profile(_file, _terms):
                     textProfileTmp[resS.end():]
             elif len(terms) == 3:
                 if terms[1].strip():
-                    replaceInt = equalTerm(terms[1],sNum,localVars)
+                    replaceInt = equalTerm(terms[1],sNum,sMD,localVars)
                     localVars[nameLocVar] = replaceInt
                     replace = str(replaceInt)
                 else:
                     replaceInt = localVars[nameLocVar]
                     replace = ""
                 localVars[nameLocVar] =  equalTerm(terms[2],
-                                                   sNum, localVars)
+                                                   sNum,sMD,localVars)
                 textProfileTmp = textProfileTmp[:resS.start()] + replace +\
                     textProfileTmp[resS.end():]
             else:
@@ -1978,7 +1998,10 @@ class profile(_file, _terms):
 
         # Локальные переменные
         localVars = {}
+        # Регулярное выражние для сложения
         sNum = re.compile("\-[^\-\+]+|[^\-\+]+")
+        # Регулярное выражение для умножениея и деления
+        sMD = re.compile("[^\-\+\*\/]+")
         resS = self._reFunc.search(textProfile)
         textProfileTmp = textProfile
         while resS: