write a python script to generate tarball

richardcypher · richardcypher · commit f4f09789db62 · 2013-09-12T10:54:54.000+08:00
diff --git a/tools/make-nw-headers.py b/tools/make-nw-headers.py
@@ -0,0 +1,91 @@
+#!/usr/bin/env python
+import os
+import tarfile
+import sys
+import getnwversion
+import shutil
+import distutils.core
+import re
+
+script_dir = os.path.dirname(__file__)
+nw_root  = os.path.normpath(os.path.join(script_dir, os.pardir))
+project_root = os.path.normpath(os.path.join(nw_root, os.pardir, os.pardir)) 
+third_party_dir = os.path.normpath(os.path.join(project_root, 'third_party'))
+tmp_dir = tmp_dir = os.path.normpath(os.path.join(nw_root, 'tmp'))
+
+nw_version = getnwversion.nw_version
+
+#parse command line arguments
+'''
+-t, the version of nw-headers to download
+'''
+if '-t' in sys.argv:
+  nw_version = sys.argv[sys.argv.index('-t') + 1]
+tarname = 'node-v' + nw_version + '.tar.gz'
+tarpath = os.path.join(tmp_dir, tarname)
+
+#make tmpdir
+if os.path.exists(tmp_dir):
+  pass
+else:
+  os.mkdir(tmp_dir)
+
+# prepare the files to compress
+print 'Begin copy file'
+base = os.path.join(third_party_dir, 'node')
+for dirpath, dirnames, filenames in os.walk(base):
+  relpath = dirpath.replace(third_party_dir + os.sep, '')
+  for dirs in dirnames:
+    if dirs =='gyp' or dirs == 'gyp_addon':
+      try:
+        shutil.copytree(os.path.join(dirpath, dirs), os.path.join(tmp_dir, relpath, dirs))
+      except:
+        distutils.dir_util.copy_tree(os.path.join(dirpath, dirs), os.path.join(tmp_dir, relpath, dirs))
+  for files in filenames:
+    if files.endswith('.h') or files.endswith('.gypi') or files == 'gyp' or files == 'gyp_addon':
+      if not os.path.exists(os.path.join(tmp_dir, relpath)):
+        os.makedirs(os.path.join(tmp_dir, relpath))
+      shutil.copyfile(os.path.join(dirpath, files), os.path.join(tmp_dir, relpath, files))
+shutil.rmtree(os.path.join(tmp_dir, 'node', 'deps', 'v8'))
+base = os.path.join(project_root, 'v8')
+for dirpath, dirnames, filenames in os.walk(base):
+  relpath = dirpath.replace(project_root + os.sep, '')
+  relpath = os.path.join('node', 'deps', relpath)
+  for dirs in dirnames:
+    if dirs == 'gyp' or dirs == 'gyp_addon':
+      try:
+        shutil.copytree(os.path.join(dirpath, dirs), os.path.join(tmp_dir, relpath, dirs))
+      except:
+        distutils.dir_util.copy_tree(os.path.join(dirpath, dirs), os.path.join(tmp_dir, relpath, dirs))
+  for files in filenames:
+    if files.endswith('.h') or files.endswith('.gypi') or files == 'gyp' or files == 'gyp_addon':
+      if not os.path.exists(os.path.join(tmp_dir, relpath)):
+        os.makedirs(os.path.join(tmp_dir, relpath))
+      shutil.copyfile(os.path.join(dirpath, files), os.path.join(tmp_dir, relpath, files))
+if os.path.exists(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'build')):
+  shutil.rmtree(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'build'))
+if os.path.exists(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'test')):   
+  shutil.rmtree(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'test'))
+if os.path.exists(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'out')):
+  shutil.rmtree(os.path.join(tmp_dir, 'node', 'deps', 'v8', 'out'))
+if os.path.exists(os.path.join(tmp_dir, 'node', 'deps', 'npm', 'node_modules')):
+  shutil.rmtree(os.path.join(tmp_dir, 'node', 'deps', 'npm', 'node_modules'))
+
+rfile = open(os.path.join(tmp_dir, 'node', 'src', 'node.h'), 'r')
+filer = rfile.read()
+sub = re.sub('third_party/node/deps/uv/include/uv.h', 'uv.h', filer, 0)
+rfile.close()
+wfile = open(os.path.join(tmp_dir, 'node', 'src', 'node.h'), 'w')
+wfile.write(sub)
+wfile.close()
+
+print 'copy file end'
+print 'Begin compress file'
+
+tar = tarfile.open(tarpath, 'w:gz')
+for dirpath, dirnames, filenames in os.walk(tmp_dir):
+  for name in filenames:
+    path = os.path.normpath(os.path.join(dirpath, name))
+    tar.add(path, path.replace(tmp_dir + os.sep, ''))
+tar.close()
+print 'compress end'
