diff --git a/jperm/ansible_api.py b/jperm/ansible_api.py index 2e89008fc..0625a599d 100644 --- a/jperm/ansible_api.py +++ b/jperm/ansible_api.py @@ -19,7 +19,6 @@ API_DIR = os.path.dirname(os.path.abspath(__file__)) ANSIBLE_DIR = os.path.join(API_DIR, 'playbooks') - class AnsibleError(StandardError): """ the base AnsibleError which contains error(required), @@ -115,7 +114,7 @@ class Command(MyInventory): super(Command, self).__init__(*args, **kwargs) self.results = '' - def run(self, command, module_name="command", timeout=5, forks=10, group='my_group'): + def run(self, command, module_name="command", timeout=5, forks=10, pattern='*'): """ run command from andible ad-hoc. command : 必须是一个需要执行的命令字符串, 比如 @@ -128,7 +127,7 @@ class Command(MyInventory): module_args=command, timeout=timeout, inventory=self.inventory, - subset=group, + pattern=pattern, forks=forks ) self.results = hoc.run() diff --git a/jperm/utils.py b/jperm/utils.py index 12756b773..d2cda15f2 100644 --- a/jperm/utils.py +++ b/jperm/utils.py @@ -4,7 +4,7 @@ import random import os.path from paramiko.rsakey import RSAKey -from os import chmod, mkdir +from os import chmod, makedirs from uuid import uuid4 from jumpserver.settings import KEY_DIR @@ -45,8 +45,8 @@ def gen_keys(): :return: 返回目录名(uuid) """ key_basename = "key-" + uuid4().hex - key_path_dir = os.path.join(KEY_DIR, key_basename) - mkdir(key_path_dir, 0700) + key_path_dir = os.path.join(KEY_DIR, 'role_key', key_basename) + makedirs(key_path_dir, 0775) key = RSAKey.generate(2048) private_key = os.path.join(key_path_dir, 'id_rsa') @@ -62,8 +62,6 @@ def gen_keys(): return key_path_dir - - if __name__ == "__main__": print gen_keys() diff --git a/jumpserver/api.py b/jumpserver/api.py index 97fc659bb..f7543d01f 100644 --- a/jumpserver/api.py +++ b/jumpserver/api.py @@ -14,10 +14,8 @@ from django.core.paginator import Paginator, EmptyPage, InvalidPage from django.http import HttpResponse, Http404 from django.template import RequestContext from juser.models import User, UserGroup +from jlog.models import Log from jasset.models import Asset, AssetGroup -# from jlog.models import Log -from jlog.models import Log, TtyLog -from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned from django.http import HttpResponseRedirect from django.shortcuts import render_to_response from django.core.mail import send_mail @@ -414,4 +412,3 @@ def my_render(template, data, request): CRYPTOR = PyCrypt(KEY) logger = set_log(LOG_LEVEL) -KEY_DIR = os.path.join(BASE_DIR, 'keys') diff --git a/jumpserver/settings.py b/jumpserver/settings.py index cc3f5a8c7..f3042349a 100644 --- a/jumpserver/settings.py +++ b/jumpserver/settings.py @@ -18,7 +18,7 @@ config = ConfigParser.ConfigParser() BASE_DIR = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) config.read(os.path.join(BASE_DIR, 'jumpserver.conf')) -KEY_DIR = os.path.join(BASE_DIR, 'role_keys') +KEY_DIR = os.path.join(BASE_DIR, 'keys') DB_HOST = config.get('db', 'host') DB_PORT = config.getint('db', 'port') diff --git a/jumpserver/views.py b/jumpserver/views.py index a907ced4f..00f28616c 100644 --- a/jumpserver/views.py +++ b/jumpserver/views.py @@ -15,7 +15,6 @@ from jumpserver.api import * from jumpserver.models import Setting from django.contrib.auth import authenticate, login, logout from django.contrib.auth.decorators import login_required -from settings import BASE_DIR from jlog.models import Log