def _create_system_account(self, username):
if not AccountService.get_by_owner(username):
logger.debug('create owner')
AccountService.create(
username=username,
account_type=AccountType.NORMAL,
email='*****@*****.**' % (username),
password=username,
phone='18511076831',
name=username,
)
def handle(self, *args, **options):
self.stdout.write("Create a admin_ account:")
self.stdout.flush()
email = raw_input("Input your email:")
if not email:
logger.info("Email should not be empty")
sys.exit(1)
password = raw_input("Input your password:"******"Password should not be empty")
sys.exit(1)
phone = raw_input("Input your cell phone:")
if not phone:
logger.info("Cell phone should not be empty")
sys.exit(1)
account, error = AccountService.create(
email=email,
password=password,
phone=phone,
name=email[:email.find('@')],
account_type=AccountType.HANKOU,
)
if error is not None:
logger.error("Create admin_ account failed, %s" % error)
sys.exit(1)
account.user.is_superuser = True
account.user.save()
logger.info("Create admin_ account successful!")
def post(self, request, *args, **kwargs):
request_data = request.data.get('data', {})
validator = CreateDepartmentMembersValidator(data=request_data)
if not validator.is_valid():
return Response(console_response(code=1, msg=validator.errors))
department_id = validator.validated_data.get('department_id')
member_type = validator.validated_data.get('member_type')
validator = self.get_create_member_validator(member_type)(
data=request_data)
if not validator.is_valid():
return Response(console_response(code=1, msg=validator.errors))
member_info = validator.validated_data.get('member_info')
member, error = AccountService.create(**member_info)
if error:
return Response(
console_response(code=1, ret_set={},
msg='成员创建失败' + str(error)))
elif DepartmentMemberService.join([member.user], department_id):
# 创建成功 就加入到指定部门
if member.type in AccountType.FINANCE_TYPES:
permission_group_id_list = validator.validated_data.get(
'permission_group_id_list')
new_member = AccountService.get_by_owner(member.user)
groups = PermissionGroupService.gets(permission_group_id_list)
count = UserPermissionService.append_groups(new_member, groups)
ret_set = DepartmentMemberService.to_dict(new_member)
if count == len(permission_group_id_list):
# create_payload = {
# "owner": member.user.username,
# "zone": request_data.get("zone", "bj")
# }
# create_resp = create_base_net(create_payload)
# if create_resp.get("ret_code"):
# return Response(console_response(code=1, msg="创建成员子网失败"))
# ret_set["sub_net"] = create_resp.get("ret_set")[0]
#
return Response(console_response(ret_set=ret_set))
else:
return Response(
console_response(code=1,
ret_set=ret_set,
msg='成员初始化权限失败'))
elif member.type in AccountType.PORTAL_TYPES:
new_member = AccountService.get_by_owner(member.user)
ret_set = DepartmentMemberService.to_dict(new_member)
return Response(console_response(ret_set=ret_set))
else:
AccountService.delete_by_username(member.user.username,
really_delete=True)
return Response(
console_response(code=1, ret_set={}, msg='成员加入部门失败'))
def setUp(self):
telphone = 18901177755
self.creator = AccountService.create(
'*****@*****.**',
'admin',
telphone
)[0]
self.users = [
AccountService.create(
'*****@*****.**' % (i + 1),
'admin',
telphone + i + 1
)[0]
for i in range(16)
]
self.permissions = list()
for i in range(16):
permission = Permission(name='UnitTest-Permission-%02d' % (i + 1))
permission.save()
self.permissions.append(permission)
super(BasePermissionHelperTestCase, self).setUp()
def setUp(self):
super(AbstractTestCase, self).setUp()
zone = ZoneModel.objects.first()
self.zone = zone.name
email = '*****@*****.**'
password = '******'
phone = '106900810045'
name = '单元测试'
account, err = AccountService.create(email, password, phone, name=name)
account.save()
self.account = account
self.owner = account.user.username
self.client = Client(enforce_csrf_checks=False)
self.client.login(username=account.user.username, password=password)