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))
