Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- from Nelly.models import Album, Image
- from django.core.files.base import File
- def import_albums(path):
- items = os.listdir(path) # находим все элементы базовой папки
- dirs = [] # тут будут папки с фотографиями (альбомы)
- for item in items: # проходимся по всем элементам
- if os.path.isdir(os.path.join(path, item)): # выбираем только папки
- dirs.append(os.path.join(path, item))
- for directory in dirs: # проходимся по папкам
- album = Album() # создаем новый альбом
- album.name = os.path.split(directory)[-1] # в имя альбома ложим название папки
- album.save() # сохраняем альбом
- img_list = os.listdir(directory) # получаем список изображений в альбоме
- for img in img_list: # проходимся по изображениям
- image = Image() # создаем изображение в базе
- with open(os.path.join(directory, img), 'rb') as f: # открываем локальное изображение
- my_file = File(f)
- image.image.save(img, my_file)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement