lulzette/twvdscli
2
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

Fixed access_token

Browse Source
This commit is contained in:
Lulzette 2022-04-07 16:00:14 +03:00
parent 3898468084
commit ca25bbf388
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
twvdscli.py
View File

@ -8,6 +8,7 @@ import typer
import os import os
import configparser import configparser
import base64 import base64
import prettytable
app = typer.Typer() app = typer.Typer()
servers_app = typer.Typer() servers_app = typer.Typer()
@ -100,16 +101,14 @@ def vds_list():
print('{0:6} {1:15} {2:20} {3:16} {4:2} {5:5} {6:4} '.format(i['id'], state, i['name'], i['ip'], i['configuration']['cpu'], i['configuration']['ram'], i['configuration']['disk_size'])) print('{0:6} {1:15} {2:20} {3:16} {4:2} {5:5} {6:4} '.format(i['id'], state, i['name'], i['ip'], i['configuration']['cpu'], i['configuration']['ram'], i['configuration']['disk_size']))
def auth(login, passwd): def auth(based):
""" """
Convert login:pass into base64 and make a request Get access token
:param login: :param based:
:param passwd:
:return: :return:
""" """
based = base64.b64encode(str(login+":"+passwd).encode('utf-8')) headers = {"Authorization": "Basic " + based}
headers = {"Authorization": "Basic " + based.decode('utf-8')}
result = requests.post( result = requests.post(
'https://public-api.timeweb.com/api/v2/auth', 'https://public-api.timeweb.com/api/v2/auth',
@ -122,29 +121,34 @@ def auth(login, passwd):
result = result.content.decode('utf-8') result = result.content.decode('utf-8')
result = json.loads(result) result = json.loads(result)
result = result['access_token'] result = result['access_token']
print(result)
return result return result
def get_api_key(): def get_api_key():
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(os.path.join(os.getenv('HOME'), '.config', 'twvdscli.ini')) config.read(os.path.join(os.getenv('HOME'), '.config', 'twvdscli.ini'))
result = config.get('api', 'key', fallback=None) based = config.get('api', 'key', fallback=None)
if result is None: if based is None:
login = input("Enter login ") login = input("Enter login ")
passwd = input("Enter passwd ") passwd = input("Enter passwd ")
result = auth(login, passwd) based = base64.b64encode(str(login + ":" + passwd).encode('utf-8'))
based = based.decode('utf-8')
config.add_section('api') config.add_section('api')
config.set('api', 'key', result) config.set('api', 'key', based)
with open(os.path.join(os.getenv('HOME'), '.config', 'twvdscli.ini'), 'w') as configfile: with open(os.path.join(os.getenv('HOME'), '.config', 'twvdscli.ini'), 'w') as configfile:
config.write(configfile) config.write(configfile)
result = auth(based)
return result return result
if __name__ == '__main__': if __name__ == '__main__':
apikey = get_api_key() apikey = get_api_key()
if apikey is None:
print(typer.style("Auth Error", fg=typer.colors.RED))
sys.exit(1)
reqHeader = {"Authorization": "Bearer " + apikey} reqHeader = {"Authorization": "Bearer " + apikey}
app() app()