Add main() function
This commit is contained in:
parent
ff8eaad69a
commit
f4ba7e9803
3
setup.py
3
setup.py
@ -20,13 +20,14 @@ setup(
|
||||
python_requires = '>=3.6',
|
||||
py_modules = ['twvdscli'],
|
||||
install_requires = [
|
||||
'click==7.1.1'
|
||||
'typer==0.4.0',
|
||||
'prettytable==3.2.0',
|
||||
'wcwidth==0.2.5'
|
||||
],
|
||||
entry_points = {
|
||||
'console_scripts': [
|
||||
'twvdscli = twvdscli'
|
||||
'twvdscli = twvdscli:main'
|
||||
]
|
||||
}
|
||||
)
|
||||
|
@ -323,11 +323,13 @@ def get_api_key():
|
||||
result = auth(based)
|
||||
return result
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
def main():
|
||||
apikey = get_api_key()
|
||||
if apikey is None:
|
||||
print(typer.style("Auth Error", fg=typer.colors.RED))
|
||||
sys.exit(1)
|
||||
reqHeader = {"Authorization": "Bearer " + apikey}
|
||||
app()
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
Loading…
Reference in New Issue
Block a user