copy_header.py

import shutil
import subprocess
import sys
import os
from util import *

def GetUserPath():
  user = os.getcwd().split('/')[1]
  if user == 'root':
    user_path = '/root/'
  elif user == 'home':
    user_path = '/home/' + os.getcwd().split('/')[2] + '/'
  else:
    print('unknown user!%s' %(os.getcwd()))
    sys.exit(1)
  return user_path

def Copy():
  if not os.path.exists('./lib-hft'):
    print("%s not existed!"%('./lib-hft'))
    return
  dir_list = ['struct', 'util', 'core']
  header_path = 'lib-hft/include/'
  target_path = 'external/common/include/'
  for dl in dir_list:
    if not os.path.exists(header_path+dl):
      continue
    for f in os.listdir(header_path+dl):
      if f.split('.')[-1] == 'h' or f.split('.')[-1] == 'hpp':
        print('copying %s from %s to %s' % (f, header_path+dl, target_path+dl))
        command = 'cp -rf %s %s' % (header_path+dl+"/"+f, target_path+dl)
        os.system(command)
  if not os.path.exists(header_path):
    return
  for f in os.listdir(header_path):
    if f.split('.')[-1] != 'h' and f.split('.')[-1] != 'hpp':
      continue
    shutil.copy(header_path+f, target_path)
    print('copying %s to %s' % (header_path+f, target_path))
  if not os.path.exists("./lib-hft/make_dir/libnick.so"):
    return
  shutil.copy("./lib-hft/make_dir/libnick.so", './external/common/lib')

def Install(topic='lib-hft'):
  if not os.path.exists(topic):
    RedPrint("%s not existed!"%(topic))
    return
  if os.path.getsize(topic) < 100:
    RedPrint(topic, 'size is too small', os.path.getsize(topic))
    return
  #os.system('cd %s; cmake .; make install' %(topic))
  os.system('cd %s; install -d build; cd build; cmake ..; make install' %(topic))

if __name__ == '__main__':
  Install()