import os import django import random from datetime import datetime import sys def setup_django(): """设置Django环境""" # project_path = "D:/Document/code/vue/greater_wms" project_path = "D:/code/vue/greater_wms" sys.path.append(project_path) os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'greaterwms.settings') django.setup() print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Django环境已设置") # 定义系统中的页面和组件 PAGES = [ # 库存管理 {"primary_page": "stock","path": "/stock/management", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/stocklist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/stockbinlist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/emptybin", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/occupiedbin", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/binset", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "stock","path": "/stock/handcount", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # ERP管理 {"primary_page": "erp","path": "/erp/erpasn", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "erp","path": "/erp/erpasnmaterial", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "erp","path": "/erp/erpdnmaterial", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "erp","path": "/erp/erpasnaudit", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "erp","path": "/erp/erpdn", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "erp","path": "/erp/erpsortstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 入库管理 {"primary_page": "inbound","path": "/inbound/asn", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/predeliverystock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/preloadstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/presortstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/sortstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/shortage", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/more", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "inbound","path": "/inbound/asnfinish", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 托盘管理 {"primary_page":"container","path": "/container/containerlist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"container","path": "/container/containerdetail", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"container","path": "/container/containercategory", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"container","path": "/container/containeroperate", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 仪表盘 {"primary_page": "dashboard","path": "/dashboard/inboundAndOutbound", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/flows_statements", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/flows", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/flows_complex", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/batchlog", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/ContainerDetailLogModel", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page": "dashboard","path": "/dashboard/MaterialChangeHistory", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 统计管理 {"primary_page":"count","path": "/count/detaillog", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/batchoperatelog", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/countbatchlog", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/presortstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/sortstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/shortage", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/containerDetail", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/batch", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"count","path": "/count/asnfinish", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 出库管理 {"primary_page":"outbound","path": "/outbound/dn", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/freshorder", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/neworder", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/pickstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/pickedstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/pickinglist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/shippedstock", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/backorder", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/pod", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"outbound","path": "/outbound/container_check", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 商品管理 {"primary_page":"goods","path": "/goods/goodslist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsunit", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsclass", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsbrand", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodscolor", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsspecs", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsshape", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"goods","path": "/goods/goodsorigin", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 任务管理 {"primary_page":"taskpage","path": "/taskpage/task", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 仓库管理 {"primary_page":"warehouse","path": "/warehouse/warehouseset", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/department", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/boundcodetype", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/boundtype", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/boundbusiness", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/status", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"warehouse","path": "/warehouse/product", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 员工管理 {"primary_page": "staff", "path": "/staff/roles", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"staff","path": "/staff/stafflist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"staff","path": "/staff/stafflist_check_code", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"staff","path": "/staff/stafftype", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 上传中心 {"primary_page":"uploadcenter","path": "/uploadcenter/initializeupload", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"uploadcenter","path": "/uploadcenter/addupload", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, # 下载中心 {"primary_page":"downloadcenter","path": "/downloadcenter/downloadinbound", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"downloadcenter","path": "/downloadcenter/downloadoutbound", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"downloadcenter","path": "/downloadcenter/downloadstocklist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"downloadcenter","path": "/downloadcenter/downloadgoodslist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, {"primary_page":"downloadcenter","path": "/downloadcenter/downloadbinlist", "components": ["view", "edit", "add", "delete", "export", "confirm","adjust","download"]}, ] # 定义角色及其权限配置 ROLES = { "管理员": { "description": "系统管理员，拥有所有权限", "page_access": "all", # 所有页面 "component_access": "all" # 所有组件 }, "经理": { "description": "部门经理，拥有大部分管理权限", "page_access": [ "/stock/management","/stock/stockbinlist","/stock/stocklist", "/erp/erpasn","/erp/erpasnmaterial","/erp/erpdnmaterial","/erp/erpdn","/erp/erpsortstock", "/inbound/asn","/inbound/predeliverystock","/inbound/sortstock", "/container/containerlist","/container/containerdetail","/container/containercategory","/container/containeroperate", "/outbound/dn","/outbound/backorder","/outbound/container_check", "/taskpage/task", "/count/batch","/count/countbatchlog","/count/detaillog","/count/batchoperatelog", "/dashboard/flows_statements","/dashboard/flows","/dashboard/MaterialChangeHistory","/dashboard/batchlog","/dashboard/ContainerDetailLogModel", "/warehouse/department","/warehouse/boundcodetype","/warehouse/boundtype","/warehouse/boundbusiness","/warehouse/status","/warehouse/product", "/staff/stafflist","/staff/stafflist_check_code","/staff/stafftype" ], "component_access": "all" # 所有组件 }, "主管": { "description": "仓库主管，负责日常运营管理", "page_access": [ "/stock/management","/stock/stockbinlist","/stock/stocklist", "/erp/erpasn","/erp/erpasnmaterial","/erp/erpdnmaterial","/erp/erpdn","/erp/erpsortstock", "/inbound/asn","/inbound/predeliverystock","/inbound/sortstock", "/container/containerlist","/container/containerdetail","/container/containercategory", "/outbound/dn","/outbound/backorder","/outbound/container_check", "/taskpage/task", "/count/batch","/count/countbatchlog","/count/detaillog","/count/batchoperatelog", "/dashboard/flows_statements","/dashboard/flows","/dashboard/MaterialChangeHistory","/dashboard/batchlog","/dashboard/ContainerDetailLogModel", # "/warehouse/department","/warehouse/boundcodetype","/warehouse/boundtype","/warehouse/boundbusiness","/warehouse/status","/warehouse/product", "/staff/stafflist","/staff/stafftype" ], "component_access": [ "view", "edit", "add", "delete", "export", "confirm","adjust","download" ] }, "操作员": { "description": "仓库操作员，负责具体操作", "page_access": [ "/stock/management","/stock/stockbinlist","/stock/stocklist", "/erp/erpasn","/erp/erpasnmaterial","/erp/erpdnmaterial","/erp/erpdn","/erp/erpsortstock", "/inbound/asn","/inbound/predeliverystock","/inbound/sortstock", "/container/containerlist","/container/containercategory", "/outbound/dn","/outbound/backorder","/outbound/container_check", "/taskpage/task", "/count/batch","/count/countbatchlog","/count/detaillog","/count/batchoperatelog", "/dashboard/flows_statements","/dashboard/flows" # "/warehouse/department","/warehouse/boundcodetype","/warehouse/boundtype","/warehouse/boundbusiness","/warehouse/status","/warehouse/product", # "/staff/stafflist","/staff/stafftype" ], "component_access": [ "view", "edit", "add","download","confirm" ] }, "查看员": { "description": "数据查看员，只能查看数据", "page_access": [ "/stock/management","/stock/stockbinlist", "/erp/erpasn","/erp/erpasnmaterial","/erp/erpdnmaterial","/erp/erpdn","/erp/erpsortstock", "/inbound/asn","/inbound/predeliverystock","/inbound/sortstock", "/container/containerlist","/container/containercategory", "/outbound/dn","/outbound/backorder","/outbound/container_check", "/taskpage/task", "/count/batch","/count/countbatchlog","/count/detaillog","/count/batchoperatelog", "/dashboard/flows_statements","/dashboard/flows" # "/warehouse/department","/warehouse/boundcodetype","/warehouse/boundtype","/warehouse/boundbusiness","/warehouse/status","/warehouse/product", # "/staff/stafflist","/staff/stafftype" ], "component_access": ["view","download"] } } def create_permissions(): """创建所有页面和组件的权限""" from staff.models import Permission print("🚀 开始创建权限...") created_count = 0 for page_info in PAGES: primary_page = page_info.get("primary_page") page_path = page_info["path"] components = page_info["components"] # 创建页面访问权限 page_permission, created = Permission.objects.get_or_create( primary_page=primary_page, page=page_path, component=None, defaults={ "name": f"{page_path} 页面访问", "description": f"访问 {page_path} 页面的权限", "enabled": True } ) if created: created_count += 1 print(f"✅ 创建页面权限: {page_path}") # 创建组件权限 for component in components: comp_permission, created = Permission.objects.get_or_create( primary_page=primary_page, page=page_path, component=component, defaults={ "name": f"{page_path} - {component} 组件", "description": f"在 {page_path} 页面使用 {component} 组件的权限", "enabled": True } ) if created: created_count += 1 print(f" ✅ 创建组件权限: {page_path} - {component}") print(f"📊 权限创建完成! 共创建 {created_count} 个权限") return created_count def create_roles_and_assign_permissions(): """创建角色并分配权限""" from staff.models import ListModel, Permission, Role print("\n🚀 开始创建角色并分配权限...") role_count = 0 permission_count = 0 # 获取所有权限 all_permissions = Permission.objects.all() for role_name, role_config in ROLES.items(): # 创建角色 role, created = Role.objects.get_or_create( name=role_name, defaults={ "description": role_config["description"] } ) if role: role_count += 1 print(f"✅ 创建角色: {role_name}") # 分配权限 if role_config["page_access"] == "all": # 分配所有权限 role.permissions.set(all_permissions) permission_count += all_permissions.count() print(f" 🔄 分配所有权限 ({all_permissions.count()} 个)") else: # 分配特定权限 assigned_permissions = [] # 处理页面访问权限 for page_path in role_config["page_access"]: # 获取精确匹配该路径的所有页面权限 page_perms = all_permissions.filter( page=page_path, # 使用精确匹配 component=None ) assigned_permissions.extend(page_perms) # 获取精确匹配该路径的组件权限 comp_perms = all_permissions.filter( page=page_path, # 使用精确匹配 component__isnull=False ) # 根据组件访问规则过滤 if role_config["component_access"] == "all": assigned_permissions.extend(comp_perms) else: for perm in comp_perms: # 检查组件权限是否在允许的列表中 if any(access in perm.component for access in role_config["component_access"]): assigned_permissions.append(perm) # 分配权限 role.permissions.set(assigned_permissions) permission_count += len(assigned_permissions) print(f" 🔄 分配 {len(assigned_permissions)} 个权限") print(f"📊 角色创建完成! 共创建 {role_count} 个角色，分配 {permission_count} 个权限") return role_count, permission_count def main(): setup_django() print("=" * 50) print("🔐 权限与角色批量创建工具") print("=" * 50) print("此脚本将根据预定义的页面和角色配置批量创建:") print("1. 所有页面的访问权限") print("2. 所有组件的操作权限") print("3. 不同角色的权限配置") print("=" * 50) confirm = input("是否开始批量创建? (y/n): ").lower() if confirm == 'y': start_time = datetime.now() # 创建权限 perm_count = create_permissions() # 创建角色并分配权限 role_count, perm_assigned_count = create_roles_and_assign_permissions() end_time = datetime.now() duration = (end_time - start_time).total_seconds() print("\n" + "=" * 50) print("🎉 批量创建完成!") print(f"⏱️ 耗时: {duration:.2f} 秒") print(f"✅ 创建权限: {perm_count} 个") print(f"👥 创建角色: {role_count} 个") print(f"🔗 分配权限: {perm_assigned_count} 次") print("=" * 50) else: print("操作已取消") if __name__ == "__main__": main()