python / django将目录文件添加到db

时间:2017-09-02 06:09:14

标签: django python-3.x django-views

我有一个包含几个pdf的文件夹,我想添加到我的数据库中,但是使用下面的代码,每次刷新页面时,相同的数据再次添加到db,是否有办法避免这种情况和只有新数据添加到数据库?谢谢

 def index(request):
        path = "/Users/Sam/Desktop/docs"
        mylist1 = os.listdir(path)
        mylist2 = glob.glob(path + "//*.pdf")
        mylist1.pop(0)
        file = []
        for z in mylist1:
            file.append(os.path.splitext(z)[0])
        for (x, y) in zip(file, mylist2):
            p = File(file_name=x, file_path=y)
            p.save()
        all_files = File.objects.all().order_by('file_name')
        return render(request, 'form/index.html', {'all_files': all_files})

1 个答案:

答案 0 :(得分:1)

您可以尝试get or create

for (x, y) in zip(file, mylist2):
    File.objects.get_or_create(file_name=x, file_path=y)

而不是

for (x, y) in zip(file, mylist2):
    p = File(file_name=x, file_path=y)
    p.save()