【PHP学习笔记】用户注册模块用户类以及验证码类

wuyijie

　　最近正在学习《PHP&MySQL范例精解》，刚刚看到第一张，关于用户注册模块的设计，这本书提供了很多可重用类，便于用于其它项目中。
　　所以，把第一章，可重用类的代码贴出来，便于以后查阅以及供给有需要的朋友。
　　1：User类，包括读取和设置数据库，以及保存更改交互
　　



  1 <?php
  2     class User{
  3         private $uid;
  4         private $fields;
  5         
  6         public function __construct(){
  7             $this->uid=null;
  8             $this->fields=array('username'=>'','password'=>'','emailAddr'=>'','isActive'=>false);
  9         }
10         
11         public function __get($field){
12             if($field=='userId'){
13                 return $this->uid;
14             }else{
15                 return $this->fields[$field];
16             }
17         }
18         
19         public function __set($field,$value){
20             if(array_key_exists($field,$this->fields)){
21                 $this->fields[$field]=$value;
22             }
23         }
24         
25         //return if username is valid format
26         public static function validateUsername($username){
27             return preg_match('/^[A-Z0-9]{2,20}$/i',$username);
28         }
29         
30         //return if email address is valid format
31         public static function validateEmailAddr($email){
32             return filter_var($email,FILTER_VALIDATE_EMAIL);
33         }
34         
35         //return an object populated based on the record‘s user id
36         public static function getById($user_id){
37             $user=new User();
38             $query=sprintf('SELECT USERNAME,PASSWORD,EMAIL_ADDR,IS_ACTIVE '.
39                 'FROM %sUSER WHERE USER_ID=%d',DB_TBL_PREFIX,$user_id);
40             $result=mysql_query($query,$GLOBALS['DB']);
41             if(mysql_num_rows($result)){
42                 $row=mysql_fetch_assoc($result);
43                 $user->username=$row['USERNAME'];
44                 $user->password=$row['PASSWORD'];
45                 $user->emailAddr=$row['EMAIL_ADDR'];
46                 $user->isActive=$row['IS_ACTIVE'];
47                 ChromePhp::log($user_id);
48                 $user->uid=$user_id;
49             }
50             mysql_free_result($result);
51             return $user;
52         }
53         
54         //return an object populated based on the record’s username
55         public static function getByUsername($username){
56             $user=new User();
57             $query=sprintf('SELECT USER_ID,PASSWORD,EMAIL_ADDR,IS_ACTIVE '.
58                 'FROM %sUSER WHERE USERNAME="%s"',DB_TBL_PREFIX,mysql_real_escape_string($username,$GLOBALS['DB']));
59             $result=mysql_query($query,$GLOBALS['DB']);
60             if(mysql_num_rows($result)){
61                 $row=mysql_fetch_assoc($result);
62                 $user->username=$username;
63                 $user->password=$row['PASSWORD'];
64                 $user->emailAddr=$row['EMAIL_ADDR'];
65                 $user->isActive=$row['IS_ACTIVE'];
66                 $user->uid=$row['USER_ID'];
67             }
68             mysql_free_result($result);
69             return $user;
70         }
71         
72         //save the record to the database
73         public function save(){
74             //update existing user's information
75             if($this->uid){
76                 $query = sprintf('UPDATE %sUSER SET USERNAME = "%s", ' .
77                     'PASSWORD = "%s", EMAIL_ADDR = "%s", IS_ACTIVE = %d ' .
78                     'WHERE USER_ID = %d',
79                     DB_TBL_PREFIX,
80                     mysql_real_escape_string($this->username, $GLOBALS['DB']),
81                     mysql_real_escape_string($this->password, $GLOBALS['DB']),
82                     mysql_real_escape_string($this->emailAddr, $GLOBALS['DB']),
83                     $this->isActive,
84                     $this->userId);
85                 return mysql_query($query, $GLOBALS['DB']);
86             }else{
87                 //create a new user
88                 $query=sprintf('INSERT INTO %sUSER(USERNAME,PASSWORD,' .
89                         'EMAIL_ADDR,IS_ACTIVE) VALUES ("%s","%s","%s",%d)',
90                         DB_TBL_PREFIX,
91                         mysql_real_escape_string($this->username,$GLOBALS['DB']),
92                         mysql_real_escape_string($this->password,$GLOBALS['DB']),
93                         mysql_real_escape_string($this->emailAddr,$GLOBALS['DB']),
94                         $this->isActive);
95                 if(mysql_query($query,$GLOBALS['DB'])){
96                     $this->uid=mysql_insert_id($GLOBALS['DB']);
97                     return true;
98                 }else{
99                     return false;   
100                 }
101             }
102         }
103         
104         //set the record as inactive and return an activation token
105         public function setInactive(){
106             $this->isActive=false;
107             $this->save();
108             $token=random_text(5);
109             $query=sprintf('INSERT INTO %sPENDING (USER_ID,TOKEN)' .
110                     'VALUES (%d,"%s")',DB_TBL_PREFIX,$this->uid,$token);
111             return (mysql_query($query,$GLOBALS['DB']))?$token:false;
112         }
113         
114         //clear the user's pending status and set the record as active
115         public function setActive($token){
116             $query=sprintf('SELECT TOKEN FROM %sPENDING WHERE USER_ID=%d ' .
117                     'AND TOKEN="%s"',DB_TBL_PREFIX,$this->uid,mysql_real_escape_string($token,$GLOBALS['DB']));
118             $result=mysql_query($query,$GLOBALS['DB']);
119             if(!mysql_num_rows(($result))){
120                 mysql_free_result($result);
121                 return false;
122             }else{
123                 mysql_free_result($result);
124                 $query=sprintf('DELETE FROM %sPENDING WHERE USER_ID=%d ' .
125                         'AND TOKEN="%s"',DB_TBL_PREFIX,$this->uid,mysql_real_escape_string($token,$GLOBALS['DB']));
126                 if(!mysql_query($query,$GLOBALS['DB'])){
127                     return false;
128                 }else{
129                     $this->isActive=true;
130                     return $this->save();
131                 }                        
132             }   
133         }
134     }
135 ?>
　　如何使用：



