forked from mbj4668/pyang
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
89 lines (82 loc) · 3.48 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
from setuptools import setup
from setuptools import Distribution
from os.path import join
import pyang
import glob
import os
import re
import sys
modules_iana = glob.glob(os.path.join('modules', 'iana', '*.yang'))
modules_ietf = glob.glob(os.path.join('modules', 'ietf', '*.yang'))
xslt = glob.glob(os.path.join('xslt', '*.xsl'))
schema = glob.glob(os.path.join('schema', '*.rng'))
images = glob.glob(os.path.join('tools', 'images', '*'))
man1 = glob.glob(os.path.join('man', 'man1', '*.1'))
class PyangDist(Distribution):
"""The purpose of this subclass of Distribution is to extend the
install procedure with preprocessing of shell scripts and man
pages so that they reflect the actual installation prefix, which
may be changed through the --prefix option.
"""
def preprocess_files(self, prefix):
"""Change the installation prefix where necessary.
"""
if prefix is None: return
files = ("bin/yang2dsdl", "man/man1/yang2dsdl.1",
"pyang/plugins/jsonxsl.py")
regex = re.compile("^(.*)/usr/local(.*)$")
for f in files:
inf = open(f)
cnt = inf.readlines()
inf.close()
ouf = open(f,"w")
for line in cnt:
mo = regex.search(line)
if mo is None:
ouf.write(line)
else:
ouf.write(mo.group(1) + prefix + mo.group(2) +
"\n")
ouf.close()
def run_commands(self):
opts = self.command_options
if "install" in opts:
self.preprocess_files(opts["install"].get("prefix",
("", None))[1])
Distribution.run_commands(self)
# If the installation is on windows, place pyang.bat file in Scripts directory
if os.sep == '\\':
script_dir = join(sys.prefix, 'Scripts')
pyang_file = join(script_dir, 'pyang')
path = join(script_dir, 'pyang.bat')
with open(path, 'w') as script:
script.write('@echo off\n%s %s %%*\n' % ('python', pyang_file))
setup(name='pyang',
version=pyang.__version__,
author='Martin Bjorklund',
author_email='[email protected]',
description="A YANG (RFC 6020) validator and converter",
long_description="An extensible YANG (RFC 6020) validator. Provides a framwork for plugins that can convert YANG modules to other formats.",
url='https://github.com/mbj4668/pyang',
license='BSD',
classifiers=[
'Development Status :: 5 - Production/Stable',
'License :: OSI Approved :: BSD License',
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
],
keywords='YANG validator',
distclass=PyangDist,
scripts=['bin/pyang', 'bin/yang2html', 'bin/yang2dsdl', 'bin/json2xml'],
packages=['pyang', 'pyang.plugins', 'pyang.translators'],
data_files=[
('share/man/man1', man1),
('share/yang/modules/iana', modules_iana),
('share/yang/modules/ietf', modules_ietf),
('share/yang/xslt', xslt),
('share/yang/images', images),
('share/yang/schema', schema),
('etc/bash_completion.d', ['etc/bash_completion.d/pyang']),
]
)