greater_wms/data_base/test_permission.py

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": "staff", "path": "/permission/roles", "components": ["add_role", "edit_role", "delete_role"]},
    
    
    # 库存管理
    {"primary_page": "stock","path": "/stock/management", "components": ["create_stock", "adjust_stock", "delete_stock"]},
    {"primary_page": "stock","path": "/stock/stocklist", "components": ["view_stock", "export_stock"]},
    {"primary_page": "stock","path": "/stock/stockbinlist", "components": ["view_bin", "edit_bin"]},
    {"primary_page": "stock","path": "/stock/emptybin", "components": ["view_empty", "mark_empty"]},
    {"primary_page": "stock","path": "/stock/occupiedbin", "components": ["view_occupied", "release_bin"]},
    {"primary_page": "stock","path": "/stock/binset", "components": ["add_bin", "edit_bin", "delete_bin"]},
    {"primary_page": "stock","path": "/stock/handcount", "components": ["start_count", "record_count", "finish_count"]},

    # ERP管理
    {"primary_page": "erp","path": "/erp/erpasn", "components": ["create_asn", "edit_asn", "delete_asn"]},
    {"primary_page": "erp","path": "/erp/erpasnmaterial", "components": ["add_material", "edit_material", "delete_material"]},
    {"primary_page": "erp","path": "/erp/erpdnmaterial", "components": ["add_dn_material", "edit_dn_material", "delete_dn_material"]},
    {"primary_page": "erp","path": "/erp/erpasnaudit", "components": ["approve_asn", "reject_asn"]},
    {"primary_page": "erp","path": "/erp/erpdn", "components": ["create_dn", "edit_dn", "delete_dn"]},
    {"primary_page": "erp","path": "/erp/erpsortstock", "components": ["view_sorted", "export_sorted"]},
    
    # 入库管理
    {"primary_page": "inbound","path": "/inbound/asn", "components": ["create_asn", "edit_asn", "delete_asn"]},
    {"primary_page": "inbound","path": "/inbound/predeliverystock", "components": ["view_pre_delivery", "confirm_delivery"]},
    {"primary_page": "inbound","path": "/inbound/preloadstock", "components": ["view_pre_load", "confirm_load"]},
    {"primary_page": "inbound","path": "/inbound/presortstock", "components": ["view_pre_sort", "confirm_sort"]},
    {"primary_page": "inbound","path": "/inbound/sortstock", "components": ["view_sorted", "export_sorted"]},
    {"primary_page": "inbound","path": "/inbound/shortage", "components": ["report_shortage", "resolve_shortage"]},
    {"primary_page": "inbound","path": "/inbound/more", "components": ["view_additional", "add_additional"]},
    {"primary_page": "inbound","path": "/inbound/asnfinish", "components": ["confirm_finish", "export_finish"]},


    # 容器管理
    {"primary_page":"container","path": "/container/containerlist", "components": ["view_container", "add_container"]},
    {"primary_page":"container","path": "/container/containerdetail", "components": ["view_detail", "edit_detail"]},
    {"primary_page":"container","path": "/container/containercategory", "components": ["view_category", "add_category"]},
    {"primary_page":"container","path": "/container/containeroperate", "components": ["operate_container", "confirm_operation"]},
    # 仪表盘
    {"primary_page": "dashboard","path": "/dashboard/inboundAndOutbound", "components": ["view_chart", "export_data"]},
    {"primary_page": "dashboard","path": "/dashboard/flows_statements", "components": ["view_flow", "filter_flow"]},
    {"primary_page": "dashboard","path": "/dashboard/flows", "components": ["view_flow_detail", "export_flow"]},
    {"primary_page": "dashboard","path": "/dashboard/flows_complex", "components": ["view_complex_flow", "analyze_flow"]},
    {"primary_page": "dashboard","path": "/dashboard/batchlog", "components": ["view_log", "search_log"]},
    {"primary_page": "dashboard","path": "/dashboard/ContainerDetailLogModel", "components": ["view_container_log", "export_log"]},
    {"primary_page": "dashboard","path": "/dashboard/MaterialChangeHistory", "components": ["view_history", "compare_history"]},
    

    

    
    # 统计管理
    {"primary_page":"count","path": "/count/detaillog", "components": ["view_log", "search_log"]},
    {"primary_page":"count","path": "/count/batchoperatelog", "components": ["view_batch_log", "export_log"]},
    {"primary_page":"count","path": "/count/countbatchlog", "components": ["view_count_log", "analyze_log"]},
    {"primary_page":"count","path": "/count/presortstock", "components": ["view_pre_sort", "export_pre_sort"]},
    {"primary_page":"count","path": "/count/sortstock", "components": ["view_sorted", "export_sorted"]},
    {"primary_page":"count","path": "/count/shortage", "components": ["view_shortage", "resolve_shortage"]},
    {"primary_page":"count","path": "/count/containerDetail", "components": ["view_container", "export_container"]},
    {"primary_page":"count","path": "/count/batch", "components": ["view_batch", "manage_batch"]},
    {"primary_page":"count","path": "/count/asnfinish", "components": ["view_finish", "export_finish"]},
    

    
    # 出库管理
    {"primary_page":"outbound","path": "/outbound/dn", "components": ["create_dn", "edit_dn", "delete_dn"]},
    {"primary_page":"outbound","path": "/outbound/freshorder", "components": ["view_fresh", "process_fresh"]},
    {"primary_page":"outbound","path": "/outbound/neworder", "components": ["view_new", "process_new"]},
    {"primary_page":"outbound","path": "/outbound/pickstock", "components": ["view_pick", "confirm_pick"]},
    {"primary_page":"outbound","path": "/outbound/pickedstock", "components": ["view_picked", "confirm_picked"]},
    {"primary_page":"outbound","path": "/outbound/pickinglist", "components": ["view_list", "export_list"]},
    {"primary_page":"outbound","path": "/outbound/shippedstock", "components": ["view_shipped", "confirm_shipped"]},
    {"primary_page":"outbound","path": "/outbound/backorder", "components": ["view_backorder", "process_backorder"]},
    {"primary_page":"outbound","path": "/outbound/pod", "components": ["view_pod", "confirm_pod"]},
    {"primary_page":"outbound","path": "/outbound/container_check", "components": ["check_container", "confirm_check"]},
    
    # 商品管理
    {"primary_page":"goods","path": "/goods/goodslist", "components": ["view_goods", "add_goods"]},
    {"primary_page":"goods","path": "/goods/goodsunit", "components": ["view_unit", "add_unit"]},
    {"primary_page":"goods","path": "/goods/goodsclass", "components": ["view_class", "add_class"]},
    {"primary_page":"goods","path": "/goods/goodsbrand", "components": ["view_brand", "add_brand"]},
    {"primary_page":"goods","path": "/goods/goodscolor", "components": ["view_color", "add_color"]},
    {"primary_page":"goods","path": "/goods/goodsspecs", "components": ["view_specs", "add_specs"]},
    {"primary_page":"goods","path": "/goods/goodsshape", "components": ["view_shape", "add_shape"]},
    {"primary_page":"goods","path": "/goods/goodsorigin", "components": ["view_origin", "add_origin"]},
    
    # 任务管理
    {"primary_page":"taskpage","path": "/taskpage/task", "components": ["view_task", "create_task", "assign_task"]},
    
    # 仓库管理
    {"primary_page":"warehouse","path": "/warehouse/warehouseset", "components": ["view_setting", "edit_setting"]},
    {"primary_page":"warehouse","path": "/warehouse/department", "components": ["view_department", "add_department"]},
    {"primary_page":"warehouse","path": "/warehouse/boundcodetype", "components": ["view_code_type", "add_code_type"]},
    {"primary_page":"warehouse","path": "/warehouse/boundtype", "components": ["view_bound_type", "add_bound_type"]},
    {"primary_page":"warehouse","path": "/warehouse/boundbusiness", "components": ["view_business", "add_business"]},
    {"primary_page":"warehouse","path": "/warehouse/status", "components": ["view_status", "edit_status"]},
    {"primary_page":"warehouse","path": "/warehouse/product", "components": ["view_product", "add_product"]},
    
    # 员工管理
    {"primary_page":"staff","path": "/staff/stafflist", "components": ["view_staff", "add_staff"]},
    {"primary_page":"staff","path": "/staff/stafflist_check_code", "components": ["check_code", "reset_code"]},
    {"primary_page":"staff","path": "/staff/stafftype", "components": ["view_type", "add_type"]},
    
    # 上传中心
    {"primary_page":"uploadcenter","path": "/uploadcenter/initializeupload", "components": ["init_upload", "process_init"]},
    {"primary_page":"uploadcenter","path": "/uploadcenter/addupload", "components": ["add_upload", "process_add"]},
    
    # 下载中心
    {"primary_page":"downloadcenter","path": "/downloadcenter/downloadinbound", "components": ["download_inbound"]},
    {"primary_page":"downloadcenter","path": "/downloadcenter/downloadoutbound", "components": ["download_outbound"]},
    {"primary_page":"downloadcenter","path": "/downloadcenter/downloadstocklist", "components": ["download_stock"]},
    {"primary_page":"downloadcenter","path": "/downloadcenter/downloadgoodslist", "components": ["download_goods"]},
    {"primary_page":"downloadcenter","path": "/downloadcenter/downloadbinlist", "components": ["download_bin"]},

]

