之前的通讯录进行文件保存

妖狐艹你老母 2022-07-12 01:15 165阅读 0赞

上次做完通讯录后，总是有点遗憾，就是没有把电话数据保存到本地，没有实质上的实现通讯录功能，今天，我把通讯录重新拿出来做了一次，对它进行文件保存。

做完之后，感觉其实很简单，就是在数据存入的时候，不要再添加结点到链表中了，直接把这个结构体数据保存到文件中，查看的时候就是从文件中把数据读出来，这个时候需要用到链表了，因为不知道文件中数据的数量，所以链表是一个不错的选择。读出来之后，再把链表打印一下。稍微麻烦一点的就是删除了，先把数据读出来，然后进行遍历删除，接着把原文件删除，然后再把链表一个一个存进文件。

/**********************************************************
    File Name:         通讯录项目(文件保存)
    Author:                      Date：2017-2-6
    Description： 实现通讯录功能
    Fuction List:       show()   	 显示最初布局
    					Init_list()	 添加好友信息
    					Print()		 列表好友信息
    					Search()	 搜索好友
    					Delete() 	 删除好友
    ************************************************************/
    #include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <sys/types.h>
    #include <sys/stat.h>
    #include <fcntl.h>
    #include <unistd.h>
    
    
    #define ok       	   0
    #define error    	  -1
    #define malloc_error  -2
    #define N             20
    
    typedef struct node
    {
    	char ID;
    	char name[N];
    	char tel[N ];
    	char address[N];
    	char company_phone[N];
    	struct node *next;
    }Address;
    typedef Address* PAddress;
    
    int i = 1;
    
    //显示最初布局
    void show()
    {
    	system("clear");
    	printf("\t\t*********************************************\n");
    	printf("\t\t* \twelcome to address book             *\n");
    	printf("\t\t*                                           *\n");
    	printf("\t\t*\tA) Add the information of friends   *\n");
    	printf("\t\t*\tB) Show all friends information     *\n");
    	printf("\t\t*\tC) Search for the friend            *\n");
    	printf("\t\t*\tD) Remove the friend       	    *\n");
    	printf("\t\t*                                           *\n");
    	printf("\t\t*********************************************\n");
    	printf("                                         \n");
    	printf("                                         \n");
    	printf("                                         \n");
    	printf("                                         \n");
    	printf("         please input your choice:");
    }
    
    //添加好友信息
    int Init_list(PAddress h)
    {
    	if(h == NULL)
    	{
    		return error;
    	}
    	
    	PAddress p = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	if (p == NULL)
    	{
    		return malloc_error;
    	}
    	
    	system("clear");
    	
    	p->ID = '0'+i;                             
    	i++;
    	
    	printf("\n		please input the name:");
    	scanf("%s", p->name);                             //输入姓名
    
    	printf("\n		please input the tel:");
    	scanf("%s", p->tel);                           //输入手机号
    	
    	int j = 0;
    	int len = 0;
    	while(p->tel[j++] != '\0')
    	{
    		len++;
    	}
    	while(len != 11)                //判断输入的手机号码是不是11位的
    	{
    		
    		printf("\n		please input 11 numbers,input again:");
    		scanf("%s", p->tel);
    		len = 0;
    		j = 0;
    		while(p->tel[j++] != '\0')
    		{
    			len++;
    		}
    	}
    
    	printf("\n		please input the address:");
    	scanf("%s", p->address);
    
    	printf("\n		please input the company phone:");
    	scanf("%s", p->company_phone);
    	
    	j = 0;
    	len = 0;
    	while(p->company_phone[j++] != '\0')
    	{
    		len++;
    	}
    	while(len != 8)             //判断输入的电话号码是不是8位的
    	{
    		
    		printf("\n		please input 8 numbers,input again:");
    		scanf("%s", p->company_phone);
    		len = 0;
    		j = 0;
    		while(p->company_phone[j++] != '\0')
    		{
    			len++;
    		}
    	}
    			
    	p->next = NULL;
    	
    	int fd;
    	int ret;
    	fd = open("address_book.txt", O_CREAT | O_RDWR | O_APPEND, S_IRUSR | S_IWUSR);
    	if(fd == -1)
    	{
    		perror("open");
    		return;
    	}
    	
    	ret = write(fd, p, sizeof(Address));
    	if(ret == -1)
    	{
    		perror("write");
    		return;
    	}
    	close(fd);
    	
    	return ok;
    }
    
    //列表好友信息
    void Print()
    {
    	int fd;
    	int ret;
    
    	fd = open("address_book.txt", O_RDONLY, S_IRUSR | S_IWUSR);
    	if(fd == -1)
    	{
    		printf("There is no address_book.txt!\n");
    		return;
    	}
    	
    	PAddress h = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	PAddress p = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	
    	ret = read(fd, p, sizeof(Address));
    	if(ret == 0)
    	{
    		printf("There is no friend!\n");
    		return;
    	}
    	h->next = p;
    	while(1)
    	{
    		PAddress t = (PAddress)malloc(sizeof(Address)/sizeof(char));
    		
    		ret = read(fd, t, sizeof(Address));
    		if(ret == 0)
    		{
    			p->next = NULL;
    			break;
    		}
    		p->next = t;
    		p = p->next;
    	}
    	
    	PAddress temp = h->next;
    	while(temp)
    	{
    		printf("Num:%c name:%s  tel:%s  address:%s  company phone:%s\n", 
    		temp->ID , temp->name, temp->tel, temp->address, temp->company_phone);
    		temp = temp->next;
    	}
    	printf("\n");
    	
    	close(fd);
    }
    
    //搜索好友
    int Search(int i)
    {
    	int fd;
    	int ret;
    	int flag = 1;
    	fd = open("address_book.txt", O_RDONLY, S_IRUSR | S_IWUSR);
    	if(fd == -1)
    	{
    		printf("There is no address_book.txt!\n");
    		return;
    	}
    	
    	PAddress h = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	PAddress p = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	
    	ret = read(fd, p, sizeof(Address));
    	if(ret == 0)
    	{
    		printf("There is no friend!\n");
    		return;
    	}
    	h->next = p;
    	while(1)
    	{
    		PAddress t = (PAddress)malloc(sizeof(Address)/sizeof(char));
    		
    		ret = read(fd, t, sizeof(Address));
    		if(ret == 0)
    		{
    			p->next = NULL;
    			break;
    		}
    		p->next = t;
    		p = p->next;
    	}
    	
    	PAddress temp = h->next;
    	while(temp)
    	{
    		if(temp->ID == i)
    		{
    			flag = 0;
    			printf("\n\n");
    			printf("Num:%c  name:%s  tel:%s  address:%s  company phone:%s\n", 
    			temp->ID, temp->name, temp->tel, temp->address, temp->company_phone);
    			printf("\n\n");
    		}
    		temp = temp->next;
    	}
    	if(flag)
    	{
    		printf("\n\n");
    		printf("Don't have the friend\n");
    		printf("\n\n");
    	}
    	
    	return ok;
    }
    
    //删除好友
    int Delete(int i)
    {
    	int fd;
    	int ret;
    	
    	fd = open("address_book.txt", O_RDONLY, S_IRUSR | S_IWUSR);
    	if(fd == -1)
    	{
    		printf("There is no address_book.txt!\n");
    		return;
    	}
    	
    	PAddress h = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	PAddress p = (PAddress)malloc(sizeof(Address)/sizeof(char));
    	
    	ret = read(fd, p, sizeof(Address));
    	if(ret == 0)
    	{
    		printf("There is no friend!\n");
    		return;
    	}
    	h->next = p;
    	while(1)
    	{
    		PAddress t = (PAddress)malloc(sizeof(Address)/sizeof(char));
    		
    		ret = read(fd, t, sizeof(Address));
    		if(ret == 0)
    		{
    			p->next = NULL;
    			break;
    		}
    		p->next = t;
    		p = p->next;
    	}
    	close(fd);
    	
    	PAddress temp = h;
    	
    	if(temp->next->next == NULL)             //一个结点的表
    	{
    		if(temp->next->ID == i)
    		{
    			PAddress tmp = temp->next;
    			temp->next = NULL;
    			free(tmp);
    			
    			ret = remove("address_book.txt");
    			if(ret == -1)
    			{
    				perror("remove");
    				return;
    			}
    			int fd2 = open("address_book.txt", O_CREAT | O_RDWR | O_APPEND, S_IRUSR | S_IWUSR);
    			close(fd2);
    			
    			printf("\n\n");
    			printf("Deleted successfully\n");
    			printf("\n\n");
    			
    			return ok;
    		}
    		else
    		{
    			return error;
    		}
    	} 
    	
    	while(temp)                     // 此处temp 不能写temp->next  要考虑删除的是最后一个结点
    	{
    		if(temp->next->ID == i)
    		{
    			PAddress p = temp->next;
    			temp->next = p->next;
    			free(p);
    			printf("\n\n");
    			printf("Deleted successfully\n");
    			printf("\n\n");
    			break;
    		}
    		if(temp->next->next == NULL )         //考虑i超出了表中元素的时候
    		{
    			return error;
    		}			
    		temp = temp->next;
    	}
    	
    	int fd1;
    	fd1 = open("address_book.txt",  O_RDWR | O_TRUNC | O_APPEND, S_IRUSR | S_IWUSR);
    	
    	temp = h->next;
    	while(temp)
    	{
    		ret = write(fd1, temp, sizeof(Address));
    		if(ret == -1)
    		{
    			perror("write");
    			return;
    		}
    		temp = temp->next;
    	}
    	close(fd1);
    	return ok;
    }
    
    int main()
    {
    	char option[2];                              //定义一个字符串存放输入的选项  去除回车键对循环的影响
    	int back = 1;
    	char id[3];
    	
    	PAddress head_node = (PAddress)malloc(sizeof(Address)/sizeof(char));     //定义头结点
    	if(head_node == NULL)
    	{
    		return malloc_error;
    	}
    	head_node->next = NULL;
    	
    	while(back--)
    	{
    		show();
    		scanf("%s", option);
    		
    		 switch (option[0])
    		 {
    			 case 'A' :
    			 {
    				Init_list(head_node);
    		
    				system("clear");
    				printf("\n\n\n\n\n\n");
    				printf("		Congratulations on your success to create contacts!\n");
    				printf("\n\n\n\n\n\n");
    				printf("		If you want to back,please input (1):");
    				scanf("%d", &back);
    				break;
    			}
    			
    			case 'B' :
    			{
    				back = 0;
    				system("clear");
    				printf("\n\n\n\n\n\n");
    				
    				Print();
    				
    				printf("If you want to back,please input (1):");
    				scanf("%d", &back);
    				break;
    			}
    			
    			case 'C' :
    			{
    				back = 0;
    				system("clear");
    				printf("\n\n\n\n\n\n");
    				printf("if you want to search your friend,please input ID:");
    				scanf("%s",id);
    				
    				Search(id[0]);
    				
    				printf("If you want to back,please input (1):");
    				scanf("%d", &back);
    				break;
    			}
    			
    			case 'D' :
    			{
    				back = 0;
    				system("clear");
    				printf("\n\n\n\n\n\n");
    				printf("if you want to delete your friend,please input ID:");
    				scanf("%s",id);
    				
    				if (Delete(id[0]) != ok)
    				{
    					printf("\n\n");
    					printf("num %d friend is not exist!", id);
    					printf("\n\n");
    				}
    				printf("If you want to back,please input (1):");
    				scanf("%d", &back);
    				break;
    			}
    			
    			default :
    			{
    				printf("\n\n");
    				system("clear");
    				printf("\n\n\n\n\n\n");
    				printf("\t\tplaese input A、B、C or D!\n");
    				printf("\n\n");
    				printf("If you want to back,please input (1):");
    				scanf("%d", &back);
    				
    				break;
    			}
    		 }
    	
    	}
    	return 0;
    }