django -- ORM实现作者增删改查
前戏
前面我们已经实现了出版社的增删改查，书的增删改查，书和出版社的对应关系。现在来写一下作者的增删改查和书的对应关系，那书和作者有什么关系呢？一个作者可以写多本书，一本书可以有多个作者，所以书和作者是多对多的关系，这样的话，我们就需要一张表来记录书和作者的关系，想一下sql语句是怎么创建表的
-- 创建作者表 create table author( id int primary key auto_increment, name varchar(32) not null );
-- 创建作者和书的关系表 create table author2book( id int primary key auto_increment, author_id int not null, book_id int not null, constraint fk_author foreign key (author_id) references author(id) on delete cascade on update cascade, constraint fk_book foreign key (book_id) references book(id) on delete cascade on update cascade );
ORM创建表
使用上面的SQL语句创建表是不是很麻烦，那我们使用ORM创建表怎么创建呢？这里有两种方法
第一种：自己创建作者表和第三张关系表
# 作者表 class Author(models.Model): id = models.AutoField(primary_key=True) # 自增id主键 name= models.CharField(max_length=32) # 作者名字 # 创建作者和书籍的关系 class Author2Book(models.Model): id = models.AutoField(primary_key=True) author = models.ForeignKey(to='Author', on_delete=models.CASCADE) book = models.ForeignKey(to='Book', on_delete=models.CASCADE)
第二种：让ORM帮我们创建第三张关系表
# 作者表 class Author(models.Model): id = models.AutoField(primary_key=True) # 自增id主键 name= models.CharField(max_length=32) # 作者名字 books = models.ManyToManyField('Book') # ManyToManyField自动帮我们创建第三张表，只是ORM层面建立的一个多对多的关系，不存在于Author表中
def __str__(self):
return self.name
books这个字段不会出现在Author表和Book表里，会另外创建一张表来存两者之间的关系，所以写在Author表或Book表里都可以，如果写在Book表里，里面就要写Author了
我们使用第二种方法来创建数据表，因为如果我们要查作者和书的关系时，第二种方法对象.books就可以查到
执行完生成数据表的命令后，查看数据库，发现ORM自动给我们生成了第三张表，我们在两张数据表里添加一些数
用ORM查询
先来看一下数据表里的数据
作者表
作者和书的对应关系表
书表
添加路由
url(r'^author_list/', views.author_list), # 作者列表
添加对应函数
def author_list(request): # 查询到所有的作者 author_data = Author.objects.all() return render(request, 'author_list.html', {'author_list': author_data})
这里需要说明的是author_data是所有作者的对象，
print(author_data) # <QuerySet [<Author: 邹邹>, <Author: 马云>, <Author: 范冰冰>, <Author: 刘亦菲>, <Author: 景甜>]>
我们去循环author_data
def author_list(request): # 查询到所有的作者 author_data = Author.objects.all() print(author_data) # <QuerySet [<Author: 邹邹>, <Author: 马云>, <Author: 范冰冰>, <Author: 刘亦菲>, <Author: 景甜>]> for author in author_data: print(author) # 取到每个出版社的作者 print(author.books) # 是一个ORM提供的桥梁，帮我找对应关系 print(author.books.all()) # 取到每个作者出版的书籍 取出来的是一个对象
print('*'*50)
结果
<QuerySet [<Author: 邹邹>, <Author: 马云>, <Author: 范冰冰>, <Author: 刘亦菲>, <Author: 景甜>]> 邹邹 appTest01.Book.None <QuerySet [<Book: Book object>, <Book: Book object>, <Book: Book object>]> 取出来的是一个对象 ************************************************** 马云 appTest01.Book.None <QuerySet [<Book: Book object>]> ************************************************** 范冰冰 appTest01.Book.None <QuerySet [<Book: Book object>]> ************************************************** 刘亦菲 appTest01.Book.None <QuerySet [<Book: Book object>]> ************************************************** 景甜 appTest01.Book.None <QuerySet []> **************************************************
在取添加author_list.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>作者列表</title> </head> <body> <table border="1"> <thead> <tr> <th>序号</th> <th>作者id</th> <th>作者姓名</th> <th>作者作品</th> </tr> </thead> <tbody> {% for author in author_list %} <tr> <td>{{ forloop.counter }}</td> <td>{{ author.id }}</td> <td>{{ author.name }}</td> <td style="text-align: center"> {% for book in author.books.all %} {% if forloop.last %} 《{{ book.title }}》 {% else %} 《{{ book.title }}》, {% endif %} {% empty %} 暂无作品 {% endfor %} </td> </tr> {% endfor %} </tbody> </table> </body> </html>
代码解释：
{% for book in author.books.all %} 在前面我们知道 author.books.all()取出来的是一个对象，因为一个作者可能有多个书，所以我们循环这个对象
{% empty %}也是for循环里的，当循环出的book是一个空对象时，执行下面的代码”暂无作品“
{% if forloop.last %}表示循环到最后一个执行下面的代码
添加作者
获取作者的功能已经实现了，那我们在写个添加作者的功能，添加作者，还要选择书，所以这里涉及了两张表，第一张是author表，第二张是author和book对应关系的那张表，先去添加路由
url(r'^add_author/', views.add_author), # 添加作者
在去添加对应的函数
# 添加作者 def add_author(request): book_data = Book.objects.all() return render(request, 'add_author.html', {"book_list": book_data})
添加add_author.html文件
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Title</title> </head> <body> <h1>添加作者</h1> <form action="" method="post"> <input type="text" name="author_name"> <select name="books" multiple> {% for book in book_list %} <option value="{{book.id }}">{{ book.title }}</option> {% endfor %} </select> <input type="submit" value="提交"> </form> </body> </html>
因为一个作者可以有多本书，所以书籍是多选的，在去修改对应的函数
# 添加作者 def add_author(request): if request.method == 'POST': new_author_name = request.POST.get('author_name') # 这个只能取到一个值 book_ids = request.POST.getlist('books') # 这个取多个值，取到的是一个列表 author_obj = Author.objects.create(name=new_author_name) #创建新的作者 # author_obj.books.add(*book_ids) # 参数是一个一个单独的书籍id值，打散传进去 author_obj.books.set(book_ids) # 参数是书籍id值的列表 return redirect('/author_list/') book_data = Book.objects.all() return render(request, 'add_author.html', {"book_list": book_data})
因为书籍是多选的，所以可能有多个书的id，所以要用到request.POST.getlist('books')
author_obj = Author.objects.create(name=new_author_name) 返回的值是作者的名字
author_obj.books.add(*book_ids)因为只能一个一个接收参数，所以这里需要打散，记得前面要加上表名
a = [1, 2, 3] print(*a) 结果： 1 2 3
打散
author_obj.books.set(book_ids)接收的是一个列表，和上面的效果是一样的
删除作者
添加路由
url(r'^delete_author/', views.delete_author), # 删除作者
添加对应的函数
# 删除作者 def delete_author(request): delete_author_id = request.GET.get('id') Author.objects.get(id=delete_author_id).delete() return redirect('/author_list/')
html文件只需要更改下author_list.html里的代码即可
<td> <a href="/delete_author/?id={{ author.id }}">删除</a> <a href="/edit_author/?id={{ author.id }}">编辑</a> </td>
编辑作者
添加路由
url(r'^edit_author/', views.edit_author), # 编辑作者
添加对应的函数
# 编辑作者 def edit_author(request): edit_author_id = request.GET.get('id') # 取到要编辑的作者id # edit_author_obj = Author.objects.get(id=edit_author_id) # 取到要编辑的作者对象 edit_author_obj = Author.objects.filter(id=edit_author_id)[0] # 取到要编辑的作者对象 if request.method == 'POST': new_book_ids = request.POST.getlist('book_ids') new_author_name = request.POST.get('author_name') edit_author_obj.name = new_author_name edit_author_obj.save() edit_author_obj.books.set(new_book_ids) # 修改作者和书的关系表 return redirect('/author_list/') book_data = Book.objects.all() return render(request, 'edit_author.html', {'author': edit_author_obj, 'book_list': book_data})
说明：
edit_author_obj.books.set(new_book_ids)意思是将这个作者和书的对应关系去掉，然后重新设置对应关系
添加edit_author.html文件
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>编辑作者</title> </head> <body> <form action="" method="post"> <input type="text" name="author_name" value="{{ author.name }}"> <select name="book_ids" multiple> {% for book in book_list %} {% if book in author.books.all %} <option selected value="{{ book.id }}">{{ book.title }}</option> {% else %} <option value="{{ book.id }}">{{ book.title }}</option> {% endif %} {% endfor %} </select> <input type="submit" value="提交"> </form> </body> </html>
代码解释
{% if book in author.books.all %} 这里的book是book这个表里的一项，在判断这一项在没在书和作者对应关系的那张表里，如果在，则设置成默认选中的
上传文件
上传文件需要在html文件里的form表单里指定 enctype="multipart/form-data"。这样Django才能正常接收文件
添加路由
url(r'^upload/', views.upload), # 上传文件
添加对应函数
# 上传文件 def upload(request): if request.method == 'POST': file_obj = request.FILES.get('file_name') # 获取文件 file_name = file_obj.name # 取到文件名+后缀 # 从上传文件对象里一点一点读取数据，写到本地 with open(file_name, 'wb') as f: for line in file_obj: f.write(line) return render(request, 'upload.html')
说明：之前获取提交的数据用的是request.POST.get或者request.GET.get，但是获取文件信息要用request.FILES.get，返回的也是一个字典，key是input输入框里的name属性
添加对应的html文件
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>上传文件</title> </head> <body> <form action="" method="post" enctype="multipart/form-data"> <input type="file" name="file_name"> <input type="submit"> </form> </body> </html>
改版：
导入settings.py文件
from django.conf import settings
如果文件存在，重命名
# 上传文件 def upload(request): if request.method == 'POST': file_obj = request.FILES.get('file_name') # 获取文件 file_name = file_obj.name # 取到文件名+后缀 # 判断文件是否存在 if os.path.exists(os.path.join(settings.BASE_DIR, file_name)): # 如果存在同名的文件 name,suffix = file_name.split('.') name += '1' file_name = name + '.'+ suffix # 从上传文件对象里一点一点读取数据，写到本地 with open(file_name, 'wb') as f: for line in file_obj: f.write(line) return render(request, 'upload.html')
