From 52f21e22b26d6d652e3fcdbb09c8d57c50aea143 Mon Sep 17 00:00:00 2001
From: Lulzette <ivan@lulzette.ru>
Date: Sun, 27 Mar 2022 20:12:52 +0300
Subject: [PATCH] Working Init

---
 main.py | 86 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..74e7839
--- /dev/null
+++ b/main.py
@@ -0,0 +1,86 @@
+#!/usr/bin/python3
+import json
+import configparser
+import requests
+import typer
+import os
+
+app = typer.Typer()
+
+
+class Servers:
+    @staticmethod
+    def list():
+        response = requests.get("https://api.cloudvps.reg.ru/v1/reglets", headers=reqHeader)
+        return response.json()
+
+    @staticmethod
+    def get_systems():
+        response = requests.get("https://api.cloudvps.reg.ru/v1/images?type=distribution", headers=reqHeader)
+        return response.json()
+
+    @staticmethod
+    def create(name, tariff, image):
+        response = requests.post("https://api.cloudvps.reg.ru/v1/reglets",
+                                 headers=reqHeader,
+                                 json={"name": name,
+                                       "size": tariff,
+                                       "image": image
+                                       }
+                                 )
+        return response.json()
+
+
+def get_balance():
+    response = requests.get("https://api.cloudvps.reg.ru/v1/balance_data", headers=reqHeader)
+    return response.json()
+
+
+@app.command()
+def get_tariffs():
+    response = requests.get("https://api.cloudvps.reg.ru/v1/prices", headers=reqHeader)
+    print(json.dumps(response.json(),indent=4))
+
+
+@app.command()
+def balance():
+    print(get_balance())
+
+
+@app.command()
+def list_servers():
+    print(Servers.list())
+
+
+@app.command()
+def list_os():
+    systems_list = Servers.get_systems()['images']
+    for i in systems_list:
+        print(i['slug'])
+
+
+@app.command()
+def create_server(name: str = typer.Option(...), tariff: str = typer.Option(...), image: str = typer.Option(...)):
+    print(Servers.create(name=name, tariff=tariff, image=image))
+
+
+def get_api_key():
+    config = configparser.ConfigParser()
+    config.read(os.path.join(os.getenv('XDG_CONFIG_HOME'), 'regructl.ini'))
+    result = config.get('api', 'key', fallback=None)
+    if result is None:
+        result = input("Enter API Key ")
+        config.add_section('api')
+        config.set('api', 'key', result)
+
+        with open(os.path.join(os.getenv('XDG_CONFIG_HOME'), 'regructl.ini'), 'w') as configfile:
+            config.write(configfile)
+
+    return result
+
+
+if __name__ == '__main__':
+    apikey = get_api_key()
+    reqHeader = {"Authorization": "Bearer " + apikey}
+
+    app()