user.py 3.52 KB
Newer Older
danfengliu's avatar
danfengliu committed
1 2 3 4 5 6 7 8
# -*- coding: utf-8 -*-

import base
import swagger_client

class User(base.Base):

    def create_user(self, name=None,
9
        email = None, user_password=None, realname = None, role_id = None, **kwargs):
danfengliu's avatar
danfengliu committed
10 11 12 13 14 15 16 17
        if name is None:
            name = base._random_name("user")
        if realname is None:
            realname = base._random_name("realname")
        if email is None:
            email = '%s@%s.com' % (realname,"vmware")
        if user_password is None:
            user_password = "Harbor12345678"
18 19
        if role_id is None:
            role_id = 0
danfengliu's avatar
danfengliu committed
20 21

        client = self._get_client(**kwargs)
22
        user = swagger_client.User(username = name, email = email, password = user_password, realname = realname, role_id = role_id)
danfengliu's avatar
danfengliu committed
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
        _, status_code, header = client.users_post_with_http_info(user)

        base._assert_status_code(201, status_code)

        return base._get_id_from_header(header), name

    def get_users(self, username=None, email=None, page=None, page_size=None, **kwargs):
        client = self._get_client(**kwargs)
        params={}
        if username is not None:
            params["username"] = username
        if email is not None:
            params["email"] = email
        if page is not None:
            params["page"] = page
        if page_size is not None:
            params["page_size"] = page_size
        data, status_code, _ = client.users_get_with_http_info(**params)
        base._assert_status_code(200, status_code)
        return data

    def get_user(self, user_id, **kwargs):
        client = self._get_client(**kwargs)
        data, status_code, _ = client.users_user_id_get_with_http_info(user_id)
        base._assert_status_code(200, status_code)
        print "data in lib:", data
        return data


    def get_user_current(self, **kwargs):
        client = self._get_client(**kwargs)
        data, status_code, _ = client.users_current_get_with_http_info()
        base._assert_status_code(200, status_code)
        return data

58
    def delete_user(self, user_id, expect_status_code = 200, **kwargs):
danfengliu's avatar
danfengliu committed
59 60
        client = self._get_client(**kwargs)
        _, status_code, _ = client.users_user_id_delete_with_http_info(user_id)
61
        base._assert_status_code(expect_status_code, status_code)
danfengliu's avatar
danfengliu committed
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
        return user_id

    def update_user_pwd(self, user_id, new_password=None, old_password=None, **kwargs):
        if old_password is None:
            old_password  = ""
        password = swagger_client.Password(old_password, new_password)
        client = self._get_client(**kwargs)
        _, status_code, _ = client.users_user_id_password_put_with_http_info(user_id, password)
        base._assert_status_code(200, status_code)
        return user_id

    def update_uesr_profile(self, user_id, email=None, realname=None, comment=None, **kwargs):
        client = self._get_client(**kwargs)
        user_rofile = swagger_client.UserProfile(email, realname, comment)
        _, status_code, _ = client.users_user_id_put_with_http_info(user_id, user_rofile)
        base._assert_status_code(200, status_code)
        return user_id

    def update_uesr_role_as_sysadmin(self, user_id, IsAdmin, **kwargs):
        client = self._get_client(**kwargs)
        has_admin_role = swagger_client.HasAdminRole(IsAdmin)
        print "has_admin_role:", has_admin_role
        _, status_code, _ = client.users_user_id_sysadmin_put_with_http_info(user_id, has_admin_role)
        base._assert_status_code(200, status_code)
        return user_id