Advertisement
Guest User

Untitled

a guest
Apr 27th, 2015
207
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.31 KB | None | 0 0
  1. import os
  2. from Nelly.models import Album, Image
  3. from django.core.files.base import File
  4. def import_albums(path):
  5.     items = os.listdir(path)  # находим все элементы базовой папки
  6.     dirs = []  # тут будут папки с фотографиями (альбомы)
  7.     for item in items:  # проходимся по всем элементам
  8.         if os.path.isdir(os.path.join(path, item)):  # выбираем только папки
  9.             dirs.append(os.path.join(path, item))
  10.     for directory in dirs:  # проходимся по папкам
  11.         album = Album()  # создаем новый альбом
  12.         album.name = os.path.split(directory)[-1]  # в имя альбома ложим название папки
  13.         album.save()  # сохраняем альбом
  14.         img_list = os.listdir(directory)  # получаем список изображений в альбоме
  15.         for img in img_list:  # проходимся по изображениям
  16.             image = Image()  # создаем изображение в базе
  17.             with open(os.path.join(directory, img), 'rb') as f:  # открываем локальное изображение
  18.                 my_file = File(f)
  19.                 image.image.save(img, my_file)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement