wizard/create_preset.py

   1 import pickle
   2 import sys
   3 import os
   4 import shutil
   5 import pprint
   6 import glob
   7 import re
   8
   9 def remove(path):
  10     if os.path.isdir(path):
  11         shutil.rmtree(path, ignore_errors=True)
  12     else:
  13         try:
  14             os.remove(path)
  15         except OSError:
  16             pass
  17
  18 def findPath(start, target):
  19     pa = start
  20     while pa != "/":
  21         pa = os.path.abspath(pa + "/..")
  22         print pa
  23         if os.path.exists(pa + "/" + target):
  24             return os.path.relpath(pa, start)
  25
  26 if len(sys.argv) < 3:
  27     print "Usage: %s <project_dir> <preset_dir>" % sys.argv[0]
  28     exit(0)
  29
  30 prj_dir = sys.argv[1]
  31 preset_dir = sys.argv[2]
  32
  33 if not os.path.exists(prj_dir + "/project.bertos"):
  34     print "%s does not seem a Wizard created project." % prj_dir
  35     exit(1)
  36
  37 p = open(prj_dir + "/project.bertos")
  38 s = pickle.load(p)
  39 if s["WIZARD_VERSION"] < 3:
  40     print "Project version too old."
  41     exit(1)
  42 pname = s["PROJECT_NAME"]
  43 preset_dir += pname + ".test"
  44
  45 hw_path = findPath(preset_dir, "hw")
  46 if not hw_path:
  47     print "hw/ path not found in parent directories of %s" % preset_dir
  48     exit(1)
  49
  50 bertos_path = findPath(preset_dir, "bertos")
  51 if not bertos_path:
  52     print "bertos/ path not found in parent directories of %s" % preset_dir
  53     exit(1)
  54
  55 shutil.copytree(prj_dir, preset_dir)
  56 remove(preset_dir + "/bertos")
  57 remove(preset_dir + "/images")
  58 remove(preset_dir + "/obj")
  59 remove(preset_dir + "/Makefile")
  60 remove(preset_dir + "/buildrev.h")
  61 remove(preset_dir + "/VERSION")
  62 remove(preset_dir + "/" + pname + ".project")
  63 remove(preset_dir + "/" + pname + ".workspace")
  64
  65 l = glob.glob(preset_dir + "/" + pname + "/*")
  66 for f in l:
  67     shutil.move(f, preset_dir)
  68
  69 remove(preset_dir + "/" + pname)
  70 remove(preset_dir + "/hw")
  71
  72 assert(os.path.exists(preset_dir + "/" + hw_path     + "/hw"))
  73 assert(os.path.exists(preset_dir + "/" + bertos_path + "/bertos"))
  74
  75 s["BERTOS_PATH"] = bertos_path
  76 s["PRESET"] = True
  77 s["PROJECT_HW_PATH"] = hw_path
  78 s["PROJECT_SRC_PATH"] = "."
  79 s["PRESET"] = True
  80 pprint.pprint(s)
  81 p = open(preset_dir + "/project.bertos", "w")
  82 pickle.dump(s, p)
  83
  84 p = open(preset_dir + "/.spec", "w")
  85 p.write("name = '%s preset'" % pname)
  86 p.close()
  87
  88 bertos_path = os.path.abspath(preset_dir + "/" + bertos_path)
  89 hw_path = os.path.abspath(preset_dir + "/" + hw_path)
  90
  91 src_path = os.path.relpath(preset_dir, bertos_path)
  92 hw_path  = os.path.relpath(hw_path, bertos_path)
  93
  94 mk = open(preset_dir + "/" + pname + ".mk").read()
  95 mk = re.sub(r"(%s_SRC_PATH\s*=\s*).*" % pname, r"\1%s" % src_path, mk)
  96 mk = re.sub(r"(%s_HW_PATH\s*=\s*).*" % pname, r"\1%s" % hw_path, mk)
  97
  98 #remove absolute path from toolchain
  99 mk = re.sub(r'(%s_PREFIX\s*=\s*").*?([^/]*")' % pname, r'\1\2', mk)
 100 open(preset_dir + "/" + pname + ".mk", "w").write(mk)