Packaging

MANIFEST.in

@@ -1 +1 @@
-include README.md HISTORY.rst templates/*
+include README.md HISTORY.rst

ntc_templates/parse.py

@@ -1,15 +1,11 @@
 """ntc_templates.parse."""
-import os
-import sys
+import pkg_resources
 from textfsm.clitable import CliTableError
 import textfsm.clitable as clitable
 
 
 def _get_template_dir():
-    ntc_template_abspath = os.path.abspath(sys.modules['ntc_templates'].__file__)
-    base_dir = os.path.dirname(ntc_template_abspath)
-    template_dir = '%s%s%s' % (base_dir, os.sep, 'templates')
-    return template_dir
+    return pkg_resources.resource_filename("ntc_templates", "templates")
 
 
 def _clitable_to_dict(cli_table):

ntc_templates/templates

@@ -0,0 +1 @@
+../templates

setup.py

@@ -1,43 +1,20 @@
-"""setup.py file."""
-import re
+from ntc_templates import __version__
 from codecs import open
-from glob import glob
 from setuptools import setup
-import os
-import shutil
-
-version = ''
-with open('ntc_templates/__init__.py', 'r') as fd:
-    version = re.search(r'^__version__\s*=\s*[\'"]([^\'"]*)[\'"]',
-                        fd.read(), re.MULTILINE).group(1)
-
-if not version:
-    raise RuntimeError('Cannot find version information')
 
 with open('README.md', 'r', 'utf-8') as f:
     readme = f.read()
 
 with open('HISTORY.rst', 'r', 'utf-8') as f:
     history = f.read()
 
-
 long_description = readme + '\n\n' + history
 
-
-template_files = glob('templates/*')
-
-if os.path.islink('ntc_templates/templates'):
-    os.unlink('ntc_templates/templates')
-elif os.path.isdir('ntc_templates/templates'):
-    shutil.rmtree('ntc_templates/templates')
-
-os.symlink('../templates', 'ntc_templates/templates')
 config = {
     'name': 'ntc_templates',
-    # 'package_dir': {'': 'lib'},
     'packages': ['ntc_templates'],
-    'version': version,
-    'package_data': {'ntc_templates': template_files},
+    'version': __version__,
+    'package_data': {'ntc_templates': ['templates/*']},
     'description': 'Package to return structured data from the output of network devices.',
     'long_description': long_description,
     'author': 'network.toCode()',
@@ -50,10 +27,8 @@
     'classifiers': ['Development Status :: 4 - Beta',
                     'Intended Audience :: Developers',
                     'Intended Audience :: System Administrators',
-                    'Programming Language :: Python :: 2.7']
+                    'Programming Language :: Python :: 2.7'],
+    'zip_safe': False
 }
 
 setup(**config)
-
-if os.path.islink('ntc_templates/templates'):
-    os.unlink('ntc_templates/templates')