mirror of
https://github.com/YunoHost/moulinette.git
synced 2024-09-03 20:06:31 +02:00
134 lines
4.3 KiB
Python
Executable file
134 lines
4.3 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import gettext
|
|
import getpass
|
|
try:
|
|
import yaml
|
|
except ImportError:
|
|
sys.stderr.write('Error: Yunohost CLI Require yaml lib\n')
|
|
sys.stderr.write('apt-get install python-yaml\n')
|
|
sys.exit(1)
|
|
import json
|
|
if not __debug__:
|
|
import traceback
|
|
|
|
gettext.install('YunoHost')
|
|
|
|
try:
|
|
from yunohost import YunoHostError, YunoHostLDAP, str_to_func, colorize, pretty_print_dict, display_error, validate, win, parse_dict
|
|
except ImportError:
|
|
sys.stderr.write('Error: Yunohost CLI Require YunoHost lib\n')
|
|
sys.exit(1)
|
|
|
|
|
|
def main():
|
|
"""
|
|
Main instructions
|
|
|
|
Parse the action_dict and execute the action-specific function,
|
|
then print json or pretty result if executed in a tty :)
|
|
|
|
Returns:
|
|
int -- 0 or error code
|
|
|
|
"""
|
|
|
|
if len(sys.argv) < 2:
|
|
sys.argv.append('-h')
|
|
|
|
with open('action_map.yml') as f:
|
|
action_map = yaml.load(f)
|
|
|
|
admin_password_provided = False
|
|
json_print = False
|
|
write_ldap = True
|
|
postinstall = False
|
|
|
|
for key, arg in enumerate(sys.argv):
|
|
if arg == '--admin-password':
|
|
admin_password_provided = True
|
|
admin_password = sys.argv[key+1]
|
|
sys.argv.pop(key)
|
|
sys.argv.pop(key)
|
|
if arg == '--no-ldap':
|
|
write_ldap = False
|
|
sys.argv.pop(key)
|
|
if arg == '--json':
|
|
json_print = True
|
|
sys.argv.pop(key)
|
|
|
|
try:
|
|
try:
|
|
with open('/etc/yunohost/installed') as f: pass
|
|
except IOError:
|
|
postinstall = True
|
|
if len(sys.argv) < 3 or sys.argv[1] != 'tools' or sys.argv[2] != 'postinstall':
|
|
raise YunoHostError(17, _("YunoHost is not correctly installed, please execute 'yunohost tools postinstall'"))
|
|
|
|
args = parse_dict(action_map)
|
|
args_dict = vars(args).copy()
|
|
for key in args_dict.keys():
|
|
sanitized_key = key.replace('-', '_')
|
|
if sanitized_key is not key:
|
|
args_dict[sanitized_key] = args_dict[key]
|
|
del args_dict[key]
|
|
del args_dict['func']
|
|
try:
|
|
with open('/etc/yunohost/passwd') as f:
|
|
admin_password = f.read()
|
|
admin_password_provided = True
|
|
except IOError: pass
|
|
if postinstall:
|
|
result = args.func(**args_dict)
|
|
elif admin_password_provided:
|
|
with YunoHostLDAP(password=admin_password):
|
|
result = args.func(**args_dict)
|
|
elif os.isatty(1) and write_ldap:
|
|
admin_password = getpass.getpass(colorize(_('Admin Password: '), 'yellow'))
|
|
with YunoHostLDAP(password=admin_password):
|
|
try:
|
|
with open('/var/run/yunohost.pid', 'r'):
|
|
raise YunoHostError(1, _("A YunoHost command is already running"))
|
|
except IOError:
|
|
with open('/var/run/yunohost.pid', 'w') as f:
|
|
f.write('ldap')
|
|
os.system('chmod 400 /var/run/yunohost.pid')
|
|
with open('/etc/yunohost/passwd', 'w') as f:
|
|
f.write(admin_password)
|
|
os.system('chmod 400 /etc/yunohost/passwd')
|
|
try:
|
|
result = args.func(**args_dict)
|
|
except KeyboardInterrupt, EOFError:
|
|
raise YunoHostError(125, _("Interrupted"))
|
|
finally:
|
|
os.remove('/etc/yunohost/passwd')
|
|
os.remove('/var/run/yunohost.pid')
|
|
else:
|
|
with YunoHostLDAP(anonymous=True):
|
|
result = args.func(**args_dict)
|
|
#except TypeError, error:
|
|
#if not __debug__ :
|
|
#traceback.print_exc()
|
|
#print(_("Not (yet) implemented function"))
|
|
#return 1
|
|
except YunoHostError, error:
|
|
display_error(error, json_print)
|
|
return error.code
|
|
else:
|
|
if json_print or not os.isatty(1) and result is not None:
|
|
if len(win) > 0:
|
|
result['success'] = win
|
|
print(json.dumps(result))
|
|
elif result is not None:
|
|
pretty_print_dict(result)
|
|
else:
|
|
pass
|
|
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|