bin/addvars2notebook_GEOM.py

   1
   2 def addvars2notebook(filename, vars_and_values):
   3     stream = open(filename)
   4     lines = stream.readlines()
   5     stream.close()
   6     newlines = []
   7     for line in lines:
   8         if line.find("= geompy.") >= 0:
   9             name = line.split('=')[0]
  10             name = name.strip()
  11             vals = line
  12             fields = vals.split("(")
  13             if len(fields) == 2:
  14                 begin = fields[0] + "("
  15                 vals = fields[1]
  16                 fields = vals.split(")")
  17                 if len(fields) == 2:
  18                     vals = fields[0]
  19                     end = ")" + fields[1]
  20                     vals = vals.split(',')
  21                     newline  = begin
  22                     newvals = []
  23                     for i in range(len(vals)):
  24                         valname = name + "_val_%s"%(i+1)
  25                         val = vals[i]
  26                         vvv = val.strip()
  27                         try:
  28                             iii = int(vvv)
  29                             vars_and_values.append([valname, val])
  30                             val = val.replace(vvv, valname.__repr__())
  31                         except ValueError:
  32                             try:
  33                                 fff = float(vvv)
  34                                 vars_and_values.append([valname, val])
  35                                 val = val.replace(vvv, valname.__repr__())
  36                             except ValueError:
  37                                 pass
  38                             pass
  39                         newvals.append(val)
  40                         pass
  41                     newline += ','.join(newvals)
  42                     newline += end
  43                     line = newline
  44                     pass
  45                 pass
  46             pass
  47         newlines.append(line)
  48         pass
  49     content = "".join(newlines)
  50     return content