191 lines
4.2 KiB
Plaintext
191 lines
4.2 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 1,
|
|
"id": "b82fd271",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"import pymysql\n",
|
|
"import snowflake\n",
|
|
"from faker import Faker\n",
|
|
"import random\n",
|
|
"import datetime\n",
|
|
"fake = Faker('zh_CN')\n",
|
|
"import random\n",
|
|
"# 创建数据库连接\n",
|
|
"connection = pymysql.connect(\n",
|
|
"host='192.168.0.244',\n",
|
|
"port=3306,\n",
|
|
"user='root',\n",
|
|
"password='Abcd@123456',\n",
|
|
"database='hongxx',\n",
|
|
"charset='utf8'\n",
|
|
")\n",
|
|
"# 创建游标对象\n",
|
|
"cursor = connection.cursor()"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 2,
|
|
"id": "759c550a",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"def create_table(cursor,result):\n",
|
|
" \n",
|
|
" data = []\n",
|
|
" for i in result:\n",
|
|
" list_list = list(i)\n",
|
|
" des=cursor.description # 获取表详情，字段名，长度，属性等\n",
|
|
" t = \",\".join([item[0] for item in des])\n",
|
|
" table_head = t.split(',') # # 查询表列名 用,分割\n",
|
|
" \n",
|
|
" dict_result = dict(zip(table_head, list_list)) # 打包为元组的列表 再转换为字典\n",
|
|
" data.append(dict_result) # 将字典添加到list_result中\n",
|
|
" return data"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 26,
|
|
"id": "b717043a",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"sql = 'select * from sys_user where dept_id=100'\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 27,
|
|
"id": "b4b373b8",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"cursor.execute(sql)\n",
|
|
"rows = cursor.fetchall()\n",
|
|
"result = create_table(cursor,rows)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 28,
|
|
"id": "394495d3",
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"data": {
|
|
"text/plain": [
|
|
"[2,\n",
|
|
" 3,\n",
|
|
" 4,\n",
|
|
" 6,\n",
|
|
" 24,\n",
|
|
" 31,\n",
|
|
" 308,\n",
|
|
" 909,\n",
|
|
" 917,\n",
|
|
" 918,\n",
|
|
" 1145,\n",
|
|
" 3690,\n",
|
|
" 3698,\n",
|
|
" 3828,\n",
|
|
" 3842,\n",
|
|
" 3843,\n",
|
|
" 3844,\n",
|
|
" 3895,\n",
|
|
" 3899,\n",
|
|
" 3912,\n",
|
|
" 3913,\n",
|
|
" 3935,\n",
|
|
" 3936,\n",
|
|
" 3988,\n",
|
|
" 5440,\n",
|
|
" 5492,\n",
|
|
" 7696,\n",
|
|
" 7713,\n",
|
|
" 7714,\n",
|
|
" 7723,\n",
|
|
" 7728,\n",
|
|
" 2059454179031199746]"
|
|
]
|
|
},
|
|
"execution_count": 28,
|
|
"metadata": {},
|
|
"output_type": "execute_result"
|
|
}
|
|
],
|
|
"source": [
|
|
"[i[\"user_id\"] for i in result]\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 25,
|
|
"id": "325e4664",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"for item in result:\n",
|
|
" userid = item[\"user_id\"]\n",
|
|
" \n",
|
|
" sql2 = f\"update sys_user set dept_id=100, team_user_type ='BACKEND' where user_id={userid}\"\n",
|
|
" cursor.execute(sql2)\n",
|
|
"connection.commit()"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 24,
|
|
"id": "f62b4e7d",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"for item in result:\n",
|
|
" userid = item[\"user_id\"]\n",
|
|
" sql2 = f\"update sys_user_dept set dept_id=100, person_type='hx_sys' where user_id={userid}\"\n",
|
|
" cursor.execute(sql2)\n",
|
|
"connection.commit()\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "7dcf47b8",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"for item in result:\n",
|
|
" userid = item[\"user_id\"]\n",
|
|
" sql2 = f\"update sys_user_role set dept_id=100 where user_id={userid}\"\n",
|
|
" cursor.execute(sql2)\n",
|
|
"connection.commit()"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Python 3",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.12.4"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|