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