def _add_to_place_album(self):
"""
将照片自动放入对应的地点相册，国内按市分组，国外按国家分组
"""
if self.addr_text is not None:
if self.addr_country == "中国":
album_addr_key = self.addr_city
addr_level = 3
else:
album_addr_key = self.addr_country
addr_level = 1
from album.models import Album
place_albums = Album.objects.filter(type="A",
addr_name=album_addr_key)
if len(place_albums) == 0:
album = Album(title=album_addr_key,
auto=True,
type="A",
addr_name=album_addr_key,
addr_level=addr_level)
album.save()
else:
album = place_albums[0]
album.photos.add(self)
album.save()
def create_album(request):
if request.method == 'POST':
user = request.user
if user.is_authenticated():
name = request.POST['album_name_create']
description = request.POST['album_description_create']
album = Album(owner=user,
name=name,
description=description,
firstpage=0)
album.save()
return redirect('view_album', album_id=album.id)
else:
return redirect('/')
def setUp(self):
super().setUp()
self.genre = Genre(description='test', genre_image=self._get_image())
self.band = Band(name='test',
genre=self.genre,
band_image=self._get_image())
self.album = Album(name='test',
band=self.band,
release_date=2020,
cover_image=self._get_image())
self.genre.save()
self.band.save()
self.album.save()
def post(self, request, *args, **kwargs):
form_class = self.get_form_class()
form = self.get_form(form_class)
album = Album()
if form.is_valid():
album.artist_name = form.cleaned_data['artist_name']
album.album_name = form.cleaned_data['album_name']
album.released_date = form.cleaned_data['released_date']
album.length = form.cleaned_data['length']
album.added_by = request.user
album.save()
for item in form.cleaned_data['genre']:
album.genre.add(item)
return HttpResponseRedirect(self.get_success_url())
else:
return render(request, self.template_name, {'form': form})
