tools/install-decoders

   1 #!/usr/bin/env python3
   2 #
   3 # This file is part of the libsigrokdecode project.
   4 #
   5 # Copyright (C) 2012 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):
  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         worklist.append((pd, pd_dir, install_list))
  43
  44     print("Installing %d protocol decoders:" % len(worklist))
  45     col = 0
  46     for pd, pd_dir, install_list in worklist:
  47         msg = pd + ' '
  48         if (col + len(msg) > 80):
  49             print()
  50             col = 0
  51         print(msg, end='')
  52         col += len(msg)
  53         pd_dst = dstdir + '/' + pd
  54         try:
  55             os.mkdir(pd_dst)
  56         except FileExistsError:
  57             pass
  58         for f in install_list:
  59             copy(pd_dir + '/' + f, pd_dst)
  60     print()
  61
  62
  63 def config_get_extra_install(config_file):
  64     install_list = []
  65     for line in open(config_file).read().split('\n'):
  66         line = line.strip()
  67         if len(line) == 0 or line[0] == '#':
  68             continue
  69         words = line.split()
  70         if words[0] != 'extra-install':
  71             continue
  72         install_list.extend(words[1:])
  73
  74     return install_list
  75
  76
  77 def usage(msg=None):
  78     if msg:
  79         print(msg)
  80         ret = 1
  81     else:
  82         ret = 0
  83     print("""Usage:
  84     install-decoders [-i <decoder source>] -o <install path>""")
  85     sys.exit(ret)
  86
  87
  88 #
  89 # main
  90 #
  91
  92 src = 'decoders'
  93 dst = None
  94 try:
  95     opts, args = getopt(sys.argv[1:], 'i:o:')
  96     for opt, arg in opts:
  97         if opt == '-i':
  98             src = arg
  99         elif opt == '-o':
 100             dst = arg
 101 except Exception as e:
 102     usage(str(e))
 103
 104 if len(args) != 0 or dst is None:
 105     usage()
 106
 107 install(src, dst)
 108
 109