Add main() function

This commit is contained in:
ge 2022-04-08 21:27:45 +03:00
parent ff8eaad69a
commit f4ba7e9803
2 changed files with 6 additions and 3 deletions

View File

@ -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'
]
}
)

View File

@ -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()