<?php
    //create user instance
    $u=new User();
    $u->username='jack';
    $u->password=sha1('gogo');
    $u->emailAddr='zjczoo@gmail.com';
    $u->save();//save this user
?>
　　



1 <?php
2     $u=User::getByUsername('jack');//update user('jack')
3     $u->password=sha1('newgogo');
4     $u->save();//save new jack
5 ?>
　　
　　2:验证码类：这个比较简单，你可以自己加个图片==



1 <?php
2     //must start or continue session and save CAPTCHA string in $_SESSION for
3     //it to be available to other requests
4     if(!isset($_SESSION)){
5         session_start();
6         header('Cache-control:private');
7     }
8     
9     //create a 65*20 pixel image
10     $width=65;
11     $height=20;
12     $image=imagecreate(65,20);
13     
14     //fill the image background color
15     $bg_color=imagecolorallocate($image,0x33,0x66,0xFF);
16     imagefilledrectangle($image,0,0,$width,$height,$bg_color);
17     
18     //fetch random text
19     $text=random_text(5);
20     
21     //determine x and y coordinates for centering text
22     $font=5;
23     $x=imagesx($image)/2-strlen($text)*imagefontwidth($font)/2;
24     $y=imagesy($image)/2-imagefontheight($font)/2;
25     
26     //write text on image
27     $fg_color=imagecolorallocate($image,0xFF,0xFF,0xFF);
28     imagestring($image,$font,$x,$y,$text,$fg_color);
29     
30     //save the CAPTCHA string for later comparison
31     $_SESSION['captcha']=$text;
32     
33     //output the image
34     header('Content-type:image/png');
35     imagepng($image);
36     
37     imagedestroy($image);
38 ?>
　　另外，该类用到了random_text()函数，代码如下：



1 <?php
2     function random_text($count,$rm_similar=false){
3         $chars=array_flip(array_merge(range(0,9),range('A','Z')));
4         
5         if($rm_similar){
6             unset($chars[0],$chars[1],$chars[2],$chars[5],$chars[8],$chars['B'],$chars['I'],$chars['O'],$chars['Q'],$chars['S'],$chars['V'],$chars['Z']);
7         }
8         
9         for($i=0,$text='';$i<$count;$i++){
10             $text.=array_rand($chars);
11         }
12         return $text;
13     }
14 ?>
　　
　　连接数据库类：



<?php
// database connection and schema constants
define('DB_HOST', 'localhost');
define('DB_USER', 'username');
define('DB_PASSWORD', 'yourpassword');
define('DB_SCHEMA', 'WROX_DATABASE');
define('DB_TBL_PREFIX', 'WROX_');
// establish a connection to the database server
if (!$GLOBALS['DB'] = mysql_connect(DB_HOST, DB_USER, DB_PASSWORD))
{
    die('Error: Unable to connect to database server.');
}
if (!mysql_select_db(DB_SCHEMA, $GLOBALS['DB']))
{
    mysql_close($GLOBALS['DB']);
    die('Error: Unable to select database schema.');
}
?>
　　
　　sql语句：



1 DROP TABLE IF EXISTS WROX_PENDING;
2 DROP TABLE IF EXISTS WROX_USER;
3
4 CREATE TABLE WROX_USER (
5     USER_ID    INTEGER UNSIGNED  NOT NULL  AUTO_INCREMENT,
6     USERNAME   VARCHAR(20)       NOT NULL,
7     PASSWORD   CHAR(40)          NOT NULL,
8     EMAIL_ADDR VARCHAR(100)      NOT NULL,
9     IS_ACTIVE  TINYINT(1)        DEFAULT 0,
10
11     PRIMARY KEY (USER_ID)
12 )
13 ENGINE=MyISAM DEFAULT CHARACTER SET gb2312
14     COLLATE gb2312_chinese_ci AUTO_INCREMENT=0;
15
16 CREATE TABLE WROX_PENDING (
17     USER_ID       INTEGER UNSIGNED  PRIMARY KEY NOT NULL,
18     TOKEN         CHAR(10)          NOT NULL,
19     CREATED_DATE  TIMESTAMP         DEFAULT  CURRENT_TIMESTAMP,
20
21     FOREIGN KEY (USER_ID)
22         REFERENCES WROX_USER(USER_ID)
23 )
24 ENGINE=MyISAM DEFAULT CHARACTER SET gb2312
25     COLLATE gb2312_chinese_ci;