tools/install-decoders

   1 #!/usr/bin/env python3
   2 ##
   3 ## This file is part of the libsigrokdecode project.
   4 ##
   5 ## Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
   6 ##
   7 ## This program is free software; you can redistribute it and/or modify
   8 ## it under the terms of the GNU General Public License as published by
   9 ## the Free Software Foundation; either version 3 of the License, or
  10 ## (at your option) any later version.
  11 ##
  12 ## This program is distributed in the hope that it will be useful,
  13 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 ## GNU General Public License for more details.
  16 ##
  17 ## You should have received a copy of the GNU General Public License
  18 ## along with this program; if not, see <http://www.gnu.org/licenses/>.
  19 ##
  20
  21 import os
  22 import sys
  23 from shutil import copy
  24 from getopt import getopt
  25
  26
  27 def install(srcdir, dstdir, s):
  28     worklist = []
  29     for pd in os.listdir(srcdir):
  30         pd_dir = srcdir + '/' + pd
  31         if not os.path.isdir(pd_dir):
  32             continue
  33         install_list = []
  34         for f in os.listdir(pd_dir):
  35             pd_file = pd_dir + '/' + f
  36             if not os.path.isfile(pd_file):
  37                 continue
  38             if f == 'config':
  39                 install_list.extend(config_get_extra_install(pd_file))
  40             elif f[-3:] == '.py':
  41                 install_list.append(f)
  42         if install_list:
  43             worklist.append((pd, pd_dir, install_list))
  44
  45     print("Installing %d %s:" % (len(worklist), s))
  46     col = 0
  47     for pd, pd_dir, install_list in worklist:
  48         msg = pd + ' '
  49         if (col + len(msg) > 80):
  50             print()
  51             col = 0
  52         print(msg, end='')
  53         col += len(msg)
  54         pd_dst = os.path.join(dstdir, pd)
  55         try:
  56             os.mkdir(pd_dst)
  57         except OSError as e:
  58             if e.errno != os.errno.EEXIST:
  59                 raise
  60             else:
  61                 pass
  62         for f in install_list:
  63             copy(os.path.join(pd_dir, f), pd_dst)
  64     print()
  65
  66
  67 def config_get_extra_install(config_file):
  68     install_list = []
  69     for line in open(config_file).read().split('\n'):
  70         line = line.strip()
  71         if len(line) == 0 or line[0] == '#':
  72             continue
  73         words = line.split()
  74         if words[0] != 'extra-install':
  75             continue
  76         install_list.extend(words[1:])
  77
  78     return install_list
  79
  80
  81 def usage(msg=None):
  82     if msg:
  83         print(msg)
  84         ret = 1
  85     else:
  86         ret = 0
  87     print("""Usage:
  88     install-decoders [-i <decoder source>] -o <install path>""")
  89     sys.exit(ret)
  90
  91
  92 #
  93 # main
  94 #
  95
  96 src = 'decoders'
  97 dst = None
  98 try:
  99     opts, args = getopt(sys.argv[1:], 'i:o:')
 100     for opt, arg in opts:
 101         if opt == '-i':
 102             src = arg
 103         elif opt == '-o':
 104             dst = arg
 105 except Exception as e:
 106     usage(str(e))
 107
 108 if len(args) != 0 or dst is None:
 109     usage()
 110
 111 install(src, dst, 'protocol decoders')
 112 install(src + '/common', dst + '/common', 'common modules')
 113
 114