148 lines
6.5 KiB
Python
148 lines
6.5 KiB
Python
|
from stat import S_IMODE, S_ISDIR, S_ISSOCK
|
||
|
|
from os import lstat, getuid, getgid
|
||
|
|
from os.path import exists
|
||
|
|
from pwd import getpwuid
|
||
|
|
from grp import getgrgid
|
||
|
|
from tiramisu import FilenameOption, UsernameOption, GroupnameOption, IntOption, BoolOption, ChoiceOption, \
|
||
|
|
OptionDescription, Leadership, Config, Calculation, Params, ParamSelfOption, ParamOption, ParamValue, \
|
||
|
|
calc_value
|
||
|
|
from tiramisu.error import LeadershipError, PropertiesOptionError
|
||
|
|
|
||
|
|
|
||
|
|
def get_username(filename, exists, create=False):
|
||
|
|
if exists:
|
||
|
|
uid = lstat(filename).st_uid
|
||
|
|
elif create:
|
||
|
|
# the current uid
|
||
|
|
uid = getuid()
|
||
|
|
else:
|
||
|
|
return
|
||
|
|
|
||
|
|
return getpwuid(uid).pw_name
|
||
|
|
|
||
|
|
|
||
|
|
def get_grpname(filename, exists, create=False):
|
||
|
|
if exists:
|
||
|
|
gid = lstat(filename).st_gid
|
||
|
|
elif create:
|
||
|
|
# the current gid
|
||
|
|
gid = getgid()
|
||
|
|
else:
|
||
|
|
return
|
||
|
|
return getgrgid(gid).gr_name
|
||
|
|
|
||
|
|
|
||
|
|
def calc_type(filename, is_exists):
|
||
|
|
if is_exists:
|
||
|
|
mode = lstat(filename).st_mode
|
||
|
|
if S_ISSOCK(mode):
|
||
|
|
return 'socket'
|
||
|
|
elif S_ISDIR(mode):
|
||
|
|
return 'directory'
|
||
|
|
return 'file'
|
||
|
|
|
||
|
|
|
||
|
|
def calc_mode(filename, is_exists, type):
|
||
|
|
if is_exists:
|
||
|
|
return int(oct(S_IMODE(lstat(filename).st_mode))[2:])
|
||
|
|
if type == 'file':
|
||
|
|
return 644
|
||
|
|
elif type == 'directory':
|
||
|
|
return 755
|
||
|
|
elif type == 'socket':
|
||
|
|
return 444
|
||
|
|
|
||
|
|
|
||
|
|
filename = FilenameOption('filename',
|
||
|
|
'Filename',
|
||
|
|
multi=True,
|
||
|
|
properties=('mandatory',))
|
||
|
|
exists_ = BoolOption('exists',
|
||
|
|
'This file exists',
|
||
|
|
Calculation(exists, Params(ParamOption(filename))),
|
||
|
|
multi=True,
|
||
|
|
properties=('mandatory', 'frozen', 'force_default_on_freeze', 'advanced'))
|
||
|
|
create = BoolOption('create',
|
||
|
|
'Create automaticly the file',
|
||
|
|
multi=True,
|
||
|
|
default_multi=True,
|
||
|
|
properties=(Calculation(calc_value,
|
||
|
|
Params(ParamValue('disabled'),
|
||
|
|
kwargs={'condition': ParamOption(exists_),
|
||
|
|
'expected': ParamValue(True)})),))
|
||
|
|
type_ = ChoiceOption('type',
|
||
|
|
'The file type',
|
||
|
|
('file', 'directory', 'socket'),
|
||
|
|
Calculation(calc_type, Params((ParamOption(filename),
|
||
|
|
ParamOption(exists_)))),
|
||
|
|
multi=True,
|
||
|
|
properties=('force_default_on_freeze', 'mandatory',
|
||
|
|
Calculation(calc_value,
|
||
|
|
Params(ParamValue('hidden'),
|
||
|
|
kwargs={'condition': ParamOption(exists_),
|
||
|
|
'expected': ParamValue(True)})),
|
||
|
|
Calculation(calc_value,
|
||
|
|
Params(ParamValue('frozen'),
|
||
|
|
kwargs={'condition': ParamOption(exists_),
|
||
|
|
'expected': ParamValue(True)}))))
|
||
|
|
username = UsernameOption('user',
|
||
|
|
'User',
|
||
|
|
default_multi=Calculation(get_username, Params((ParamOption(filename),
|
||
|
|
ParamOption(exists_),
|
||
|
|
ParamOption(create, notraisepropertyerror=True)))),
|
||
|
|
multi=True,
|
||
|
|
properties=('force_store_value',
|
||
|
|
Calculation(calc_value,
|
||
|
|
Params(ParamValue('mandatory'),
|
||
|
|
kwargs={'condition': ParamOption(create, notraisepropertyerror=True),
|
||
|
|
'expected': ParamValue(True),
|
||
|
|
'no_condition_is_invalid': ParamValue(True)})),))
|
||
|
|
grpname = GroupnameOption('group',
|
||
|
|
'Group',
|
||
|
|
default_multi=Calculation(get_grpname, Params((ParamOption(filename),
|
||
|
|
ParamOption(exists_),
|
||
|
|
ParamOption(create, notraisepropertyerror=True)))),
|
||
|
|
multi=True,
|
||
|
|
properties=('force_store_value',
|
||
|
|
Calculation(calc_value,
|
||
|
|
Params(ParamValue('mandatory'),
|
||
|
|
kwargs={'condition': ParamOption(create, notraisepropertyerror=True),
|
||
|
|
'expected': ParamValue(True),
|
||
|
|
'no_condition_is_invalid': ParamValue(True)})),))
|
||
|
|
mode = IntOption('mode',
|
||
|
|
'Mode',
|
||
|
|
default_multi=Calculation(calc_mode, Params((ParamOption(filename), ParamOption(exists_), ParamOption(type_)))),
|
||
|
|
multi=True,
|
||
|
|
properties=('mandatory', 'advanced', 'force_store_value'))
|
||
|
|
|
||
|
|
new = Leadership('new',
|
||
|
|
'Add new file',
|
||
|
|
[filename, exists_, create, type_, username, grpname, mode])
|
||
|
|
|
||
|
|
root = OptionDescription('root', 'root', [new])
|
||
|
|
|
||
|
|
config = Config(root)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#config.option('new.create', 1).value.set(False)
|
||
|
|
#config.option('new.type', 1).value.set('file')
|
||
|
|
#config.option('new.type', 2).value.set('file')
|
||
|
|
#print(config.value.dict())
|
||
|
|
#config.option('new.type', 2).value.set('directory')
|
||
|
|
#print(config.value.dict())
|
||
|
|
#print(config.unrestraint.option('new.mode', 0).owner.isdefault())
|
||
|
|
#print(config.unrestraint.option('new.mode', 1).owner.isdefault())
|
||
|
|
#print(config.unrestraint.option('new.mode', 2).owner.isdefault())
|
||
|
|
#config.property.read_only()
|
||
|
|
#print(config.option('new.mode', 0).owner.isdefault())
|
||
|
|
#print(config.option('new.mode', 1).owner.isdefault())
|
||
|
|
#print(config.option('new.mode', 2).owner.isdefault())
|
||
|
|
#print(config.value.dict())
|
||
|
|
#config.property.read_write()
|
||
|
|
#config.option('new.type', 2).value.set('file')
|
||
|
|
#print(config.value.dict())
|
||
|
|
#config.option('new.mode', 2).value.reset()
|
||
|
|
#print(config.value.dict())
|