weiphp 投票插件的主控制器部分代码2

<?php
public function edit() {        //编辑方法
        // 获取模型信息
        $id = I ( 'id', 0, 'intval' );
        
        if (IS_POST) {        //判断处理传参
            $_POST ['mTime'] = time ();        //获取当前时间戳
            
            $Model = D ( parse_name ( get_table_name ( $this->model ['id'] ), 1 ) );    //实例化模型
            // 获取模型的字段信息
            $Model = $this->checkAttr ( $Model, $this->model ['id'] );
            if ($Model->create () && $Model->save ()) {
                // 增加选项
                D ( 'Addons://Vote/VoteOption' )->set ( I ( 'post.id' ), I ( 'post.' ) );
                
                // 保存关键词
                D ( 'Common/Keyword' )->set ( I ( 'post.keyword' ), 'Vote', I ( 'post.id' ) );
                
                $this->success ( '保存' . $this->model ['title'] . '成功！', U ( 'lists?model=' . $this->model ['name'] ) );
            } else {
                $this->error ( $Model->getError () );
            }
        } else {
            $fields = get_model_attribute ( $this->model ['id'] );        //获取模型信息并缓存
            
            // 获取数据
            $data = M ( get_table_name ( $this->model ['id'] ) )->find ( $id );
            $data || $this->error ( '数据不存在！' );        //判断若没有数据，报错
            
            $token = get_token ();        //获取token
            if (isset ( $data ['token'] ) && $token != $data ['token'] && defined ( 'ADDON_PUBLIC_PATH' )) {        //判断数据中token值
                $this->error ( '非法访问！' );
            }
            
            $option_list = M ( 'vote_option' )->where ( 'vote_id=' . $id )->order ( '`order` asc' )->select ();        //获取数据
            $this->assign ( 'option_list', $option_list );        //向模版发送option_list数据
            
            $this->assign ( 'fields', $fields );        //向模版发送fields数据
            $this->assign ( 'data', $data );        //向模版发送data数据
            $this->meta_title = '编辑' . $this->model ['title'];    //赋值    
            $this->display ( T ( 'Addons://Vote@Vote/edit' ) );        //调用edit模版
        }
    }
    public function add() {        //添加方法
        if (IS_POST) {
            // 自动补充token
            $_POST ['token'] = get_token ();
            $Model = D ( parse_name ( get_table_name ( $this->model ['id'] ), 1 ) );
            // 获取模型的字段信息
            $Model = $this->checkAttr ( $Model, $this->model ['id'] );
            if ($Model->create () && $vote_id = $Model->add ()) {
                // 增加选项
                D ( 'Addons://Vote/VoteOption' )->set ( $vote_id, I ( 'post.' ) );
                
                // 保存关键词
                D ( 'Common/Keyword' )->set ( I ( 'keyword' ), 'Vote', $vote_id );
                
                $this->success ( '添加' . $this->model ['title'] . '成功！', U ( 'lists?model=' . $this->model ['name'] ) );    //提示成功及相应信息
            } else {
                $this->error ( $Model->getError () );    //失败报错
            }
        } else {
            
            $vote_fields = get_model_attribute ( $this->model ['id'] );        //获取模型信息并缓存
            $this->assign ( 'fields', $vote_fields );        //数据交给模版
            // 选项表
            $option_fields = get_model_attribute ( $this->option ['id'] );        //获取信息并缓存
            $this->assign ( 'option_fields', $option_fields );
            
            $this->meta_title = '新增' . $this->model ['title'];        //赋值
            $this->display ( $this->model ['template_add'] ? $this->model ['template_add'] : T ( 'Addons://Vote@Vote/add' ) );        //判断并选择模版显示
        }
    }
    protected function checkAttr($Model, $model_id) {        //定义方法检查字符串
        $fields = get_model_attribute ( $model_id, false );        //获取信息并缓存
        $validate = $auto = array ();        //设置数组
        foreach ( $fields as $key => $attr ) {        //循环fields数组
            if ($attr ['is_must']) { // 必填字段
                $validate [] = array (
                        $attr ['name'],
                        'require',
                        $attr ['title'] . '必须!' 
                );
            }
            // 自动验证规则
            if (! empty ( $attr ['validate_rule'] ) || $attr ['validate_type'] == 'unique') {
                //组装数组
                $validate [] = array (
                        $attr ['name'],
                        $attr ['validate_rule'],
                        $attr ['error_info'] ? $attr ['error_info'] : $attr ['title'] . '验证错误',
                        0,
                        $attr ['validate_type'],
                        $attr ['validate_time'] 
                );
            }
            // 自动完成规则
            if (! empty ( $attr ['auto_rule'] )) {
                //组装数组
                $auto [] = array (
                        $attr ['name'],
                        $attr ['auto_rule'],
                        $attr ['auto_time'],
                        $attr ['auto_type'] 
                );
            } elseif ('checkbox' == $attr ['type']) { // 多选型
                $auto [] = array (
                        $attr ['name'],
                        'arr2str',
                        3,
                        'function' 
                );
            } elseif ('datetime' == $attr ['type']) { // 日期型
                $auto [] = array (
                        $attr ['name'],
                        'strtotime',
                        3,
                        'function' 
                );
            }
        }
        return $Model->validate ( $validate )->auto ( $auto );        //返回模型的信息
    }