fluzzi32/connpy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update big

Browse Source
This commit is contained in:
fluzzi
2022-03-18 15:32:48 -03:00
parent 4a593f2016
commit 255b2bd4ef
3 changed files with 173 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
conn/configfile.py
View File

@ -15,7 +15,7 @@ from Crypto.PublicKey import RSA
class configfile:
def __init__(self, conf = None):
def __init__(self, conf = None, *, key = None):
home = os.path.expanduser("~")
self.defaultdir = home + '/.config/conn'
self.defaultfile = self.defaultdir + '/config.yaml'
@ -24,8 +24,11 @@ class configfile:
self.dir = self.defaultdir
self.file = self.defaultfile
else:
self.dir = os.path.dirname(conf)
self.file = conf
if key == None:
self.key = self.defaultkey
else:
self.key = key
if os.path.exists(self.file):
config = self.loadconfig(self.file)
else:
@ -33,23 +36,85 @@ class configfile:
self.config = config["config"]
self.connections = config["connections"]
self.profiles = config["profiles"]
if not os.path.exists(self.key):
self.createkey(self.key)
self.privatekey = RSA.import_key(open(self.key).read())
self.publickey = self.privatekey.publickey()
def loadconfig(self, conf):
ymlconf = open(conf)
return yaml.load(ymlconf.read(), Loader=yaml.FullLoader)
return yaml.load(ymlconf.read(), Loader=yaml.CLoader)
def createconfig(self, conf):
defaultconfig = {'config': {'case': False, 'frun': False, 'idletime': 30}, 'connections': {}, 'profiles': {}}
defaultconfig = {'config': {'case': False, 'frun': False, 'idletime': 30}, 'connections': {}, 'profiles': { "default": { "host":"", "protocol":"ssh", "port":"", "user":"", "password":"", "options":"", "logs":"" }}}
if not os.path.exists(conf):
with open(conf, "w") as f:
yaml.dump(defaultconfig, f, explicit_start=True)
yaml.dump(defaultconfig, f, explicit_start=True, Dumper=yaml.CDumper)
f.close()
ymlconf = open(conf)
return yaml.load(ymlconf.read(), Loader=yaml.FullLoader)
return yaml.load(ymlconf.read(), Loader=yaml.CLoader)
def saveconfig(self, conf):
newconfig = {"config":{}, "connections": {}, "profiles": {}}
newconfig["config"] = self.config
newconfig["connections"] = self.connections
newconfig["profiles"] = self.profiles
with open(conf, "w") as f:
yaml.dump(newconfig, f, explicit_start=True, Dumper=yaml.CDumper)
f.close()
def createkey(self, keyfile):
key = RSA.generate(2048)
with open('mykey.pem','wb') as f:
with open(keyfile,'wb') as f:
f.write(key.export_key('PEM'))
f.close()
def _explode_unique(self, unique):
uniques = unique.split("@")
result = {"id": uniques[0]}
if len(uniques) == 2:
result["folder"] = uniques[1]
elif len(uniques) == 3:
result["folder"] = uniques[2]
result["subfolder"] = uniques[1]
return result
def _connections_add(self,*, id, host, folder='', subfolder='', options='', logs='', password='', port='', protocol='', user='' ):
if folder == '':
self.connections[id] = {"host": host, "options": options, "logs": logs, "password": password, "port": port, "protocol": protocol, "user": user, "type": "connection"}
elif folder != '' and subfolder == '':
self.connections[folder][id] = {"host": host, "options": options, "logs": logs, "password": password, "port": port, "protocol": protocol, "user": user, "type": "connection"}
elif folder != '' and subfolder != '':
self.connections[folder][subfolder][id] = {"host": host, "options": options, "logs": logs, "password": password, "port": port, "protocol": protocol, "user": user, "type": "connection"}
def _connections_del(self,*, id, folder='', subfolder=''):
if folder == '':
del self.connections[id]
elif folder != '' and subfolder == '':
del self.connections[folder][id]
elif folder != '' and subfolder != '':
del self.connections[folder][subfolder][id]
def _folder_add(self,*, folder, subfolder = ''):
if subfolder == '':
if folder not in self.connections:
self.connections[folder] = {"type": "folder"}
else:
if subfolder not in self.connections[folder]:
self.connections[folder][subfolder] = {"type": "subfolder"}
def _folder_del(self,*, folder, subfolder=''):
if subfolder == '':
del self.connections[folder]
else:
del self.connections[folder][subfolder]
def _profiles_add(self,*, id, host = '', options='', logs='', password='', port='', protocol='', user='' ):
self.profiles[id] = {"host": host, "options": options, "logs": logs, "password": password, "port": port, "protocol": protocol, "user": user}
def _profiles_del(self,*, id ):
del self.profiles[id]