102 lines
3.7 KiB
Python
102 lines
3.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
#-*- coding: utf-8 -*-
|
||
|
|
# impra/ini.py
|
||
|
|
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
|
||
|
|
#
|
||
|
|
# software : ImpraStorage <http://kirmah.sourceforge.net/>
|
||
|
|
# version : 1.01
|
||
|
|
# date : 2014
|
||
|
|
# licence : GPLv3.0 <http://www.gnu.org/licenses/>
|
||
|
|
# author : a-Sansara <[a-sansara]at[clochardprod]dot[net]>
|
||
|
|
# copyright : pluie.org <http://www.pluie.org/>
|
||
|
|
#
|
||
|
|
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
|
||
|
|
#
|
||
|
|
# This file is part of ImpraStorage.
|
||
|
|
#
|
||
|
|
# ImpraStorage is free software (free as in speech) : you can redistribute it
|
||
|
|
# and/or modify it under the terms of the GNU General Public License as
|
||
|
|
# published by the Free Software Foundation, either version 3 of the License,
|
||
|
|
# or (at your option) any later version.
|
||
|
|
#
|
||
|
|
# ImpraStorage is distributed in the hope that it will be useful, but WITHOUT
|
||
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||
|
|
# more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with ImpraStorage. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
#
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ module ini ~~
|
||
|
|
|
||
|
|
from re import split as regsplit
|
||
|
|
from psr.sys import Sys, Io, Const
|
||
|
|
from psr.ini import IniFile
|
||
|
|
from psr.log import Log
|
||
|
|
from kirmah.crypt import KeyGen
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class IniFile ~~
|
||
|
|
|
||
|
|
class KiniFile(IniFile):
|
||
|
|
"""Read and write inifile"""
|
||
|
|
|
||
|
|
@Log(Const.LOG_BUILD)
|
||
|
|
def __init__(self, path, keyPath=None):
|
||
|
|
""""""
|
||
|
|
self.path = path
|
||
|
|
self.dic = {}
|
||
|
|
self.keyPath = path+'.key' if keyPath is None else keyPath
|
||
|
|
if not Io.file_exists(self.keyPath) :
|
||
|
|
kg = KeyGen()
|
||
|
|
Io.set_data(self.keyPath, kg.key)
|
||
|
|
if not Io.file_exists(path) :
|
||
|
|
self.set('profile' , 'default', 'main')
|
||
|
|
self.set('key' ,kg.key,'default.keys')
|
||
|
|
self.set('mark' ,kg.mark,'default.keys')
|
||
|
|
self.set('salt' ,'-*-ImpraStorage-*-','default.keys')
|
||
|
|
self.save()
|
||
|
|
self.read()
|
||
|
|
|
||
|
|
|
||
|
|
@Log()
|
||
|
|
def save(self,path=None,notAssign=False):
|
||
|
|
""""""
|
||
|
|
path = path if path is not None else self.path
|
||
|
|
Io.set_data(path, '# last updated : '+str(Sys.datetime.now())+Const.LF+self.toString())
|
||
|
|
call = ' '.join(['python3', 'kirmah-cli.py', 'enc', '-qf', path, '-z', '-r', '-m', '-o', path+'.kmh', '-k', self.keyPath ])
|
||
|
|
Sys.sysCall(call)
|
||
|
|
Io.removeFile(path)
|
||
|
|
if not notAssign : self.path = path
|
||
|
|
|
||
|
|
|
||
|
|
@Log()
|
||
|
|
def read(self):
|
||
|
|
""""""
|
||
|
|
try:
|
||
|
|
call = ' '.join([Sys.executable, 'kirmah-cli.py', 'dec', '-qf', self.path+'.kmh', '-z', '-r', '-m', '-o', self.path, '-k', self.keyPath ])
|
||
|
|
Sys.sysCall(call)
|
||
|
|
with Io.rfile(self.path, False) as fi:
|
||
|
|
csection = 'main'
|
||
|
|
self.dic[csection] = {}
|
||
|
|
for l in fi:
|
||
|
|
l = l.rstrip().lstrip()
|
||
|
|
if len(l) > 0 and not l[0]=='#' :
|
||
|
|
d = regsplit(' *= *', l , 1)
|
||
|
|
if len(d)> 1:
|
||
|
|
self.dic[csection][d[0]] = d[1] if d[1] is not None else ''
|
||
|
|
elif len(l)>0 and l[0]=='[':
|
||
|
|
csection = l.strip('[]')
|
||
|
|
self.dic[csection] = {}
|
||
|
|
Io.removeFile(self.path)
|
||
|
|
except IOError :
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@Log()
|
||
|
|
def delete(self):
|
||
|
|
Io.removeFile(self.path+'.kmh')
|
||
|
|
self.dic = {}
|