all_resource_keys = list(db2.query(conn, 'select resource_key from resources')) random.shuffle(all_resource_keys) def random_resource_for_authority(auth, count): for i in range(0, count): yield (auth, all_resource_keys[i]['RESOURCE_KEY']) populate_authority_resource(conn, random_resource_for_authority(-1, 4)) populate_authority_resource(conn, random_resource_for_authority(-2, 3)) if __name__ == '__main__': #src_conn = db2.connect('HAIAN', '10.3.1.249', '60000', 'db2inst1', '1qaz2wsx') conn = db2.connect('KBMASTER', '10.98.0.192', '50000', 'db2inst1', 'password') ibm_db.exec_immediate(conn, 'set schema db2inst1') initialize_resource(conn) initialize_test_data(conn) ''' db2.clear(conn, 'role_authority', 'authority_resource', 'menus', 'authorities', 'resources', )#, 'metadatas') apps = ['SC-CORE'] for app in apps: populate_resources(conn, get_resources(conn, app)) populate_authorities(conn, get_authorities(conn)) populate_authority_resource(conn, get_authority_resource(conn, app))
for role_info in roles_info: yield (next_pk_val, role_info[0], 1, role_info[1], role_info[2], datetime.datetime.now()) next_pk_val += 1 def extra_user_role(conn): user = get_user(conn, 'admin') roles = ('sso_admin',) for role in roles: yield (user['USER_KEY'], get_role(conn, role)['ROLE_KEY']) if __name__ == '__main__': src_conn = db2.connect('HAIAN', '10.3.1.249', '60000', 'db2inst1', '1qaz2wsx') dest_conn = db2.connect('KBMASTER', '10.98.0.192', '50000', 'db2inst1', 'password') ibm_db.exec_immediate(dest_conn, 'set schema db2inst1') # Truncate target database tables. ibm_db.exec_immediate(dest_conn, 'delete from sso_user_role') ibm_db.exec_immediate(dest_conn, 'delete from sso_roles') ibm_db.exec_immediate(dest_conn, 'delete from sso_users') ibm_db.exec_immediate(dest_conn, 'delete from SSO_TENANTS') populate_tenant(dest_conn, sso_tenants(src_conn)) populate_user(dest_conn, sso_users(src_conn)) populate_role(dest_conn, sso_roles(src_conn)) populate_user_role(dest_conn, sso_user_role(src_conn))