From f4ba7e980384eddd0a04eea3ffca694384a599c9 Mon Sep 17 00:00:00 2001
From: ge <gechandev@gmail.com>
Date: Fri, 8 Apr 2022 21:27:45 +0300
Subject: [PATCH] Add main() function

---
 setup.py    | 3 ++-
 twvdscli.py | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 27bac68..87f3f9a 100644
--- a/setup.py
+++ b/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'
         ]
     }
 )
diff --git a/twvdscli.py b/twvdscli.py
index e088be5..95e999b 100755
--- a/twvdscli.py
+++ b/twvdscli.py
@@ -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()