# 定义角色及其权限配置
ROLES = {
    "管理员": {
        "description": "系统管理员，拥有所有权限",
        "page_access": "all",  # 所有页面
        "component_access": "all"  # 所有组件
    },
    "经理": {
        "description": "部门经理，拥有大部分管理权限",
        "page_access": [
            "/dashboard", "/stock", "/inbound", "/outbound", 
            "/count", "/container", "/goods", "/warehouse", "/staff"
        ],
        "component_access": "all"  # 所有组件
    },
    "主管": {
        "description": "仓库主管，负责日常运营管理",
        "page_access": [
            "/stock", "/inbound", "/outbound", "/count", "/container"
        ],
        "component_access": [
            "view", "edit", "add", "delete", "export", "confirm"
        ]
    },
    "操作员": {
        "description": "仓库操作员，负责具体操作",
        "page_access": [
            "/stock/stocklist", "/stock/stockbinlist", "/inbound/asn", 
            "/outbound/pickstock", "/outbound/pickedstock"
        ],
        "component_access": [
            "view", "confirm", "process"
        ]
    },
    "查看员": {
        "description": "数据查看员，只能查看数据",
        "page_access": [
            "/dashboard", "/stock/stocklist", "/count/detaillog", "/count","/erp"
        ],
        "component_access": ["view"]
    }
}

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__startswith=page_path,
                        component=None
                    )
                    assigned_permissions.extend(page_perms)
                    
                    # 获取匹配该路径的组件权限
                    comp_perms = all_permissions.filter(
                        page__startswith=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()