wordpress 文件上传设置文件类型和大小(角色权限)
发布:smiling 来源: PHP粉丝网 添加日期:2014-03-20 13:43:45 浏览: 评论:0
我们最初是简单的给所有用户有上传文件并且设置文件上传可以使用media_handle_upload和wp_handle_upload 类似的函数,其实都是调用wp_handle_upload函数
这个函数的第4个参数,支持数组,代码如下:
- $access = array('test_form' => false, 'mimes' => array('jpeg' => 'image/jpeg',
- 'png' => 'image/png','gif' => 'image/gif','jpg' => 'image/jpeg'));
这样写就可以限制文件的类型,但是貌似对伪造的文件不好使,经过测试可以使用一个过滤器,完美解决这个问题,过滤器名称:wp_handle_upload_prefilter,代码如下:
- add_filter('wp_handle_upload_prefilter', 'custom_upload_filter' );
- function custom_upload_filter($file)
- {
- $fileSize = $file['size'];//获取文件大小 做自定义处理
- $info =getimagesize($file['tmp_name']);//获取文件属性 如果不是图片将返回false
- return $file;
- }
让用户拥有上传文件的权限
默认情况下,有些用户是不允许上传文件的,你可以在主题的 functions.php 添加下面的代码:
- //允许用户投稿时上传文件
- if ( current_user_can('contributor') && !current_user_can('upload_files') )
- add_action('admin_init', 'allow_contributor_uploads');
- function allow_contributor_uploads() {
- $contributor = get_role('contributor');
- $contributor->add_cap('upload_files');
- }
上面的代码就是给 'contributor' 这个用户角色添加了 'upload_files'(上传文件)的权限。
限制用户上传文件的类型
首先,大家可以先了解一下 WordPress 默认允许上传的文件类型,打开WordPress的 /wp-includes/functions.php 文件,然后搜索 function wp_get_mime_types 定位到那里,你就会看到详细的文件类型:
- function wp_get_mime_types() {
- // Accepted MIME types are set here as PCRE unless provided.
- return apply_filters( 'mime_types', array(
- // Image formats
- 'jpg|jpeg|jpe' => 'image/jpeg',
- 'gif' => 'image/gif',
- 'png' => 'image/png',
- 'bmp' => 'image/bmp',
- 'tif|tiff' => 'image/tiff',
- 'ico' => 'image/x-icon',
- // Video formats
- 'asf|asx|wax|wmv|wmx' => 'video/asf',
- 'avi' => 'video/avi',
- 'divx' => 'video/divx',
- 'flv' => 'video/x-flv',
- 'mov|qt' => 'video/quicktime',
- 'mpeg|mpg|mpe' => 'video/mpeg',
- 'mp4|m4v' => 'video/mp4',
- 'ogv' => 'video/ogg',
- 'mkv' => 'video/x-matroska',
- // Text formats
- 'txt|asc|c|cc|h' => 'text/plain',
- 'csv' => 'text/csv',
- 'tsv' => 'text/tab-separated-values',
- 'ics' => 'text/calendar',
- 'rtx' => 'text/richtext',
- 'css' => 'text/css',
- 'htm|html' => 'text/html',
- // Audio formats
- 'mp3|m4a|m4b' => 'audio/mpeg',
- 'ra|ram' => 'audio/x-realaudio',
- 'wav' => 'audio/wav',
- 'ogg|oga' => 'audio/ogg',
- 'mid|midi' => 'audio/midi',
- 'wma' => 'audio/wma',
- 'mka' => 'audio/x-matroska',
- // Misc application formats
- 'rtf' => 'application/rtf',
- 'js' => 'application/javascript',
- 'pdf' => 'application/pdf',
- 'swf' => 'application/x-shockwave-flash',
- 'class' => 'application/java',
- 'tar' => 'application/x-tar',
- 'zip' => 'application/zip',
- 'gz|gzip' => 'application/x-gzip',
- 'rar' => 'application/rar',
- '7z' => 'application/x-7z-compressed',
- 'exe' => 'application/x-msdownload',
- // MS Office formats
- 'doc' => 'application/msword',
- 'pot|pps|ppt' => 'application/vnd.ms-powerpoint',
- 'wri' => 'application/vnd.ms-write',
- 'xla|xls|xlt|xlw' => 'application/vnd.ms-excel',
- 'mdb' => 'application/vnd.ms-access',
- 'mpp' => 'application/vnd.ms-project',
- 'docx' => 'application/vnd.openxmlformats-
- officedocument.wordprocessingml.document',
- 'docm' => 'application/vnd.ms-word.document.macroEnabled.12',
- 'dotx' => 'application/vnd.openxmlformats-
- officedocument.wordprocessingml.template',
- 'dotm' => 'application/vnd.ms-word.template.macroEnabled.12',
- 'xlsx' => 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
- 'xlsm' => 'application/vnd.ms-excel.sheet.macroEnabled.12',
- 'xlsb' => 'application/vnd.ms-excel.sheet.binary.macroEnabled.12',
- 'xltx' => 'application/vnd.openxmlformats-officedocument.spreadsheetml.template',
- 'xltm' => 'application/vnd.ms-excel.template.macroEnabled.12',
- 'xlam' => 'application/vnd.ms-excel.addin.macroEnabled.12',
- 'pptx' => 'application/vnd.openxmlformats-
- officedocument.presentationml.presentation',
- 'pptm' => 'application/vnd.ms-powerpoint.presentation.macroEnabled.12',
- 'ppsx' => 'application/vnd.openxmlformats-officedocument.presentationml.slideshow',
- 'ppsm' => 'application/vnd.ms-powerpoint.slideshow.macroEnabled.12',
- 'potx' => 'application/vnd.openxmlformats-officedocument.presentationml.template',
- 'potm' => 'application/vnd.ms-powerpoint.template.macroEnabled.12',
- 'ppam' => 'application/vnd.ms-powerpoint.addin.macroEnabled.12',
- 'sldx' => 'application/vnd.openxmlformats-officedocument.presentationml.slide',
- 'sldm' => 'application/vnd.ms-powerpoint.slide.macroEnabled.12',
- 'onetoc|onetoc2|onetmp|onepkg' => 'application/onenote',
- // OpenOffice formats
- 'odt' => 'application/vnd.oasis.opendocument.text',
- 'odp' => 'application/vnd.oasis.opendocument.presentation',
- 'ods' => 'application/vnd.oasis.opendocument.spreadsheet',
- 'odg' => 'application/vnd.oasis.opendocument.graphics',
- 'odc' => 'application/vnd.oasis.opendocument.chart',
- 'odb' => 'application/vnd.oasis.opendocument.database',
- 'odf' => 'application/vnd.oasis.opendocument.formula',
- // WordPerfect formats
- 'wp|wpd' => 'application/wordperfect',
- ) );
=> 的前面为格式,后面为格式描述,如果你要禁止上传其中的某些类型,可以参考下面的例子,将下面的代码添加到主题的 functions.php 文件:
- //禁止上传avi和mp4格式的文件
- add_filter('upload_mimes', 'custom_upload_mimes');
- function custom_upload_mimes ( $existing_mimes=array() ) {
- unset ($existing_mimes['avi']);
- unset ($existing_mimes['mp4']);
- return $existing_mimes;
- }
如果你还要禁止更多,可以按照 unset ($existing_mimes['格式']); 样例添加即可,如果你仅仅只需要允许用户上传几种类型而已,还可以通过下面的更简洁的方法,代码添加到主题的functions.php 文件:
- //只允许上传图片文件
- add_filter('upload_mimes', 'custom_upload_mimes');
- function custom_upload_mimes ( $existing_mimes=array() ) {
- unset ($existing_mimes);//禁止上传任何文件
- $existing_mimes['jpg|jpeg|gif|png']='image/image';//允许用户上传jpg,gif,png文件
- return $existing_mimes;
- }
如果你还要允许上传其他格式,重复使用 $existing_mimes['格式']='描述';即可.
限制用户上传的文件大小
同样在主题的 functions.php 文件中,添加下面的代码:
- //限制上传文件的最大体积
- function max_up_size() {
- return 500*1024; // 500 kb
- }
- add_filter('upload_size_limit', 'max_up_size');
上面的例子是限制所有用户上传的文件的最大体积为 500 kb (1M =1024*1024)。
注意:主机空间和WordPress本身一般设置了允许上传的文件的最大体积,所以在这里设置需要考虑到这点.
限制不同用户角色可上传的文件类型及大小
其实上面已经给出了限制类型和大小的方法,要根据不同用户角色来限制,只需要添加角色判断代码即可,倡萌举个综合的例子:
不同用户上传的类型,代码如下:
- function custom_upload_mimes ( $existing_mimes=array() ) {
- unset ($existing_mimes);//禁止上传任何文件
- if( current_user_can( 'publish_posts' ) && !current_user_can( 'publish_pages' ) ) {
- //允许作者(Author)上传的类型
- $existing_mimes['jpg|jpeg|gif|png']='image/image';//允许用户上传jpg,gif,png
- 文件
- $existing_mimes['zip']='application/zip'; //允许用户上传zip压缩包
- $existing_mimes['pdf']='application/pdf'; //允许用户上传pdf文件
- }elseif( current_user_can( 'edit_posts' ) && !current_user_can( 'publish_posts' ) )
- {
- //允许投稿者(Contributor)上传的类型
- $existing_mimes['jpg|jpeg|gif|png']='image/image';
- $existing_mimes['pdf']='application/pdf';
- }else{
- //其他用户角色上传的类型
- $existing_mimes['jpg|jpeg|gif|png']='image/image';
- }
- return $existing_mimes;
- }
- //不同用户上传的大小
- function max_up_size() {
- if( current_user_can( 'publish_posts' ) && !current_user_can( 'publish_pages' ) ) {
- return 2048*1024; // 允许作者(Author)上传 2M
- }elseif( current_user_can( 'edit_posts' ) && !current_user_can( 'publish_posts' ) )
- {
- return 1024*1024; // 允许投稿者(Contributor)上传 1M
- }else{
- return 500*1024; // 其他用户角色上传 500 kb
- }
- }
- //只对非管理员执行这两个函数(即:对管理员不生效)
- if( !current_user_can( 'manage_options' ) ) {
- add_filter('upload_mimes', 'custom_upload_mimes');
- add_filter('upload_size_limit', 'max_up_size');
- }
大家只要灵活使用 if 语句判断不同的角色赋予不同的权限即可.
Tags: wordpress 文件上传 类型
相关文章
- ·WordPress初级教程1:什么是博客?(2013-11-11)
- ·WordPress初级教程-2: 什么是WordPress?(2013-11-11)
- ·WordPress初级教程-3: WordPress的功能和特点(2013-11-11)
- ·WordPress初级教程-4: 选择WordPress博客的主机和域名(2013-11-11)
- ·WordPress初级教程-5: 安装WordPress(2013-11-11)
- ·WordPress初级教程-6: 本地安装WordPress(2013-11-11)
- ·WordPress初级教程-7: 一个数据库中安装多个WordPress博客(2013-11-11)
- ·WordPress初级教程-8: WordPress控制面板/ Dashboard(2013-11-11)
- ·WordPress初级教程-9: WordPress用户设置/ Users(2013-11-11)
- ·WordPress初级教程-10: WordPress博客配置/ Settings(2013-11-11)
- ·关于wordpress上传图片不显示的原因(2013-11-11)
- ·WordPress程序的脆弱点你知道吗 (2013-11-11)
- ·总结八大Wordpress网站百度收录实现秒收的方法绝招 (2013-11-11)
- ·WordPress如何网站投稿者也可以上传图片(2014-03-18)
- ·WordPress怎么修改新用户注册邮件内容(2014-03-18)
- ·WordPress怎么添加前台注册功能(2014-03-18)
推荐文章
热门文章
最新评论文章
- 写给考虑创业的年轻程序员(10)
- PHP新手上路(一)(7)
- 惹恼程序员的十件事(5)
- PHP邮件发送例子,已测试成功(5)
- 致初学者:PHP比ASP优秀的七个理由(4)
- PHP会被淘汰吗?(4)
- PHP新手上路(四)(4)
- 如何去学习PHP?(2)
- 简单入门级php分页代码(2)
- php中邮箱email 电话等格式的验证(2)