first commit

conn/configfile.py

@@ -0,0 +1,55 @@
+#!/usr/bin/python3
+#Imports
+import yaml
+import os
+import re
+from . import tools
+from Crypto.PublicKey import RSA
+
+
+#Constants
+
+#Variables
+
+#functions and clsses
+
+class configfile:
+    
+    def __init__(self, conf = None):
+        home = os.path.expanduser("~")
+        self.defaultdir = home + '/.config/conn'
+        self.defaultfile = self.defaultdir + '/config.yaml'
+        self.defaultkey = self.defaultdir + '/.osk'
+        if conf == None:
+            self.dir = self.defaultdir
+            self.file = self.defaultfile
+        else:
+            self.dir = os.path.dirname(conf)
+            self.file = conf
+        if os.path.exists(self.file):
+            config = self.loadconfig(self.file)
+        else:
+            config = self.createconfig(self.file)
+        self.config = config["config"]
+        self.connections = config["connections"]
+        self.profiles = config["profiles"]
+
+
+    def loadconfig(self, conf):
+        ymlconf = open(conf)
+        return yaml.load(ymlconf.read(), Loader=yaml.FullLoader)
+
+    def createconfig(self, conf):
+        defaultconfig = {'config': {'case': False, 'frun': False, 'idletime': 30}, 'connections': {}, 'profiles': {}}
+        if not os.path.exists(conf):
+            with open(conf, "w") as f:
+                yaml.dump(defaultconfig, f, explicit_start=True)
+                f.close()
+        ymlconf = open(conf)
+        return yaml.load(ymlconf.read(), Loader=yaml.FullLoader)
+
+    def createkey(self, keyfile):
+        key = RSA.generate(2048)
+        with open('mykey.pem','wb') as f:
+            f.write(key.export_key('PEM'))
+            f.close()