Add wv_cli.py, add Click requirement, change entry point

This commit is contained in:
Daniel Rojas 2021-10-02 11:40:36 +02:00 committed by Laurier Loiselle
parent 628c80fb4d
commit ef6af32a01
No known key found for this signature in database
GPG Key ID: 345920CC72089A3F
3 changed files with 34 additions and 1 deletions

View File

@ -1,3 +1,4 @@
click
graphviz
pillow
pyyaml

View File

@ -17,6 +17,7 @@ setup(
long_description=open(README_PATH).read(),
long_description_content_type='text/markdown',
install_requires=[
'click',
'pyyaml',
'pillow',
'graphviz',
@ -27,7 +28,9 @@ setup(
package_dir={'': 'src'},
packages=find_packages('src'),
entry_points={
'console_scripts': ['wireviz=wireviz.wireviz:main'],
'console_scripts': [
'wireviz=wireviz.wv_cli:main',
],
},
classifiers=[
'Development Status :: 4 - Beta',

29
src/wireviz/wv_cli.py Normal file
View File

@ -0,0 +1,29 @@
import os
import sys
import click
import wireviz.wireviz
@click.command()
@click.argument('filepath', nargs=-1)
@click.option('-p', '--prepend', default=None)
@click.option('-o', '--output', default='hpst')
@click.option('-V', '--version', is_flag=True, default=False)
def main(filepath, prepend, output, version):
print('WireViz!')
# get list of files
try:
_ = iter(filepath)
except TypeError:
filepaths = [filepath]
else:
filepaths = list(filepath)
for f in filepaths:
print(f)
print()
if __name__ == '__main__':
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
wireviz()